maestro-agent 0.0.1 → 0.0.2

package/src/gateway/index.ts

@@ -0,0 +1,31 @@
+/**
+ * Gateway 入口 — 初始化并启动所有配置的平台 Adapter
+ */
+
+import type { HubContext } from "../core/server";
+import { getConfig } from "../api/config";
+import { GatewayManager } from "./manager";
+import { TelegramAdapter } from "./adapters/telegram";
+
+export async function startGateway(ctx: HubContext): Promise<GatewayManager | null> {
+  // 优先从 config 表读取，环境变量作为 fallback
+  const botToken = getConfig(ctx, "telegram_bot_token") || process.env.MAESTRO_TELEGRAM_BOT_TOKEN;
+
+  // 如果没有配置任何平台 token，不启动网关
+  if (!botToken) {
+    return null;
+  }
+
+  const manager = new GatewayManager(ctx);
+
+  // Telegram Adapter
+  const telegram = new TelegramAdapter({
+    botToken,
+    pollingTimeout: 30,
+    onMessage: manager.getMessageHandler(),
+  });
+  manager.register(telegram);
+
+  await manager.connectAll();
+  return manager;
+}

package/src/gateway/manager.ts

@@ -0,0 +1,176 @@
+/**
+ * GatewayManager — 统一消息网关调度层
+ *
+ * 管理所有平台 Adapter 的生命周期，统一消息路由。
+ * 参考 Hermes 的 gateway/run.py 和 OpenClaw 的 server-plugins.ts。
+ */
+
+import { handleTelegramCommand, isTelegramCommand } from "../bridges/telegram";
+import type { HubContext } from "../core/server";
+import type { InboundMessage, MessageHandler, PlatformAdapter } from "./types";
+
+export class GatewayManager {
+  private adapters: Map<string, PlatformAdapter> = new Map();
+  private ctx: HubContext;
+  private messageHandler: MessageHandler;
+
+  constructor(ctx: HubContext) {
+    this.ctx = ctx;
+    this.messageHandler = this.handleInboundMessage.bind(this);
+
+    // Agent 回复推送：监听 chat.message 事件，路由到对应平台
+    ctx.bus.on("chat.message", (event: any) => {
+      this.handleAgentReply(event).catch((err) => {
+        console.error("[gateway] agent reply push error:", err.message);
+      });
+    });
+  }
+
+  /** 注册一个平台 Adapter */
+  register(adapter: PlatformAdapter) {
+    this.adapters.set(adapter.platform, adapter);
+  }
+
+  /** 启动所有已注册的 Adapter */
+  async connectAll(): Promise<void> {
+    for (const [platform, adapter] of this.adapters) {
+      try {
+        const ok = await adapter.connect();
+        if (ok) {
+          console.log(`[gateway] ✅ ${platform} connected`);
+        } else {
+          console.warn(`[gateway] ⚠️ ${platform} connection returned false`);
+        }
+      } catch (err: any) {
+        console.error(`[gateway] ❌ ${platform} connect failed:`, err.message);
+      }
+    }
+  }
+
+  /** 停止所有 Adapter */
+  async disconnectAll(): Promise<void> {
+    for (const [platform, adapter] of this.adapters) {
+      try {
+        await adapter.disconnect();
+        console.log(`[gateway] ${platform} disconnected`);
+      } catch (err: any) {
+        console.error(`[gateway] ${platform} disconnect error:`, err.message);
+      }
+    }
+  }
+
+  /** 获取消息处理回调（供 Adapter 内部调用） */
+  getMessageHandler(): MessageHandler {
+    return this.messageHandler;
+  }
+
+  /** 获取指定平台的 Adapter */
+  getAdapter(platform: string): PlatformAdapter | undefined {
+    return this.adapters.get(platform);
+  }
+
+  /** 统一入站消息处理 */
+  private async handleInboundMessage(msg: InboundMessage): Promise<void> {
+    const { platform, chatId, text } = msg;
+
+    // Telegram 特殊处理：slash 命令
+    if (platform === "telegram" && isTelegramCommand(text)) {
+      const response = handleTelegramCommand(this.ctx.db, text, chatId);
+      const adapter = this.adapters.get("telegram");
+      if (adapter && response?.payload?.text) {
+        await adapter.send(chatId, response.payload.text, { parseMode: "markdown" });
+      }
+      return;
+    }
+
+    // 查找 binding（chat_id → workspace_id）
+    const binding = this.ctx.db.query(
+      "SELECT workspace_id FROM telegram_binding WHERE chat_id = ?"
+    ).get(chatId) as any;
+
+    if (!binding) {
+      // 未绑定，发送帮助信息
+      const adapter = this.adapters.get(platform);
+      if (adapter) {
+        await adapter.send(chatId, "⚠️ This chat is not bound to any workspace. Use /bind <workspace_name> to link it.");
+      }
+      return;
+    }
+
+    // 桥接到 workspace chat (lazy import to avoid circular deps)
+    const { insertChatMessage, runAgentLoop } = await import("../api/chat");
+    const humanMsg = insertChatMessage(this.ctx, binding.workspace_id, "human", platform, text);
+    this.ctx.bus.publish("chat.message", {
+      workspace_id: binding.workspace_id,
+      message_id: humanMsg.id,
+      seq: humanMsg.seq,
+      sender_type: "human",
+      source: platform,
+      tg_chat_id: chatId,
+    });
+
+    // 发送 typing 指示器
+    const adapter = this.adapters.get(platform);
+    adapter?.sendTyping(chatId).catch(() => {});
+
+    // 触发 agent loop
+    runAgentLoop(this.ctx, binding.workspace_id).catch((err) => {
+      console.error(`[gateway] agent loop error for ${binding.workspace_id}:`, err.message);
+    });
+  }
+
+  /** Agent 回复推送回对应平台 */
+  private async handleAgentReply(event: any) {
+    if (event.sender_type !== "agent") return;
+    if (event.source && this.adapters.has(event.source)) return; // 避免回声
+
+    const workspaceId = event.workspace_id;
+    if (!workspaceId) return;
+
+    // 查找该 workspace 绑定的所有 chat
+    const bindings = this.ctx.db.query(
+      "SELECT chat_id FROM telegram_binding WHERE workspace_id = ?"
+    ).all(workspaceId) as any[];
+    if (!bindings?.length) return;
+
+    // 获取消息内容
+    const msg = event.message_id
+      ? this.ctx.db.query("SELECT content, sender_id FROM chat_message WHERE id = ?").get(event.message_id) as any
+      : null;
+    if (!msg?.content) return;
+
+    const formattedText = `🤖 *${msg.sender_id || "MAESTRO"}*\n${msg.content}`;
+
+    // 推送到所有绑定的 chat（目前只有 telegram，后续扩展）
+    for (const binding of bindings) {
+      // 根据 binding 确定平台（目前 telegram_binding 表只存 telegram）
+      const adapter = this.adapters.get("telegram");
+      if (!adapter) continue;
+
+      // 消息分片
+      const chunks = chunkText(formattedText, adapter.textChunkLimit);
+      for (const chunk of chunks) {
+        await adapter.send(binding.chat_id, chunk, { parseMode: "markdown" });
+      }
+    }
+  }
+}
+
+/** 消息分片：按 limit 切分长文本 */
+export function chunkText(text: string, limit: number): string[] {
+  if (text.length <= limit) return [text];
+  const chunks: string[] = [];
+  let remaining = text;
+  while (remaining.length > 0) {
+    if (remaining.length <= limit) {
+      chunks.push(remaining);
+      break;
+    }
+    // 尝试在换行符处切分
+    let splitAt = remaining.lastIndexOf("\n", limit);
+    if (splitAt < limit * 0.5) splitAt = limit; // 没有合适的换行，硬切
+    chunks.push(remaining.slice(0, splitAt));
+    remaining = remaining.slice(splitAt);
+  }
+  return chunks;
+}

