@llblab/pi-telegram 0.7.1 → 0.8.0

package/index.ts

@@ -5,18 +5,19 @@
  */
 
 import * as Api from "./lib/api.ts";
-import * as AttachmentHandlers from "./lib/attachment-handlers.ts";
-import * as Attachments from "./lib/attachments.ts";
-import * as Commands from "./lib/commands.ts";
+import * as OutboundAttachments from "./lib/outbound-attachments.ts";
 import * as CommandTemplates from "./lib/command-templates.ts";
+import * as Commands from "./lib/commands.ts";
 import * as Config from "./lib/config.ts";
+import * as InboundHandlers from "./lib/inbound-handlers.ts";
 import * as Keyboard from "./lib/keyboard.ts";
 import * as Lifecycle from "./lib/lifecycle.ts";
 import * as Locks from "./lib/locks.ts";
 import * as Media from "./lib/media.ts";
-import * as Menu from "./lib/menu.ts";
 import * as MenuQueue from "./lib/menu-queue.ts";
+import * as Menu from "./lib/menu.ts";
 import * as Model from "./lib/model.ts";
+import * as OutboundHandlers from "./lib/outbound-handlers.ts";
 import * as Pi from "./lib/pi.ts";
 import * as Polling from "./lib/polling.ts";
 import * as Preview from "./lib/preview.ts";
@@ -24,11 +25,11 @@ import * as PromptTemplates from "./lib/prompt-templates.ts";
 import * as Prompts from "./lib/prompts.ts";
 import * as Queue from "./lib/queue.ts";
 import * as Replies from "./lib/replies.ts";
-import * as Runtime from "./lib/runtime.ts";
 import * as Routing from "./lib/routing.ts";
+import * as Runtime from "./lib/runtime.ts";
 import * as Setup from "./lib/setup.ts";
-import * as OutboundHandlers from "./lib/outbound-handlers.ts";
 import * as Status from "./lib/status.ts";
+import * as TextGroups from "./lib/text-groups.ts";
 
 type ActivePiModel = NonNullable<Pi.ExtensionContext["model"]>;
 type RuntimeTelegramQueueItem = Queue.TelegramQueueItem<Pi.ExtensionContext>;
@@ -67,6 +68,10 @@ export default function (pi: Pi.ExtensionAPI) {
     Api.TelegramMessage,
     Pi.ExtensionContext
   >();
+  const textGroupRuntime = TextGroups.createTelegramTextGroupController<
+    Api.TelegramMessage,
+    Pi.ExtensionContext
+  >();
   const telegramQueueStore =
     Queue.createTelegramQueueStore<Pi.ExtensionContext>();
   const deferredQueueDispatchRuntime =
@@ -109,15 +114,13 @@ export default function (pi: Pi.ExtensionAPI) {
         queue.incrementNextPriorityReactionOrder,
       updateStatus,
     });
-  const attachmentHandlerRuntime =
-    AttachmentHandlers.createTelegramAttachmentHandlerRuntime<Pi.ExtensionContext>(
-      {
-        getHandlers: configStore.getAttachmentHandlers,
-        execCommand: CommandTemplates.execCommandTemplate,
-        getCwd: Pi.getExtensionContextCwd,
-        recordRuntimeEvent,
-      },
-    );
+  const inboundHandlerRuntime =
+    InboundHandlers.createTelegramInboundHandlerRuntime<Pi.ExtensionContext>({
+      getHandlers: configStore.getInboundHandlers,
+      execCommand: CommandTemplates.execCommandTemplate,
+      getCwd: Pi.getExtensionContextCwd,
+      recordRuntimeEvent,
+    });
 
   // --- Telegram API ---
 
