mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-20 14:30:10 +08:00
128 lines
3.5 KiB
Go
128 lines
3.5 KiB
Go
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)
|
||
}
|
||
}
|