Merge pull request #158 from renhongchao/feature/bee

feat:添加小蜜蜂接入

Author: liuzhq1986

README.md

@@ -260,6 +260,7 @@ LobsterAI can bridge the Agent to multiple IM platforms. Send a message from you
 | Telegram | grammY | Bot API integration |
 | Discord | discord.js | Discord bot integration |
 | NetEase IM | node-nim V2 SDK | NetEase IM P2P messaging |
+| NetEase Bee | node-nim V2 SDK | NetEase Bee Personal Digital Assistant |
 
 Configure the corresponding platform Token/Secret in the Settings panel to enable. Once set up, you can send instructions directly to the Agent from your phone IM (e.g., "analyze this dataset", "make a weekly summary PPT"), and the Agent will execute on the desktop and return results.
 

README_zh.md

@@ -261,6 +261,7 @@ LobsterAI 支持将 Agent 桥接到多种 IM 平台。在手机上通过 IM 发
 | Telegram | grammY | Bot API 接入 |
 | Discord | discord.js | Discord Bot 接入 |
 | 云信 IM | node-nim V2 SDK | 网易云信 IM P2P 私聊 |
+| 网易小蜜蜂 | node-nim V2 SDK | [网易小蜜蜂个人数字助理](https://wp.m.163.com/163/html/bee/lobsterai_guide/index.html) |
 
 在设置面板中配置对应平台的 Token/密钥即可启用。配置完成后，你可以在手机 IM 中直接对 Agent 下达指令（如「帮我分析这份数据」「做一份本周工作汇报 PPT」），Agent 会在桌面端自动执行并返回结果。
 

electron-builder.json

@@ -111,4 +111,4 @@
     "node_modules/node-nim/**"
   ],
   "asar": true
-}
+}

public/xiaomifeng.png

@@ -9,6 +9,7 @@ import { FeishuGateway } from './feishuGateway';
 import { TelegramGateway } from './telegramGateway';
 import { DiscordGateway } from './discordGateway';
 import { NimGateway } from './nimGateway';
+import { XiaomifengGateway } from './xiaomifengGateway';
 import { IMChatHandler } from './imChatHandler';
 import { IMCoworkHandler } from './imCoworkHandler';
 import { IMStore } from './imStore';
