🎨 优化目录结构

plugin/manager/timer/parse.go

@@ -0,0 +1,166 @@
+package timer
+
+import (
+	"crypto/md5"
+	"encoding/binary"
+	"fmt"
+	"strconv"
+	"strings"
+	"time"
+	"unicode"
+
+	"github.com/sirupsen/logrus"
+	"github.com/wdvxdr1123/ZeroBot/utils/helper"
+)
+
+// GetTimerInfo 获得标准化定时字符串
+func (t *Timer) GetTimerInfo() string {
+	if t.Cron != "" {
+		return fmt.Sprintf("[%d]%s", t.GrpID, t.Cron)
+	}
+	return fmt.Sprintf("[%d]%d月%d日%d周%d:%d", t.GrpID, t.Month(), t.Day(), t.Week(), t.Hour(), t.Minute())
+}
+
+// GetTimerID 获得标准化 ID
+func (t *Timer) GetTimerID() uint32 {
+	key := t.GetTimerInfo()
+	m := md5.Sum(helper.StringToBytes(key))
+	return binary.LittleEndian.Uint32(m[:4])
+}
+
+// GetFilledCronTimer 获得以cron填充好的ts
+func GetFilledCronTimer(croncmd string, alert string, img string, botqq, gid int64) *Timer {
+	var t Timer
+	t.Alert = alert
+	t.Cron = croncmd
+	t.URL = img
+	t.SelfID = botqq
+	t.GrpID = gid
+	return &t
+}
+
+// GetFilledTimer 获得填充好的ts
+func GetFilledTimer(dateStrs []string, botqq, grp int64, matchDateOnly bool) *Timer {
+	monthStr := []rune(dateStrs[1])
+	dayWeekStr := []rune(dateStrs[2])
+	hourStr := []rune(dateStrs[3])
+	minuteStr := []rune(dateStrs[4])
+
+	var t Timer
+	mon := time.Month(chineseNum2Int(monthStr))
+	if (mon != -1 && mon <= 0) || mon > 12 { // 月份非法
+		t.Alert = "月份非法！"
+		return &t
+	}
+	t.SetMonth(mon)
+	lenOfDW := len(dayWeekStr)
+	switch {
+	case lenOfDW == 4: // 包括末尾的"日"
+		dayWeekStr = []rune{dayWeekStr[0], dayWeekStr[2]} // 去除中间的十
+		d := chineseNum2Int(dayWeekStr)
+		if (d != -1 && d <= 0) || d > 31 { // 日期非法
+			t.Alert = "日期非法1！"
+			return &t
+		}
+		t.SetDay(d)
+	case dayWeekStr[lenOfDW-1] == rune('日'): // xx日
+		dayWeekStr = dayWeekStr[:lenOfDW-1]
+		d := chineseNum2Int(dayWeekStr)
+		if (d != -1 && d <= 0) || d > 31 { // 日期非法
+			t.Alert = "日期非法2！"
+			return &t
+		}
+		t.SetDay(d)
+	case dayWeekStr[0] == rune('每'): // 每周
+		t.SetWeek(-1)
+	default: // 周x
+		w := chineseNum2Int(dayWeekStr[1:])
+		if w == 7 { // 周天是0
+			w = 0
+		}
+		if w < 0 || w > 6 { // 星期非法
+			t.Alert = "星期非法！"
+			return &t
+		}
+		t.SetWeek(time.Weekday(w))
+	}
+	if len(hourStr) == 3 {
+		hourStr = []rune{hourStr[0], hourStr[2]} // 去除中间的十
+	}
+	h := chineseNum2Int(hourStr)
+	if h < -1 || h > 23 { // 小时非法
+		t.Alert = "小时非法！"
+		return &t
+	}
+	t.SetHour(h)
+	if len(minuteStr) == 3 {
+		minuteStr = []rune{minuteStr[0], minuteStr[2]} // 去除中间的十
+	}
+	min := chineseNum2Int(minuteStr)
+	if min < -1 || min > 59 { // 分钟非法
+		t.Alert = "分钟非法！"
+		return &t
+	}
+	t.SetMinute(min)
+	if !matchDateOnly {
+		urlStr := dateStrs[5]
+		if urlStr != "" { // 是图片url
+			t.URL = urlStr[3:] // utf-8下用为3字节
+			logrus.Println("[群管]" + t.URL)
+			if !strings.HasPrefix(t.URL, "http") {
+				t.URL = "illegal"
+				logrus.Println("[群管]url非法！")
+				return &t
+			}
+		}
+		t.Alert = dateStrs[6]
+		t.SetEn(true)
+	}
+	t.SelfID = botqq
+	t.GrpID = grp
+	return &t
+}
+
+// chineseNum2Int 汉字数字转int，仅支持-10～99，最多两位数，其中"每"解释为-1，"每二"为-2，以此类推
+func chineseNum2Int(rs []rune) int {
+	r := -1
+	l := len(rs)
+	mai := rune('每')
+	if unicode.IsDigit(rs[0]) { // 默认可能存在的第二位也为int
+		r, _ = strconv.Atoi(string(rs))
+	} else {
+		switch {
+		case rs[0] == mai:
+			if l == 2 {
+				r = -chineseChar2Int(rs[1])
+			}
+		case l == 1:
+			r = chineseChar2Int(rs[0])
+		default:
+			ten := chineseChar2Int(rs[0])
+			if ten != 10 {
+				ten *= 10
+			}
+			ge := chineseChar2Int(rs[1])
+			if ge == 10 {
+				ge = 0
+			}
+			r = ten + ge
+		}
+	}
+	return r
+}
+
+// chineseChar2Int 处理单个字符的映射0~10
+func chineseChar2Int(c rune) int {
+	if c == rune('日') || c == rune('天') { // 周日/周天
+		return 7
+	}
+	match := []rune("零一二三四五六七八九十")
+	for i, m := range match {
+		if c == m {
+			return i
+		}
+	}
+	return 0
+}