@@ -152,19 +155,23 @@ export default function (pi: Pi.ExtensionAPI) {
 
   // --- Reply Runtime Wiring ---
 
-  const {
-    replyTransport,
-    sendTextReply,
-    sendMarkdownReply,
-    editInteractiveMessage,
-    sendInteractiveMessage,
-  } =
+  const replyRuntime =
     Replies.createTelegramRenderedMessageDeliveryRuntime<Keyboard.TelegramInlineKeyboardMarkup>(
       {
         sendMessage,
         editMessage: editTelegramMessageText,
       },
     );
+  const { replyTransport, editInteractiveMessage, sendInteractiveMessage } =
+    replyRuntime;
+  const { sendTextReply, sendMarkdownReply } =
+    OutboundHandlers.createTelegramOutboundTextReplyRuntime({
+      sendTextReply: replyRuntime.sendTextReply,
+      sendMarkdownReply: replyRuntime.sendMarkdownReply,
+      execCommand: CommandTemplates.execCommandTemplate,
+      getHandlers: configStore.getOutboundHandlers,
+      recordRuntimeEvent,
+    });
   const dispatchNextQueuedTelegramTurn =
     Queue.createTelegramQueueDispatchRuntime<Pi.ExtensionContext>({
       ...telegramQueueStore,
@@ -193,6 +200,13 @@ export default function (pi: Pi.ExtensionAPI) {
     editMessageText: editTelegramMessageText,
     ...replyTransport,
   });
+  const { finalizeMarkdownPreview } =
+    OutboundHandlers.createTelegramOutboundTextPreviewRuntime({
+      finalizeMarkdownPreview: previewRuntime.finalizeMarkdown,
+      execCommand: CommandTemplates.execCommandTemplate,
+      getHandlers: configStore.getOutboundHandlers,
+      recordRuntimeEvent,
+    });
 
   // --- Bridge Setup ---
 
@@ -213,9 +227,8 @@ export default function (pi: Pi.ExtensionAPI) {
       appendQueuedItem: queueMutationRuntime.append,
       updateStatus,
     });
-  const getQueueItemCount = Queue.createTelegramQueueItemCountGetter(
-    telegramQueueStore,
-  );
+  const getQueueItemCount =
+    Queue.createTelegramQueueItemCountGetter(telegramQueueStore);
   const getPromptTemplateCommands =
     PromptTemplates.createTelegramPromptTemplateCommandGetter({
       getCommands,
@@ -277,6 +290,7 @@ export default function (pi: Pi.ExtensionAPI) {
     bridgeRuntime,
     activeTurnRuntime,
     mediaGroupRuntime,
+    textGroupRuntime,
     telegramQueueStore,
     queueMutationRuntime,
     modelMenuRuntime,
@@ -286,7 +300,7 @@ export default function (pi: Pi.ExtensionAPI) {
     openQueueMenu: queueMenuRuntime.openQueueMenu,
     queueMenuCallbackHandler: queueMenuRuntime.handleCallbackQuery,
     buttonActionStore,
-    attachmentHandlerRuntime,
+    inboundHandlerRuntime,
     updateStatus,
     dispatchNextQueuedTelegramTurn,
     answerCallbackQuery,
@@ -342,7 +356,10 @@ export default function (pi: Pi.ExtensionAPI) {
     prepareTempDir,
     updateStatus,
     unbindDeferredDispatchContext: deferredQueueDispatchRuntime.unbind,
-    clearPendingMediaGroups: mediaGroupRuntime.clear,
+    clearPendingMediaGroups: TextGroups.createTelegramGroupedInputClearer({
+      clearMediaGroups: mediaGroupRuntime.clear,
+      clearTextGroups: textGroupRuntime.clear,
+    }),
     clearModelMenuState: modelMenuRuntime.clear,
     getActiveTurnChatId: activeTurnRuntime.getChatId,
     clearPreview: previewRuntime.clear,
@@ -358,7 +375,7 @@ export default function (pi: Pi.ExtensionAPI) {
 
   // --- Extension API Bindings ---
 
-  Attachments.registerTelegramAttachmentTool(pi, {
+  OutboundAttachments.registerTelegramOutboundAttachmentTool(pi, {
     getActiveTurn: activeTurnRuntime.get,
     recordRuntimeEvent,
   });
@@ -393,7 +410,7 @@ export default function (pi: Pi.ExtensionAPI) {
     clearDispatchPending: lifecycle.clearDispatchPending,
   });
   const queuedAttachmentSender =
-    Attachments.createTelegramQueuedAttachmentSender({
+    OutboundAttachments.createTelegramQueuedOutboundAttachmentSender({
       sendMultipart: callMultipart,
       sendTextReply,
       recordRuntimeEvent,
@@ -428,14 +445,15 @@ export default function (pi: Pi.ExtensionAPI) {
     updateStatus,
     getActiveTurn: activeTurnRuntime.get,
     extractAssistant: Replies.extractLatestAssistantMessageText,
-    getPreserveQueuedTurnsAsHistory: lifecycle.shouldPreserveQueuedTurnsAsHistory,
+    getPreserveQueuedTurnsAsHistory:
+      lifecycle.shouldPreserveQueuedTurnsAsHistory,
     resetRuntimeState: agentEndResetter,
     dispatchNextQueuedTelegramTurn,
     requestDeferredDispatchNextQueuedTelegramTurn:
       deferredQueueDispatchRuntime.request,
     clearPreview: previewRuntime.clear,
     setPreviewPendingText: previewRuntime.setPendingText,
-    finalizeMarkdownPreview: previewRuntime.finalizeMarkdown,
+    finalizeMarkdownPreview,
     sendMarkdownReply,
     sendTextReply,
     sendQueuedAttachments: queuedAttachmentSender,
@@ -447,7 +465,9 @@ export default function (pi: Pi.ExtensionAPI) {
   });
   // Wire transport-level reply dedup reset via lifecycle
   Lifecycle.setResetTransportReplyDedup(Replies.resetTransportReplyDedup);
-  const agentStartWithDedupReset = Lifecycle.createAgentStartDedupHook(agentLifecycleHooks.onAgentStart);
+  const agentStartWithDedupReset = Lifecycle.createAgentStartDedupHook(
+    agentLifecycleHooks.onAgentStart,
+  );
   Lifecycle.registerTelegramLifecycleHooks(pi, {
     ...sessionLifecycleRuntime,
     ...agentLifecycleHooks,

package/lib/config.ts

@@ -9,7 +9,7 @@ import { chmod, mkdir, readFile, rename, writeFile } from "node:fs/promises";
 import { homedir } from "node:os";
 import { join, resolve } from "node:path";
 
-import type { TelegramAttachmentHandlerConfig } from "./attachment-handlers.ts";
+import type { TelegramInboundHandlerConfig } from "./inbound-handlers.ts";
 import type { CommandTemplateObjectConfig } from "./command-templates.ts";
 
 function getAgentDir(): string {
@@ -39,7 +39,8 @@ export interface TelegramConfig {
   botId?: number;
   allowedUserId?: number;
   lastUpdateId?: number;
-  attachmentHandlers?: TelegramAttachmentHandlerConfig[];
+  inboundHandlers?: TelegramInboundHandlerConfig[];
+  attachmentHandlers?: TelegramInboundHandlerConfig[];
   outboundHandlers?: TelegramOutboundHandlerConfig[];
 }
 
@@ -50,7 +51,8 @@ export interface TelegramConfigStore {
   getBotToken: () => string | undefined;
   hasBotToken: () => boolean;
   getAllowedUserId: () => number | undefined;
-  getAttachmentHandlers: () => TelegramAttachmentHandlerConfig[] | undefined;
+  getInboundHandlers: () => TelegramInboundHandlerConfig[] | undefined;
+  getAttachmentHandlers: () => TelegramInboundHandlerConfig[] | undefined;
   getOutboundHandlers: () => TelegramOutboundHandlerConfig[] | undefined;
   setAllowedUserId: (userId: number) => void;
   load: () => Promise<void>;
@@ -104,6 +106,10 @@ export function createTelegramConfigStore(
     getBotToken: () => config.botToken,
     hasBotToken: () => !!config.botToken,
     getAllowedUserId: () => config.allowedUserId,
+    getInboundHandlers: () => [
+      ...(config.inboundHandlers ?? []),
+      ...(config.attachmentHandlers ?? []),
+    ],
     getAttachmentHandlers: () => config.attachmentHandlers,
     getOutboundHandlers: () => config.outboundHandlers,
     setAllowedUserId: (userId) => {