mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-20 14:30:10 +08:00
147 lines
3.7 KiB
Go
147 lines
3.7 KiB
Go
package moyu
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"fmt"
|
||
"os"
|
||
"strconv"
|
||
"time"
|
||
|
||
"github.com/FloatTech/ZeroBot-Plugin/control"
|
||
"github.com/robfig/cron"
|
||
|
||
zero "github.com/wdvxdr1123/ZeroBot"
|
||
"github.com/wdvxdr1123/ZeroBot/message"
|
||
)
|
||
|
||
//开启的群
|
||
type Kq struct {
|
||
GroupID []int64 `json:"群号"` //群号
|
||
}
|
||
|
||
//默认
|
||
var MY = Kq{
|
||
GroupID: []int64{639865824},
|
||
}
|
||
|
||
func init() { // 插件主体
|
||
os.MkdirAll("config/", 0777)
|
||
_, err := os.Stat(`config/moyu.json`)
|
||
if err == nil {
|
||
f, _ := os.Open("config/moyu.json")
|
||
defer f.Close()
|
||
err := json.NewDecoder(f).Decode(&MY)
|
||
if err != nil {
|
||
fmt.Print("config.json格式错误! 请检查!\n")
|
||
time.Sleep(5 * time.Second)
|
||
return
|
||
}
|
||
} else {
|
||
fp, _ := os.Create("config/moyu.json")
|
||
defer fp.Close()
|
||
data, _ := json.Marshal(MY)
|
||
var out bytes.Buffer
|
||
json.Indent(&out, data, "", "\t")
|
||
out.WriteTo(fp)
|
||
}
|
||
|
||
FansDaily() // 开启提醒
|
||
engine := control.Register("moyu", &control.Options{
|
||
DisableOnDefault: false,
|
||
Help: "moyu\n" +
|
||
"- 删除提醒\n" +
|
||
"- 添加提醒\n",
|
||
})
|
||
|
||
engine.OnRegex(`^(添加|删除)提醒$`).
|
||
SetBlock(true).SetPriority(20).Handle(func(ctx *zero.Ctx) {
|
||
fp, _ := os.Create("config/moyu.json")
|
||
defer fp.Close()
|
||
for i, v := range MY.GroupID {
|
||
if v == ctx.Event.GroupID {
|
||
MY.GroupID = append(MY.GroupID[:i], MY.GroupID[i+1:]...)
|
||
break
|
||
}
|
||
}
|
||
if ctx.State["regex_matched"].([]string)[1] == "添加" {
|
||
MY.GroupID = append(MY.GroupID, ctx.Event.GroupID)
|
||
ctx.Send(message.Text("添加成功！"))
|
||
} else {
|
||
ctx.Send(message.Text("删除成功！"))
|
||
}
|
||
data, _ := json.Marshal(MY)
|
||
var out bytes.Buffer
|
||
json.Indent(&out, data, "", "\t")
|
||
out.WriteTo(fp)
|
||
})
|
||
|
||
}
|
||
|
||
// 定时任务每天10点执行一次
|
||
func FansDaily() {
|
||
c := cron.New()
|
||
_ = c.AddFunc("0 0 10 * * ?", func() { fansData() })
|
||
c.Start()
|
||
}
|
||
|
||
// 获取数据拼接消息链并发送
|
||
func fansData() {
|
||
for _, v := range MY.GroupID {
|
||
zero.RangeBot(func(id int64, ctx *zero.Ctx) bool {
|
||
ctx.SendGroupMessage(
|
||
v,
|
||
[]message.MessageSegment{
|
||
message.Text(time.Now().Format("2006-01-02") +
|
||
"上午好，摸鱼人！\n工作再累，一定不要忘记摸鱼哦！有事没事起身去茶水间，去厕所，去廊道走走别老在工位上坐着，钱是老板的,但命是自己的。" +
|
||
Zm() + "\n" +
|
||
Moyu("元旦", 2022, 1, 1) + "\n" +
|
||
Moyu("春节", 2022, 1, 31) + "\n" +
|
||
Moyu("清明节", 2022, 4, 3) + "\n" +
|
||
Moyu("劳动节", 2022, 4, 30) + "\n" +
|
||
Moyu("端午节", 2022, 6, 3) + "\n" +
|
||
Moyu("中秋节", 2022, 9, 10) + "\n" +
|
||
Moyu("国庆节", 2022, 10, 1) + "\n" +
|
||
"\n\n上班是帮老板赚钱，摸鱼是赚老板的钱！最后，祝愿天下所有摸鱼人，都能愉快的渡过每一天…",
|
||
),
|
||
},
|
||
)
|
||
return true
|
||
})
|
||
|
||
}
|
||
}
|
||
|
||
// 获取两个时间相差
|
||
func Moyu(text string, year int, month time.Month, day int) string {
|
||
currentTime := time.Now()
|
||
t1 := time.Date(year, month, day, 0, 0, 0, 0, time.Local)
|
||
t2 := time.Date(currentTime.Year(), currentTime.Month(), currentTime.Day(), 0, 0, 0, 0, time.Local)
|
||
tt := int(t1.Sub(t2).Hours() / 24)
|
||
if tt >= 0 {
|
||
return "距离" + text + "还有: " + strconv.Itoa(tt) + " 天!"
|
||
} else {
|
||
return "好好享受 " + text + " 假期吧!"
|
||
}
|
||
}
|
||
|
||
func Zm() string {
|
||
t := time.Now().Weekday().String()
|
||
switch {
|
||
case t == "Sunday":
|
||
return "\n好好享受周末吧！"
|
||
case t == "Monday":
|
||
return "\n距离周末还有:4天！"
|
||
case t == "Tuesday":
|
||
return "\n距离周末还有:3天！"
|
||
case t == "Wednesday":
|
||
return "\n距离周末还有:2天！"
|
||
case t == "Thursday":
|
||
return "\n距离周末还有:1天！"
|
||
case t == "Friday":
|
||
return "\n距离周末还有:0天！"
|
||
default:
|
||
return "\n好好享受周末吧！"
|
||
}
|
||
}
|