feishu-bridge 1.0.6 → 1.0.8

package/dist/cli/index.js

@@ -616,9 +616,6 @@ ${error.substring(0, 2e3)}`;
   }
 };
 
-// src/feishu/websocket.ts
-var import_ws = require("ws");
-
 // src/core/message-router.ts
 init_logger();
 var MessageRouter = class {
@@ -761,163 +758,109 @@ var MessageRouter = class {
 // src/feishu/websocket.ts
 init_logger();
 init_config();
+var Lark = __toESM(require("@larksuiteoapi/node-sdk"));
 var FeishuWebSocketHandler = class {
   constructor(client, commandExecutor, adapters) {
     this.client = client;
     this.commandExecutor = commandExecutor;
     this.adapters = adapters;
-    this.ws = null;
-    this.reconnectAttempts = 0;
-    this.maxReconnectAttempts = 5;
-    this.reconnectInterval = 5e3;
-    // 5秒
-    this.heartbeatInterval = null;
+    this.wsClient = null;
+    this.eventDispatcher = null;
     this.isRunning = false;
     this.logger = new Logger();
     this.messageRouter = new MessageRouter(adapters);
   }
-  /**
-   * 启动WebSocket连接
-   */
   async start() {
     if (this.isRunning) {
       this.logger.warn("WebSocket handler already running");
       return;
     }
-    this.isRunning = true;
-    this.reconnectAttempts = 0;
-    await this.connect();
-  }
-  /**
-   * 停止WebSocket连接
-   */
-  async stop() {
-    this.isRunning = false;
-    if (this.heartbeatInterval) {
-      clearInterval(this.heartbeatInterval);
-      this.heartbeatInterval = null;
-    }
-    if (this.ws) {
-      this.ws.close();
-      this.ws = null;
-    }
-    this.logger.info("WebSocket handler stopped");
-  }
-  /**
-   * 建立WebSocket连接
-   */
-  async connect() {
+    const config = ConfigManager.getInstance().getFeishuConfig();
     try {
-      const config = ConfigManager.getInstance().getFeishuConfig();
-      const wsUrl = config.domain === "feishu" ? "wss://ws.feishu.cn/passport/" : "wss://ws.larksuite.com/passport/";
-      this.logger.info(`Connecting to WebSocket: ${wsUrl}`);
-      this.ws = new import_ws.WebSocket(wsUrl, {
-        headers: {
-          "Authorization": `Bearer ${await this.getAccessToken()}`
-        }
+      this.logger.info("Creating Feishu WebSocket client...");
+      this.wsClient = new Lark.WSClient({
+        appId: config.appId,
+        appSecret: config.appSecret,
+        domain: config.domain === "lark" ? Lark.Domain.Lark : Lark.Domain.Feishu,
+        loggerLevel: Lark.LoggerLevel.info
       });
-      this.setupWebSocketHandlers();
+      this.eventDispatcher = new Lark.EventDispatcher({
+        encryptKey: config.encryptKey || "",
+        verificationToken: config.verificationToken || ""
+      });
+      this.registerEventHandlers();
+      this.logger.info("Starting WebSocket connection...");
+      await this.wsClient.start({ eventDispatcher: this.eventDispatcher });
+      this.isRunning = true;
+      this.logger.info("\u2705 Feishu WebSocket connected successfully!");
     } catch (error) {
-      this.logger.error("Failed to connect WebSocket:", error);
-      this.handleReconnect();
+      this.logger.error("\u274C Failed to start WebSocket handler:", error);
+      throw error;
     }
   }
-  /**
-   * 设置WebSocket事件处理器
-   */
-  setupWebSocketHandlers() {
-    if (!this.ws) return;
-    this.ws.on("open", () => {
-      this.logger.info("WebSocket connected");
-      this.reconnectAttempts = 0;
-      this.startHeartbeat();
-    });
-    this.ws.on("message", (data) => {
-      this.handleMessage(data.toString());
-    });
-    this.ws.on("close", (code, reason) => {
-      this.logger.info(`WebSocket closed: ${code} - ${reason.toString()}`);
-      this.stopHeartbeat();
-      if (this.isRunning) {
-        this.handleReconnect();
+  registerEventHandlers() {
+    if (!this.eventDispatcher) return;
+    this.eventDispatcher.register({
+      "im.message.receive_v1": async (event) => {
+        await this.handleMessageEvent(event);
+      },
+      "im.message.message_read_v1": (event) => {
+        this.logger.debug("Message read:", event);
+      },
+      "im.chat.member.bot.added_v1": (event) => {
+        this.logger.info("Bot added to chat:", event);
+      },
+      "im.chat.member.bot.deleted_v1": (event) => {
+        this.logger.info("Bot removed from chat:", event);
       }
     });
-    this.ws.on("error", (error) => {
-      this.logger.error("WebSocket error:", error);
-    });
-    this.ws.on("ping", () => {
-      this.ws?.pong();
-    });
-  }
-  /**
-   * 处理收到的消息
-   */
-  async handleMessage(data) {
-    try {
-      const event = JSON.parse(data);
-      this.logger.debug("Received WebSocket message:", event);
-      if (event.type === "url_verification") {
-        this.logger.info("Received URL verification challenge via WebSocket");
-        return;
-      }
-      if (event.type === "event_callback" && event.event) {
-        await this.handleEvent(event.event);
-      }
-    } catch (error) {
-      this.logger.error("Error handling WebSocket message:", error);
-    }
   }
-  /**
-   * 处理事件
-   */
-  async handleEvent(event) {
-    const { message } = event;
+  async handleMessageEvent(event) {
+    const { message, sender } = event;
     if (!message || message.message_type !== "text") {
       this.logger.debug("Ignoring non-text message");
       return;
     }
     try {
       const content = JSON.parse(message.content);
-      const text = content.text.trim();
-      const sender = event.sender?.sender_id?.open_id;
-      if (!sender) {
+      const text = content.text?.trim() || "";
+      const senderId = sender?.sender_id?.open_id;
+      if (!senderId) {
         this.logger.warn("Missing sender ID");
         return;
       }
-      this.logger.info(`Received message from ${sender}: ${text}`);
+      this.logger.info(`\u{1F4E8} Received message from ${senderId}: ${text}`);
       const route = await this.messageRouter.route(text);
       if (route.type === "error") {
         await this.client.sendMessage(message.chat_id, {
           msg_type: "text",
-          content: {
-            text: route.content
-          }
+          content: { text: route.content }
         });
         return;
       }
       if (route.type === "prompt_choice") {
         await this.client.sendMessage(message.chat_id, {
           msg_type: "text",
-          content: {
-            text: route.content
-          }
+          content: { text: route.content }
         });
         return;
       }
       if (route.type === "direct" && route.target) {
         const command = this.messageRouter.extractCommand(text);
-        this.logger.info(`Routing to ${route.target}: ${command}`);
+        this.logger.info(`\u{1F3AF} Routing to ${route.target}: ${command}`);
+        await this.client.sendMessage(message.chat_id, {
+          msg_type: "text",
+          content: { text: `\u23F3 \u6B63\u5728\u6267\u884C: ${command}` }
+        });
         const result = await this.commandExecutor.execute({
           target: route.target,
           command,
-          sender,
+          sender: senderId,
           chatId: message.chat_id
         });
         await this.client.sendMessage(message.chat_id, {
           msg_type: "text",
-          content: {
-            text: this.formatResponse(result)
-          }
+          content: { text: this.formatResponse(result) }
         });
       }
     } catch (error) {
@@ -925,89 +868,13 @@ var FeishuWebSocketHandler = class {
       try {
         await this.client.sendMessage(message.chat_id, {
           msg_type: "text",
-          content: {
-            text: `\u274C \u5904\u7406\u6D88\u606F\u65F6\u51FA\u9519: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}`
-          }
+          content: { text: `\u274C \u5904\u7406\u6D88\u606F\u65F6\u51FA\u9519` }
         });
       } catch (sendError) {
         this.logger.error("Failed to send error message:", sendError);
       }
     }
   }
-  /**
-   * 启动心跳保活
-   */
-  startHeartbeat() {
-    this.heartbeatInterval = setInterval(() => {
-      if (this.ws?.readyState === import_ws.WebSocket.OPEN) {
-        this.ws.ping();
-      }
-    }, 3e4);
-  }
-  /**
-   * 停止心跳
-   */
-  stopHeartbeat() {
-    if (this.heartbeatInterval) {
-      clearInterval(this.heartbeatInterval);
-      this.heartbeatInterval = null;
-    }
-  }
-  /**
-   * 处理重连
-   */
-  handleReconnect() {
-    if (this.reconnectAttempts >= this.maxReconnectAttempts) {
-      this.logger.error(`Max reconnection attempts (${this.maxReconnectAttempts}) reached`);
-      return;
-    }
-    this.reconnectAttempts++;
-    const delay = this.reconnectInterval * this.reconnectAttempts;
-    this.logger.info(`Reconnecting in ${delay}ms (attempt ${this.reconnectAttempts}/${this.maxReconnectAttempts})`);
-    setTimeout(() => {
-      if (this.isRunning) {
-        this.connect();
-      }
-    }, delay);
-  }
-  /**
-   * 获取访问令牌
-   */
-  async getAccessToken() {
-    const config = ConfigManager.getInstance().getFeishuConfig();
-    const response = await fetch(
-      `https://open.${config.domain === "lark" ? "larksuite" : "feishu"}.cn/open-apis/auth/v3/tenant_access_token/internal`,
-      {
-        method: "POST",
-        headers: { "Content-Type": "application/json" },
-        body: JSON.stringify({
-          app_id: config.appId,
-          app_secret: config.appSecret
-        })
-      }
-    );
-    const data = await response.json();
-    if (data.code !== 0) {
-      throw new Error(`Failed to get access token: ${data.msg}`);
-    }
-    return data.tenant_access_token;
-  }
-  /**
-   * 提取目标IDE
-   */
-  extractTargetIDE(text) {
-    const match = text.match(/@(vscode|cursor|trae|antigravity|kiro|opencode|claude)/i);
-    return match ? match[1].toLowerCase() : "auto";
-  }
-  /**
-   * 提取命令
-   */
-  extractCommand(text) {
-    return text.replace(/@(vscode|cursor|trae|antigravity|kiro|opencode|claude)\s*/i, "").trim();
-  }
-  /**
-   * 格式化响应
-   */
   formatResponse(result) {
     if (result.success) {
       const output = typeof result.output === "string" ? result.output.substring(0, 2e3) : JSON.stringify(result.output).substring(0, 2e3);
@@ -1021,6 +888,14 @@ ${output}`;
 ${error.substring(0, 2e3)}`;
     }
   }
