package gdb

import (
	"os"
	"os/signal"
	"runtime"
	"syscall"

	ice "shylinux.com/x/icebergs"
	"shylinux.com/x/icebergs/base/mdb"
	kit "shylinux.com/x/toolkits"
	"shylinux.com/x/toolkits/file"
)

func _signal_listen(m *ice.Message, s int, arg ...string) {
	if f, ok := m.Target().Server().(*Frame); ok {
		f.listen(m, s, arg...)
	}
}
func _signal_action(m *ice.Message, arg ...string) {
	mdb.HashSelect(m.Spawn(), arg...).Table(func(value ice.Maps) { m.Cmdy(kit.Split(value[ice.CMD])) })
}
func _signal_process(m *ice.Message, p string, s os.Signal) {
	if p == "" {
		b, _ := file.ReadFile(ice.Info.PidPath)
		p = string(b)
		if runtime.GOOS == ice.WINDOWS {
			return
		}
	}
	if p, e := os.FindProcess(kit.Int(kit.Select(kit.Format(os.Getpid()), p))); e == nil {
		p.Signal(s)
	}
}

const (
	PID = "pid"
)
const (
	DEBUG   = "debug"
	ERROR   = "error"
	START   = "start"
	RESTART = "restart"
	STOP    = "stop"
	KILL    = "kill"
)
const SIGNAL = "signal"

func init() {
	Index.MergeCommands(ice.Commands{
		SIGNAL: {Name: "signal signal auto listen", Help: "信号量", Actions: ice.MergeActions(ice.Actions{
			ice.CTX_INIT: {Hand: func(m *ice.Message, arg ...string) {
				if runtime.GOOS == ice.WINDOWS {
					return
				}
				_signal_listen(m, 1, mdb.NAME, START, ice.CMD, "runtime")
				_signal_listen(m, 2, mdb.NAME, RESTART, ice.CMD, "exit 1")
				_signal_listen(m, 3, mdb.NAME, STOP, ice.CMD, "exit 0")
			}},
			LISTEN: {Name: "listen signal name cmd", Hand: func(m *ice.Message, arg ...string) {
				_signal_listen(m, kit.Int(m.Option(SIGNAL)), arg...)
			}},
			HAPPEN: {Name: "happen signal", Help: "触发", Hand: func(m *ice.Message, arg ...string) {
				_signal_action(m, m.Option(SIGNAL))
			}},
			RESTART: {Name: "restart pid", Hand: func(m *ice.Message, arg ...string) {
				_signal_process(m, m.Option(PID), syscall.SIGINT)
			}},
			STOP: {Name: "stop pid", Hand: func(m *ice.Message, arg ...string) {
				_signal_process(m, m.Option(PID), syscall.SIGQUIT)
			}},
			KILL: {Name: "kill pid signal", Hand: func(m *ice.Message, arg ...string) {
				_signal_process(m, m.Option(PID), syscall.Signal(kit.Int(kit.Select("9", m.Option(SIGNAL)))))
			}},
		}, mdb.HashAction(mdb.SHORT, SIGNAL, mdb.FIELD, "time,signal,name,cmd", mdb.ACTION, HAPPEN), mdb.ClearOnExitHashAction()), Hand: func(m *ice.Message, arg ...string) {
			mdb.HashSelect(m, arg...).Sort(SIGNAL)
		}},
	})
}

func SignalNotify(m *ice.Message, sig syscall.Signal, cb func()) {
	ch := make(chan os.Signal)
	signal.Notify(ch, sig)
	m.Go(func() {
		for {
			if _, ok := <-ch; ok {
				cb()
			}
		}
	})
}