@@ -53,6 +54,7 @@ export class IMGatewayManager extends EventEmitter {
   private telegramGateway: TelegramGateway;
   private discordGateway: DiscordGateway;
   private nimGateway: NimGateway;
+  private xiaomifengGateway: XiaomifengGateway;
   private imStore: IMStore;
   private chatHandler: IMChatHandler | null = null;
   private coworkHandler: IMCoworkHandler | null = null;
@@ -72,6 +74,7 @@ export class IMGatewayManager extends EventEmitter {
     this.telegramGateway = new TelegramGateway();
     this.discordGateway = new DiscordGateway();
     this.nimGateway = new NimGateway();
+    this.xiaomifengGateway = new XiaomifengGateway();
 
     // Store Cowork dependencies if provided
     if (options?.coworkRunner && options?.coworkStore) {
@@ -167,6 +170,24 @@ export class IMGatewayManager extends EventEmitter {
     this.nimGateway.on('message', (message: IMMessage) => {
       this.emit('message', message);
     });
+
+    // Xiaomifeng events
+    this.xiaomifengGateway.on('status', () => {
+      this.emit('statusChange', this.getStatus());
+    });
+    this.xiaomifengGateway.on('connected', () => {
+      this.emit('statusChange', this.getStatus());
+    });
+    this.xiaomifengGateway.on('disconnected', () => {
+      this.emit('statusChange', this.getStatus());
+    });
+    this.xiaomifengGateway.on('error', (error) => {
+      this.emit('error', { platform: 'xiaomifeng', error });
+      this.emit('statusChange', this.getStatus());
+    });
+    this.xiaomifengGateway.on('message', (message: IMMessage) => {
+      this.emit('message', message);
+    });
   }
 
   /**
@@ -200,6 +221,11 @@ export class IMGatewayManager extends EventEmitter {
       console.log('[IMGatewayManager] Reconnecting NIM...');
       this.nimGateway.reconnectIfNeeded();
     }
+
+    if (this.xiaomifengGateway && !this.xiaomifengGateway.isConnected()) {
+      console.log('[IMGatewayManager] Reconnecting Xiaomifeng...');
+      this.xiaomifengGateway.reconnectIfNeeded();
+    }
   }
 
   /**
@@ -268,6 +294,7 @@ export class IMGatewayManager extends EventEmitter {
     this.telegramGateway.setMessageCallback(messageHandler);
     this.discordGateway.setMessageCallback(messageHandler);
     this.nimGateway.setMessageCallback(messageHandler);
+    this.xiaomifengGateway.setMessageCallback(messageHandler);
   }
 
   /**
@@ -396,6 +423,7 @@ export class IMGatewayManager extends EventEmitter {
       telegram: this.telegramGateway.getStatus(),
       discord: this.discordGateway.getStatus(),
       nim: this.nimGateway.getStatus(),
+      xiaomifeng: this.xiaomifengGateway.getStatus(),
     };
   }
 
@@ -616,6 +644,8 @@ export class IMGatewayManager extends EventEmitter {
       await this.discordGateway.start(config.discord);
     } else if (platform === 'nim') {
       await this.nimGateway.start(config.nim);
+    } else if (platform === 'xiaomifeng') {
+      await this.xiaomifengGateway.start(config.xiaomifeng);
     }
 
     // Restore persisted notification target
@@ -636,6 +666,8 @@ export class IMGatewayManager extends EventEmitter {
       await this.discordGateway.stop();
     } else if (platform === 'nim') {
       await this.nimGateway.stop();
+    } else if (platform === 'xiaomifeng') {
+      await this.xiaomifengGateway.stop();
     }
   }
 
@@ -684,6 +716,14 @@ export class IMGatewayManager extends EventEmitter {
         console.error(`[IMGatewayManager] Failed to start NIM: ${error.message}`);
       }
     }
+
+    if (config.xiaomifeng?.enabled && config.xiaomifeng?.clientId && config.xiaomifeng?.secret) {
+      try {
+        await this.startGateway('xiaomifeng');
+      } catch (error: any) {
+        console.error(`[IMGatewayManager] Failed to start Xiaomifeng: ${error.message}`);
+      }
+    }
   }
 
   /**
@@ -696,14 +736,15 @@ export class IMGatewayManager extends EventEmitter {
       this.telegramGateway.stop(),
       this.discordGateway.stop(),
       this.nimGateway.stop(),
+      this.xiaomifengGateway.stop(),
     ]);
   }
 
   /**
    * Check if any gateway is connected
    */
   isAnyConnected(): boolean {
-    return this.dingtalkGateway.isConnected() || this.feishuGateway.isConnected() || this.telegramGateway.isConnected() || this.discordGateway.isConnected() || this.nimGateway.isConnected();
+    return this.dingtalkGateway.isConnected() || this.feishuGateway.isConnected() || this.telegramGateway.isConnected() || this.discordGateway.isConnected() || this.nimGateway.isConnected() || this.xiaomifengGateway.isConnected();
   }
 
   /**
@@ -722,6 +763,9 @@ export class IMGatewayManager extends EventEmitter {
     if (platform === 'nim') {
       return this.nimGateway.isConnected();
     }
+    if (platform === 'xiaomifeng') {
+      return this.xiaomifengGateway.isConnected();
+    }
     return this.feishuGateway.isConnected();
   }
 
@@ -793,6 +837,7 @@ export class IMGatewayManager extends EventEmitter {
       telegram: { ...current.telegram, ...(configOverride.telegram || {}) },
       discord: { ...current.discord, ...(configOverride.discord || {}) },
       nim: { ...current.nim, ...(configOverride.nim || {}) },
+      xiaomifeng: { ...current.xiaomifeng, ...(configOverride.xiaomifeng || {}) },
       settings: { ...current.settings, ...(configOverride.settings || {}) },
     };
   }
@@ -820,6 +865,12 @@ export class IMGatewayManager extends EventEmitter {
       if (!config.nim.token) fields.push('token');
       return fields;
     }
+    if (platform === 'xiaomifeng') {
+      const fields: string[] = [];
+      if (!config.xiaomifeng?.clientId) fields.push('clientId');
+      if (!config.xiaomifeng?.secret) fields.push('secret');
+      return fields;
+    }
     return config.discord.botToken ? [] : ['botToken'];
   }
 
@@ -872,13 +923,28 @@ export class IMGatewayManager extends EventEmitter {
       // check will happen when the user enables the gateway and the SDK logs in.
       return `云信配置已填写（Account: ${config.nim.account}）。请启用渠道，SDK 登录时将完成实际凭证验证。`;
     }
-    const response = await fetchJsonWithTimeout<DiscordUserResponse>('https://discord.com/api/v10/users/@me', {
-      headers: {
-        Authorization: `Bot ${config.discord.botToken}`,
-      },
-    }, CONNECTIVITY_TIMEOUT_MS);
-    const username = response.username ? `${response.username}#${response.discriminator || '0000'}` : 'unknown';
-    return `Discord 鉴权通过（Bot: ${username}）。`;
+
+    if (platform === 'xiaomifeng') {
+      // 小蜜蜂使用网易云信 NIM SDK，鉴权是通过 SDK 登录验证的
+      // 这里我们只做配置完整性检查，实际登录验证在 start 时进行
+      const { clientId, secret } = config.xiaomifeng;
+      if (!clientId || !secret) {
+        throw new Error('配置不完整');
+      }
+      return `小蜜蜂配置已就绪（Client ID: ${clientId}）。`;
+    }
+
+    if (platform === 'discord') {
+      const response = await fetchJsonWithTimeout<DiscordUserResponse>('https://discord.com/api/v10/users/@me', {
+        headers: {
+          Authorization: `Bot ${config.discord.botToken}`,
+        },
+      }, CONNECTIVITY_TIMEOUT_MS);
+      const username = response.username ? `${response.username}#${response.discriminator || '0000'}` : 'unknown';
+      return `Discord 鉴权通过（Bot: ${username}）。`;
+    }
+
+    return '未知平台。';
   }
 
   private resolveFeishuDomain(domain: string, Lark: any): any {
@@ -906,6 +972,7 @@ export class IMGatewayManager extends EventEmitter {
     if (platform === 'dingtalk') return status.dingtalk.startedAt;
     if (platform === 'telegram') return status.telegram.startedAt;
     if (platform === 'nim') return status.nim.startedAt;
+    if (platform === 'xiaomifeng') return status.xiaomifeng.startedAt;
     return status.discord.startedAt;
   }
 
@@ -914,6 +981,7 @@ export class IMGatewayManager extends EventEmitter {
     if (platform === 'feishu') return status.feishu.lastInboundAt;
     if (platform === 'telegram') return status.telegram.lastInboundAt;
     if (platform === 'nim') return status.nim.lastInboundAt;
+    if (platform === 'xiaomifeng') return status.xiaomifeng.lastInboundAt;
     return status.discord.lastInboundAt;
   }
 
@@ -922,6 +990,7 @@ export class IMGatewayManager extends EventEmitter {
     if (platform === 'feishu') return status.feishu.lastOutboundAt;
     if (platform === 'telegram') return status.telegram.lastOutboundAt;
     if (platform === 'nim') return status.nim.lastOutboundAt;
+    if (platform === 'xiaomifeng') return status.xiaomifeng.lastOutboundAt;
     return status.discord.lastOutboundAt;
   }
 
@@ -930,6 +999,7 @@ export class IMGatewayManager extends EventEmitter {
     if (platform === 'feishu') return status.feishu.error;
     if (platform === 'telegram') return status.telegram.lastError;
     if (platform === 'nim') return status.nim.lastError;
+    if (platform === 'xiaomifeng') return status.xiaomifeng.lastError;
     return status.discord.lastError;
   }
 

src/main/im/imGatewayManager.ts

@@ -11,6 +11,7 @@ import {
   TelegramConfig,
   DiscordConfig,
   NimConfig,
+  XiaomifengConfig,
   IMSettings,
   IMPlatform,
   IMSessionMapping,
@@ -19,6 +20,7 @@ import {
   DEFAULT_TELEGRAM_CONFIG,
   DEFAULT_DISCORD_CONFIG,
   DEFAULT_NIM_CONFIG,
+  DEFAULT_XIAOMIFENG_CONFIG,
   DEFAULT_IM_SETTINGS,
 } from './types';
 
@@ -61,7 +63,7 @@ export class IMStore {
    * Migrate existing IM configs to ensure stable defaults.
    */
   private migrateDefaults(): void {
-    const platforms = ['dingtalk', 'feishu', 'telegram', 'discord', 'nim'] as const;
+    const platforms = ['dingtalk', 'feishu', 'telegram', 'discord', 'nim', 'xiaomifeng'] as const;
     let changed = false;
 
     for (const platform of platforms) {
@@ -162,6 +164,7 @@ export class IMStore {
     const telegram = this.getConfigValue<TelegramConfig>('telegram') ?? DEFAULT_TELEGRAM_CONFIG;
     const discord = this.getConfigValue<DiscordConfig>('discord') ?? DEFAULT_DISCORD_CONFIG;
     const nim = this.getConfigValue<NimConfig>('nim') ?? DEFAULT_NIM_CONFIG;
+    const xiaomifeng = this.getConfigValue<XiaomifengConfig>('xiaomifeng') ?? DEFAULT_XIAOMIFENG_CONFIG;
     const settings = this.getConfigValue<IMSettings>('settings') ?? DEFAULT_IM_SETTINGS;
 
     // Resolve enabled field: default to false for safety
@@ -181,6 +184,7 @@ export class IMStore {
       telegram: resolveEnabled(telegram, DEFAULT_TELEGRAM_CONFIG),
       discord: resolveEnabled(discord, DEFAULT_DISCORD_CONFIG),
       nim: resolveEnabled(nim, DEFAULT_NIM_CONFIG),
+      xiaomifeng: resolveEnabled(xiaomifeng, DEFAULT_XIAOMIFENG_CONFIG),
       settings: { ...DEFAULT_IM_SETTINGS, ...settings },
     };
   }
@@ -201,6 +205,9 @@ export class IMStore {
     if (config.nim) {
       this.setNimConfig(config.nim);
     }
+    if (config.xiaomifeng) {
+      this.setXiaomifengConfig(config.xiaomifeng);
+    }
     if (config.settings) {
       this.setIMSettings(config.settings);
     }
@@ -266,6 +273,18 @@ export class IMStore {
     this.setConfigValue('nim', { ...current, ...config });
   }
 
+  // ==================== Xiaomifeng Config ====================
+
+  getXiaomifengConfig(): XiaomifengConfig {
+    const stored = this.getConfigValue<XiaomifengConfig>('xiaomifeng');
+    return { ...DEFAULT_XIAOMIFENG_CONFIG, ...stored };
+  }
+
+  setXiaomifengConfig(config: Partial<XiaomifengConfig>): void {
+    const current = this.getXiaomifengConfig();
+    this.setConfigValue('xiaomifeng', { ...current, ...config });
+  }
+
   // ==================== IM Settings ====================
 
   getIMSettings(): IMSettings {
@@ -298,7 +317,8 @@ export class IMStore {
     const hasTelegram = !!config.telegram.botToken;
     const hasDiscord = !!config.discord.botToken;
     const hasNim = !!(config.nim.appKey && config.nim.account && config.nim.token);
-    return hasDingTalk || hasFeishu || hasTelegram || hasDiscord || hasNim;
+    const hasXiaomifeng = !!(config.xiaomifeng?.clientId && config.xiaomifeng?.secret);
+    return hasDingTalk || hasFeishu || hasTelegram || hasDiscord || hasNim || hasXiaomifeng;
   }
 
   // ==================== Notification Target Persistence ====================