+  async stop() {
+    this.isRunning = false;
+    if (this.wsClient) {
+      this.wsClient.close();
+      this.wsClient = null;
+    }
+    this.logger.info("WebSocket handler stopped");
+  }
 };
 
 // src/feishu/client.ts
@@ -3445,18 +3320,42 @@ var ReverseChannel = class extends import_events.EventEmitter {
 };
 
 // src/skills/builtin/feishu.ts
+init_config();
 var FeishuSkills = class {
   constructor(client) {
+    this.accessToken = null;
+    this.tokenExpireTime = 0;
     this.client = client;
+    const config = ConfigManager.getInstance().getFeishuConfig();
+    this.baseUrl = config.domain === "lark" ? "https://open.larksuite.com/open-apis" : "https://open.feishu.cn/open-apis";
+  }
+  async getAccessToken() {
+    const now = Date.now();
+    if (this.accessToken && now < this.tokenExpireTime) {
+      return this.accessToken;
+    }
+    const config = ConfigManager.getInstance().getFeishuConfig();
+    const response = await fetch(`${this.baseUrl}/auth/v3/tenant_access_token/internal`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({
+        app_id: config.appId,
+        app_secret: config.appSecret
+      })
+    });
+    const data = await response.json();
+    if (data.code !== 0) {
+      throw new Error(`Failed to get access token: ${data.msg}`);
+    }
+    this.accessToken = data.tenant_access_token;
+    this.tokenExpireTime = now + (data.expire - 300) * 1e3;
+    return this.accessToken;
   }
