feat: 增加设置指定时间的一次性提醒任务

Author: yqchilde

plugins/manager/README.md

@@ -12,6 +12,7 @@
         * `设置每周的提醒`，例如：设置每周三10:00:00的提醒
         * `设置每天的提醒`，例如：设置每天10:00:00的提醒
         * `设置每隔多久的提醒`，例如：设置每隔1小时的提醒
+        * `设置指定时间的一次性提醒任务`，例如：设置2023-01-01 15:00:00的提醒
         * `列出所有任务`
         * `删除任务 任务ID`
         * `删除全部任务`

plugins/manager/cronjob.go

@@ -17,10 +17,11 @@ const (
 	JobTypeFunc   = "func"   // 函数类任务
 	JobTypePlugin = "plugin" // 插件类任务
 
-	RegexOfRemindEveryMonth = `^设置每月(0?[1-9]|[12][0-9]|3[01])号(([01]?[0-9]|2[0-3]):[0-5][0-9]:[0-5][0-9])的提醒$`
-	RegexOfRemindEveryWeek  = `^设置每周(一|二|三|四|五|六|七|日)(([01]?[0-9]|2[0-3]):[0-5][0-9]:[0-5][0-9])的提醒$`
-	RegexOfRemindEveryDay   = `^设置每天(([01]?[0-9]|2[0-3]):[0-5][0-9]:[0-5][0-9])的提醒$`
-	RegexOfRemindInterval   = `^设置每隔(\d+)(s|秒|m|分|分钟|h|时|d|小时)的提醒$`
+	RegexOfRemindEveryMonth  = `^设置每月(0?[1-9]|[12][0-9]|3[01])号(([01]?[0-9]|2[0-3]):[0-5][0-9]:[0-5][0-9])的提醒$`
+	RegexOfRemindEveryWeek   = `^设置每周(一|二|三|四|五|六|七|日)(([01]?[0-9]|2[0-3]):[0-5][0-9]:[0-5][0-9])的提醒$`
+	RegexOfRemindEveryDay    = `^设置每天(([01]?[0-9]|2[0-3]):[0-5][0-9]:[0-5][0-9])的提醒$`
+	RegexOfRemindInterval    = `^设置每隔(\d+)(s|秒|m|分|分钟|h|时|d|小时)的提醒$`
+	RegexOfRemindSpecifyTime = `^设置((20[2-9][0-9]|2100)-(0[1-9]|1[0-2])-(0[1-9]|[12][0-9]|3[01])\s([01][0-9]|2[0-3]):[0-5][0-9]:[0-5][0-9])的提醒$`
 )
 
 type CronJob struct {
@@ -58,7 +59,7 @@ func registerCronjob() {
 					if _, err := AddRemindOfEveryMonth(ctx, cronJob.Tag, matched, func() {
 						ctx.SendText(cronJob.GroupId, cronJob.Remind)
 					}); err != nil {
-						log.Errorf("恢复每月提醒任务失败: jobId: %s, error: %v", cronJob.Id, err)
+						log.Errorf("恢复每月提醒任务失败: jobId: %d, error: %v", cronJob.Id, err)
 					}
 				}
 
@@ -67,7 +68,7 @@ func registerCronjob() {
 					if _, err := AddRemindOfEveryWeek(ctx, cronJob.Tag, matched, func() {
 						ctx.SendText(cronJob.GroupId, cronJob.Remind)
 					}); err != nil {
-						log.Errorf("恢复每周提醒任务失败: jobId: %s, error: %v", cronJob.Id, err)
+						log.Errorf("恢复每周提醒任务失败: jobId: %d, error: %v", cronJob.Id, err)
 					}
 				}
 
@@ -76,7 +77,7 @@ func registerCronjob() {
 					if _, err := AddRemindOfEveryDay(ctx, cronJob.Tag, matched, func() {
 						ctx.SendText(cronJob.GroupId, cronJob.Remind)
 					}); err != nil {
-						log.Errorf("恢复每天提醒任务失败: jobId: %s, error: %v", cronJob.Id, err)
+						log.Errorf("恢复每天提醒任务失败: jobId: %d, error: %v", cronJob.Id, err)
 					}
 				}
 
@@ -85,7 +86,16 @@ func registerCronjob() {
 					if _, err := AddRemindForInterval(ctx, cronJob.Tag, matched, func() {
 						ctx.SendText(cronJob.GroupId, cronJob.Remind)
 					}); err != nil {
-						log.Errorf("恢复间隔提醒任务失败: jobId: %s, error: %v", cronJob.Id, err)
+						log.Errorf("恢复间隔提醒任务失败: jobId: %d, error: %v", cronJob.Id, err)
+					}
+				}
+
+				// 恢复指定时间提醒任务
+				if matched := regexp.MustCompile(RegexOfRemindSpecifyTime).FindStringSubmatch(cronJob.Desc); matched != nil {
+					if _, err := AddRemindForSpecifyTime(ctx, cronJob.Tag, matched, func() {
+						ctx.SendText(cronJob.GroupId, cronJob.Remind)
+					}); err != nil {
+						log.Errorf("恢复指定时间提醒任务失败: jobId: %d, error: %v", cronJob.Id, err)
 					}
 				}
 			}
