@lmcl/ailo-mcp-feishu 0.0.6 → 0.0.8

package/dist/feishu-handler.js

@@ -1,4 +1,5 @@
 import * as fs from "node:fs";
+import * as os from "node:os";
 import * as path from "node:path";
 import * as Lark from "@larksuiteoapi/node-sdk";
 import { getWorkDir } from "@lmcl/ailo-mcp-sdk";
@@ -490,73 +491,43 @@ export class FeishuHandler {
         return "p2p";
     }
     /**
-     * 下载消息资源到本地缓存，返回 path。失败时返回 null（调用方传 ref）。
-     * 路径按年/月组织：blobs/YYYY/MM/
-     * 文件名：时间戳_原始file_name（必须提供 file_name，否则不保存，走 ref）
+     * 保存附件到本地，返回绝对路径。
+     * 收到飞书消息时，先调用此函数保存图片/视频/文件，再传给 LLM。
+     * 失败返回 null，不传 ref。
      */
-    async downloadToCache(messageId, fileKey, resourceType, aidoType, fileName) {
-        const trimmed = fileName?.trim();
-        if (!trimmed)
-            return null;
-        const workDir = getWorkDir();
-        if (!workDir)
-            return null;
+    async saveResourceToLocal(messageId, fileKey, resourceType, aidoType, fileName) {
+        const workDir = getWorkDir() ?? path.join(os.tmpdir(), "ailo-mcp-feishu-blobs");
         const now = new Date();
-        const year = String(now.getFullYear());
-        const month = String(now.getMonth() + 1).padStart(2, "0");
-        const cacheDir = path.join(workDir, "blobs", year, month);
+        const cacheDir = path.join(workDir, "blobs", String(now.getFullYear()), String(now.getMonth() + 1).padStart(2, "0"));
+        await fs.promises.mkdir(cacheDir, { recursive: true });
+        const sanitized = fileName.replace(/[/\\?*:|"<>]/g, "_").slice(0, 200);
+        const outPath = path.join(cacheDir, `${Date.now()}_${sanitized}`);
+        let buffer = null;
         try {
-            await fs.promises.mkdir(cacheDir, { recursive: true });
-        }
-        catch {
-            return null;
-        }
-        const sanitized = trimmed.replace(/[/\\?*:|"<>]/g, "_").slice(0, 200);
-        const filename = `${Date.now()}_${sanitized}`;
-        const outPath = path.join(cacheDir, filename);
-        const tryMessageResource = async () => {
             const res = await this.client.im.v1.messageResource.get({
                 params: { type: resourceType },
                 path: { message_id: messageId, file_key: fileKey },
             });
-            if (!res?.getReadableStream)
-                return null;
-            return streamToBuffer(res.getReadableStream());
-        };
-        try {
-            let buffer = await tryMessageResource();
-            if (!buffer && resourceType === "image" && aidoType === "image") {
+            if (res?.getReadableStream)
+                buffer = await streamToBuffer(res.getReadableStream());
+        }
+        catch {
+            // messageResource 可能失败，图片可尝试 image.get
+        }
+        if (!buffer && resourceType === "image" && aidoType === "image") {
+            try {
                 const res = await this.client.im.v1.image.get({ path: { image_key: fileKey } });
-                if (res?.getReadableStream) {
+                if (res?.getReadableStream)
                     buffer = await streamToBuffer(res.getReadableStream());
-                }
             }
-            if (!buffer)
-                return null;
-            await fs.promises.writeFile(outPath, buffer);
-            return outPath;
+            catch {
+                // ignore
+            }
         }
-        catch (err) {
-            console.warn(`[feishu] downloadToCache failed (${resourceType}):`, err);
+        if (!buffer)
             return null;
-        }
-    }
-    /**
-     * 获取消息资源并转为 attachment。有 file_name 时下载到 cache 传 path；无则传 ref（不保存）。
-     */
-    async fetchMessageResourceAttachment(messageId, fileKey, resourceType, aidoType, fileName) {
-        if (!fileName?.trim()) {
-            return { type: aidoType, ref: `im:${messageId}:${fileKey}:${aidoType}`, channel: "feishu" };
-        }
-        const pathOrNull = await this.downloadToCache(messageId, fileKey, resourceType, aidoType, fileName);
-        if (pathOrNull) {
-            return {
-                type: aidoType,
-                path: pathOrNull,
-                name: path.basename(pathOrNull),
-            };
-        }
-        return { type: aidoType, ref: `im:${messageId}:${fileKey}:${aidoType}`, channel: "feishu" };
+        await fs.promises.writeFile(outPath, buffer);
+        return path.resolve(outPath);
     }
     /**
      * 使用飞书 WebSocket 长连接接收事件，无需配置回调地址。
@@ -654,18 +625,11 @@ export class FeishuHandler {
                 else if (messageType === "post") {
                     text = extractTextFromPostContent(rawContent);
                     const postImageKeys = [...new Set(extractImageKeysFromPostContent(rawContent))];
-                    if (postImageKeys.length > 0) {
-                        console.error(`[feishu] post: fetching ${postImageKeys.length} image(s) from message ${messageId}`);
-                        for (const imageKey of postImageKeys) {
-                            const attach = await this.fetchMessageResourceAttachment(messageId, imageKey, "image", "image");
-                            if (attach) {
-                                attachments.push(attach);
-                                console.error(`[feishu] post: fetched image ok (${attach.path ? "path" : "ref"})`);
-                            }
-                            else {
-                                console.warn(`[feishu] post: failed to fetch image ${imageKey}`);
-                            }
-                        }
+                    for (const imageKey of postImageKeys) {
+                        const fileName = `image_${imageKey.slice(-12)}.png`;
+                        const absPath = await this.saveResourceToLocal(messageId, imageKey, "image", "image", fileName);
+                        if (absPath)
+                            attachments.push({ type: "image", path: absPath, name: path.basename(absPath) });
                     }
                 }
                 else {
@@ -675,20 +639,18 @@ export class FeishuHandler {
                             const content = JSON.parse(rawContent || "{}");
                             const fileKey = content[mediaConfig.contentKey];
                             if (fileKey) {
-                                const fileName = content["file_name"] ?? content["fileName"];
-                                const attach = await this.fetchMessageResourceAttachment(messageId, fileKey, mediaConfig.resourceType, mediaConfig.aidoType, fileName);
-                                if (attach)
-                                    attachments.push(attach);
+                                const fileName = content["file_name"] ?? content["fileName"] ?? `${mediaConfig.aidoType}_${fileKey.slice(-12)}.${mediaConfig.aidoType === "image" ? "png" : mediaConfig.aidoType === "video" ? "mp4" : mediaConfig.aidoType === "audio" ? "mp3" : "bin"}`;
+                                const absPath = await this.saveResourceToLocal(messageId, fileKey, mediaConfig.resourceType, mediaConfig.aidoType, fileName);
+                                if (absPath)
+                                    attachments.push({ type: mediaConfig.aidoType, path: absPath, name: path.basename(absPath) });
                                 else
                                     text = `[无法获取${mediaConfig.aidoType}资源]`;
                             }
                             else {
-                                console.warn(`[feishu] ${messageType} message missing ${mediaConfig.contentKey} in content`);
                                 text = "[无法解析的媒体消息]";
                             }
                         }
-                        catch (err) {
-                            console.warn(`[feishu] failed to parse ${messageType} message content:`, err);
+                        catch {
                             text = "[无法解析的媒体消息]";
                         }
                     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lmcl/ailo-mcp-feishu",
-  "version": "0.0.6",
+  "version": "0.0.8",
   "description": "Ailo 飞书/Lark 通道 MCP",
   "type": "module",
   "main": "dist/index.js",

package/src/feishu-handler.ts

@@ -1,4 +1,5 @@
 import * as fs from "node:fs";
+import * as os from "node:os";
 import * as path from "node:path";
 import * as Lark from "@larksuiteoapi/node-sdk";
 import { getWorkDir, type BridgeHandler, type BridgeMessage, type ContextTag } from "@lmcl/ailo-mcp-sdk";
@@ -579,88 +580,45 @@ export class FeishuHandler implements BridgeHandler {
   }
 
   /**
-   * 下载消息资源到本地缓存，返回 path。失败时返回 null（调用方传 ref）。
-   * 路径按年/月组织：blobs/YYYY/MM/
-   * 文件名：时间戳_原始file_name（必须提供 file_name，否则不保存，走 ref）
+   * 保存附件到本地，返回绝对路径。
+   * 收到飞书消息时，先调用此函数保存图片/视频/文件，再传给 LLM。
+   * 失败返回 null，不传 ref。
    */
-  private async downloadToCache(
+  private async saveResourceToLocal(
     messageId: string,
     fileKey: string,
     resourceType: string,
     aidoType: "image" | "audio" | "video" | "file",
     fileName: string
   ): Promise<string | null> {
-    const trimmed = fileName?.trim();
-    if (!trimmed) return null;
-    const workDir = getWorkDir();
-    if (!workDir) return null;
+    const workDir = getWorkDir() ?? path.join(os.tmpdir(), "ailo-mcp-feishu-blobs");
     const now = new Date();
-    const year = String(now.getFullYear());
-    const month = String(now.getMonth() + 1).padStart(2, "0");
-    const cacheDir = path.join(workDir, "blobs", year, month);
-    try {
-      await fs.promises.mkdir(cacheDir, { recursive: true });
-    } catch {
-      return null;
-    }
-    const sanitized = trimmed.replace(/[/\\?*:|"<>]/g, "_").slice(0, 200);
-    const filename = `${Date.now()}_${sanitized}`;
-    const outPath = path.join(cacheDir, filename);
+    const cacheDir = path.join(workDir, "blobs", String(now.getFullYear()), String(now.getMonth() + 1).padStart(2, "0"));
+    await fs.promises.mkdir(cacheDir, { recursive: true });
+    const sanitized = fileName.replace(/[/\\?*:|"<>]/g, "_").slice(0, 200);
+    const outPath = path.join(cacheDir, `${Date.now()}_${sanitized}`);
 
-    const tryMessageResource = async (): Promise<Buffer | null> => {
+    let buffer: Buffer | null = null;
+    try {
       const res = await this.client.im.v1.messageResource.get({
         params: { type: resourceType },
         path: { message_id: messageId, file_key: fileKey },
       });
-      if (!res?.getReadableStream) return null;
-      return streamToBuffer(res.getReadableStream());
-    };
-
-    try {
-      let buffer: Buffer | null = await tryMessageResource();
-      if (!buffer && resourceType === "image" && aidoType === "image") {
+      if (res?.getReadableStream) buffer = await streamToBuffer(res.getReadableStream());
+    } catch {
+      // messageResource 可能失败，图片可尝试 image.get
+    }
+    if (!buffer && resourceType === "image" && aidoType === "image") {
+      try {
         const res = await this.client.im.v1.image.get({ path: { image_key: fileKey } });
-        if (res?.getReadableStream) {
-          buffer = await streamToBuffer(res.getReadableStream());
-        }
+        if (res?.getReadableStream) buffer = await streamToBuffer(res.getReadableStream());
+      } catch {
+        // ignore
       }
-      if (!buffer) return null;
-      await fs.promises.writeFile(outPath, buffer);
-      return outPath;
-    } catch (err) {
-      console.warn(`[feishu] downloadToCache failed (${resourceType}):`, err);
-      return null;
     }
-  }
-
-  /**
-   * 获取消息资源并转为 attachment。有 file_name 时下载到 cache 传 path；无则传 ref（不保存）。
-   */
-  private async fetchMessageResourceAttachment(
-    messageId: string,
-    fileKey: string,
-    resourceType: string,
-    aidoType: "image" | "audio" | "video" | "file",
-    fileName?: string
-  ): Promise<FeishuAttachment | null> {
-    if (!fileName?.trim()) {
-      return { type: aidoType, ref: `im:${messageId}:${fileKey}:${aidoType}`, channel: "feishu" };
-    }
-    const pathOrNull = await this.downloadToCache(
-      messageId,
-      fileKey,
-      resourceType,
-      aidoType,
-      fileName
-    );
-    if (pathOrNull) {
-      return {
-        type: aidoType,
-        path: pathOrNull,
-        name: path.basename(pathOrNull),
-      };
-    }
-    return { type: aidoType, ref: `im:${messageId}:${fileKey}:${aidoType}`, channel: "feishu" };
+    if (!buffer) return null;
+    await fs.promises.writeFile(outPath, buffer);
+    return path.resolve(outPath);
   }
 
   /**
@@ -776,22 +734,10 @@ export class FeishuHandler implements BridgeHandler {
         } else if (messageType === "post") {
           text = extractTextFromPostContent(rawContent);
           const postImageKeys = [...new Set(extractImageKeysFromPostContent(rawContent))];
-          if (postImageKeys.length > 0) {
-            console.error(`[feishu] post: fetching ${postImageKeys.length} image(s) from message ${messageId}`);
-            for (const imageKey of postImageKeys) {
-              const attach = await this.fetchMessageResourceAttachment(
-                messageId,
-                imageKey,
-                "image",
-                "image"
-              );
-              if (attach) {
-                attachments.push(attach);
-                console.error(`[feishu] post: fetched image ok (${attach.path ? "path" : "ref"})`);
-              } else {
-                console.warn(`[feishu] post: failed to fetch image ${imageKey}`);
-              }
-            }
+          for (const imageKey of postImageKeys) {
+            const fileName = `image_${imageKey.slice(-12)}.png`;
+            const absPath = await this.saveResourceToLocal(messageId, imageKey, "image", "image", fileName);
+            if (absPath) attachments.push({ type: "image", path: absPath, name: path.basename(absPath) });
           }
         } else {
           const mediaConfig = MEDIA_MESSAGE_CONFIG[messageType];
@@ -800,24 +746,20 @@ export class FeishuHandler implements BridgeHandler {
               const content = JSON.parse(rawContent || "{}") as Record<string, string>;
               const fileKey = content[mediaConfig.contentKey];
               if (fileKey) {
-                const fileName = content["file_name"] ?? content["fileName"];
-                const attach = await this.fetchMessageResourceAttachment(
+                const fileName = content["file_name"] ?? content["fileName"] ?? `${mediaConfig.aidoType}_${fileKey.slice(-12)}.${mediaConfig.aidoType === "image" ? "png" : mediaConfig.aidoType === "video" ? "mp4" : mediaConfig.aidoType === "audio" ? "mp3" : "bin"}`;
+                const absPath = await this.saveResourceToLocal(
                   messageId,
                   fileKey,
                   mediaConfig.resourceType,
                   mediaConfig.aidoType,
                   fileName
                 );
-                if (attach) attachments.push(attach);
+                if (absPath) attachments.push({ type: mediaConfig.aidoType, path: absPath, name: path.basename(absPath) });
                 else text = `[无法获取${mediaConfig.aidoType}资源]`;
               } else {
-                console.warn(
-                  `[feishu] ${messageType} message missing ${mediaConfig.contentKey} in content`
-                );
                 text = "[无法解析的媒体消息]";
               }
-            } catch (err) {
-              console.warn(`[feishu] failed to parse ${messageType} message content:`, err);
+            } catch {
               text = "[无法解析的媒体消息]";
             }
           }