@geminixiang/mama 0.2.0-beta.1 → 0.2.0-beta.10

package/dist/adapters/telegram/bot.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"bot.d.ts","sourceRoot":"","sources":["../../../src/adapters/telegram/bot.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAUhF,MAAM,WAAW,aAAc,SAAQ,QAAQ;IAC7C,IAAI,EAAE,SAAS,GAAG,SAAS,CAAC;IAC5B,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AA2DD,qBAAa,WAAY,YAAW,GAAG;IACrC,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,OAAO,CAAa;IAC5B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,SAAS,CAAuB;IACxC,OAAO,CAAC,WAAW,CAAuB;IAC1C,OAAO,CAAC,MAAM,CAAmC;IACjD,OAAO,CAAC,WAAW,CAAa;IAEhC,YAAY,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,EAQ7E;IAMK,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAiB3B;IAEK,WAAW,CAAC,cAAc,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAGvE;IAEK,aAAa,CAAC,cAAc,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAWnF;IAED,YAAY,CAAC,KAAK,EAAE,QAAQ,GAAG,OAAO,CAmBrC;IAED,eAAe,IAAI,YAAY,CAQ9B;IAMK,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAGlE;IAEK,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAElE;IAEK,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAMvF;IAEK,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAEvE;IAEK,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE9C;IAEK,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAIjF;IAED,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,CAI9C;IAED,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,IAAI,CAS9D;IAED;;;;OAIG;IACG,kBAAkB,CACtB,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,GAAG,GACX,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC,CAyBhD;YAKa,mBAAmB;IAgDjC,OAAO,CAAC,QAAQ;IAShB,OAAO,CAAC,qBAAqB;IAuB7B,OAAO,CAAC,gBAAgB;IAMxB,OAAO,CAAC,SAAS;IAKjB,OAAO,CAAC,kBAAkB;CA0I3B","sourcesContent":["import { appendFileSync, existsSync, mkdirSync, readFileSync, writeFileSync } from \"fs\";\nimport { basename, join } from \"path\";\nimport { Bot as GrammyBot, InputFile } from \"grammy\";\nimport type { Bot, BotEvent, BotHandler, PlatformInfo } from \"../../adapter.js\";\nimport { parseLoginCommand } from \"../../login.js\";\nimport * as log from \"../../log.js\";\nimport { formatAlreadyWorking, formatNothingRunning } from \"../../ui-copy.js\";\nimport { createTelegramAdapters } from \"./context.js\";\n\n// ============================================================================\n// Types\n// ============================================================================\n\nexport interface TelegramEvent extends BotEvent {\n  type: \"message\" | \"command\";\n  userName?: string;\n}\n\ninterface MessageContext {\n  msg: any;\n  text: string;\n  chatId: string;\n  chatType: string;\n  userId: string;\n  userName: string;\n  msgId: string;\n  threadTs: string | undefined;\n  sessionKey: string;\n}\n\nconst TELEGRAM_COMMANDS = [\n  { command: \"start\", description: \"Show the welcome message\" },\n  { command: \"help\", description: \"Show help\" },\n  { command: \"stop\", description: \"Stop the current task\" },\n  { command: \"new\", description: \"Start a new conversation\" },\n  { command: \"login\", description: \"Open your secure login page\" },\n] as const;\n\n// ============================================================================\n// Per-channel queue for sequential processing\n// ============================================================================\n\ntype QueuedWork = () => Promise<void>;\n\nclass ChannelQueue {\n  private queue: QueuedWork[] = [];\n  private processing = false;\n\n  enqueue(work: QueuedWork): void {\n    this.queue.push(work);\n    this.processNext();\n  }\n\n  size(): number {\n    return this.queue.length;\n  }\n\n  private async processNext(): Promise<void> {\n    if (this.processing || this.queue.length === 0) return;\n    this.processing = true;\n    const work = this.queue.shift()!;\n    try {\n      await work();\n    } catch (err) {\n      log.logWarning(\"Telegram queue error\", err instanceof Error ? err.message : String(err));\n    }\n    this.processing = false;\n    this.processNext();\n  }\n}\n\n// ============================================================================\n// TelegramBot\n// ============================================================================\n\nexport class TelegramBot implements Bot {\n  private client: GrammyBot;\n  private handler: BotHandler;\n  private botToken: string;\n  private workingDir: string;\n  private botUserId: string | null = null;\n  private botUsername: string | null = null;\n  private queues = new Map<string, ChannelQueue>();\n  private startupTime: number = 0;\n\n  constructor(handler: BotHandler, config: { token: string; workingDir: string }) {\n    this.handler = handler;\n    this.botToken = config.token;\n    this.workingDir = config.workingDir;\n    this.client = new GrammyBot(config.token);\n    this.client.catch((err) => {\n      log.logWarning(\"Telegram error\", err instanceof Error ? err.message : String(err));\n    });\n  }\n\n  // ==========================================================================\n  // Public API (implements Bot)\n  // ==========================================================================\n\n  async start(): Promise<void> {\n    const me = await this.client.api.getMe();\n    this.botUserId = String(me.id);\n    this.botUsername = me.username ?? null;\n    this.startupTime = Date.now();\n\n    await this.client.api.setMyCommands([...TELEGRAM_COMMANDS]);\n\n    this.setupEventHandlers();\n\n    // Start polling in background (bot.start() runs indefinitely)\n    this.client.start().catch((err) => {\n      log.logWarning(\"Telegram polling error\", err instanceof Error ? err.message : String(err));\n    });\n\n    log.logConnected();\n    log.logInfo(`Telegram bot started as @${this.botUsername ?? this.botUserId}`);\n  }\n\n  async postMessage(conversationId: string, text: string): Promise<string> {\n    const result = await this.postMessageRaw(parseInt(conversationId), text);\n    return String(result);\n  }\n\n  async updateMessage(conversationId: string, ts: string, text: string): Promise<void> {\n    try {\n      await this.client.api.editMessageText(parseInt(conversationId), parseInt(ts), text, {\n        parse_mode: \"HTML\",\n      });\n    } catch (err) {\n      const msg = err instanceof Error ? err.message : String(err);\n      if (!msg.includes(\"message is not modified\")) {\n        throw err;\n      }\n    }\n  }\n\n  enqueueEvent(event: BotEvent): boolean {\n    const queue = this.getQueue(event.conversationId);\n    if (queue.size() >= 5) {\n      log.logWarning(\n        `Event queue full for ${event.conversationId}, discarding: ${event.text.substring(0, 50)}`,\n      );\n      return false;\n    }\n    log.logInfo(`Enqueueing event for ${event.conversationId}: ${event.text.substring(0, 50)}`);\n    queue.enqueue(() => {\n      const telegramEvent: TelegramEvent = {\n        ...event,\n        type: \"message\",\n        conversationId: event.conversationId,\n      };\n      const adapters = createTelegramAdapters(telegramEvent, this, true);\n      return this.handler.handleEvent(event, this, adapters, true);\n    });\n    return true;\n  }\n\n  getPlatformInfo(): PlatformInfo {\n    return {\n      name: \"telegram\",\n      formattingGuide:\n        '## Telegram Formatting (HTML mode)\\nBold: <b>text</b>, Italic: <i>text</i>, Code: <code>code</code>, Pre: <pre>code</pre>\\nLinks: <a href=\"url\">text</a>',\n      channels: [],\n      users: [],\n    };\n  }\n\n  // ==========================================================================\n  // Internal helpers (used by context.ts)\n  // ==========================================================================\n\n  async postMessageRaw(chatId: number, text: string): Promise<number> {\n    const result = await this.client.api.sendMessage(chatId, text, { parse_mode: \"HTML\" });\n    return result.message_id;\n  }\n\n  async postPlainMessage(chatId: number, text: string): Promise<void> {\n    await this.client.api.sendMessage(chatId, text);\n  }\n\n  async postReply(chatId: number, replyToMessageId: number, text: string): Promise<number> {\n    const result = await this.client.api.sendMessage(chatId, text, {\n      parse_mode: \"HTML\",\n      reply_parameters: { message_id: replyToMessageId },\n    });\n    return result.message_id;\n  }\n\n  async deleteMessageRaw(chatId: number, messageId: number): Promise<void> {\n    await this.client.api.deleteMessage(chatId, messageId);\n  }\n\n  async sendTyping(chatId: number): Promise<void> {\n    await this.client.api.sendChatAction(chatId, \"typing\");\n  }\n\n  async uploadFile(channel: string, filePath: string, title?: string): Promise<void> {\n    const fileName = title ?? basename(filePath);\n    const fileContent = readFileSync(filePath);\n    await this.client.api.sendDocument(parseInt(channel), new InputFile(fileContent, fileName));\n  }\n\n  logToFile(channel: string, entry: object): void {\n    const channelDir = join(this.workingDir, channel);\n    if (!existsSync(channelDir)) mkdirSync(channelDir, { recursive: true });\n    appendFileSync(join(channelDir, \"log.jsonl\"), `${JSON.stringify(entry)}\\n`);\n  }\n\n  logBotResponse(channel: string, text: string, ts: string): void {\n    this.logToFile(channel, {\n      date: new Date().toISOString(),\n      ts,\n      user: \"bot\",\n      text,\n      attachments: [],\n      isBot: true,\n    });\n  }\n\n  /**\n   * Process attachments from a Telegram message\n   * Downloads files before returning metadata so the agent can read them immediately\n   * Returns format compatible with ChatMessage: { name: string, localPath: string }[]\n   */\n  async processAttachments(\n    chatId: string,\n    message: any,\n  ): Promise<{ name: string; localPath: string }[]> {\n    const downloads: Array<Promise<{ name: string; localPath: string } | null>> = [];\n\n    // Handle photos (take the largest size for best quality)\n    if (message.photo && message.photo.length > 0) {\n      const photos = message.photo;\n      const photo = photos[photos.length - 1]; // Largest photo\n      const fileId = photo.file_id;\n\n      downloads.push(this.processTelegramFile(chatId, fileId, `photo_${message.message_id}.jpg`));\n    }\n\n    // Handle documents\n    if (message.document) {\n      const doc = message.document;\n      const fileId = doc.file_id;\n      const fileName = doc.file_name ?? `document_${message.message_id}`;\n\n      downloads.push(this.processTelegramFile(chatId, fileId, fileName));\n    }\n\n    const attachments = await Promise.all(downloads);\n    return attachments.filter(\n      (attachment): attachment is { name: string; localPath: string } => attachment !== null,\n    );\n  }\n\n  /**\n   * Download a file from Telegram and return attachment metadata\n   */\n  private async processTelegramFile(\n    chatId: string,\n    fileId: string,\n    originalName: string,\n  ): Promise<{ name: string; localPath: string } | null> {\n    try {\n      // Get file info from Telegram\n      const file = await this.client.api.getFile(fileId);\n      if (!file.file_path) {\n        log.logWarning(\"Telegram file has no path\", fileId);\n        return null;\n      }\n\n      // Generate local filename\n      const ts = Date.now();\n      const sanitizedName = originalName.replace(/[^a-zA-Z0-9._-]/g, \"_\");\n      const filename = `${ts}_${sanitizedName}`;\n      const localPath = `${chatId}/attachments/${filename}`;\n      const attachmentsDir = join(this.workingDir, chatId, \"attachments\");\n\n      if (!existsSync(attachmentsDir)) mkdirSync(attachmentsDir, { recursive: true });\n\n      // Construct download URL\n      const downloadUrl = `https://api.telegram.org/file/bot${this.botToken}/${file.file_path}`;\n\n      // Download the file\n      const response = await fetch(downloadUrl);\n      if (!response.ok) {\n        throw new Error(`HTTP ${response.status}: ${response.statusText}`);\n      }\n\n      const buffer = await response.arrayBuffer();\n      writeFileSync(join(attachmentsDir, filename), Buffer.from(buffer));\n\n      return {\n        name: originalName,\n        localPath: localPath,\n      };\n    } catch (err) {\n      log.logWarning(`Failed to process Telegram file`, `${originalName}: ${err}`);\n      return null;\n    }\n  }\n\n  // ==========================================================================\n  // Private - Event Handlers\n  // ==========================================================================\n\n  private getQueue(channelId: string): ChannelQueue {\n    let queue = this.queues.get(channelId);\n    if (!queue) {\n      queue = new ChannelQueue();\n      this.queues.set(channelId, queue);\n    }\n    return queue;\n  }\n\n  private extractMessageContext(msg: any): MessageContext | null {\n    if (!msg) return null;\n    if (msg.date * 1000 < this.startupTime) return null;\n    if (msg.from?.is_bot) return null;\n\n    const text = msg.text ?? msg.caption ?? \"\";\n    if (!text && !msg.document && !msg.photo) return null;\n\n    const chatId = String(msg.chat.id);\n    const chatType = msg.chat.type;\n    const userId = String(msg.from?.id ?? \"unknown\");\n    const userName = msg.from?.username ?? msg.from?.first_name ?? userId;\n    const msgId = String(msg.message_id);\n    const replyToId = msg.reply_to_message?.message_id;\n    const threadTs = replyToId ? String(replyToId) : undefined;\n\n    // Private chats: single session per chat (no per-message splitting)\n    // Groups: per-thread sessions (use reply chain or unique message id)\n    const sessionKey = chatType === \"private\" ? chatId : `${chatId}:${threadTs ?? msgId}`;\n\n    return { msg, text, chatId, chatType, userId, userName, msgId, threadTs, sessionKey };\n  }\n\n  private isAddressedToBot(text: string, chatType: string): boolean {\n    if (chatType === \"private\") return true;\n    if (!this.botUsername) return false;\n    return text.toLowerCase().includes(`@${this.botUsername.toLowerCase()}`);\n  }\n\n  private cleanText(text: string): string {\n    if (!this.botUsername) return text.trim();\n    return text.replace(new RegExp(`@${this.botUsername}`, \"gi\"), \"\").trim();\n  }\n\n  private setupEventHandlers(): void {\n    // --- Slash commands (registered before catch-all so grammY intercepts them) ---\n\n    this.client.command(\"start\", async (ctx) => {\n      const mc = this.extractMessageContext(ctx.message);\n      if (!mc) return;\n      await this.postMessageRaw(\n        parseInt(mc.chatId),\n        [\n          \"<b>Welcome!</b>\",\n          \"\",\n          \"I'm an AI coding agent. Send me a message or use these commands:\",\n          \"\",\n          \"/new — Start a new conversation\",\n          \"/stop — Stop the current task\",\n          \"/help — Show help\",\n          \"/login — Open your secure login page\",\n        ].join(\"\\n\"),\n      );\n    });\n\n    this.client.command(\"help\", async (ctx) => {\n      const mc = this.extractMessageContext(ctx.message);\n      if (!mc) return;\n      await this.postMessageRaw(\n        parseInt(mc.chatId),\n        [\n          \"<b>Available commands:</b>\",\n          \"\",\n          \"/start — Show the welcome message\",\n          \"/help — Show help\",\n          \"/stop — Stop the current task\",\n          \"/new — Start a new conversation\",\n          \"/login — Open your secure login page\",\n          \"\",\n          \"You can also send a regular message to chat with the agent.\",\n        ].join(\"\\n\"),\n      );\n    });\n\n    this.client.command(\"stop\", async (ctx) => {\n      const mc = this.extractMessageContext(ctx.message);\n      if (!mc) return;\n      if (this.handler.isRunning(mc.sessionKey)) {\n        await this.handler.handleStop(mc.sessionKey, mc.chatId, this);\n      } else {\n        await this.postMessage(mc.chatId, formatNothingRunning(\"telegram\"));\n      }\n    });\n\n    this.client.command(\"new\", async (ctx) => {\n      const mc = this.extractMessageContext(ctx.message);\n      if (!mc) return;\n      await this.handler.handleNew(mc.sessionKey, mc.chatId, this);\n    });\n\n    this.client.command(\"login\", async (ctx) => {\n      const mc = this.extractMessageContext(ctx.message);\n      if (!mc) return;\n      await this.handler.handleLogin(\n        \"telegram\",\n        mc.userId,\n        mc.chatId,\n        this,\n        mc.text,\n        mc.chatType === \"private\",\n      );\n    });\n\n    // --- Catch-all for regular (non-command) messages ---\n\n    this.client.on(\"message\", async (ctx) => {\n      const mc = this.extractMessageContext(ctx.message);\n      if (!mc) return;\n\n      // In groups, only respond when addressed to bot\n      if (!this.isAddressedToBot(mc.text, mc.chatType)) return;\n\n      const cleanedText = this.cleanText(mc.text);\n\n      if (parseLoginCommand(cleanedText)) {\n        await this.handler.handleLogin(\n          \"telegram\",\n          mc.userId,\n          mc.chatId,\n          this,\n          cleanedText,\n          mc.chatType === \"private\",\n        );\n        return;\n      }\n\n      // Process attachments\n      const processedAttachments = await this.processAttachments(mc.chatId, mc.msg);\n\n      const event: TelegramEvent = {\n        type: \"message\",\n        conversationId: mc.chatId,\n        ts: mc.msgId,\n        thread_ts: mc.threadTs,\n        sessionKey: mc.sessionKey,\n        user: mc.userId,\n        userName: mc.userName,\n        text: cleanedText,\n        attachments: processedAttachments,\n      };\n\n      // Log the message\n      this.logToFile(mc.chatId, {\n        date: new Date(mc.msg.date * 1000).toISOString(),\n        ts: mc.msgId,\n        user: mc.userId,\n        userName: mc.userName,\n        text: cleanedText,\n        attachments: processedAttachments,\n        isBot: false,\n      });\n\n      // Handle bare \"stop\" text (backward compat)\n      if (cleanedText.toLowerCase() === \"stop\") {\n        if (this.handler.isRunning(mc.sessionKey)) {\n          await this.handler.handleStop(mc.sessionKey, mc.chatId, this);\n        } else {\n          await this.postMessage(mc.chatId, formatNothingRunning(\"telegram\"));\n        }\n        return;\n      }\n\n      if (this.handler.isRunning(mc.sessionKey)) {\n        await this.postMessage(mc.chatId, formatAlreadyWorking(\"telegram\", \"/stop\"));\n      } else {\n        this.getQueue(mc.sessionKey).enqueue(() => {\n          const adapters = createTelegramAdapters(event, this, false);\n          return this.handler.handleEvent(event, this, adapters, false);\n        });\n      }\n    });\n  }\n}\n"]}
+{"version":3,"file":"bot.d.ts","sourceRoot":"","sources":["../../../src/adapters/telegram/bot.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AA8BhF,MAAM,WAAW,aAAc,SAAQ,QAAQ;IAC7C,IAAI,EAAE,SAAS,GAAG,SAAS,CAAC;IAC5B,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAuBD,qBAAa,WAAY,YAAW,GAAG;IACrC,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,OAAO,CAAa;IAC5B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,SAAS,CAAuB;IACxC,OAAO,CAAC,WAAW,CAAuB;IAC1C,OAAO,CAAC,MAAM,CAAmC;IACjD,OAAO,CAAC,WAAW,CAAa;IAEhC,YAAY,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,EAQ7E;IAMK,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAwB3B;IAEK,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAGhE;IAEK,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAwB5E;IAED,YAAY,CAAC,KAAK,EAAE,QAAQ,GAAG,OAAO,CAerC;IAED,eAAe,IAAI,YAAY,CAQ9B;IAMK,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAgBlE;IAEK,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAIlE;IAEK,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAoBvF;IAEK,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAEvE;IAEK,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE9C;IAEK,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAMjF;IAED,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,CAE9C;IAED,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,IAAI,CAE9D;IAED;;;;OAIG;IACG,kBAAkB,CACtB,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,GAAG,GACX,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC,CAyBhD;YAKa,mBAAmB;IAgDjC,OAAO,CAAC,QAAQ;IAShB,OAAO,CAAC,qBAAqB;IAsC7B,OAAO,CAAC,gBAAgB;IAMxB,OAAO,CAAC,SAAS;IAKjB,OAAO,CAAC,UAAU;IAIlB,OAAO,CAAC,iBAAiB;IAUzB,OAAO,CAAC,kBAAkB;CA+H3B","sourcesContent":["import { existsSync, mkdirSync, readFileSync, writeFileSync } from \"fs\";\nimport { basename, join } from \"path\";\nimport { Bot as GrammyBot, InputFile } from \"grammy\";\nimport type { Bot, BotEvent, BotHandler, PlatformInfo } from \"../../adapter.js\";\nimport * as log from \"../../log.js\";\nimport { resolveChatSessionKey } from \"../../session-policy.js\";\nimport { evaluateAutoReplyPolicy } from \"../../trigger.js\";\nimport { formatAlreadyWorking, formatNothingRunning } from \"../../ui-copy.js\";\nimport {\n  appendBotResponseLog,\n  appendChannelLog,\n  ChannelQueue,\n  resolveOnlyScopedStopTarget,\n  resolveStopTarget,\n  withRetry,\n} from \"../shared.js\";\nimport { createTelegramAdapters } from \"./context.js\";\nimport { escapeTelegramHtml } from \"./html.js\";\n\n// grammY surfaces Telegram errors as `GrammyError` with `error_code` mirroring\n// the Bot API. 429 is the rate-limit status; the response also includes\n// `parameters.retry_after` but exponential backoff is good enough here.\nfunction telegramIsRateLimited(err: Error): boolean {\n  return (err as { error_code?: number }).error_code === 429;\n}\n\nconst telegramRetry = <T>(fn: () => Promise<T>): Promise<T> =>\n  withRetry(fn, { isRateLimited: telegramIsRateLimited });\n\n// ============================================================================\n// Types\n// ============================================================================\n\nexport interface TelegramEvent extends BotEvent {\n  type: \"message\" | \"command\";\n  userName?: string;\n}\n\ninterface MessageContext {\n  msg: any;\n  text: string;\n  chatId: string;\n  chatType: string;\n  conversationKind: \"direct\" | \"shared\";\n  userId: string;\n  userName: string;\n  msgId: string;\n  threadTs: string | undefined;\n  sessionKey: string;\n}\n\n// ============================================================================\n// TelegramBot\n// ============================================================================\n\nfunction isTelegramHtmlParseError(message: string): boolean {\n  return message.includes(\"can't parse entities\");\n}\n\nexport class TelegramBot implements Bot {\n  private client: GrammyBot;\n  private handler: BotHandler;\n  private botToken: string;\n  private workingDir: string;\n  private botUserId: string | null = null;\n  private botUsername: string | null = null;\n  private queues = new Map<string, ChannelQueue>();\n  private startupTime: number = 0;\n\n  constructor(handler: BotHandler, config: { token: string; workingDir: string }) {\n    this.handler = handler;\n    this.botToken = config.token;\n    this.workingDir = config.workingDir;\n    this.client = new GrammyBot(config.token);\n    this.client.catch((err) => {\n      log.logWarning(\"Telegram error\", err instanceof Error ? err.message : String(err));\n    });\n  }\n\n  // ==========================================================================\n  // Public API (implements Bot)\n  // ==========================================================================\n\n  async start(): Promise<void> {\n    const me = await this.client.api.getMe();\n    this.botUserId = String(me.id);\n    this.botUsername = me.username ?? null;\n    this.startupTime = Date.now();\n\n    await this.client.api.setMyCommands([\n      { command: \"login\", description: \"Store credentials in your private vault\" },\n      { command: \"session\", description: \"Open the current session in the web viewer\" },\n      { command: \"model\", description: \"Switch this conversation's LLM model\" },\n      { command: \"sandbox\", description: \"Show or boost sandbox limits\" },\n      { command: \"stop\", description: \"Stop ongoing conversation\" },\n      { command: \"new\", description: \"Reset conversation history and start fresh\" },\n    ]);\n\n    this.setupEventHandlers();\n\n    // Start polling in background (bot.start() runs indefinitely)\n    this.client.start().catch((err) => {\n      log.logWarning(\"Telegram polling error\", err instanceof Error ? err.message : String(err));\n    });\n\n    log.logConnected(\"Telegram\");\n    log.logInfo(`Telegram bot started as @${this.botUsername ?? this.botUserId}`);\n  }\n\n  async postMessage(channel: string, text: string): Promise<string> {\n    const result = await this.postMessageRaw(parseInt(channel), text);\n    return String(result);\n  }\n\n  async updateMessage(channel: string, ts: string, text: string): Promise<void> {\n    return telegramRetry(async () => {\n      try {\n        await this.client.api.editMessageText(parseInt(channel), parseInt(ts), text, {\n          parse_mode: \"HTML\",\n        });\n      } catch (err) {\n        const msg = err instanceof Error ? err.message : String(err);\n        if (msg.includes(\"message is not modified\")) {\n          return;\n        }\n        if (!isTelegramHtmlParseError(msg)) {\n          throw err;\n        }\n        await this.client.api.editMessageText(\n          parseInt(channel),\n          parseInt(ts),\n          escapeTelegramHtml(text),\n          {\n            parse_mode: \"HTML\",\n          },\n        );\n      }\n    });\n  }\n\n  enqueueEvent(event: BotEvent): boolean {\n    const conversationId = event.conversationId;\n    const queue = this.getQueue(conversationId);\n    if (queue.size() >= 5) {\n      log.logWarning(\n        `Event queue full for ${conversationId}, discarding: ${event.text.substring(0, 50)}`,\n      );\n      return false;\n    }\n    log.logInfo(`Enqueueing event for ${conversationId}: ${event.text.substring(0, 50)}`);\n    queue.enqueue(() => {\n      const adapters = createTelegramAdapters(event as TelegramEvent, this, true);\n      return this.handler.handleEvent(event, this, adapters, true);\n    });\n    return true;\n  }\n\n  getPlatformInfo(): PlatformInfo {\n    return {\n      name: \"telegram\",\n      formattingGuide:\n        '## Telegram Formatting (HTML mode)\\nBold: <b>text</b>, Italic: <i>text</i>, Code: <code>code</code>, Pre: <pre>code</pre>\\nLinks: <a href=\"url\">text</a>',\n      channels: [],\n      users: [],\n    };\n  }\n\n  // ==========================================================================\n  // Internal helpers (used by context.ts)\n  // ==========================================================================\n\n  async postMessageRaw(chatId: number, text: string): Promise<number> {\n    return telegramRetry(async () => {\n      try {\n        const result = await this.client.api.sendMessage(chatId, text, { parse_mode: \"HTML\" });\n        return result.message_id;\n      } catch (err) {\n        const msg = err instanceof Error ? err.message : String(err);\n        if (!isTelegramHtmlParseError(msg)) {\n          throw err;\n        }\n        const result = await this.client.api.sendMessage(chatId, escapeTelegramHtml(text), {\n          parse_mode: \"HTML\",\n        });\n        return result.message_id;\n      }\n    });\n  }\n\n  async postPlainMessage(chatId: number, text: string): Promise<void> {\n    return telegramRetry(async () => {\n      await this.client.api.sendMessage(chatId, text);\n    });\n  }\n\n  async postReply(chatId: number, replyToMessageId: number, text: string): Promise<number> {\n    return telegramRetry(async () => {\n      try {\n        const result = await this.client.api.sendMessage(chatId, text, {\n          parse_mode: \"HTML\",\n          reply_parameters: { message_id: replyToMessageId },\n        });\n        return result.message_id;\n      } catch (err) {\n        const msg = err instanceof Error ? err.message : String(err);\n        if (!isTelegramHtmlParseError(msg)) {\n          throw err;\n        }\n        const result = await this.client.api.sendMessage(chatId, escapeTelegramHtml(text), {\n          parse_mode: \"HTML\",\n          reply_parameters: { message_id: replyToMessageId },\n        });\n        return result.message_id;\n      }\n    });\n  }\n\n  async deleteMessageRaw(chatId: number, messageId: number): Promise<void> {\n    await this.client.api.deleteMessage(chatId, messageId);\n  }\n\n  async sendTyping(chatId: number): Promise<void> {\n    await this.client.api.sendChatAction(chatId, \"typing\");\n  }\n\n  async uploadFile(channel: string, filePath: string, title?: string): Promise<void> {\n    return telegramRetry(async () => {\n      const fileName = title ?? basename(filePath);\n      const fileContent = readFileSync(filePath);\n      await this.client.api.sendDocument(parseInt(channel), new InputFile(fileContent, fileName));\n    });\n  }\n\n  logToFile(channel: string, entry: object): void {\n    appendChannelLog(this.workingDir, channel, entry);\n  }\n\n  logBotResponse(channel: string, text: string, ts: string): void {\n    appendBotResponseLog(this.workingDir, channel, text, ts);\n  }\n\n  /**\n   * Process attachments from a Telegram message\n   * Downloads files before returning metadata so the agent can read them immediately\n   * Returns format compatible with ChatMessage: { name: string, localPath: string }[]\n   */\n  async processAttachments(\n    chatId: string,\n    message: any,\n  ): Promise<{ name: string; localPath: string }[]> {\n    const downloads: Array<Promise<{ name: string; localPath: string } | null>> = [];\n\n    // Handle photos (take the largest size for best quality)\n    if (message.photo && message.photo.length > 0) {\n      const photos = message.photo;\n      const photo = photos[photos.length - 1]; // Largest photo\n      const fileId = photo.file_id;\n\n      downloads.push(this.processTelegramFile(chatId, fileId, `photo_${message.message_id}.jpg`));\n    }\n\n    // Handle documents\n    if (message.document) {\n      const doc = message.document;\n      const fileId = doc.file_id;\n      const fileName = doc.file_name ?? `document_${message.message_id}`;\n\n      downloads.push(this.processTelegramFile(chatId, fileId, fileName));\n    }\n\n    const attachments = await Promise.all(downloads);\n    return attachments.filter(\n      (attachment): attachment is { name: string; localPath: string } => attachment !== null,\n    );\n  }\n\n  /**\n   * Download a file from Telegram and return attachment metadata\n   */\n  private async processTelegramFile(\n    chatId: string,\n    fileId: string,\n    originalName: string,\n  ): Promise<{ name: string; localPath: string } | null> {\n    try {\n      // Get file info from Telegram\n      const file = await this.client.api.getFile(fileId);\n      if (!file.file_path) {\n        log.logWarning(\"Telegram file has no path\", fileId);\n        return null;\n      }\n\n      // Generate local filename\n      const ts = Date.now();\n      const sanitizedName = originalName.replace(/[^a-zA-Z0-9._-]/g, \"_\");\n      const filename = `${ts}_${sanitizedName}`;\n      const localPath = `${chatId}/attachments/${filename}`;\n      const fullDir = join(this.workingDir, chatId, \"attachments\");\n\n      if (!existsSync(fullDir)) mkdirSync(fullDir, { recursive: true });\n\n      // Construct download URL\n      const downloadUrl = `https://api.telegram.org/file/bot${this.botToken}/${file.file_path}`;\n\n      // Download the file\n      const response = await fetch(downloadUrl);\n      if (!response.ok) {\n        throw new Error(`HTTP ${response.status}: ${response.statusText}`);\n      }\n\n      const buffer = await response.arrayBuffer();\n      writeFileSync(join(fullDir, filename), Buffer.from(buffer));\n\n      return {\n        name: originalName,\n        localPath: localPath,\n      };\n    } catch (err) {\n      log.logWarning(`Failed to process Telegram file`, `${originalName}: ${err}`);\n      return null;\n    }\n  }\n\n  // ==========================================================================\n  // Private - Event Handlers\n  // ==========================================================================\n\n  private getQueue(channelId: string): ChannelQueue {\n    let queue = this.queues.get(channelId);\n    if (!queue) {\n      queue = new ChannelQueue(\"Telegram\");\n      this.queues.set(channelId, queue);\n    }\n    return queue;\n  }\n\n  private extractMessageContext(msg: any): MessageContext | null {\n    if (!msg) return null;\n    if (msg.date * 1000 < this.startupTime) return null;\n    if (msg.from?.is_bot) return null;\n\n    const text = msg.text ?? msg.caption ?? \"\";\n    if (!text && !msg.document && !msg.photo) return null;\n\n    const chatId = String(msg.chat.id);\n    const chatType = msg.chat.type;\n    const userId = String(msg.from?.id ?? \"unknown\");\n    const userName = msg.from?.username ?? msg.from?.first_name ?? userId;\n    const msgId = String(msg.message_id);\n    const replyToId = msg.reply_to_message?.message_id;\n    const threadTs = replyToId ? String(replyToId) : undefined;\n    const conversationKind = chatType === \"private\" ? \"direct\" : \"shared\";\n\n    const sessionKey = resolveChatSessionKey({\n      conversationId: chatId,\n      conversationKind,\n      messageId: msgId,\n      threadTs,\n    });\n\n    return {\n      msg,\n      text,\n      chatId,\n      chatType,\n      conversationKind,\n      userId,\n      userName,\n      msgId,\n      threadTs,\n      sessionKey,\n    };\n  }\n\n  private isAddressedToBot(text: string, chatType: string): boolean {\n    if (chatType === \"private\") return true;\n    if (!this.botUsername) return false;\n    return text.toLowerCase().includes(`@${this.botUsername.toLowerCase()}`);\n  }\n\n  private cleanText(text: string): string {\n    if (!this.botUsername) return text.trim();\n    return text.replace(new RegExp(`@${this.botUsername}`, \"gi\"), \"\").trim();\n  }\n\n  private isStopText(text: string): boolean {\n    return /^\\/?stop(?:@\\w+)?$/i.test(text.trim());\n  }\n\n  private resolveStopTarget(mc: MessageContext): string | null {\n    const directTarget = resolveStopTarget({\n      handler: this.handler,\n      conversationId: mc.chatId,\n      sessionKey: mc.sessionKey,\n    });\n    if (directTarget) return directTarget;\n    return resolveOnlyScopedStopTarget(this.handler, mc.chatId);\n  }\n\n  private setupEventHandlers(): void {\n    // --- Slash commands (registered before catch-all so grammY intercepts them) ---\n\n    this.client.command(\"stop\", async (ctx) => {\n      const mc = this.extractMessageContext(ctx.message);\n      if (!mc) return;\n      const stopTarget = this.resolveStopTarget(mc);\n      if (stopTarget) {\n        await this.handler.handleStop(stopTarget, mc.chatId, this);\n      } else {\n        await this.postMessage(mc.chatId, formatNothingRunning(\"telegram\"));\n      }\n    });\n\n    this.client.command(\"new\", async (ctx) => {\n      const mc = this.extractMessageContext(ctx.message);\n      if (!mc) return;\n      await this.handler.handleNewCommand(mc.sessionKey, mc.chatId, this);\n    });\n\n    this.client.command(\"sandbox\", async (ctx) => {\n      const mc = this.extractMessageContext(ctx.message);\n      if (!mc) return;\n      const cleanedText = this.cleanText(mc.text).replace(/^\\/sandbox(?:@\\w+)?/i, \"/pi-sandbox\");\n      const event: TelegramEvent = {\n        type: \"command\",\n        conversationId: mc.chatId,\n        conversationKind: mc.conversationKind,\n        ts: mc.msgId,\n        thread_ts: mc.threadTs,\n        sessionKey: mc.sessionKey,\n        user: mc.userId,\n        userName: mc.userName,\n        text: cleanedText,\n        attachments: [],\n      };\n      this.logToFile(mc.chatId, {\n        date: new Date(mc.msg.date * 1000).toISOString(),\n        ts: mc.msgId,\n        ...(mc.conversationKind === \"shared\" && mc.threadTs ? { threadTs: mc.threadTs } : {}),\n        user: mc.userId,\n        userName: mc.userName,\n        text: cleanedText,\n        attachments: [],\n        isBot: false,\n      });\n      const adapters = createTelegramAdapters(event, this, false);\n      await this.handler.handleEvent(event, this, adapters, false);\n    });\n\n    // --- Catch-all for regular (non-command) messages ---\n\n    this.client.on(\"message\", async (ctx) => {\n      const mc = this.extractMessageContext(ctx.message);\n      if (!mc) return;\n\n      const cleanedText = this.cleanText(mc.text);\n      const addressedToBot = this.isAddressedToBot(mc.text, mc.chatType);\n\n      if (this.isStopText(cleanedText)) {\n        this.logToFile(mc.chatId, {\n          date: new Date(mc.msg.date * 1000).toISOString(),\n          ts: mc.msgId,\n          user: mc.userId,\n          userName: mc.userName,\n          text: cleanedText,\n          attachments: [],\n          isBot: false,\n        });\n\n        const stopTarget = this.resolveStopTarget(mc);\n        if (stopTarget) {\n          await this.handler.handleStop(stopTarget, mc.chatId, this);\n        } else if (addressedToBot || mc.chatType === \"private\") {\n          await this.postMessage(mc.chatId, formatNothingRunning(\"telegram\"));\n        }\n        return;\n      }\n\n      const isAutoReplyCandidate = mc.chatType !== \"private\" && !addressedToBot;\n\n      const eventBase: TelegramEvent = {\n        type: \"message\",\n        conversationId: mc.chatId,\n        conversationKind: mc.conversationKind,\n        ts: mc.msgId,\n        thread_ts: mc.threadTs,\n        sessionKey: mc.sessionKey,\n        user: mc.userId,\n        userName: mc.userName,\n        text: cleanedText,\n      };\n\n      const triggerResult = isAutoReplyCandidate\n        ? await evaluateAutoReplyPolicy({ event: eventBase, workingDir: this.workingDir })\n        : ({ trigger: true, reason: \"addressed\" } as const);\n\n      const logEntryBase = {\n        date: new Date(mc.msg.date * 1000).toISOString(),\n        ts: mc.msgId,\n        ...(mc.conversationKind === \"shared\" && mc.threadTs ? { threadTs: mc.threadTs } : {}),\n        user: mc.userId,\n        userName: mc.userName,\n        text: cleanedText,\n        isBot: false,\n      };\n\n      if (!triggerResult.trigger) {\n        this.logToFile(mc.chatId, { ...logEntryBase, attachments: [] });\n        return;\n      }\n\n      const processedAttachments = await this.processAttachments(mc.chatId, mc.msg);\n      const event: TelegramEvent = { ...eventBase, attachments: processedAttachments };\n\n      this.logToFile(mc.chatId, { ...logEntryBase, attachments: processedAttachments });\n\n      if (this.handler.isRunning(mc.sessionKey)) {\n        await this.postMessage(mc.chatId, formatAlreadyWorking(\"telegram\", \"/stop\"));\n      } else {\n        this.getQueue(mc.sessionKey).enqueue(() => {\n          const adapters = createTelegramAdapters(event, this, false);\n          return this.handler.handleEvent(event, this, adapters, false);\n        });\n      }\n    });\n  }\n}\n"]}

package/dist/adapters/telegram/bot.js

@@ -1,47 +1,26 @@
-import { appendFileSync, existsSync, mkdirSync, readFileSync, writeFileSync } from "fs";
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "fs";
 import { basename, join } from "path";
 import { Bot as GrammyBot, InputFile } from "grammy";
-import { parseLoginCommand } from "../../login.js";
 import * as log from "../../log.js";
+import { resolveChatSessionKey } from "../../session-policy.js";
+import { evaluateAutoReplyPolicy } from "../../trigger.js";
 import { formatAlreadyWorking, formatNothingRunning } from "../../ui-copy.js";
+import { appendBotResponseLog, appendChannelLog, ChannelQueue, resolveOnlyScopedStopTarget, resolveStopTarget, withRetry, } from "../shared.js";
 import { createTelegramAdapters } from "./context.js";
-const TELEGRAM_COMMANDS = [
-    { command: "start", description: "Show the welcome message" },
-    { command: "help", description: "Show help" },
-    { command: "stop", description: "Stop the current task" },
-    { command: "new", description: "Start a new conversation" },
-    { command: "login", description: "Open your secure login page" },
-];
-class ChannelQueue {
-    constructor() {
-        this.queue = [];
-        this.processing = false;
-    }
-    enqueue(work) {
-        this.queue.push(work);
-        this.processNext();
-    }
-    size() {
-        return this.queue.length;
-    }
-    async processNext() {
-        if (this.processing || this.queue.length === 0)
-            return;
-        this.processing = true;
-        const work = this.queue.shift();
-        try {
-            await work();
-        }
-        catch (err) {
-            log.logWarning("Telegram queue error", err instanceof Error ? err.message : String(err));
-        }
-        this.processing = false;
-        this.processNext();
-    }
+import { escapeTelegramHtml } from "./html.js";
+// grammY surfaces Telegram errors as `GrammyError` with `error_code` mirroring
+// the Bot API. 429 is the rate-limit status; the response also includes
+// `parameters.retry_after` but exponential backoff is good enough here.
+function telegramIsRateLimited(err) {
+    return err.error_code === 429;
 }
+const telegramRetry = (fn) => withRetry(fn, { isRateLimited: telegramIsRateLimited });
 // ============================================================================
 // TelegramBot
 // ============================================================================
+function isTelegramHtmlParseError(message) {
+    return message.includes("can't parse entities");
+}
 export class TelegramBot {
     constructor(handler, config) {
         this.botUserId = null;
@@ -64,46 +43,57 @@ export class TelegramBot {
         this.botUserId = String(me.id);
         this.botUsername = me.username ?? null;
         this.startupTime = Date.now();
-        await this.client.api.setMyCommands([...TELEGRAM_COMMANDS]);
+        await this.client.api.setMyCommands([
+            { command: "login", description: "Store credentials in your private vault" },
+            { command: "session", description: "Open the current session in the web viewer" },
+            { command: "model", description: "Switch this conversation's LLM model" },
+            { command: "sandbox", description: "Show or boost sandbox limits" },
+            { command: "stop", description: "Stop ongoing conversation" },
+            { command: "new", description: "Reset conversation history and start fresh" },
+        ]);
         this.setupEventHandlers();
         // Start polling in background (bot.start() runs indefinitely)
         this.client.start().catch((err) => {
             log.logWarning("Telegram polling error", err instanceof Error ? err.message : String(err));
         });
-        log.logConnected();
+        log.logConnected("Telegram");
         log.logInfo(`Telegram bot started as @${this.botUsername ?? this.botUserId}`);
     }
-    async postMessage(conversationId, text) {
-        const result = await this.postMessageRaw(parseInt(conversationId), text);
+    async postMessage(channel, text) {
+        const result = await this.postMessageRaw(parseInt(channel), text);
         return String(result);
     }
-    async updateMessage(conversationId, ts, text) {
-        try {
-            await this.client.api.editMessageText(parseInt(conversationId), parseInt(ts), text, {
-                parse_mode: "HTML",
-            });
-        }
-        catch (err) {
-            const msg = err instanceof Error ? err.message : String(err);
-            if (!msg.includes("message is not modified")) {
-                throw err;
+    async updateMessage(channel, ts, text) {
+        return telegramRetry(async () => {
+            try {
+                await this.client.api.editMessageText(parseInt(channel), parseInt(ts), text, {
+                    parse_mode: "HTML",
+                });
             }
-        }
+            catch (err) {
+                const msg = err instanceof Error ? err.message : String(err);
+                if (msg.includes("message is not modified")) {
+                    return;
+                }
+                if (!isTelegramHtmlParseError(msg)) {
+                    throw err;
+                }
+                await this.client.api.editMessageText(parseInt(channel), parseInt(ts), escapeTelegramHtml(text), {
+                    parse_mode: "HTML",
+                });
+            }
+        });
     }
     enqueueEvent(event) {
-        const queue = this.getQueue(event.conversationId);
+        const conversationId = event.conversationId;
+        const queue = this.getQueue(conversationId);
         if (queue.size() >= 5) {
-            log.logWarning(`Event queue full for ${event.conversationId}, discarding: ${event.text.substring(0, 50)}`);
+            log.logWarning(`Event queue full for ${conversationId}, discarding: ${event.text.substring(0, 50)}`);
             return false;
         }
-        log.logInfo(`Enqueueing event for ${event.conversationId}: ${event.text.substring(0, 50)}`);
+        log.logInfo(`Enqueueing event for ${conversationId}: ${event.text.substring(0, 50)}`);
         queue.enqueue(() => {
-            const telegramEvent = {
-                ...event,
-                type: "message",
-                conversationId: event.conversationId,
-            };
-            const adapters = createTelegramAdapters(telegramEvent, this, true);
+            const adapters = createTelegramAdapters(event, this, true);
             return this.handler.handleEvent(event, this, adapters, true);
         });
         return true;
@@ -120,18 +110,49 @@ export class TelegramBot {
     // Internal helpers (used by context.ts)
     // ==========================================================================
     async postMessageRaw(chatId, text) {
-        const result = await this.client.api.sendMessage(chatId, text, { parse_mode: "HTML" });
-        return result.message_id;
+        return telegramRetry(async () => {
+            try {
+                const result = await this.client.api.sendMessage(chatId, text, { parse_mode: "HTML" });
+                return result.message_id;
+            }
+            catch (err) {
+                const msg = err instanceof Error ? err.message : String(err);
+                if (!isTelegramHtmlParseError(msg)) {
+                    throw err;
+                }
+                const result = await this.client.api.sendMessage(chatId, escapeTelegramHtml(text), {
+                    parse_mode: "HTML",
+                });
+                return result.message_id;
+            }
+        });
     }
     async postPlainMessage(chatId, text) {
-        await this.client.api.sendMessage(chatId, text);
+        return telegramRetry(async () => {
+            await this.client.api.sendMessage(chatId, text);
+        });
     }
     async postReply(chatId, replyToMessageId, text) {
-        const result = await this.client.api.sendMessage(chatId, text, {
-            parse_mode: "HTML",
-            reply_parameters: { message_id: replyToMessageId },
+        return telegramRetry(async () => {
+            try {
+                const result = await this.client.api.sendMessage(chatId, text, {
+                    parse_mode: "HTML",
+                    reply_parameters: { message_id: replyToMessageId },
+                });
+                return result.message_id;
+            }
+            catch (err) {
+                const msg = err instanceof Error ? err.message : String(err);
+                if (!isTelegramHtmlParseError(msg)) {
+                    throw err;
+                }
+                const result = await this.client.api.sendMessage(chatId, escapeTelegramHtml(text), {
+                    parse_mode: "HTML",
+                    reply_parameters: { message_id: replyToMessageId },
+                });
+                return result.message_id;
+            }
         });
-        return result.message_id;
     }
     async deleteMessageRaw(chatId, messageId) {
         await this.client.api.deleteMessage(chatId, messageId);
@@ -140,25 +161,17 @@ export class TelegramBot {
         await this.client.api.sendChatAction(chatId, "typing");
     }
     async uploadFile(channel, filePath, title) {
-        const fileName = title ?? basename(filePath);
-        const fileContent = readFileSync(filePath);
-        await this.client.api.sendDocument(parseInt(channel), new InputFile(fileContent, fileName));
+        return telegramRetry(async () => {
+            const fileName = title ?? basename(filePath);
+            const fileContent = readFileSync(filePath);
+            await this.client.api.sendDocument(parseInt(channel), new InputFile(fileContent, fileName));
+        });
     }
     logToFile(channel, entry) {
-        const channelDir = join(this.workingDir, channel);
-        if (!existsSync(channelDir))
-            mkdirSync(channelDir, { recursive: true });
-        appendFileSync(join(channelDir, "log.jsonl"), `${JSON.stringify(entry)}\n`);
+        appendChannelLog(this.workingDir, channel, entry);
     }
     logBotResponse(channel, text, ts) {
-        this.logToFile(channel, {
-            date: new Date().toISOString(),
-            ts,
-            user: "bot",
-            text,
-            attachments: [],
-            isBot: true,
-        });
+        appendBotResponseLog(this.workingDir, channel, text, ts);
     }
     /**
      * Process attachments from a Telegram message
@@ -200,9 +213,9 @@ export class TelegramBot {
             const sanitizedName = originalName.replace(/[^a-zA-Z0-9._-]/g, "_");
             const filename = `${ts}_${sanitizedName}`;
             const localPath = `${chatId}/attachments/${filename}`;
-            const attachmentsDir = join(this.workingDir, chatId, "attachments");
-            if (!existsSync(attachmentsDir))
-                mkdirSync(attachmentsDir, { recursive: true });
+            const fullDir = join(this.workingDir, chatId, "attachments");
+            if (!existsSync(fullDir))
+                mkdirSync(fullDir, { recursive: true });
             // Construct download URL
             const downloadUrl = `https://api.telegram.org/file/bot${this.botToken}/${file.file_path}`;
             // Download the file
@@ -211,7 +224,7 @@ export class TelegramBot {
                 throw new Error(`HTTP ${response.status}: ${response.statusText}`);
             }
             const buffer = await response.arrayBuffer();
-            writeFileSync(join(attachmentsDir, filename), Buffer.from(buffer));
+            writeFileSync(join(fullDir, filename), Buffer.from(buffer));
             return {
                 name: originalName,
                 localPath: localPath,
@@ -228,7 +241,7 @@ export class TelegramBot {
     getQueue(channelId) {
         let queue = this.queues.get(channelId);
         if (!queue) {
-            queue = new ChannelQueue();
+            queue = new ChannelQueue("Telegram");
             this.queues.set(channelId, queue);
         }
         return queue;
@@ -250,10 +263,25 @@ export class TelegramBot {
         const msgId = String(msg.message_id);
         const replyToId = msg.reply_to_message?.message_id;
         const threadTs = replyToId ? String(replyToId) : undefined;
-        // Private chats: single session per chat (no per-message splitting)
-        // Groups: per-thread sessions (use reply chain or unique message id)
-        const sessionKey = chatType === "private" ? chatId : `${chatId}:${threadTs ?? msgId}`;
-        return { msg, text, chatId, chatType, userId, userName, msgId, threadTs, sessionKey };
+        const conversationKind = chatType === "private" ? "direct" : "shared";
+        const sessionKey = resolveChatSessionKey({
+            conversationId: chatId,
+            conversationKind,
+            messageId: msgId,
+            threadTs,
+        });
+        return {
+            msg,
+            text,
+            chatId,
+            chatType,
+            conversationKind,
+            userId,
+            userName,
+            msgId,
+            threadTs,
+            sessionKey,
+        };
     }
     isAddressedToBot(text, chatType) {
         if (chatType === "private")
@@ -267,45 +295,28 @@ export class TelegramBot {
             return text.trim();
         return text.replace(new RegExp(`@${this.botUsername}`, "gi"), "").trim();
     }
+    isStopText(text) {
+        return /^\/?stop(?:@\w+)?$/i.test(text.trim());
+    }
+    resolveStopTarget(mc) {
+        const directTarget = resolveStopTarget({
+            handler: this.handler,
+            conversationId: mc.chatId,
+            sessionKey: mc.sessionKey,
+        });
+        if (directTarget)
+            return directTarget;
+        return resolveOnlyScopedStopTarget(this.handler, mc.chatId);
+    }
     setupEventHandlers() {
         // --- Slash commands (registered before catch-all so grammY intercepts them) ---
-        this.client.command("start", async (ctx) => {
-            const mc = this.extractMessageContext(ctx.message);
-            if (!mc)
-                return;
-            await this.postMessageRaw(parseInt(mc.chatId), [
-                "<b>Welcome!</b>",
-                "",
-                "I'm an AI coding agent. Send me a message or use these commands:",
-                "",
-                "/new — Start a new conversation",
-                "/stop — Stop the current task",
-                "/help — Show help",
-                "/login — Open your secure login page",
-            ].join("\n"));
-        });
-        this.client.command("help", async (ctx) => {
-            const mc = this.extractMessageContext(ctx.message);
-            if (!mc)
-                return;
-            await this.postMessageRaw(parseInt(mc.chatId), [
-                "<b>Available commands:</b>",
-                "",
-                "/start — Show the welcome message",
-                "/help — Show help",
-                "/stop — Stop the current task",
-                "/new — Start a new conversation",
-                "/login — Open your secure login page",
-                "",
-                "You can also send a regular message to chat with the agent.",
-            ].join("\n"));
-        });
         this.client.command("stop", async (ctx) => {
             const mc = this.extractMessageContext(ctx.message);
             if (!mc)
                 return;
-            if (this.handler.isRunning(mc.sessionKey)) {
-                await this.handler.handleStop(mc.sessionKey, mc.chatId, this);
+            const stopTarget = this.resolveStopTarget(mc);
+            if (stopTarget) {
+                await this.handler.handleStop(stopTarget, mc.chatId, this);
             }
             else {
                 await this.postMessage(mc.chatId, formatNothingRunning("telegram"));
@@ -315,60 +326,95 @@ export class TelegramBot {
             const mc = this.extractMessageContext(ctx.message);
             if (!mc)
                 return;
-            await this.handler.handleNew(mc.sessionKey, mc.chatId, this);
+            await this.handler.handleNewCommand(mc.sessionKey, mc.chatId, this);
         });
-        this.client.command("login", async (ctx) => {
+        this.client.command("sandbox", async (ctx) => {
             const mc = this.extractMessageContext(ctx.message);
             if (!mc)
                 return;
-            await this.handler.handleLogin("telegram", mc.userId, mc.chatId, this, mc.text, mc.chatType === "private");
+            const cleanedText = this.cleanText(mc.text).replace(/^\/sandbox(?:@\w+)?/i, "/pi-sandbox");
+            const event = {
+                type: "command",
+                conversationId: mc.chatId,
+                conversationKind: mc.conversationKind,
+                ts: mc.msgId,
+                thread_ts: mc.threadTs,
+                sessionKey: mc.sessionKey,
+                user: mc.userId,
+                userName: mc.userName,
+                text: cleanedText,
+                attachments: [],
+            };
+            this.logToFile(mc.chatId, {
+                date: new Date(mc.msg.date * 1000).toISOString(),
+                ts: mc.msgId,
+                ...(mc.conversationKind === "shared" && mc.threadTs ? { threadTs: mc.threadTs } : {}),
+                user: mc.userId,
+                userName: mc.userName,
+                text: cleanedText,
+                attachments: [],
+                isBot: false,
+            });
+            const adapters = createTelegramAdapters(event, this, false);
+            await this.handler.handleEvent(event, this, adapters, false);
         });
         // --- Catch-all for regular (non-command) messages ---
         this.client.on("message", async (ctx) => {
             const mc = this.extractMessageContext(ctx.message);
             if (!mc)
                 return;
-            // In groups, only respond when addressed to bot
-            if (!this.isAddressedToBot(mc.text, mc.chatType))
-                return;
             const cleanedText = this.cleanText(mc.text);
-            if (parseLoginCommand(cleanedText)) {
-                await this.handler.handleLogin("telegram", mc.userId, mc.chatId, this, cleanedText, mc.chatType === "private");
+            const addressedToBot = this.isAddressedToBot(mc.text, mc.chatType);
+            if (this.isStopText(cleanedText)) {
+                this.logToFile(mc.chatId, {
+                    date: new Date(mc.msg.date * 1000).toISOString(),
+                    ts: mc.msgId,
+                    user: mc.userId,
+                    userName: mc.userName,
+                    text: cleanedText,
+                    attachments: [],
+                    isBot: false,
+                });
+                const stopTarget = this.resolveStopTarget(mc);
+                if (stopTarget) {
+                    await this.handler.handleStop(stopTarget, mc.chatId, this);
+                }
+                else if (addressedToBot || mc.chatType === "private") {
+                    await this.postMessage(mc.chatId, formatNothingRunning("telegram"));
+                }
                 return;
             }
-            // Process attachments
-            const processedAttachments = await this.processAttachments(mc.chatId, mc.msg);
-            const event = {
+            const isAutoReplyCandidate = mc.chatType !== "private" && !addressedToBot;
+            const eventBase = {
                 type: "message",
                 conversationId: mc.chatId,
+                conversationKind: mc.conversationKind,
                 ts: mc.msgId,
                 thread_ts: mc.threadTs,
                 sessionKey: mc.sessionKey,
                 user: mc.userId,
                 userName: mc.userName,
                 text: cleanedText,
-                attachments: processedAttachments,
             };
-            // Log the message
-            this.logToFile(mc.chatId, {
+            const triggerResult = isAutoReplyCandidate
+                ? await evaluateAutoReplyPolicy({ event: eventBase, workingDir: this.workingDir })
+                : { trigger: true, reason: "addressed" };
+            const logEntryBase = {
                 date: new Date(mc.msg.date * 1000).toISOString(),
                 ts: mc.msgId,
+                ...(mc.conversationKind === "shared" && mc.threadTs ? { threadTs: mc.threadTs } : {}),
                 user: mc.userId,
                 userName: mc.userName,
                 text: cleanedText,
-                attachments: processedAttachments,
                 isBot: false,
-            });
-            // Handle bare "stop" text (backward compat)
-            if (cleanedText.toLowerCase() === "stop") {
-                if (this.handler.isRunning(mc.sessionKey)) {
-                    await this.handler.handleStop(mc.sessionKey, mc.chatId, this);
-                }
-                else {
-                    await this.postMessage(mc.chatId, formatNothingRunning("telegram"));
-                }
+            };
+            if (!triggerResult.trigger) {
+                this.logToFile(mc.chatId, { ...logEntryBase, attachments: [] });
                 return;
             }
+            const processedAttachments = await this.processAttachments(mc.chatId, mc.msg);
+            const event = { ...eventBase, attachments: processedAttachments };
+            this.logToFile(mc.chatId, { ...logEntryBase, attachments: processedAttachments });
             if (this.handler.isRunning(mc.sessionKey)) {
                 await this.postMessage(mc.chatId, formatAlreadyWorking("telegram", "/stop"));
             }