@@ -265,6 +275,49 @@ func registerCronjob() {
 		}
 	})
 
+	// 设置指定时间的一次性提醒任务
+	// 设置2023-01-01 15:00:00的提醒
+	engine.OnRegex(RegexOfRemindSpecifyTime, robot.UserOrGroupAdmin).SetBlock(true).Handle(func(ctx *robot.Ctx) {
+		matched := ctx.State["regex_matched"].([]string)
+		jobDesc := ctx.MessageString()
+		recv, cancel := ctx.EventChannel(ctx.CheckUserSession()).Repeat()
+		defer cancel()
+		ctx.ReplyText("请问需要提醒什么呢？")
+		for {
+			select {
+			case <-time.After(20 * time.Second):
+				ctx.ReplyTextAndAt("操作时间太久了，请重新设置")
+				return
+			case c := <-recv:
+				jobId := mid.UniqueId()
+				jobTag := strconv.Itoa(int(jobId))
+				remind := c.Event.Message.Content
+
+				// 设置定时任务
+				if _, err := AddRemindForSpecifyTime(ctx, jobTag, matched, func() { ctx.ReplyText(remind) }); err != nil {
+					ctx.ReplyText(fmt.Errorf("设置失败: %v", err).Error())
+					return
+				}
+
+				// 存起来便于服务启动恢复
+				if err := db.Orm.Table("cronjob").Create(&CronJob{
+					Id:      jobId,
+					Tag:     jobTag,
+					Type:    JobTypeRemind,
+					GroupId: ctx.Event.FromUniqueID,
+					Desc:    jobDesc,
+					Remind:  remind,
+				}).Error; err != nil {
+					ctx.ReplyTextAndAt(fmt.Errorf("设置失败: %v", err).Error())
+					return
+				}
+				ctx.ReplyTextAndAt(fmt.Sprintf("已为您%s: %s", jobDesc, remind))
+				job.StartAsync()
+				return
+			}
+		}
+	})
+
 	// 列出当前所有定时任务
 	engine.OnFullMatch("列出所有任务").SetBlock(true).Handle(func(ctx *robot.Ctx) {
 		var cronJobs []CronJob

plugins/manager/cronjob_add.go

@@ -66,3 +66,15 @@ func AddRemindForInterval(ctx *robot.Ctx, jobTag string, matched []string, f fun
 	}
 	return job.Tag(jobTag).Do(func() { f() })
 }
+
+// AddRemindForSpecifyTime 添加指定时间提醒
+func AddRemindForSpecifyTime(ctx *robot.Ctx, jobTag string, matched []string, f func()) (*gocron.Job, error) {
+	parseTime, _ := time.ParseInLocation("2006-01-02 15:04:05", matched[1], time.Local)
+	if parseTime.Before(time.Now()) {
+		return nil, fmt.Errorf("请不要设置过去的时间")
+	}
+	return job.Every(1).Tag(jobTag).LimitRunsTo(1).StartAt(parseTime).Do(func() {
+		f()
+		db.Orm.Table("cronjob").Where("tag = ?", jobTag).Delete(&CronJob{})
+	})
+}