@johpaz/hive-agents 0.0.34 → 0.0.36

package/packages/core/src/channels/telegram.ts

@@ -0,0 +1,612 @@
+import { Bot, GrammyError, InputFile, type Context } from "grammy";
+import { BaseChannel, type ChannelConfig, type IncomingMessage, type OutboundMessage } from "./base.ts";
+import { logger } from "../utils/logger.ts";
+import { getDb } from "../storage/sqlite.ts";
+
+export interface TelegramConfig extends ChannelConfig {
+  botToken: string;
+  groups?: boolean;
+}
+
+export class TelegramChannel extends BaseChannel {
+  name = "telegram";
+  accountId: string;
+  config: TelegramConfig;
+
+  private bot?: Bot;
+  private log = logger.child("telegram");
+  private chatIdCache: Map<string, number> = new Map();
+  private messageIdCache: Map<string, number> = new Map();
+  // Deduplication: records recently processed message_ids to avoid double sends
+  private recentlyProcessed: Map<number, number> = new Map();
+
+  constructor(accountId: string, config: TelegramConfig) {
+    super();
+    this.accountId = accountId;
+    this.config = {
+      ...config,
+      dmPolicy: config.dmPolicy ?? "open",
+      allowFrom: config.allowFrom ?? [],
+      enabled: config.enabled ?? true,
+    };
+  }
+
+  async start(): Promise<void> {
+    if (this.running) {
+      this.log.warn("Telegram bot is already running, skipping start");
+      return;
+    }
+
+    if (!this.config.botToken) {
+      throw new Error("Telegram bot token not configured");
+    }
+
+    this.bot = new Bot(this.config.botToken);
+
+    this.bot.on("message", async (ctx: Context) => {
+      await this.handleTelegramMessage(ctx);
+    });
+
+    // Note: edited_message intentionally NOT handled — editing a message
+    // should not trigger a new agent response (was causing double sends).
+
+    this.bot.catch((err: Error) => {
+      this.log.error(`Telegram error: ${err.message}`);
+    });
+
+    this.bot.start({
+      onStart: () => {
+        this.running = true;
+        this.log.info(`Telegram bot started: @${this.bot?.botInfo?.username ?? "unknown"}`);
+        try {
+          getDb().query(`UPDATE channels SET status = 'connected' WHERE id = ?`).run(this.accountId);
+        } catch { /* ignore DB errors */ }
+      },
+    }).catch((error: Error) => {
+      this.log.error(`Telegram bot error: ${error.message}`);
+      this.running = false;
+      try {
+        getDb().query(`UPDATE channels SET status = 'error' WHERE id = ?`).run(this.accountId);
+      } catch { /* ignore DB errors */ }
+    });
+  }
+
+  private async handleTelegramMessage(ctx: Context): Promise<void> {
+    const message = ctx.message;
+    if (!message) return;
+
+    const chatId = message.chat.id.toString();
+    const userId = message.from?.id?.toString() ?? "unknown";
+    const isGroup = message.chat.type === "group" || message.chat.type === "supergroup";
+    const kind = isGroup ? "group" : "direct";
+    const peerId = isGroup
+      ? `${message.chat.id}:${message.from?.id ?? "unknown"}`
+      : chatId;
+    const messageId = message.message_id;
+
+    if (message.from?.is_bot) {
+      return;
+    }
+
+    // Deduplication: ignore message_ids already processed in the last 60 seconds
+    const now = Date.now();
+    if (this.recentlyProcessed.has(messageId)) {
+      this.log.debug(`Duplicate message_id ${messageId} ignored`);
+      return;
+    }
+    this.recentlyProcessed.set(messageId, now);
+    // Clean up old entries (> 60s) to prevent unbounded growth
+    for (const [id, ts] of this.recentlyProcessed) {
+      if (now - ts > 60_000) this.recentlyProcessed.delete(id);
+    }
+
+    const text = message.text;
+    const isCommand = text?.startsWith("/") ?? false;
+
+    if (text === "/myid" || text?.startsWith("/myid@")) {
+      await ctx.reply(
+        `🆔 Tu Telegram ID es: <code>${userId}</code>\n\n` +
+        `Para autorizarte, ejecuta:\n` +
+        `<code>hive config set channels.telegram.accounts.default.allowFrom.+ "tg:${userId}"</code>`,
+        { parse_mode: "HTML" }
+      );
+      return;
+    }
+
+    if (text === "/start" || text?.startsWith("/start@")) {
+      const agentName = "Bee";
+      await ctx.reply(
+        `¡Hola! Soy ${agentName}, tu asistente personal.\n\n` +
+        `Tu Telegram ID: <code>${userId}</code>\n\n` +
+        `Para empezar a usar el bot, asegúrate de estar autorizado.`,
+        { parse_mode: "HTML" }
+      );
+      return;
+    }
+
+    if (text === "/help" || text?.startsWith("/help@")) {
+      await ctx.reply(this.getHelpMessage(userId), { parse_mode: "HTML" });
+      return;
+    }
+
+    if (text === "/stop" || text?.startsWith("/stop@")) {
+      await ctx.reply("⏹ Detención actual cancelada.", { parse_mode: "HTML" });
+      return;
+    }
+
+    if (text === "/new" || text?.startsWith("/new@")) {
+      await ctx.reply("🔄 Sesión reiniciada.", { parse_mode: "HTML" });
+      return;
+    }
+
+    if (!isGroup && !this.isUserAllowed(chatId)) {
+      this.log.debug(`Message from unauthorized user: ${chatId}`);
+      const rejectMsg = this.config.dmPolicy === "allowlist"
+        ? `⛔ No estás autorizado.\n\n` +
+        `Tu Telegram ID: <code>${userId}</code>\n\n` +
+        `Para autorizarte:\n` +
+        `1. Ejecuta en el servidor: <code>hive config edit</code>\n` +
+        `2. Añade bajo channels.telegram.accounts.default.allowFrom:\n` +
+        `<pre>  - "tg:${userId}"</pre>\n` +
+        `3. Ejecuta: <code>hive reload</code>`
+        : `⛔ No estás autorizado para usar este bot.\n\n` +
+        `Tu Telegram ID: <code>${userId}</code>`;
+      await ctx.reply(rejectMsg, { parse_mode: "HTML" });
+      return;
+    }
+
+    if (isGroup && !(this.config.groups ?? false)) {
+      return;
+    }
+
+  let content = text;
+  let contentType = "text";
+  let image: IncomingMessage["image"];
+  let document_: IncomingMessage["document"];
+
+  if (message.photo && !text) {
+    const caption = message.caption ?? "";
+    contentType = "photo";
+    try {
+      const photos = message.photo;
+      const largest = photos[photos.length - 1];
+      if (largest && this.bot) {
+        const file = await this.bot.api.getFile(largest.file_id);
+        if (file.file_path) {
+          image = {
+            url: `https://api.telegram.org/file/bot${this.config.botToken}/${file.file_path}`,
+            mimeType: "image/jpeg",
+            caption: caption || undefined,
+          };
+        }
+      }
+    } catch (err) {
+      this.log.warn(`Failed to download photo: ${(err as Error).message}`);
+    }
+    content = caption || "";
+  }
+
+    if (message.voice) {
+      const voice = message.voice;
+      const fileId = voice.file_id;
+
+      let audioBuffer: Buffer | undefined;
+      let audioUrl: string | undefined;
+
+      try {
+        const file = await this.bot!.api.getFile(fileId);
+        const filePath = file.file_path;
+        if (filePath) {
+          audioUrl = `https://api.telegram.org/file/bot${this.config.botToken}/${filePath}`;
+        }
+      } catch (error) {
+        this.log.error(`Failed to get voice file: ${(error as Error).message}`);
+      }
+
+      const msgSessionId = this.formatSessionId(peerId, kind);
+
+      const incomingMessage: IncomingMessage = {
+        sessionId: msgSessionId,
+        channel: "telegram",
+        accountId: this.accountId,
+        peerId,
+        peerKind: kind,
+        content: "",
+        audio: audioBuffer ? { buffer: audioBuffer } : audioUrl ? { url: audioUrl, mimeType: "audio/ogg" } : undefined,
+        metadata: {
+          telegram: {
+            chatId: message.chat.id,
+            userId: message.from?.id,
+            username: message.from?.username,
+            messageId,
+            chatType: message.chat.type,
+            contentType: "voice",
+          },
+        },
+        replyToId: message.reply_to_message
+          ? `tg:${message.reply_to_message.message_id}`
+          : undefined,
+      };
+
+      await this.handleMessage(incomingMessage);
+      return;
+    }
+
+    if (message.sticker) {
+      return;
+    }
+
+  if (message.document && !text) {
+    const docName = (message.document as any).file_name ?? "documento";
+    const caption = message.caption ?? "";
+    contentType = "document";
+    try {
+      if (this.bot) {
+        const file = await this.bot.api.getFile(message.document.file_id);
+        if (file.file_path) {
+          document_ = {
+            url: `https://api.telegram.org/file/bot${this.config.botToken}/${file.file_path}`,
+            mimeType: message.document.mime_type || "application/octet-stream",
+            fileName: docName,
+          };
+        }
+      }
+    } catch (err) {
+      this.log.warn(`Failed to download document: ${(err as Error).message}`);
+    }
+    content = caption || "";
+  }
+
+    const sessionId = this.formatSessionId(peerId, kind);
+    this.chatIdCache.set(sessionId, message.chat.id);
+    this.messageIdCache.set(sessionId, messageId);
+
+  const incomingMessage: IncomingMessage = {
+  sessionId,
+  channel: "telegram",
+  accountId: this.accountId,
+  peerId,
+  peerKind: kind,
+  content: content ?? "",
+  image,
+  document: document_,
+  metadata: {
+        telegram: {
+          chatId: message.chat.id,
+          userId: message.from?.id,
+          username: message.from?.username,
+          messageId,
+          chatType: message.chat.type,
+          contentType,
+        },
+      },
+      replyToId: message.reply_to_message
+        ? `tg:${message.reply_to_message.message_id}`
+        : undefined,
+    };
+
+    await this.handleMessage(incomingMessage);
+  }
+
+  private getHelpMessage(_userId: string): string {
+    return `📚 <b>Comandos disponibles:</b>
+
+<code>/myid</code> - Muestra tu Telegram ID
+<code>/start</code> - Iniciar conversación
+<code>/help</code> - Mostrar esta ayuda
+<code>/stop</code> - Detener tarea actual
+<code>/new</code> - Reiniciar sesión
+
+💡 <i>Envía un mensaje para comenzar.</i>`;
+  }
+
+  async stop(): Promise<void> {
+    if (this.bot) {
+      await this.bot.stop();
+      this.running = false;
+      this.log.info("Telegram bot stopped");
+      try {
+        getDb().query(`UPDATE channels SET status = 'disconnected' WHERE id = ?`).run(this.accountId);
+      } catch { /* ignore DB errors */ }
+    }
+  }
+
+  private getChatIdFromSession(sessionId: string): number {
+    const cached = this.chatIdCache.get(sessionId);
+    if (cached) return cached;
+
+    // Group format: "chatId:userId" (e.g. "-1001234567890:123456789")
+    // The chat ID is the first segment before the colon.
+    const colonIdx = sessionId.indexOf(":");
+    if (colonIdx > 0) {
+      const parsed = Number(sessionId.slice(0, colonIdx));
+      if (!isNaN(parsed) && parsed !== 0) return parsed;
+    }
+
+    // Direct format: sessionId is the raw chatId (e.g. stored in user_identities)
+    const direct = Number(sessionId);
+    if (!isNaN(direct) && direct !== 0) return direct;
+
+    return 0;
+  }
+
+  private getMessageIdFromSession(sessionId: string): number | undefined {
+    return this.messageIdCache.get(sessionId);
+  }
+
+  async startTyping(sessionId: string): Promise<void> {
+    if (!this.bot) return;
+
+    const chatId = this.getChatIdFromSession(sessionId);
+    if (isNaN(chatId)) return;
+
+    await this.bot.api.sendChatAction(chatId, "typing");
+
+    const interval = setInterval(async () => {
+      try {
+        await this.bot!.api.sendChatAction(chatId, "typing");
+      } catch {
+        this.stopTyping(sessionId);
+      }
+    }, 4000);
+
+    this.typingIntervals.set(sessionId, interval);
+  }
+
+  async stopTyping(sessionId: string): Promise<void> {
+    const interval = this.typingIntervals.get(sessionId);
+    if (interval) {
+      clearInterval(interval);
+      this.typingIntervals.delete(sessionId);
+    }
+  }
+
+  async send(sessionId: string, message: OutboundMessage): Promise<void> {
+    if (!this.bot) {
+      throw new Error("Telegram bot not started");
+    }
+
+    await this.stopTyping(sessionId);
+
+    const chatId = this.getChatIdFromSession(sessionId);
+
+    if (isNaN(chatId)) {
+      throw new Error(`Invalid chat ID from session: ${sessionId}`);
+    }
+
+    const content = message.content ?? "";
+
+    if (!content || content.trim().length === 0) {
+      this.log.warn(`Empty response from agent, skipping send`, { sessionId, chatId });
+      return;
+    }
+
+    const replyToId = this.getMessageIdFromSession(sessionId);
+    const maxLength = 4096;
+
+    try {
+      if (content.length <= maxLength) {
+        await this.sendWithRetry(chatId, content, replyToId);
+      } else {
+        const chunks = this.chunkMessage(content, maxLength);
+        for (let i = 0; i < chunks.length; i++) {
+          await this.sendWithRetry(chatId, chunks[i]!, i === 0 ? replyToId : undefined);
+          if (i < chunks.length - 1) {
+            await new Promise((resolve) => setTimeout(resolve, 300));
+          }
+        }
+      }
+    } catch (error: unknown) {
+      if (error instanceof GrammyError) {
+        this.log.error(`Telegram API error: ${error.description}`);
+
+        if (error.error_code === 403) {
+          this.log.warn(`Bot was blocked by user: ${chatId}`);
+          return;
+        }
+      } else if (error instanceof Error) {
+        this.log.error(`Telegram send error: ${error.message}`);
+      } else {
+        this.log.error(`Telegram send error: ${String(error)}`);
+      }
+      throw error;
+    }
+  }
+
+  async sendAudio(sessionId: string, audio: Buffer, mimeType: string): Promise<void> {
+    if (!this.bot) {
+      throw new Error("Telegram bot not started");
+    }
+
+    const chatId = this.getChatIdFromSession(sessionId);
+
+    if (isNaN(chatId)) {
+      throw new Error(`Invalid chat ID from session: ${sessionId}`);
+    }
+
+    // Retry logic for sendVoice with exponential backoff
+    const maxRetries = 2;
+    const backoffMs = [3000, 6000];
+
+    for (let attempt = 0; attempt < maxRetries; attempt++) {
+      // Use explicit timeout for sendVoice (30 seconds)
+      const controller = new AbortController();
+      const timeoutId = setTimeout(() => controller.abort(), 30000);
+
+      try {
+        const inputFile = new InputFile(audio, "voice.ogg");
+
+        // Use type assertion to bypass grammY type limitations - signal is supported at runtime
+        // via the underlying fetch API but not exposed in grammy's type definitions
+        await this.bot!.api.sendVoice(chatId, inputFile, {
+          signal: controller.signal,
+        } as any);
+
+        this.log.info(`✅ Voice sent to ${chatId}`);
+        return;
+      } catch (error: unknown) {
+        const err = error as Error & { error_code?: number };
+
+        // Don't retry on client errors (4xx)
+        if (err.error_code === 400) {
+          this.log.error(`Bad Request: ${err.message}`);
+          throw error;
+        }
+
+        if (attempt < maxRetries - 1) {
+          this.log.warn(`sendVoice attempt ${attempt + 1} failed, retrying in ${backoffMs[attempt]}ms: ${err.message}`);
+          await new Promise(resolve => setTimeout(resolve, backoffMs[attempt]));
+        } else {
+          this.log.error(`Telegram sendVoice failed after ${maxRetries} attempts: ${err.message}`);
+          throw error;
+        }
+      } finally {
+        // Always clear the timeout to prevent resource leaks
+        clearTimeout(timeoutId);
+      }
+    }
+  }
+
+  private async sendWithRetry(
+    chatId: number,
+    text: string,
+    replyToId?: number
+  ): Promise<void> {
+    const maxRetries = 3;
+    const backoffMs = [1000, 2000, 4000];
+
+    for (let attempt = 0; attempt < maxRetries; attempt++) {
+      try {
+        const html = this.markdownToHTML(text);
+        const options: any = { parse_mode: "HTML" };
+        if (replyToId) {
+          options.reply_parameters = { message_id: replyToId };
+        }
+        await this.bot!.api.sendMessage(chatId, html, options);
+        return;
+      } catch (error: unknown) {
+        const err = error as Error & { error_code?: number; parameters?: { retry_after?: number } };
+
+        if (err.error_code === 400 && err.message.includes("can't parse entities")) {
+          this.log.warn(`Markdown parsing failed, falling back to plain text for chatId: ${chatId}`);
+          await this.bot!.api.sendMessage(chatId, text, {
+            reply_parameters: replyToId ? { message_id: replyToId } : undefined
+          });
+          return;
+        }
+
+        if (err.error_code === 400) {
+          this.log.error(`Bad Request: ${err.message}`);
+          throw error;
+        }
+
+        if (err.error_code === 429) {
+          const retryAfter = err.parameters?.retry_after ?? 1;
+          this.log.warn(`Rate limited, waiting ${retryAfter}s`);
+          await new Promise((resolve) => setTimeout(resolve, retryAfter * 1000));
+          continue;
+        }
+
+        if (attempt < maxRetries - 1) {
+          this.log.warn(`Send failed, retrying in ${backoffMs[attempt]}ms (attempt ${attempt + 1}/${maxRetries})`);
+          await new Promise((resolve) => setTimeout(resolve, backoffMs[attempt]));
+        } else {
+          throw error;
+        }
+      }
+    }
+  }
+
+  private chunkMessage(content: string, maxLength: number): string[] {
+    const chunks: string[] = [];
+    let remaining = content;
+
+    while (remaining.length > 0) {
+      if (remaining.length <= maxLength) {
+        chunks.push(remaining);
+        break;
+      }
+
+      let splitPoint = remaining.lastIndexOf("\n\n", maxLength);
+      if (splitPoint === -1 || splitPoint < maxLength * 0.5) {
+        splitPoint = remaining.lastIndexOf("\n", maxLength);
+      }
+      if (splitPoint === -1 || splitPoint < maxLength * 0.5) {
+        splitPoint = remaining.lastIndexOf(" ", maxLength);
+      }
+      if (splitPoint === -1 || splitPoint < maxLength * 0.5) {
+        splitPoint = maxLength;
+      }
+
+      chunks.push(remaining.slice(0, splitPoint));
+      remaining = remaining.slice(splitPoint).trim();
+    }
+
+    return chunks;
+  }
+
+  private markdownToHTML(text: string): string {
+    // ── Step 1: extract code blocks before any escaping ────────────────────
+    // Prevents code content from being HTML-escaped or markdown-converted.
+    const codeBlocks: string[] = [];
+    const inlineCodes: string[] = [];
+
+    let out = text
+      // Fenced code blocks (``` ... ```) — strip optional language hint
+      .replace(/```(?:[^\n]*)\n?([\s\S]*?)```/g, (_m, code: string) => {
+        const idx = codeBlocks.push(code.trim()) - 1;
+        return `\x00BLOCK${idx}\x00`;
+      })
+      // Inline code (`...`)
+      .replace(/`([^`\n]+)`/g, (_m, code: string) => {
+        const idx = inlineCodes.push(code) - 1;
+        return `\x00INLINE${idx}\x00`;
+      });
+
+    // ── Step 2: escape HTML entities in the remaining text ─────────────────
+    out = out
+      .replace(/&/g, "&amp;")
+      .replace(/</g, "&lt;")
+      .replace(/>/g, "&gt;");
+
+    // ── Step 3: block-level conversions ────────────────────────────────────
+    // Headers: ### h3, ## h2, # h1 → <b>text</b>
+    out = out.replace(/^#{1,6}\s+(.+)$/gm, "<b>$1</b>");
+
+    // Horizontal rules → blank line
+    out = out.replace(/^---+$/gm, "");
+
+    // ── Step 4: inline conversions ─────────────────────────────────────────
+    // Bold **text** or __text__
+    out = out.replace(/\*\*(.+?)\*\*/gs, "<b>$1</b>");
+    out = out.replace(/__(.+?)__/gs, "<b>$1</b>");
+
+    // Italic *text* (single star, not double) — avoid greedy cross-line
+    out = out.replace(/(?<!\*)\*(?!\*)(.+?)(?<!\*)\*(?!\*)/gs, "<i>$1</i>");
+
+    // Italic _text_ — only match when surrounded by non-word chars (avoids snake_case)
+    out = out.replace(/(?<!\w)_(?!\s)(.+?)(?<!\s)_(?!\w)/gs, "<i>$1</i>");
+
+    // Strikethrough ~~text~~
+    out = out.replace(/~~(.+?)~~/gs, "<s>$1</s>");
+
+    // ── Step 5: restore code placeholders (now safely escaped) ─────────────
+    // Restore inline code
+    out = out.replace(/\x00INLINE(\d+)\x00/g, (_m, i) => {
+      const code = inlineCodes[Number(i)] ?? "";
+      return `<code>${code.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;")}</code>`;
+    });
+
+    // Restore block code
+    out = out.replace(/\x00BLOCK(\d+)\x00/g, (_m, i) => {
+      const code = codeBlocks[Number(i)] ?? "";
+      return `<pre><code>${code.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;")}</code></pre>`;
+    });
+
+    return out;
+  }
+}
+
+export function createTelegramChannel(accountId: string, config: TelegramConfig): TelegramChannel {
+  return new TelegramChannel(accountId, config);
+}