package NewHoliday

import (
	"fmt"
	"time"

	"github.com/fumiama/cron"
	zero "github.com/wdvxdr1123/ZeroBot"
	"github.com/wdvxdr1123/ZeroBot/message"

	"github.com/FloatTech/ZeroBot-Plugin/control"
)

func init() { // 插件主体
	registerNotice() // 开启提醒

	control.Register("moyu", &control.Options{
		DisableOnDefault: true,
		Help: "moyu\n" +
			"- 添加摸鱼提醒\n" +
			"- 删除摸鱼提醒\n",
	}).OnFullMatch("删除摸鱼提醒", zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(20).
		Handle(func(ctx *zero.Ctx) {
			m, ok := control.Lookup("moyu")
			if ok {
				if m.IsEnabledIn(ctx.Event.GroupID) {
					m.Disable(ctx.Event.GroupID)
					ctx.Send(message.Text("删除成功！"))
				} else {
					ctx.Send(message.Text("未启用！"))
				}
			} else {
				ctx.Send(message.Text("找不到该服务！"))
			}
		})

	zero.OnFullMatch("添加摸鱼提醒", zero.OnlyGroup, zero.AdminPermission).SetBlock(true).SetPriority(20).
		Handle(func(ctx *zero.Ctx) {
			m, ok := control.Lookup("moyu")
			if ok {
				if m.IsEnabledIn(ctx.Event.GroupID) {
					ctx.Send(message.Text("已启用！"))
				} else {
					m.Enable(ctx.Event.GroupID)
					ctx.Send(message.Text("添加成功！"))
				}
			} else {
				ctx.Send(message.Text("找不到该服务！"))
			}
		})
}

// 定时任务每天10点执行一次
func registerNotice() {
	c := cron.New()
	_, err := c.AddFunc("0 0 10 * * ?", func() { sendNotice() })
	if err != nil {
		c.Start()
	}
}

// 获取数据拼接消息链并发送
func sendNotice() {
	m, ok := control.Lookup("moyu")
	if ok {
		zero.RangeBot(func(id int64, ctx *zero.Ctx) bool {
			for _, g := range ctx.GetGroupList().Array() {
				grp := g.Get("group_id").Int()
				if m.IsEnabledIn(grp) {
					ctx.SendGroupMessage(grp,
						[]message.MessageSegment{
							message.Text(time.Now().Format("2006-01-02")),
							message.Text("上午好，摸鱼人！\n工作再累，一定不要忘记摸鱼哦！有事没事起身去茶水间，去厕所，去廊道走走别老在工位上坐着，钱是老板的,但命是自己的。"),
							message.Text(weekend()),
							message.Text("\n"),
							message.Text(NewHoliday("元旦", 2022, 1, 1)),
							message.Text("\n"),
							message.Text(NewHoliday("春节", 2022, 1, 31)),
							message.Text("\n"),
							message.Text(NewHoliday("清明节", 2022, 4, 3)),
							message.Text("\n"),
							message.Text(NewHoliday("劳动节", 2022, 4, 30)),
							message.Text("\n"),
							message.Text(NewHoliday("端午节", 2022, 6, 3)),
							message.Text("\n"),
							message.Text(NewHoliday("中秋节", 2022, 9, 10)),
							message.Text("\n"),
							message.Text(NewHoliday("国庆节", 2022, 10, 1)),
							message.Text("\n"),
							message.Text("\n\n上班是帮老板赚钱，摸鱼是赚老板的钱！最后，祝愿天下所有摸鱼人，都能愉快的渡过每一天…"),
						},
					)
				}
			}
			return true
		})
	}
}

type holiday struct {
	name string
	date time.Time
}

func NewHoliday(name string, year int, month time.Month, day int) holiday {
	return holiday{name: name, date: time.Date(year, month, day, 0, 0, 0, 0, time.Local)}
}

// 获取两个时间相差
func (h holiday) String() string {
	d := time.Until(h.date)
	if d >= 0 {
		return "距离" + h.name + "还有: " + d.String()
	} else {
		return "好好享受 " + h.name + " 假期吧!"
	}
}

func weekend() string {
	t := time.Now().Weekday()
	switch t {
	case time.Sunday, time.Saturday:
		return "\n好好享受周末吧！"
	default:
		return fmt.Sprintf("\n距离周末还有:%d天！", 5-t)
	}
}