foliko 1.1.13 → 1.1.15

package/plugins/weixin-plugin.js

@@ -11,9 +11,10 @@ const { Plugin } = require('../src/core/plugin-base')
 const { logger } = require('../src/utils/logger')
 const log = logger.child('WeChat')
 const {cleanResponse} =require('../src/utils')
+const {FileDownloader} = require('../src/utils/download')
 const removeMarkdown = require('remove-markdown');
 const { z } = require('zod')
-const { WeixinBot } = require('@chnak/weixin-bot')
+const { WeixinBot} = require('@chnak/weixin-bot')
 const fs = require('fs')
 const path = require('path')
 
@@ -54,6 +55,13 @@ class WeixinPlugin extends Plugin {
     this._qrcodeTerminal = null
     this._origStderrWrite = null
     this._initialized = false
+    const saveDir = path.join(process.cwd(), '.agent', 'data', this.name)
+    this.downloader=new FileDownloader({
+      retries: 3,
+      baseDir:saveDir
+    })
+    this.agent=null
+    this.sessionId=null
   }
 
   install(framework) {
@@ -246,7 +254,6 @@ class WeixinPlugin extends Plugin {
     // 防止重复初始化
     if (this._initialized) return this
     this._initialized = true
-
     // 获取 SessionPlugin 引用
     this._sessionPlugin = framework.pluginManager.get('session')
 
@@ -318,7 +325,9 @@ class WeixinPlugin extends Plugin {
     // 记录机器人自己的 userId（用于发送媒体）
     this._myUserId = creds.userId
     // log.info(' 已记录机器人 userId:', this._myUserId)
-
+    const {agent,sessionId} = this._getSessionAgent(creds.userId)
+    const sessionScope = agent.createSessionScope(sessionId);
+    
     // 注册消息处理
     this._bot.onMessage(async (msg) => {
       await this._handleMessage(msg)
@@ -360,12 +369,19 @@ class WeixinPlugin extends Plugin {
     return agents.length > 0 ? agents[agents.length - 1] : null
   }
 
+  _on_event_message(){
+    const {sessionId}=this.ge
+    const sessionScope = this.agent.createSessionScope(sessionId);
+    sessionScope.on('stream:chunk', ({ chunk }) => {})
+  }
+
   /**
    * 获取或创建会话Agent
    * 使用 SessionPlugin 统一管理会话历史
    */
   _getSessionAgent(userId) {
     const sessionId=`weixin_${userId}`
+    this.sessionId=sessionId
     // 检查缓存
     if (this._sessionAgents.has(userId)) {
       const agent = this._sessionAgents.get(userId)
@@ -390,13 +406,6 @@ class WeixinPlugin extends Plugin {
         metadata: { platform: 'weixin', userId }
       })
     }
-    const sessionScope = agent.createSessionScope(sessionId);
-    // sessionScope.on('stream:chunk', ({chunk}) => {
-    //     if (chunk.type === 'text') {
-    //       process.stdout.write(chunk.text);
-    //     }
-    // });
-    // log.info(` Session ready for user: ${userId}`)
     return { agent, sessionId }
   }
 
@@ -426,23 +435,26 @@ class WeixinPlugin extends Plugin {
         break
 
       case 'image':
-        // log.info(` 收到图片消息`)
-        await this._processMediaChat(userId, '[用户发送了图片]', msg)
-        break
-
       case 'file':
-        // log.info(` 收到文件消息: ${msg.fileName || 'unknown'}`)
-        await this._processMediaChat(userId, `[用户发送了文件: ${msg.fileName || 'unknown'}]`, msg)
-        break
-
       case 'video':
-        // log.info(` 收到视频消息`)
-        await this._processMediaChat(userId, '[用户发送了视频]', msg)
-        break
-
       case 'voice':
         // log.info(` 收到语音消息`)
-        await this._processMediaChat(userId, '[用户发送了语音消息]', msg)
+        const type_dict={
+          image:"图片",
+          file:"文件",
+          video:"视频",
+          voice:"音频"
+        }
+        const files=await this._save_file(msg)
+ 
+        await this._processChat(userId, `
+          [用户发送了${type_dict[msg.type]}消息]\n
+          文件列表：\n
+          ${JSON.stringify(files,null,2)} \n
+        
+          *重要：* 不要试图读取文件内容！
+        `
+        , msg)
         break
 
       default:
@@ -452,49 +464,30 @@ class WeixinPlugin extends Plugin {
         }
     }
   }
-
-  /**
-   * 处理媒体消息（图片/文件/视频/语音）
-   */
-  async _processMediaChat(userId, text, originalMsg) {
-    const sessionInfo = this._getSessionAgent(userId)
-    if (!sessionInfo) {
-      log.error(' No session agent available')
-      return
-    }
-
-    const { agent, sessionId } = sessionInfo
-
-    // 发送正在输入状态
-    try {
-      await this._bot.sendTyping(userId)
-    } catch { /* typing 失败不影响回复 */ }
-
-    await new Promise((resolve) => setTimeout(resolve, 1000))
-
-    try {
-      // 使用 continue 获取流式响应
-      const result = await agent.sendMessage(text, {
-        sessionId: sessionId,
-        priority: 1,
-      });
-      const errors=result.chunks.filter(a=>a.type==='error').map(a=>a.error)
-      if(errors.length){
-        throw new Error(errors.join('\n'))
-      }
-      if(result.content){
-        await this._sendMessageBatch(originalMsg, userId, result.content, true)
-      }else{
-        const msg=`继续进行下一步`
-        await this._sendMessageBatch(originalMsg, userId, msg, true)
-        agent.sendMessage(msg,{sessionId,priority: 1})
+  
+  async _save_file(msg){
+    const files=[]
+    try{
+      const file_list=msg?.raw?.item_list||[]
+      for(const item of file_list){
+        const file_item=item[`${msg.type}_item`]
+        const url=file_item?.media?.full_url||null
+        const aesKey=file_item?.media?.aes_key||null
+        if(url){
+          const file=await this.downloader.fromWeXin(url,aesKey)
+          files.push(file.path)
+        }
+        
       }
-    } catch (err) {
-      log.error(' Media chat error:', err)
-      await this._sendMessageBatch(originalMsg, userId, `发生错误：${err.message}`, true)
+    }catch(err){
+      console.log(err)
+    }
+    finally{
+      return files
     }
   }
 
+
   /**
    * 处理对话
    */
@@ -513,27 +506,31 @@ class WeixinPlugin extends Plugin {
     } catch { /* typing 失败不影响回复 */ }
 
     await new Promise((resolve) => setTimeout(resolve, 1000))
-
+    const sessionScope = agent.createSessionScope(sessionId);
     try {
-      const result = await agent.sendMessage(text, {
-        sessionId: sessionId,
-        priority: 1,
-      });
-      const errors=result.chunks.filter(a=>a.type==='error').map(a=>a.error)
-      if(errors.length){
-        throw new Error(errors.join('\n'))
-      }
-      if(result.content){
-        await this._sendMessageBatch(originalMsg, userId, result.content, true)
-      }else{
-        const msg=`继续进行下一步`
-        await this._sendMessageBatch(originalMsg, userId, msg, true)
-        agent.sendMessage(msg,{sessionId,priority: 1})
-      }
+      // 使用事件方式实时处理流式响应
+
+     
+      sessionScope.on('message:complete', async ({content}) => {
+        if(content){
+          await this._sendMessageBatch(originalMsg, userId, content, true)
+        }else{
+          const msg = `继续进行下一步`
+          await this._sendMessageBatch(originalMsg, userId, msg, true)
+          agent.sendMessage(msg, { sessionId, priority: 1 })
+        }
+      })
+      sessionScope.on('message:error', async ({error}) => {
+        await this._sendMessageBatch(originalMsg, userId, error, true)
+      })
+
+      const result=await agent.sendMessage(text, { sessionId, priority: 1 })
 
     } catch (err) {
       log.error(' Chat error:', err.message)
       await this._sendMessageBatch(originalMsg, userId, `发生错误：${err.message}`, true)
+    }finally{
+      sessionScope.removeAllListeners()
     }
   }