ZeroBot-Plugin/plugin_manager/timer/parse.go

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
}