🎨 优化目录结构

2026-02-11 17:50:25 +00:00 · 2022-02-25 22:15:14 +08:00
parent 5ccf753af3
commit 0cfb2e4e06
101 changed files with 116 additions and 116 deletions
--- a/plugin/manager/timer/sleep.go
+++ b/plugin/manager/timer/sleep.go
@@ -0,0 +1,165 @@
+package timer
+
+import (
+	"time"
+
+	"github.com/sirupsen/logrus"
+	zero "github.com/wdvxdr1123/ZeroBot"
+)
+
+func firstWeek(date *time.Time, week time.Weekday) (d time.Time) {
+	d = date.AddDate(0, 0, 1-date.Day())
+	for d.Weekday() != week {
+		d = d.AddDate(0, 0, 1)
+	}
+	return
+}
+
+func (t *Timer) nextWakeTime() (date time.Time) {
+	date = time.Now()
+	m := t.Month()
+	d := t.Day()
+	h := t.Hour()
+	mn := t.Minute()
+	w := t.Week()
+	var unit time.Duration
+	logrus.Debugln("[timer] unit init:", unit)
+	if mn >= 0 {
+		switch {
+		case h < 0:
+			if unit <= time.Second {
+				unit = time.Hour
+			}
+		case d < 0 || w < 0:
+			if unit <= time.Second {
+				unit = time.Hour * 24
+			}
+		case d == 0 && w >= 0:
+			delta := time.Hour * 24 * time.Duration(int(w)-int(date.Weekday()))
+			if delta < 0 {
+				delta = time.Hour * 24 * 7
+			}
+			unit += delta
+		case m < 0:
+			unit = -1
+		}
+	} else {
+		unit = time.Minute
+	}
+	logrus.Debugln("[timer] unit:", unit)
+	stable := 0
+	if mn < 0 {
+		mn = date.Minute()
+	}
+	if h < 0 {
+		h = date.Hour()
+	} else {
+		stable |= 0x8
+	}
+	switch {
+	case d < 0:
+		d = date.Day()
+	case d > 0:
+		stable |= 0x4
+	default:
+		d = date.Day()
+		if w >= 0 {
+			stable |= 0x2
+		}
+	}
+	if m < 0 {
+		m = date.Month()
+	} else {
+		stable |= 0x1
+	}
+	switch stable {
+	case 0b0101:
+		if t.Day() != time.Now().Day() || t.Month() != time.Now().Month() {
+			h = 0
+		}
+	case 0b1001:
+		if t.Month() != time.Now().Month() {
+			d = 0
+		}
+	case 0b0001:
+		if t.Month() != time.Now().Month() {
+			d = 0
+			h = 0
+		}
+	}
+	logrus.Debugln("[timer] stable:", stable)
+	logrus.Debugln("[timer] m:", m, "d:", d, "h:", h, "mn:", mn, "w:", w)
+	date = time.Date(date.Year(), m, d, h, mn, date.Second(), date.Nanosecond(), date.Location())
+	logrus.Debugln("[timer] date original:", date)
+	if unit > 0 {
+		date = date.Add(unit)
+	}
+	logrus.Debugln("[timer] date after add:", date)
+	if time.Until(date) <= 0 {
+		if t.Month() < 0 {
+			if t.Day() > 0 || (t.Day() == 0 && t.Week() >= 0) {
+				date = date.AddDate(0, 1, 0)
+			} else if t.Day() < 0 || t.Week() < 0 {
+				if t.Hour() > 0 {
+					date = date.AddDate(0, 0, 1)
+				} else if t.Minute() > 0 {
+					date = date.Add(time.Hour)
+				}
+			}
+		} else {
+			date = date.AddDate(1, 0, 0)
+		}
+	}
+	logrus.Debugln("[timer] date after fix:", date)
+	if stable&0x8 != 0 && date.Hour() != h {
+		switch {
+		case stable&0x4 == 0:
+			date = date.AddDate(0, 0, 1).Add(-time.Hour)
+		case stable&0x2 == 0:
+			date = date.AddDate(0, 0, 7).Add(-time.Hour)
+		case stable*0x1 == 0:
+			date = date.AddDate(0, 1, 0).Add(-time.Hour)
+		default:
+			date = date.AddDate(1, 0, 0).Add(-time.Hour)
+		}
+	}
+	logrus.Debugln("[timer] date after s8:", date)
+	if stable&0x4 != 0 && date.Day() != d {
+		switch {
+		case stable*0x1 == 0:
+			date = date.AddDate(0, 1, -1)
+		default:
+			date = date.AddDate(1, 0, -1)
+		}
+	}
+	logrus.Debugln("[timer] date after s4:", date)
+	if stable&0x2 != 0 && date.Weekday() != w {
+		switch {
+		case stable*0x1 == 0:
+			date = date.AddDate(0, 1, 0)
+		default:
+			date = date.AddDate(1, 0, 0)
+		}
+		date = firstWeek(&date, w)
+	}
+	logrus.Debugln("[timer] date after s2:", date)
+	if time.Until(date) <= 0 {
+		date = time.Now().Add(time.Minute)
+	}
+	return date
+}
+
+func (t *Timer) judgeHM() {
+	if t.Hour() < 0 || t.Hour() == time.Now().Hour() {
+		if t.Minute() < 0 || t.Minute() == time.Now().Minute() {
+			if t.SelfID != 0 {
+				t.sendmsg(t.GrpID, zero.GetBot(t.SelfID))
+			} else {
+				zero.RangeBot(func(id int64, ctx *zero.Ctx) (_ bool) {
+					t.sendmsg(t.GrpID, ctx)
+					return
+				})
+			}
+		}
+	}
+}