摸鱼提醒 (#79)

* Add files via upload

* Update run.go

* Update run.go

* Update run.go

plugin_moyu/run.go

@@ -0,0 +1,146 @@
+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好好享受周末吧！"
+	}
+}