package/src/gateway/types.ts

@@ -0,0 +1,77 @@
+/**
+ * Gateway Platform Adapter 统一接口
+ *
+ * 参考 Hermes (BasePlatformAdapter) 和 OpenClaw (ChannelOutboundAdapter) 的设计。
+ * 所有消息平台（Telegram、Discord、Slack、飞书等）实现此接口。
+ */
+
+export interface SendResult {
+  ok: boolean;
+  messageId?: string;
+  error?: string;
+}
+
+export interface InboundMessage {
+  platform: string;
+  chatId: string;
+  text: string;
+  senderId?: string;
+  senderName?: string;
+  threadId?: string;
+  raw?: unknown;
+}
+
+export interface SendOptions {
+  parseMode?: "markdown" | "html" | "plain";
+  threadId?: string;
+  replyToMessageId?: string;
+}
+
+/**
+ * 每个平台 Adapter 必须实现的接口
+ */
+export interface PlatformAdapter {
+  /** 平台标识 */
+  readonly platform: string;
+
+  /** 单条消息最大字符数（超过自动分片） */
+  readonly textChunkLimit: number;
+
+  /** 连接到平台，开始监听消息 */
+  connect(): Promise<boolean>;
+
+  /** 断开连接，停止监听 */
+  disconnect(): Promise<void>;
+
+  /** 发送文本消息 */
+  send(chatId: string, text: string, options?: SendOptions): Promise<SendResult>;
+
+  /** 发送 typing 指示器 */
+  sendTyping(chatId: string): Promise<void>;
+}
+
+/**
+ * Adapter 收到消息后的回调
+ */
+export type MessageHandler = (msg: InboundMessage) => Promise<void>;
+
+/**
+ * Gateway 配置
+ */
+export interface GatewayConfig {
+  telegram?: {
+    botToken: string;
+    pollingTimeout?: number;
+  };
+  discord?: {
+    botToken: string;
+  };
+  slack?: {
+    appToken: string;
+    botToken: string;
+  };
+  feishu?: {
+    appId: string;
+    appSecret: string;
+  };
+}