foliko 1.0.0

package/plugins/scheduler-plugin.js

@@ -0,0 +1,426 @@
+/**
+ * Scheduler 定时任务调度插件
+ * 支持 Cron 表达式、绝对时间、相对时间调度
+ * 任务触发时自动唤醒 Agent 发送消息
+ */
+
+const { Plugin } = require('../src/core/plugin-base')
+const { z } = require('zod')
+
+// 尝试加载 node-cron
+let cron = null
+try {
+  cron = require('node-cron')
+} catch (e) {
+  console.warn('[Scheduler] node-cron not installed, cron tasks will not work')
+}
+
+// 时间解析辅助函数
+function parseDelay(delayStr) {
+  const match = delayStr.match(/^(\d+)\s*(second|minute|hour|day|week)s?$/i)
+  if (!match) return null
+  const value = parseInt(match[1])
+  const unit = match[2].toLowerCase()
+  const multipliers = {
+    second: 1000,
+    minute: 60 * 1000,
+    hour: 60 * 60 * 1000,
+    day: 24 * 60 * 60 * 1000,
+    week: 7 * 24 * 60 * 60 * 1000
+  }
+  return value * (multipliers[unit] || 1000)
+}
+
+function parseAtTime(timeStr) {
+  const now = new Date()
+  // 简单时间格式 "12:00"
+  const timeMatch = timeStr.match(/^(\d{1,2}):(\d{2})(?::(\d{2}))?$/)
+  if (timeMatch) {
+    const date = new Date(now)
+    date.setHours(parseInt(timeMatch[1]), parseInt(timeMatch[2]), parseInt(timeMatch[3] || 0), 0)
+    if (date <= now) {
+      date.setDate(date.getDate() + 1)
+    }
+    return date
+  }
+  return new Date(timeStr)
+}
+
+/**
+ * 生成唯一 ID
+ */
+function generateId() {
+  if (require('crypto').randomUUID) {
+    return require('crypto').randomUUID()
+  }
+  return `task_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`
+}
+
+class SchedulerPlugin extends Plugin {
+  constructor(config = {}) {
+    super()
+    this.name = 'scheduler'
+    this.version = '1.0.0'
+    this.description = '定时任务调度插件，支持 Cron 表达式、绝对时间、相对时间'
+    this.priority = 15
+
+    this.config = {
+      checkInterval: config.checkInterval || 1000  // 每秒检查一次
+    }
+
+    this._framework = null
+    this._agent = null
+    this._tasks = new Map()
+    this._timer = null
+    this._taskStats = {
+      total: 0,
+      running: 0,
+      completed: 0,
+      failed: 0
+    }
+  }
+
+  install(framework) {
+    this._framework = framework
+    return this
+  }
+
+  /**
+   * 获取 Agent 实例
+   */
+  _getAgent() {
+    if (this._agent) return this._agent
+    if (this._framework._mainAgent) {
+      this._agent = this._framework._mainAgent
+    } else {
+      const agents = this._framework._agents || []
+      this._agent = agents.length > 0 ? agents[agents.length - 1] : null
+    }
+    return this._agent
+  }
+
+  start(framework) {
+    // 获取 Agent 实例
+    this._agent = this._getAgent()
+
+    // 注册调度工具
+    framework.registerTool({
+      name: 'schedule_task',
+      description: '设置定时提醒任务。支持多种时间格式：相对时间(1 minute, 2 hours)、具体时间(12:00)、Cron表达式(* * * * *)',
+      inputSchema: z.object({
+        name: z.string().optional().describe('任务名称'),
+        scheduleTime: z.string().describe('执行时间。支持格式：\n- 相对时间: "1 minute", "2 hours", "1 day"\n- 具体时间: "12:00", "14:30"\n- Cron表达式: "*/5 * * * *" (每5分钟)'),
+        message: z.string().describe('提醒消息内容'),
+        repeat: z.boolean().optional().describe('是否重复执行 (默认 false)'),
+        cronExpression: z.string().optional().describe('Cron 表达式 (当 repeat 为 true 时使用)')
+      }),
+      execute: async (args) => {
+        try {
+          const { scheduleTime, message, repeat, cronExpression } = args
+          const agent = this._getAgent()
+          if (!agent) {
+            return { success: false, error: 'Agent not available' }
+          }
+
+          const taskId = `task_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`
+          let task
+
+          // 检测是否像 Cron 表达式
+          const isCron = /^[\d*,\/-\s]+$/.test(scheduleTime) && scheduleTime.split(' ').length >= 5
+
+          if (isCron || repeat) {
+            // Cron 任务
+            if (!cron) {
+              return { success: false, error: 'node-cron not installed' }
+            }
+            task = {
+              id: taskId,
+              name: args.name || 'CronTask',
+              type: 'cron',
+              cronExpression: cronExpression || scheduleTime,
+              message,
+              enabled: true,
+              createdAt: new Date(),
+              lastRun: null,
+              runCount: 0,
+              timer: null,
+              cronTask: null
+            }
+
+            // 使用 node-cron 调度
+            task.cronTask = cron.schedule(task.cronExpression, async () => {
+              await this._executeTask(task)
+            })
+          } else if (scheduleTime.includes(':')) {
+            // 具体时间
+            const runAt = parseAtTime(scheduleTime)
+            task = {
+              id: taskId,
+              name: args.name || 'Reminder',
+              type: 'once',
+              runAt,
+              message,
+              enabled: true,
+              createdAt: new Date(),
+              lastRun: null,
+              runCount: 0,
+              timer: null
+            }
+            task.timer = setTimeout(async () => {
+              await this._executeTask(task)
+            }, runAt.getTime() - Date.now())
+          } else {
+            // 相对时间
+            const delayMs = parseDelay(scheduleTime)
+            if (!delayMs) {
+              return { success: false, error: '无效的时间格式' }
+            }
+            const runAt = new Date(Date.now() + delayMs)
+            task = {
+              id: taskId,
+              name: args.name || 'Reminder',
+              type: 'once',
+              runAt,
+              message,
+              enabled: true,
+              createdAt: new Date(),
+              lastRun: null,
+              runCount: 0,
+              timer: null
+            }
+            task.timer = setTimeout(async () => {
+              await this._executeTask(task)
+            }, delayMs)
+          }
+
+          this._tasks.set(task.id, task)
+          this._taskStats.total++
+
+          return {
+            success: true,
+            taskId: task.id,
+            name: task.name,
+            scheduleTime,
+            executeAt: task.runAt ? task.runAt.toISOString() : null,
+            cronExpression: task.cronExpression,
+            message: repeat ? '定时任务已创建 (重复执行)' : '提醒已设置'
+          }
+        } catch (err) {
+          return { success: false, error: err.message }
+        }
+      }
+    })
+
+    framework.registerTool({
+      name: 'schedule_list',
+      description: '列出所有定时任务',
+      inputSchema: z.object({}),
+      execute: async () => {
+        const tasks = Array.from(this._tasks.values()).map(t => ({
+          id: t.id,
+          name: t.name,
+          type: t.type,
+          message: t.message,
+          enabled: t.enabled,
+          nextRun: t.nextRun || t.runAt,
+          runCount: t.runCount,
+          lastRun: t.lastRun,
+          cronExpression: t.cronExpression
+        }))
+
+        return {
+          success: true,
+          tasks,
+          total: tasks.length,
+          stats: { ...this._taskStats }
+        }
+      }
+    })
+
+    framework.registerTool({
+      name: 'schedule_cancel',
+      description: '取消定时任务',
+      inputSchema: z.object({
+        taskId: z.string().describe('任务 ID')
+      }),
+      execute: async (args) => {
+        const task = this._tasks.get(args.taskId)
+        if (!task) {
+          return { success: false, error: 'Task not found' }
+        }
+
+        this._cancelTask(task)
+        return { success: true, cancelled: args.taskId }
+      }
+    })
+
+    framework.registerTool({
+      name: 'cron_examples',
+      description: '获取常用 Cron 表达式示例',
+      inputSchema: z.object({}),
+      execute: async () => {
+        return {
+          examples: [
+            { expression: '* * * * *', description: '每分钟' },
+            { expression: '*/5 * * * *', description: '每5分钟' },
+            { expression: '*/15 * * * *', description: '每15分钟' },
+            { expression: '0 * * * *', description: '每小时' },
+            { expression: '0 9 * * *', description: '每天早上9点' },
+            { expression: '0 12 * * *', description: '每天中午12点' },
+            { expression: '0 18 * * *', description: '每天下午6点' },
+            { expression: '0 9 * * 1-5', description: '工作日上午9点' },
+            { expression: '0 9 * * 0,6', description: '周末上午9点' },
+            { expression: '0 9 * * 1', description: '每周一上午9点' },
+            { expression: '0 */2 * * *', description: '每2小时' }
+          ]
+        }
+      }
+    })
+
+    // 启动调度循环（用于检查一次性任务）
+    this._startScheduler()
+
+    return this
+  }
+
+  /**
+   * 启动调度循环
+   */
+  _startScheduler() {
+    if (this._timer) {
+      clearInterval(this._timer)
+    }
+
+    this._timer = setInterval(() => {
+      this._checkTasks()
+    }, this.config.checkInterval)
+  }
+
+  /**
+   * 检查任务状态（主要用于更新 stats）
+   */
+  _checkTasks() {
+    for (const [id, task] of this._tasks) {
+      if (!task.enabled) continue
+      // 统计信息更新
+    }
+  }
+
+  /**
+   * 取消任务
+   */
+  _cancelTask(task) {
+    if (task.timer) {
+      clearTimeout(task.timer)
+      task.timer = null
+    }
+    if (task.cronTask) {
+      task.cronTask.stop()
+      task.cronTask = null
+    }
+    task.enabled = false
+  }
+
+  /**
+   * 执行任务 - 唤醒 Agent 发送消息
+   */
+  async _executeTask(task) {
+    const agent = this._getAgent()
+    if (!agent) {
+      console.error('[Scheduler] Agent not available')
+      return
+    }
+
+    console.log(`[Scheduler] Executing task: ${task.name} (${task.id})`)
+    console.log(`[Scheduler] Message: ${task.message}`)
+
+    task.lastRun = new Date()
+    task.runCount++
+    this._taskStats.running++
+
+    try {
+      // 调用 Agent 的 chat 方法发送消息
+      const result = await agent.chat(task.message, {
+        isScheduledTask: true
+      })
+
+      this._taskStats.completed++
+
+      // 输出结果
+      if (result && result.message) {
+        console.log(`\n🔔 [定时提醒] ${result.message}\n`)
+      } else if (result && result.text) {
+        console.log(`\n🔔 [定时提醒] ${result.text}\n`)
+      } else {
+        console.log(`\n🔔 [定时提醒] ${task.message}\n`)
+      }
+
+      // 一次性任务执行后清理
+      if (task.type === 'once' && !task.cronTask) {
+        this._cleanupTask(task.id)
+      }
+
+      return result
+    } catch (err) {
+      this._taskStats.failed++
+      console.error(`[Scheduler] Task ${task.name} failed: ${err.message}`)
+
+      // 一次性任务失败后也清理
+      if (task.type === 'once' && !task.cronTask) {
+        this._cleanupTask(task.id)
+      }
+
+      return { error: err.message }
+    } finally {
+      this._taskStats.running--
+    }
+  }
+
+  /**
+   * 清理任务
+   */
+  _cleanupTask(taskId) {
+    const task = this._tasks.get(taskId)
+    if (task) {
+      if (task.timer) {
+        clearTimeout(task.timer)
+        task.timer = null
+      }
+      if (task.cronTask) {
+        task.cronTask.stop()
+        task.cronTask = null
+      }
+      this._tasks.delete(taskId)
+    }
+  }
+
+  /**
+   * 停止所有任务
+   */
+  stopAll() {
+    for (const task of this._tasks.values()) {
+      this._cancelTask(task)
+    }
+
+    if (this._timer) {
+      clearInterval(this._timer)
+      this._timer = null
+    }
+  }
+
+  reload(framework) {
+    this._framework = framework
+    this._agent = this._getAgent()
+    this._startScheduler()
+  }
+
+  uninstall(framework) {
+    this.stopAll()
+    this._tasks.clear()
+    this._taskStats = { total: 0, running: 0, completed: 0, failed: 0 }
+    this._framework = null
+    this._agent = null
+  }
+}
+
+module.exports = { SchedulerPlugin }