package l_notify

import (
	"fmt"
	"gitea.cdlsxd.cn/self-tools/l_request"
	"github.com/robfig/cron/v3"
)

type Notify struct {
	Request      *l_request.Request
	DoFuc        func() (l_request.Response, error)          //测试用
	DelayList    []int32                                     //[30,60,120...]：从前往后分别延迟30秒，60秒，120秒...推送,[60]：每隔60秒推送一次
	ResultHandle func(l_request.Response, error) (stop bool) //推送结果处理函数,stop为true时表示停止推送
	cron         *cron.Cron
	stop         chan struct{}
	currenEntry  cron.EntryID
	isLoop       bool
	i            int8
}

func (n *Notify) Notify() error {
	//必要参数检测
	err := n.check()
	if err != nil {
		return err
	}

	if len(n.DelayList) == 1 {
		//如果只有一次，则默认循环推送
		n.isLoop = true
	}

	n.stop = make(chan struct{})
	//启动定时器
	n.cron = cron.New()
	n.next()
	n.cron.Start()

	return nil
}

func (n *Notify) next() {
	if n.i+1 > int8(len(n.DelayList)) {
		//当不存在下次执行，则退出定时器
		n.cron.Stop()
		close(n.stop)
		return
	}
	time := "@every " + fmt.Sprintf("%ds", n.DelayList[n.i])
	n.currenEntry, _ = n.cron.AddFunc(time, func() {
		e := n.ResultHandle(n.DoFuc())
		if n.isLoop {
			//循环推送直到成功
			if e {
				n.stop <- struct{}{}
			}
		} else {
			n.stop <- struct{}{}
		}
	})
	//开启信号监听
	n.registerListener()
	return
}

func (n *Notify) check() error {
	if n.Request == nil && n.DoFuc == nil {
		return fmt.Errorf("未初始化Request")
	}
	if len(n.DelayList) == 0 {
		return fmt.Errorf("未设置推送延迟")
	}
	return nil
}

func (n *Notify) registerListener() {
	//注册信号监听通道
	go func() {
		select {
		case <-n.stop:
			n.cron.Remove(n.currenEntry)
			n.i += 1
			n.next()
		}
	}()

}