-  /**
-   * 发送消息技能
-   */
   sendMessageSkill() {
     return {
       name: "feishu.sendMessage",
       description: "\u53D1\u9001\u6D88\u606F\u5230\u6307\u5B9A\u4F1A\u8BDD",
-      trigger: /发送消息|发信息/,
+      trigger: /发送消息|发信息|发消息/,
       riskLevel: "low",
       examples: [
         "\u53D1\u9001\u6D88\u606F\u7ED9\u5F20\u4E09\uFF1A\u4E0B\u5348\u5F00\u4F1A",
@@ -3464,32 +3363,23 @@ var FeishuSkills = class {
       ],
       execute: async (context) => {
         try {
-          const { target, content } = this.parseMessageArgs(context.message);
+          const { content } = this.parseMessageArgs(context.message);
           await this.client.sendMessage(context.chatId, {
             msg_type: "text",
             content: { text: content }
           });
-          return {
-            success: true,
-            content: "\u2705 \u6D88\u606F\u5DF2\u53D1\u9001"
-          };
+          return { success: true, content: "\u2705 \u6D88\u606F\u5DF2\u53D1\u9001" };
         } catch (error) {
-          return {
-            success: false,
-            content: `\u274C \u53D1\u9001\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}`
-          };
+          return { success: false, content: `\u274C \u53D1\u9001\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}` };
         }
       }
     };
   }
-  /**
-   * 创建日程技能
-   */
   createEventSkill() {
     return {
       name: "feishu.createEvent",
       description: "\u5728\u98DE\u4E66\u65E5\u5386\u521B\u5EFA\u65E5\u7A0B",
-      trigger: /创建日程|创建会议|添加日程/,
+      trigger: /创建日程|创建会议|添加日程|新建日程/,
       riskLevel: "medium",
       examples: [
         "\u521B\u5EFA\u65E5\u7A0B\uFF1A\u660E\u5929\u4E0B\u53483\u70B9\u56E2\u961F\u5468\u4F1A",
@@ -3497,36 +3387,57 @@ var FeishuSkills = class {
       ],
       execute: async (context) => {
         try {
-          const { title, startTime, attendees } = this.parseEventArgs(context.message);
+          const { title, startTime, endTime, description } = this.parseEventArgs(context.message);
+          const token = await this.getAccessToken();
+          const response = await fetch(`${this.baseUrl}/calendar/v4/calendars/primary/events`, {
+            method: "POST",
+            headers: {
+              "Authorization": `Bearer ${token}`,
+              "Content-Type": "application/json"
+            },
+            body: JSON.stringify({
+              summary: title,
+              description,
+              start_time: {
+                timestamp: Math.floor(startTime / 1e3).toString(),
+                timezone: "Asia/Shanghai"
+              },
+              end_time: {
+                timestamp: Math.floor(endTime / 1e3).toString(),
+                timezone: "Asia/Shanghai"
+              },
+              attendee_ability: "can_see_others",
+              free_busy_status: "busy"
+            })
+          });
+          const data = await response.json();
+          if (data.code !== 0) {
+            throw new Error(data.msg || "\u521B\u5EFA\u65E5\u7A0B\u5931\u8D25");
+          }
+          const eventLink = data.data?.event_link || "";
           return {
             success: true,
             content: `\u2705 \u65E5\u7A0B\u5DF2\u521B\u5EFA
 
-\u6807\u9898: ${title}
-\u65F6\u95F4: ${startTime}
-\u53C2\u4E0E\u8005: ${attendees.join(", ") || "\u65E0"}`,
+\u{1F4C5} ${title}
+\u{1F550} ${this.formatDateTime(startTime)} - ${this.formatDateTime(endTime)}
+
+\u{1F517} ${eventLink}`,
             actions: [
-              { label: "\u67E5\u770B\u65E5\u7A0B", action: "view_calendar" },
-              { label: "\u7F16\u8F91", action: "edit_event" }
+              { label: "\u67E5\u770B\u65E5\u7A0B", action: "view_calendar", url: eventLink }
             ]
           };
         } catch (error) {
-          return {
-            success: false,
-            content: `\u274C \u521B\u5EFA\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}`
-          };
+          return { success: false, content: `\u274C \u521B\u5EFA\u65E5\u7A0B\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}` };
         }
       }
     };
   }
-  /**
-   * 创建文档技能
-   */
   createDocSkill() {
     return {
       name: "feishu.createDoc",
       description: "\u521B\u5EFA\u98DE\u4E66\u4E91\u6587\u6863",
-      trigger: /创建文档|新建文档/,
+      trigger: /创建文档|新建文档|生成文档/,
       riskLevel: "low",
       examples: [
         "\u521B\u5EFA\u6587\u6863\uFF1A\u9879\u76EE\u9700\u6C42\u6587\u6863",
@@ -3534,89 +3445,114 @@ var FeishuSkills = class {
       ],
       execute: async (context) => {
         try {
-          const { title, content, folder } = this.parseDocArgs(context.message);
+          const { title } = this.parseDocArgs(context.message);
+          const token = await this.getAccessToken();
+          const response = await fetch(`${this.baseUrl}/docx/v1/documents`, {
+            method: "POST",
+            headers: {
+              "Authorization": `Bearer ${token}`,
+              "Content-Type": "application/json"
+            },
+            body: JSON.stringify({ title })
+          });
+          const data = await response.json();
+          if (data.code !== 0) {
+            throw new Error(data.msg || "\u521B\u5EFA\u6587\u6863\u5931\u8D25");
+          }
+          const documentId = data.data?.document?.document_id;
+          const domain = ConfigManager.getInstance().getFeishuConfig().domain === "lark" ? "larksuite.com" : "feishu.cn";
+          const documentUrl = `https://${domain}/docx/${documentId}`;
           return {
             success: true,
             content: `\u2705 \u6587\u6863\u5DF2\u521B\u5EFA
 
-\u6807\u9898: ${title}
-\u4F4D\u7F6E: ${folder || "\u6211\u7684\u7A7A\u95F4"}`,
+\u{1F4C4} ${title}
+
+\u{1F517} ${documentUrl}`,
             actions: [
-              { label: "\u6253\u5F00\u6587\u6863", action: "open_doc" },
-              { label: "\u5206\u4EAB", action: "share_doc" }
+              { label: "\u6253\u5F00\u6587\u6863", action: "open_doc", url: documentUrl }
             ]
           };
         } catch (error) {
-          return {
-            success: false,
-            content: `\u274C \u521B\u5EFA\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}`
-          };
+          return { success: false, content: `\u274C \u521B\u5EFA\u6587\u6863\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}` };
         }
       }
     };
   }
-  /**
-   * 上传文件技能
-   */
   uploadFileSkill() {
     return {
       name: "feishu.uploadFile",
       description: "\u4E0A\u4F20\u6587\u4EF6\u5230\u98DE\u4E66",
-      trigger: /上传文件|发送文件/,
+      trigger: /上传文件|发送文件|传文件/,
       riskLevel: "low",
       execute: async (context) => {
-        return {
-          success: true,
-          content: "\u{1F4CE} \u6587\u4EF6\u4E0A\u4F20\u529F\u80FD\u5F00\u53D1\u4E2D"
-        };
+        return { success: true, content: "\u{1F4CE} \u6587\u4EF6\u4E0A\u4F20\u529F\u80FD\u9700\u8981\u6587\u4EF6\u8DEF\u5F84\uFF0C\u8BF7\u4F7F\u7528\uFF1A\u4E0A\u4F20\u6587\u4EF6 /path/to/file" };
       }
     };
   }
-  /**
-   * 创建群公告技能
-   */
   createAnnouncementSkill() {
     return {
       name: "feishu.createAnnouncement",
       description: "\u53D1\u9001\u7FA4\u516C\u544A",
-      trigger: /发送公告|群公告/,
+      trigger: /发送公告|群公告|发布公告|发公告/,
       riskLevel: "medium",
       requireConfirm: true,
       execute: async (context) => {
-        return {
-          success: true,
-          content: "\u{1F4E2} \u516C\u544A\u529F\u80FD\u5F00\u53D1\u4E2D"
-        };
+        try {
+          const { title, content } = this.parseAnnouncementArgs(context.message);
+          await this.client.sendMessage(context.chatId, {
+            msg_type: "text",
+            content: { text: `\u{1F4E2} **${title}**
+
+${content}` }
+          });
+          return { success: true, content: "\u2705 \u516C\u544A\u5DF2\u53D1\u9001" };
+        } catch (error) {
+          return { success: false, content: `\u274C \u53D1\u5E03\u516C\u544A\u5931\u8D25: ${error instanceof Error ? error.message : "\u672A\u77E5\u9519\u8BEF"}` };
+        }
       }
     };
   }
-  // 辅助方法
   parseMessageArgs(message) {
-    const match = message.match(/给(.+?)：(.+)/);
-    if (match) {
-      return { target: match[1], content: match[2] };
-    }
+    const match = message.match(/给(.+?)[：:]\s*(.+)/);
+    if (match) return { target: match[1], content: match[2] };
     return { target: "current", content: message };
   }
   parseEventArgs(message) {
-    const titleMatch = message.match(/日程[：:]\s*(.+?)(?=\s|$)/);
-    const timeMatch = message.match(/(明天|今天|后天|\d{1,2}点|\d{1,2}:\d{2})/);
-    return {
-      title: titleMatch ? titleMatch[1] : "\u65B0\u65E5\u7A0B",
-      startTime: timeMatch ? timeMatch[1] : "\u672A\u6307\u5B9A\u65F6\u95F4",
-      attendees: []
-    };
+    const now = /* @__PURE__ */ new Date();
+    const titleMatch = message.match(/日程[：:]\s*(.+?)(?=(明天|今天|后天|时间|地点|$))/i);
+    const title = titleMatch ? titleMatch[1].trim() : "\u65B0\u65E5\u7A0B";
+    const timeMatch = message.match(/(明天|今天|后天)?\s*(上午|下午|晚上)?\s*(\d{1,2})[:点时]?(\d{0,2})?/);
+    let startTime = new Date(now.getTime() + 60 * 60 * 1e3);
+    if (timeMatch) {
+      const day = timeMatch[1];
+      const period = timeMatch[2];
+      let hour = parseInt(timeMatch[3]);
+      const minute = parseInt(timeMatch[4]) || 0;
+      if (day === "\u660E\u5929") startTime.setDate(startTime.getDate() + 1);
+      else if (day === "\u540E\u5929") startTime.setDate(startTime.getDate() + 2);
+      if (period === "\u4E0B\u5348" && hour < 12) hour += 12;
+      if (period === "\u665A\u4E0A" && hour < 12) hour += 12;
+      startTime.setHours(hour, minute, 0, 0);
+    }
+    const endTime = new Date(startTime.getTime() + 60 * 60 * 1e3);
+    return { title, startTime: startTime.getTime(), endTime: endTime.getTime(), description: "" };
   }
   parseDocArgs(message) {
-    const match = message.match(/文档[：:]\s*(.+?)(?=\s|$)/);
-    return {
-      title: match ? match[1] : "\u672A\u547D\u540D\u6587\u6863",
-      content: ""
-    };
+    const match = message.match(/文档[：:]\s*(.+?)(?=(内容|文件夹|$))/i);
+    return { title: match ? match[1].trim() : "\u672A\u547D\u540D\u6587\u6863" };
+  }
+  parseAnnouncementArgs(message) {
+    const titleMatch = message.match(/公告[：:]\s*(.+?)(?=内容|$)/i);
+    const title = titleMatch ? titleMatch[1].trim() : "\u7FA4\u516C\u544A";
+    const contentMatch = message.match(/内容[：:]\s*(.+)/);
+    const content = contentMatch ? contentMatch[1] : "";
+    return { title, content };
+  }
+  formatDateTime(timestamp) {
+    const date = new Date(timestamp);
+    return `${date.getMonth() + 1}\u6708${date.getDate()}\u65E5 ${date.getHours()}:${date.getMinutes().toString().padStart(2, "0")}`;
   }
-  /**
-   * 获取所有飞书技能
-   */
   getAllSkills() {
     return [
       this.sendMessageSkill(),