package system

import (
	"gitee.com/h79/goutils/common"
	"os"
	"os/signal"
	"sync"
	"syscall"
	"time"
)

func Run() {
	sys := Exit()
	for {
		select {
		case <-sys.Done():
			return
		}
	}
}

type Chan struct {
	S chan os.Signal
}

func (c *Chan) Done() <-chan os.Signal {
	return c.S
}

func Exit() *Chan {
	s := &Chan{S: make(chan os.Signal)}
	signal.Notify(s.S, os.Interrupt, os.Kill, syscall.SIGTERM, syscall.SIGQUIT)
	return s
}

func Quit() {
	timeout := time.After(2 * time.Second)
	select {
	case <-timeout:
		return
	}
}

type mainRun struct {
	ev *Event
	wg sync.WaitGroup
}

var (
	mRun *mainRun
	once sync.Once
)

func Main() *mainRun {
	once.Do(func() {
		mRun = &mainRun{
			ev: NewEvent(),
		}
	})
	return mRun
}

// Run self =true 表示主协程,子协程，select Main().Done
func (m *mainRun) Run(self bool, fn func()) {
	m.wg.Add(1)
	go func() {
		defer m.wg.Done()
		defer common.Recover()
		defer func() {
			if !self {
				return
			}
			m.ev.Fire()
		}()
		fn()
	}()
}

func (m *mainRun) Done() <-chan struct{} {
	return m.ev.Done()
}

func (m *mainRun) Wait() {
	m.wg.Wait()
}