foliko 1.0.40 → 1.0.41

package/plugins/weixin-plugin.js

@@ -9,343 +9,389 @@
 
 const { Plugin } = require('../src/core/plugin-base')
 const { z } = require('zod')
-
-module.exports = function(Plugin) {
-  return class WeixinPlugin extends Plugin {
-    constructor(config = {}) {
-      super()
-      this.name = 'weixin'
-      this.version = '1.0.0'
-      this.description = '微信对话插件，使用微信网页账号进行对话'
-      this.priority = 80
-      // 默认不启用，需要在 plugins.json 中设置 enabled: true
-      this.enabled = false
-      this.path=`.agent/data`
-      this.systemPrompt='你是一个微信助手。回复内容不要使用markdown格式文本'
-
-      this.config = {
-        forceLogin: config.forceLogin || process.env.WEIXIN_FORCE_LOGIN === 'true',
-        qrcodeTerminal: config.qrcodeTerminal !== false && process.env.WEIXIN_QRCODE_TERMINAL !== 'false',
-        allowedUsers: config.allowedUsers || []
-      }
-
-      this._framework = null
-      this._bot = null
-      this._sessionPlugin = null
-      this._sessionDeleteHandler = null
-      this._sessionAgents = new Map()  // userId -> Agent
-      this._qrcodeTerminal = null
-      this._origStderrWrite = null
-      this._initialized = false
+const { WeixinBot } = require('@chnak/weixin-bot')
+
+class WeixinPlugin extends Plugin {
+  constructor(config = {}) {
+    super()
+    this.name = 'weixin'
+    this.version = '1.0.0'
+    this.description = '微信对话插件，使用微信网页账号进行对话'
+    this.priority = 80
+    // 默认不启用，需要在 plugins.json 中设置 enabled: true
+    this.enabled = false
+    this.path=`.agent/data`
+    this.systemPrompt='你是一个微信助手。回复内容不要使用markdown格式文本'
+
+    this.config = {
+      forceLogin: config.forceLogin || process.env.WEIXIN_FORCE_LOGIN === 'true',
+      qrcodeTerminal: config.qrcodeTerminal !== false && process.env.WEIXIN_QRCODE_TERMINAL !== 'false',
+      allowedUsers: config.allowedUsers || []
     }
 
-    install(framework) {
-      this._framework = framework
-      return this
-    }
+    this._framework = null
+    this._bot = null
+    this._sessionPlugin = null
+    this._sessionDeleteHandler = null
+    this._sessionAgents = new Map()  // userId -> Agent
+    this._qrcodeTerminal = null
+    this._origStderrWrite = null
+    this._initialized = false
+  }
 
-    async start(framework) {
-      // 防止重复初始化
-      if (this._initialized) return this
-      this._initialized = true
+  install(framework) {
+    this._framework = framework
+    return this
+  }
 
-      // 获取 SessionPlugin 引用
-      this._sessionPlugin = framework.pluginManager.get('session')
+  async start(framework) {
+    // 防止重复初始化
+    if (this._initialized) return this
+    this._initialized = true
 
-      // 监听 SessionPlugin 的会话删除事件
-      if (this._sessionPlugin) {
-        this._sessionDeleteHandler = (sessionId) => {
-          // 只处理 weixin 会话的删除
-          if (sessionId.startsWith('weixin_')) {
-            console.log(`[WeChat] Session deleted: ${sessionId}`)
-          }
+    // 获取 SessionPlugin 引用
+    this._sessionPlugin = framework.pluginManager.get('session')
+
+    // 监听 SessionPlugin 的会话删除事件
+    if (this._sessionPlugin) {
+      this._sessionDeleteHandler = (sessionId) => {
+        // 只处理 weixin 会话的删除
+        if (sessionId.startsWith('weixin_')) {
+          console.log(`[WeChat] Session deleted: ${sessionId}`)
         }
-        this._sessionPlugin.on('session:deleted', this._sessionDeleteHandler)
       }
+      this._sessionPlugin.on('session:deleted', this._sessionDeleteHandler)
+    }
 
-      // 异步初始化 Bot
-      this._initBotAsync().catch(err => {
-        console.error('[WeChat] Failed to initialize bot:', err.message)
+    // 监听定时提醒事件
+    if (this._framework) {
+      this._framework.on('scheduler:reminder', async (data) => {
+        console.log('[WeChat] Received scheduler reminder:', data)
+        await this._handleScheduledReminder(data)
       })
-      return this
     }
 
-    async _initBotAsync() {
+    // 异步初始化 Bot
+    this._initBotAsync().catch(err => {
+      console.error('[WeChat] Failed to initialize bot:', err.message)
+    })
+    return this
+  }
+
+  async _initBotAsync() {
 
-      let WeixinBot
+    // 如果启用终端二维码渲染
+    if (this.config.qrcodeTerminal) {
       try {
-        const module = await import('@chnak/weixin-bot')
-        WeixinBot = module.WeixinBot
+        this._qrcodeTerminal = require('qrcode-terminal')
+        this._interceptQRCode()
       } catch (err) {
-        console.warn('[WeChat] Failed to load @chnak/weixin-bot:', err.message)
-        console.warn('[WeChat] Make sure @chnak/weixin-bot is installed: npm install @chnak/weixin-bot qrcode-terminal')
-        return
+        console.warn('[WeChat] qrcode-terminal not installed:', err.message)
       }
+    }
 
-      // 如果启用终端二维码渲染
-      if (this.config.qrcodeTerminal) {
-        try {
-          this._qrcodeTerminal = require('qrcode-terminal')
-          this._interceptQRCode()
-        } catch (err) {
-          console.warn('[WeChat] qrcode-terminal not installed:', err.message)
-        }
-      }
-
-      this._bot = new WeixinBot({
-        tokenPath:`${this.path}/${this.name}.json`,
-        onError: (err) => {
-          console.error('[WeChat] Error:', err instanceof Error ? err.stack ?? err.message : String(err))
-        },
-      })
-
-      const loginOptions = { force: this.config.forceLogin }
-      console.log('[WeChat]', this.config.forceLogin ? '强制重新扫码登录...' : '正在登录（已有凭证则自动跳过扫码）...')
+    this._bot = new WeixinBot({
+      tokenPath:`${this.path}/${this.name}.json`,
+      onError: (err) => {
+        console.error('[WeChat] Error:', err instanceof Error ? err.stack ?? err.message : String(err))
+      },
+    })
+
+    const loginOptions = { force: this.config.forceLogin }
+    console.log('[WeChat]', this.config.forceLogin ? '强制重新扫码登录...' : '正在登录（已有凭证则自动跳过扫码）...')
+
+    const creds = await this._bot.login(loginOptions)
+    console.log('[WeChat] 登录成功 — Bot ID:', creds.accountId)
+    console.log('[WeChat] 关联用户:', creds.userId)
+    console.log('[WeChat] API 地址:', creds.baseUrl)
+
+    // 注册消息处理
+    this._bot.onMessage(async (msg) => {
+      await this._handleMessage(msg)
+    })
+
+    // 启动Bot
+    await this._bot.run()
+    console.log('[WeChat] 开始接收微信消息')
+  }
 
-      const creds = await this._bot.login(loginOptions)
-      console.log('[WeChat] 登录成功 — Bot ID:', creds.accountId)
-      console.log('[WeChat] 关联用户:', creds.userId)
-      console.log('[WeChat] API 地址:', creds.baseUrl)
+  /**
+   * 拦截 SDK 的 stderr 输出，渲染二维码到终端
+   */
+  _interceptQRCode() {
+    if (!this._qrcodeTerminal) return
+
+    this._origStderrWrite = process.stderr.write.bind(process.stderr)
+    process.stderr.write = ((chunk, ...args) => {
+      const str = typeof chunk === 'string' ? chunk : chunk.toString()
+      // 检测到登录 URL，渲染二维码
+      if (str.startsWith('https://') && str.includes('qrcode=')) {
+        const url = str.trim()
+        this._qrcodeTerminal.generate(url, { small: true }, (qr) => {
+          this._origStderrWrite(qr + '\n')
+        })
+      }
+      return this._origStderrWrite(chunk, ...args)
+    })
+  }
 
-      // 注册消息处理
-      this._bot.onMessage(async (msg) => {
-        await this._handleMessage(msg)
-      })
+  /**
+   * 获取主Agent
+   */
+  _getMainAgent() {
+    if (this._framework._mainAgent) {
+      return this._framework._mainAgent
+    }
+    const agents = this._framework._agents || []
+    return agents.length > 0 ? agents[agents.length - 1] : null
+  }
 
-      // 启动Bot
-      await this._bot.run()
-      console.log('[WeChat] 开始接收微信消息')
+  /**
+   * 获取或创建会话Agent
+   * 使用 SessionPlugin 统一管理会话历史
+   */
+  _getSessionAgent(userId) {
+    // 检查缓存
+    if (this._sessionAgents.has(userId)) {
+      const agent = this._sessionAgents.get(userId)
+      console.log('[WeChat] Reusing cached session agent for userId:', userId)
+      return { agent, sessionId: `weixin_${userId}` }
     }
 
-    /**
-     * 拦截 SDK 的 stderr 输出，渲染二维码到终端
-     */
-    _interceptQRCode() {
-      if (!this._qrcodeTerminal) return
-
-      this._origStderrWrite = process.stderr.write.bind(process.stderr)
-      process.stderr.write = ((chunk, ...args) => {
-        const str = typeof chunk === 'string' ? chunk : chunk.toString()
-        // 检测到登录 URL，渲染二维码
-        if (str.startsWith('https://') && str.includes('qrcode=')) {
-          const url = str.trim()
-          this._qrcodeTerminal.generate(url, { small: true }, (qr) => {
-            this._origStderrWrite(qr + '\n')
-          })
-        }
-        return this._origStderrWrite(chunk, ...args)
+    // 创建新 agent
+    const agent = this._framework.createSessionAgent(`weixin_${userId}`, {
+      systemPrompt: this.systemPrompt
+    })
+    this._sessionAgents.set(userId, agent)
+    console.log('[WeChat] Created new session agent for userId:', userId)
+
+    // 使用 SessionPlugin 管理会话历史
+    if (this._sessionPlugin) {
+      const sessionId = `weixin_${userId}`
+      this._sessionPlugin.getOrCreateSession(sessionId, {
+        metadata: { platform: 'weixin', userId }
       })
     }
 
-    /**
-     * 获取主Agent
-     */
-    _getMainAgent() {
-      if (this._framework._mainAgent) {
-        return this._framework._mainAgent
-      }
-      const agents = this._framework._agents || []
-      return agents.length > 0 ? agents[agents.length - 1] : null
+    console.log(`[WeChat] Session ready for user: ${userId}`)
+    return { agent, sessionId: `weixin_${userId}` }
+  }
+
+  /**
+   * 处理消息
+   */
+  async _handleMessage(msg) {
+    if (!msg || !msg.userId) return
+
+    const userId = msg.userId
+    // 从 SessionPlugin 获取历史消息数量
+    let messageCount = 0
+    if (this._sessionPlugin) {
+      const session = this._sessionPlugin.getSession(`weixin_${userId}`)
+      messageCount = session?.messages?.length || 0
     }
 
-    /**
-     * 获取或创建会话Agent
-     * 使用 SessionPlugin 统一管理会话历史
-     */
-    _getSessionAgent(userId) {
-      // 检查缓存
-      if (this._sessionAgents.has(userId)) {
-        const agent = this._sessionAgents.get(userId)
-        console.log('[WeChat] Reusing cached session agent for userId:', userId)
-        return { agent, sessionId: `weixin_${userId}` }
-      }
+    console.log(`[WeChat] #${messageCount + 1} | 类型: ${msg.type} | 用户: ${userId}`)
+    console.log(`[WeChat] 内容: ${msg.text}`)
 
-      // 创建新 agent
-      const agent = this._framework.createSessionAgent(`weixin_${userId}`, {
-        systemPrompt: this.systemPrompt
-      })
-      this._sessionAgents.set(userId, agent)
-      console.log('[WeChat] Created new session agent for userId:', userId)
+    // 非文本消息暂不处理
+    if (msg.type !== 'text' || !msg.text) {
+      console.log('[WeChat] Unsupported message type or no text')
+      return
+    }
 
-      // 使用 SessionPlugin 管理会话历史
-      if (this._sessionPlugin) {
-        const sessionId = `weixin_${userId}`
-        this._sessionPlugin.getOrCreateSession(sessionId, {
-          metadata: { platform: 'weixin', userId }
-        })
-      }
+    const text = msg.text.trim()
+    await this._processChat(userId, text, msg)
+  }
 
-      console.log(`[WeChat] Session ready for user: ${userId}`)
-      return { agent, sessionId: `weixin_${userId}` }
+  /**
+   * 处理对话
+   */
+  async _processChat(userId, text, originalMsg) {
+    const sessionInfo = this._getSessionAgent(userId)
+    if (!sessionInfo) {
+      console.error('[WeChat] No session agent available')
+      return
     }
 
-    /**
-     * 处理消息
-     */
-    async _handleMessage(msg) {
-      if (!msg || !msg.userId) return
+    const { agent, sessionId } = sessionInfo
 
-      const userId = msg.userId
-      // 从 SessionPlugin 获取历史消息数量
-      let messageCount = 0
-      if (this._sessionPlugin) {
-        const session = this._sessionPlugin.getSession(`weixin_${userId}`)
-        messageCount = session?.messages?.length || 0
-      }
+    // 使用 SessionPlugin 添加用户消息到历史
+    if (this._sessionPlugin) {
+      this._sessionPlugin.addMessage(sessionId, { role: 'user', content: text })
+    }
 
-      console.log(`[WeChat] #${messageCount + 1} | 类型: ${msg.type} | 用户: ${userId}`)
-      console.log(`[WeChat] 内容: ${msg.text}`)
+    // 发送正在输入状态
+    try {
+      await this._bot.sendTyping(userId)
+    } catch { /* typing 失败不影响回复 */ }
 
-      // 非文本消息暂不处理
-      if (msg.type !== 'text' || !msg.text) {
-        console.log('[WeChat] Unsupported message type or no text')
-        return
-      }
+    await new Promise((resolve) => setTimeout(resolve, 1000))
 
-      const text = msg.text.trim()
-      await this._processChat(userId, text, msg)
-    }
+    try {
+      let fullResponse = ''
 
-    /**
-     * 处理对话
-     */
-    async _processChat(userId, text, originalMsg) {
-      const sessionInfo = this._getSessionAgent(userId)
-      if (!sessionInfo) {
-        console.error('[WeChat] No session agent available')
-        return
+      // 使用流式响应
+      for await (const chunk of agent.chatStream(text, {
+        sessionId: sessionId
+      })) {
+        if (chunk.type === 'text' && chunk.text) {
+          fullResponse += chunk.text
+        }
       }
 
-      const { agent, sessionId } = sessionInfo
-
-      // 使用 SessionPlugin 添加用户消息到历史
+      // 保存助手回复到历史（使用 SessionPlugin）
       if (this._sessionPlugin) {
-        this._sessionPlugin.addMessage(sessionId, { role: 'user', content: text })
+        this._sessionPlugin.addMessage(sessionId, { role: 'assistant', content: fullResponse })
       }
 
-      // 发送正在输入状态
-      try {
-        await this._bot.sendTyping(userId)
-      } catch { /* typing 失败不影响回复 */ }
+      // 发送回复
+      if (fullResponse) {
+        await this._bot.reply(originalMsg, fullResponse)
+        console.log(`[WeChat] 回复成功 (${fullResponse.length} 字符)`)
+      } else {
+        await this._bot.reply(originalMsg, '抱歉，我没有收到有效的回复。')
+      }
 
-      await new Promise((resolve) => setTimeout(resolve, 1000))
+    } catch (err) {
+      console.error('[WeChat] Chat error:', err)
+      await this._bot.reply(originalMsg, `发生错误：${err.message}`)
+    }
+  }
 
-      try {
-        let fullResponse = ''
-
-        // 使用流式响应
-        for await (const chunk of agent.chatStream(text, {
-          sessionId: sessionId
-        })) {
-          if (chunk.type === 'text' && chunk.text) {
-            fullResponse += chunk.text
-          }
-        }
+  /**
+   * 权限检查
+   */
+  _checkPermission(userId) {
+    if (!this.config.allowedUsers || this.config.allowedUsers.length === 0) {
+      return true
+    }
+    return this.config.allowedUsers.includes(userId)
+  }
 
-        // 保存助手回复到历史（使用 SessionPlugin）
-        if (this._sessionPlugin) {
-          this._sessionPlugin.addMessage(sessionId, { role: 'assistant', content: fullResponse })
-        }
+  /**
+   * 清除会话
+   */
+  _clearSession(userId) {
+    // 清除 SessionPlugin 中的会话（会触发 session:deleted 事件）
+    if (this._sessionPlugin) {
+      const sessionId = `weixin_${userId}`
+      this._sessionPlugin.deleteSession(sessionId)
+    }
+  }
 
-        // 发送回复
-        if (fullResponse) {
-          await this._bot.reply(originalMsg, fullResponse)
-          console.log(`[WeChat] 回复成功 (${fullResponse.length} 字符)`)
-        } else {
-          await this._bot.reply(originalMsg, '抱歉，我没有收到有效的回复。')
-        }
+  /**
+   * 处理定时提醒
+   */
+  async _handleScheduledReminder(data) {
+    const { taskName, message, sessionId } = data
 
-      } catch (err) {
-        console.error('[WeChat] Chat error:', err)
-        await this._bot.reply(originalMsg, `发生错误：${err.message}`)
-      }
+    if (!this._bot) {
+      console.warn('[WeChat] Bot not ready, cannot send reminder')
+      return
     }
 
-    /**
-     * 权限检查
-     */
-    _checkPermission(userId) {
-      if (!this.config.allowedUsers || this.config.allowedUsers.length === 0) {
-        return true
-      }
-      return this.config.allowedUsers.includes(userId)
-    }
+    // 构建提醒消息
+    const reminderText = `🔔 [${taskName}]\n\n${message}`
 
-    /**
-     * 清除会话
-     */
-    _clearSession(userId) {
-      // 清除 SessionPlugin 中的会话（会触发 session:deleted 事件）
-      if (this._sessionPlugin) {
-        const sessionId = `weixin_${userId}`
-        this._sessionPlugin.deleteSession(sessionId)
+    // 如果有 sessionId 是 weixin 类型的，发送到对应用户
+    if (sessionId && sessionId.startsWith('weixin_')) {
+      const userId = sessionId.replace('weixin_', '')
+      try {
+        await this._bot.sendText(userId, reminderText)
+        console.log(`[WeChat] Reminder sent to user ${userId}`)
+      } catch (err) {
+        console.error(`[WeChat] Failed to send reminder:`, err.message)
       }
+      return
     }
 
-    /**
-     * 获取插件状态
-     */
-    getStatus() {
-      // 从 SessionPlugin 获取 WeChat 相关会话
-      let sessions = []
-      if (this._sessionPlugin) {
-        const allSessions = this._sessionPlugin.listSessions()
-        sessions = allSessions
-          .filter(s => s.id.startsWith('weixin_'))
-          .map(s => ({
-            userId: s.id.replace('weixin_', ''),
-            historyLength: s.messageCount,
-            lastActive: s.lastActive
-          }))
-      }
+    // 其他情况（包括 null 或其他 sessionId），发送到最近的 WeChat 会话
+    if (this._sessionPlugin) {
+      const allSessions = this._sessionPlugin.listSessions()
+      const weixinSessions = allSessions
+        .filter(s => s.id.startsWith('weixin_'))
+        .sort((a, b) => new Date(b.lastActive).getTime() - new Date(a.lastActive).getTime())
 
-      return {
-        connected: !!this._bot,
-        sessionCount: sessions.length,
-        sessions,
-        config: {
-          forceLogin: this.config.forceLogin,
-          qrcodeTerminal: this.config.qrcodeTerminal
+      if (weixinSessions.length > 0) {
+        const userId = weixinSessions[0].id.replace('weixin_', '')
+        try {
+          await this._bot.sendText(userId, reminderText)
+          console.log(`[WeChat] Reminder sent to recent user ${userId}`)
+        } catch (err) {
+          console.error(`[WeChat] Failed to send reminder:`, err.message)
         }
+      } else {
+        console.warn('[WeChat] No WeChat session found to send reminder')
       }
     }
+  }
+
+  /**
+   * 获取插件状态
+   */
+  getStatus() {
+    // 从 SessionPlugin 获取 WeChat 相关会话
+    let sessions = []
+    if (this._sessionPlugin) {
+      const allSessions = this._sessionPlugin.listSessions()
+      sessions = allSessions
+        .filter(s => s.id.startsWith('weixin_'))
+        .map(s => ({
+          userId: s.id.replace('weixin_', ''),
+          historyLength: s.messageCount,
+          lastActive: s.lastActive
+        }))
+    }
 
-    /**
-     * 停止 Bot
-     */
-    stopBot() {
-      if (this._bot) {
-        this._bot.stop()
-        this._bot = null
-        console.log('[WeChat] Bot stopped')
+    return {
+      connected: !!this._bot,
+      sessionCount: sessions.length,
+      sessions,
+      config: {
+        forceLogin: this.config.forceLogin,
+        qrcodeTerminal: this.config.qrcodeTerminal
       }
     }
+  }
 
-    reload(framework) {
-      this._framework = framework
-      this._initialized = false
-      this.stopBot()
-      this.start(framework)
+  /**
+   * 停止 Bot
+   */
+  stopBot() {
+    if (this._bot) {
+      this._bot.stop()
+      this._bot = null
+      console.log('[WeChat] Bot stopped')
     }
+  }
 
-    uninstall(framework) {
-      // 销毁所有 session agents
-      for (const agent of this._sessionAgents.values()) {
-        agent.destroy()
-      }
-      this._sessionAgents.clear()
+  reload(framework) {
+    this._framework = framework
+    this._initialized = false
+    this.stopBot()
+    this.start(framework)
+  }
 
-      this.stopBot()
-      if (this._sessionPlugin && this._sessionDeleteHandler) {
-        this._sessionPlugin.off('session:deleted', this._sessionDeleteHandler)
-        this._sessionDeleteHandler = null
-      }
-      this._sessionPlugin = null
-      this._framework = null
-      // 恢复 stderr
-      if (this._origStderrWrite) {
-        process.stderr.write = this._origStderrWrite
-      }
+  uninstall(framework) {
+    // 销毁所有 session agents
+    for (const agent of this._sessionAgents.values()) {
+      agent.destroy()
+    }
+    this._sessionAgents.clear()
+
+    this.stopBot()
+    if (this._sessionPlugin && this._sessionDeleteHandler) {
+      this._sessionPlugin.off('session:deleted', this._sessionDeleteHandler)
+      this._sessionDeleteHandler = null
+    }
+    this._sessionPlugin = null
+    this._framework = null
+    // 恢复 stderr
+    if (this._origStderrWrite) {
+      process.stderr.write = this._origStderrWrite
     }
   }
 }
+
+module.exports = { WeixinPlugin }