foliko 1.0.4 → 1.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "foliko",
-  "version": "1.0.4",
+  "version": "1.0.5",
   "description": "简约的插件化 Agent 框架",
   "main": "src/index.js",
   "bin": {

package/plugins/telegram-plugin.js

@@ -12,6 +12,13 @@
 const { Plugin } = require('../src/core/plugin-base')
 const { z } = require('zod')
 
+// 转义 MarkdownV2 特殊字符
+function escapeMarkdown(text) {
+  if (!text) return ''
+  // MarkdownV2 需要转义的字符
+  return text.replace(/([\_*\[\]()~`>#+\-=|{}.!])/g, '\\$1')
+}
+
 module.exports = function(Plugin) {
   return class TelegramPlugin extends Plugin {
     constructor(config = {}) {
@@ -155,9 +162,24 @@ module.exports = function(Plugin) {
      * 处理消息
      */
     async _handleMessage(msg) {
-      if (!msg || !msg.chat || !msg.text) return
+      if (!msg || !msg.chat) return
 
       const chatId = msg.chat.id.toString()
+
+      // 处理图片消息
+      if (msg.photo) {
+        await this._handlePhoto(msg)
+        return
+      }
+
+      // 处理文档消息
+      if (msg.document) {
+        await this._handleDocument(msg)
+        return
+      }
+
+      if (!msg.text) return
+
       const text = msg.text.trim()
 
       // 命令处理
@@ -279,11 +301,13 @@ module.exports = function(Plugin) {
         // 保存助手回复到历史
         session.history.push({ role: 'assistant', content: fullResponse })
 
-        // 发送最终回复
-        await this._bot.editMessageText(fullResponse || '抱歉，我没有收到有效的回复。', {
+        // 发送最终回复（转义 Markdown 特殊字符）
+        const safeResponse = fullResponse || '抱歉，我没有收到有效的回复。'
+        await this._bot.editMessageText(safeResponse, {
           chat_id: chatId,
           message_id: thinkingMsg.message_id,
-                  })
+          parse_mode: 'MarkdownV2'
+        })
 
       } catch (err) {
         console.error('[Telegram] Chat error:', err)
@@ -318,6 +342,105 @@ module.exports = function(Plugin) {
       return this.config.allowedChats.includes(chatId)
     }
 
+    /**
+     * 处理图片消息
+     */
+    async _handlePhoto(msg) {
+      const chatId = msg.chat.id.toString()
+      const caption = msg.caption?.trim() || ''
+
+      // 权限检查
+      if (!this._checkPermission(chatId)) {
+        await this._sendMessage(chatId, '抱歉，您没有权限使用此 Bot。', msg.message_id)
+        return
+      }
+
+      // 获取最大尺寸的图片
+      const photo = msg.photo[msg.photo.length - 1]
+      console.log(`[Telegram] Photo received, file_id: ${photo.file_id}`)
+
+      try {
+        // 下载图片
+        const filePath = await this._downloadFile(photo.file_id, '.jpg', 'telegram_images')
+        console.log(`[Telegram] Photo saved: ${filePath}`)
+
+        const savedMsg = `收到图片${caption ? ': ' + caption : ''}\n已保存至: ${filePath}`
+        await this._sendMessage(chatId, savedMsg, msg.message_id)
+
+        // TODO: 调用 AI 视觉能力分析图片
+        // await this._analyzeImage(chatId, filePath, caption)
+      } catch (err) {
+        console.error('[Telegram] Failed to save photo:', err.message)
+        await this._sendMessage(chatId, '图片保存失败: ' + err.message, msg.message_id)
+      }
+    }
+
+    /**
+     * 处理文档消息
+     */
+    async _handleDocument(msg) {
+      const chatId = msg.chat.id.toString()
+      const fileName = msg.document.file_name || '未命名文件'
+      const caption = msg.caption?.trim() || ''
+
+      // 权限检查
+      if (!this._checkPermission(chatId)) {
+        await this._sendMessage(chatId, '抱歉，您没有权限使用此 Bot。', msg.message_id)
+        return
+      }
+
+      console.log(`[Telegram] Document received: ${fileName}, file_id: ${msg.document.file_id}`)
+
+      try {
+        // 根据文件扩展名确定保存目录和扩展名
+        const ext = fileName.includes('.') ? fileName.split('.').pop() : 'bin'
+        const filePath = await this._downloadFile(msg.document.file_id, ext, 'telegram_documents')
+        console.log(`[Telegram] Document saved: ${filePath}`)
+
+        const savedMsg = `收到文件: ${fileName}\n已保存至: ${filePath}${caption ? '\n\n说明: ' + caption : ''}`
+        await this._sendMessage(chatId, savedMsg, msg.message_id)
+      } catch (err) {
+        console.error('[Telegram] Failed to save document:', err.message)
+        await this._sendMessage(chatId, '文件保存失败: ' + err.message, msg.message_id)
+      }
+    }
+
+    /**
+     * 下载文件
+     * @param {string} fileId - Telegram 文件 ID
+     * @param {string} ext - 文件扩展名
+     * @param {string} subDir - 子目录名
+     * @returns {Promise<string>} 保存的文件路径
+     */
+    async _downloadFile(fileId, ext, subDir) {
+      return new Promise((resolve, reject) => {
+        const path = require('path')
+        const fs = require('fs')
+
+        // 创建保存目录
+        const saveDir = path.join(process.cwd(), '.agent', 'data', subDir)
+        if (!fs.existsSync(saveDir)) {
+          fs.mkdirSync(saveDir, { recursive: true })
+        }
+
+        // 生成文件名
+        const fileName = `${Date.now()}_${Math.random().toString(36).substring(7)}.${ext}`
+        const filePath = path.join(saveDir, fileName)
+
+        // 下载文件
+        this._bot.downloadFile(fileId, saveDir)
+          .then((savedPath) => {
+            // 重命名为期望的文件名
+            const finalPath = path.join(saveDir, fileName)
+            if (savedPath !== finalPath) {
+              fs.renameSync(savedPath, finalPath)
+            }
+            resolve(finalPath)
+          })
+          .catch(reject)
+      })
+    }
+
     /**
      * 清除会话
      */