@llblab/pi-telegram 0.2.8 → 0.2.9

package/lib/replies.ts

@@ -1,187 +1,10 @@
 /**
- * Telegram reply and preview domain helpers
- * Owns preview text decisions, preview runtime behavior, rendered-message delivery, and plain or markdown reply sending
+ * Telegram reply delivery helpers
+ * Owns rendered-message delivery, reply transport wiring, and plain or markdown final replies
  */
 
 import type { TelegramRenderedChunk, TelegramRenderMode } from "./rendering.ts";
 
-// --- Preview ---
-
-export interface TelegramPreviewStateLike {
-  mode: "draft" | "message";
-  draftId?: number;
-  messageId?: number;
-  pendingText: string;
-  lastSentText: string;
-}
-
-export interface TelegramPreviewRuntimeState extends TelegramPreviewStateLike {
-  flushTimer?: ReturnType<typeof setTimeout>;
-}
-
-export interface TelegramPreviewRuntimeDeps {
-  getState: () => TelegramPreviewRuntimeState | undefined;
-  setState: (state: TelegramPreviewRuntimeState | undefined) => void;
-  clearScheduledFlush: (state: TelegramPreviewRuntimeState) => void;
-  maxMessageLength: number;
-  renderPreviewText: (markdown: string) => string;
-  getDraftSupport: () => "unknown" | "supported" | "unsupported";
-  setDraftSupport: (support: "unknown" | "supported" | "unsupported") => void;
-  allocateDraftId: () => number;
-  sendDraft: (chatId: number, draftId: number, text: string) => Promise<void>;
-  sendMessage: (
-    chatId: number,
-    text: string,
-  ) => Promise<TelegramSentMessageLike>;
-  editMessageText: (
-    chatId: number,
-    messageId: number,
-    text: string,
-  ) => Promise<void>;
-  renderTelegramMessage: (
-    text: string,
-    options?: { mode?: TelegramRenderMode },
-  ) => TelegramRenderedChunk[];
-  sendRenderedChunks: (
-    chatId: number,
-    chunks: TelegramRenderedChunk[],
-  ) => Promise<number | undefined>;
-  editRenderedMessage: (
-    chatId: number,
-    messageId: number,
-    chunks: TelegramRenderedChunk[],
-  ) => Promise<number | undefined>;
-}
-
-export function buildTelegramPreviewFlushText(options: {
-  state: TelegramPreviewStateLike;
-  maxMessageLength: number;
-  renderPreviewText: (markdown: string) => string;
-}): string | undefined {
-  const rawText = options.state.pendingText.trim();
-  const previewText = options.renderPreviewText(rawText).trim();
-  if (!previewText || previewText === options.state.lastSentText) {
-    return undefined;
-  }
-  return previewText.length > options.maxMessageLength
-    ? previewText.slice(0, options.maxMessageLength)
-    : previewText;
-}
-
-export function buildTelegramPreviewFinalText(
-  state: TelegramPreviewStateLike,
-): string | undefined {
-  const finalText = (state.pendingText.trim() || state.lastSentText).trim();
-  return finalText || undefined;
-}
-
-export function shouldUseTelegramDraftPreview(options: {
-  draftSupport: "unknown" | "supported" | "unsupported";
-}): boolean {
-  return options.draftSupport !== "unsupported";
-}
-
-export async function clearTelegramPreview(
-  chatId: number,
-  deps: TelegramPreviewRuntimeDeps,
-): Promise<void> {
-  const state = deps.getState();
-  if (!state) return;
-  deps.clearScheduledFlush(state);
-  deps.setState(undefined);
-  if (state.mode !== "draft" || state.draftId === undefined) return;
-  try {
-    await deps.sendDraft(chatId, state.draftId, "");
-  } catch {
-    // ignore
-  }
-}
-
-export async function flushTelegramPreview(
-  chatId: number,
-  deps: TelegramPreviewRuntimeDeps,
-): Promise<void> {
-  const state = deps.getState();
-  if (!state) return;
-  state.flushTimer = undefined;
-  const truncated = buildTelegramPreviewFlushText({
-    state,
-    maxMessageLength: deps.maxMessageLength,
-    renderPreviewText: deps.renderPreviewText,
-  });
-  if (!truncated) return;
-  if (shouldUseTelegramDraftPreview({ draftSupport: deps.getDraftSupport() })) {
-    const draftId = state.draftId ?? deps.allocateDraftId();
-    state.draftId = draftId;
-    try {
-      await deps.sendDraft(chatId, draftId, truncated);
-      deps.setDraftSupport("supported");
-      state.mode = "draft";
-      state.lastSentText = truncated;
-      return;
-    } catch {
-      deps.setDraftSupport("unsupported");
-    }
-  }
-  if (state.messageId === undefined) {
-    const sent = await deps.sendMessage(chatId, truncated);
-    state.messageId = sent.message_id;
-    state.mode = "message";
-    state.lastSentText = truncated;
-    return;
-  }
-  await deps.editMessageText(chatId, state.messageId, truncated);
-  state.mode = "message";
-  state.lastSentText = truncated;
-}
-
-export async function finalizeTelegramPreview(
-  chatId: number,
-  deps: TelegramPreviewRuntimeDeps,
-): Promise<boolean> {
-  const state = deps.getState();
-  if (!state) return false;
-  await flushTelegramPreview(chatId, deps);
-  const finalText = buildTelegramPreviewFinalText(state);
-  if (!finalText) {
-    await clearTelegramPreview(chatId, deps);
-    return false;
-  }
-  if (state.mode === "draft") {
-    await deps.sendMessage(chatId, finalText);
-    await clearTelegramPreview(chatId, deps);
-    return true;
-  }
-  deps.setState(undefined);
-  return state.messageId !== undefined;
-}
-
-export async function finalizeTelegramMarkdownPreview(
-  chatId: number,
-  markdown: string,
-  deps: TelegramPreviewRuntimeDeps,
-): Promise<boolean> {
-  const state = deps.getState();
-  if (!state) return false;
-  await flushTelegramPreview(chatId, deps);
-  const chunks = deps.renderTelegramMessage(markdown, { mode: "markdown" });
-  if (chunks.length === 0) {
-    await clearTelegramPreview(chatId, deps);
-    return false;
-  }
-  if (state.mode === "draft") {
-    await deps.sendRenderedChunks(chatId, chunks);
-    await clearTelegramPreview(chatId, deps);
-    return true;
-  }
-  if (state.messageId === undefined) return false;
-  await deps.editRenderedMessage(chatId, state.messageId, chunks);
-  deps.setState(undefined);
-  return true;
-}
-
-// --- Delivery ---
-
 export interface TelegramSentMessageLike {
   message_id: number;
 }
@@ -278,8 +101,6 @@ export async function editTelegramRenderedMessage<TReplyMarkup>(
   return messageId;
 }
 
-// --- Reply Runtime ---
-
 export interface TelegramReplyRuntimeDeps {
   renderTelegramMessage: (
     text: string,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llblab/pi-telegram",
-  "version": "0.2.8",
+  "version": "0.2.9",
   "private": false,
   "description": "Better Telegram DM bridge extension for pi",
   "type": "module",

package/tests/menu.test.ts

@@ -144,7 +144,9 @@ test("Menu helpers apply menu mutations and resolve model selections", () => {
   assert.equal(state.page, 2);
   assert.equal(applyTelegramModelPageSelection(state, "2"), "unchanged");
   assert.equal(applyTelegramModelPageSelection(state, "bad"), "invalid");
-  assert.deepEqual(getTelegramModelSelection(state, "bad"), { kind: "invalid" });
+  assert.deepEqual(getTelegramModelSelection(state, "bad"), {
+    kind: "invalid",
+  });
   assert.deepEqual(getTelegramModelSelection(state, "9"), { kind: "missing" });
   assert.equal(getTelegramModelSelection(state, "0").kind, "selected");
 });
@@ -174,7 +176,11 @@ test("Menu helpers derive normalized menu pages without mutating state", () => {
 
 test("Menu helpers build model callback plans for paging, selection, and restart modes", () => {
   const modelA = { provider: "openai", id: "gpt-5", reasoning: true } as const;
-  const modelB = { provider: "anthropic", id: "claude-3", reasoning: false } as const;
+  const modelB = {
+    provider: "anthropic",
+    id: "claude-3",
+    reasoning: false,
+  } as const;
   const state = {
     chatId: 1,
     messageId: 2,
@@ -227,7 +233,8 @@ test("Menu helpers build model callback plans for paging, selection, and restart
       kind: "switch-model",
       selection: state.allModels[1],
       mode: "restart-after-tool",
-      callbackText: "Switched to claude-3. Restarting after the current tool finishes…",
+      callbackText:
+        "Switched to claude-3. Restarting after the current tool finishes…",
     },
   );
   assert.deepEqual(
@@ -254,14 +261,19 @@ test("Menu helpers route callback entry states before action handlers", async ()
       events.push(`answer:${text ?? ""}`);
     },
   });
-  await handleTelegramMenuCallbackEntry("callback-2", "status:model", undefined, {
-    handleStatusAction: async () => false,
-    handleThinkingAction: async () => false,
-    handleModelAction: async () => false,
-    answerCallbackQuery: async (_id, text) => {
-      events.push(`answer:${text ?? ""}`);
+  await handleTelegramMenuCallbackEntry(
+    "callback-2",
+    "status:model",
+    undefined,
+    {
+      handleStatusAction: async () => false,
+      handleThinkingAction: async () => false,
+      handleModelAction: async () => false,
+      answerCallbackQuery: async (_id, text) => {
+        events.push(`answer:${text ?? ""}`);
+      },
     },
-  });
+  );
   await handleTelegramMenuCallbackEntry(
     "callback-3",
     "status:model",
@@ -296,7 +308,11 @@ test("Menu helpers route callback entry states before action handlers", async ()
 test("Menu helpers execute model callback actions across update, switch, and restart paths", async () => {
   const events: string[] = [];
   const modelA = { provider: "openai", id: "gpt-5", reasoning: true } as const;
-  const modelB = { provider: "anthropic", id: "claude-3", reasoning: false } as const;
+  const modelB = {
+    provider: "anthropic",
+    id: "claude-3",
+    reasoning: false,
+  } as const;
   const state = {
     chatId: 1,
     messageId: 2,
@@ -530,8 +546,14 @@ test("Menu helpers build pure render payloads before transport", () => {
     allModels: [{ model: modelA }],
     mode: "status" as const,
   } as unknown as TelegramModelMenuState;
-  const modelPayload = buildTelegramModelMenuRenderPayload(state, modelA as never);
-  const thinkingPayload = buildTelegramThinkingMenuRenderPayload(modelA as never, "medium");
+  const modelPayload = buildTelegramModelMenuRenderPayload(
+    state,
+    modelA as never,
+  );
+  const thinkingPayload = buildTelegramThinkingMenuRenderPayload(
+    modelA as never,
+    "medium",
+  );
   const statusPayload = buildTelegramStatusMenuRenderPayload(
     "<b>Status</b>",
     modelA as never,
@@ -580,7 +602,12 @@ test("Menu helpers update and send interactive menu messages", async () => {
     },
   };
   await updateTelegramModelMenuMessage(state, modelA as never, deps);
-  await updateTelegramThinkingMenuMessage(state, modelA as never, "medium", deps);
+  await updateTelegramThinkingMenuMessage(
+    state,
+    modelA as never,
+    "medium",
+    deps,
+  );
   await updateTelegramStatusMessage(
     state,
     "<b>Status</b>",
@@ -595,7 +622,11 @@ test("Menu helpers update and send interactive menu messages", async () => {
     "medium",
     deps,
   );
-  const sentModelId = await sendTelegramModelMenuMessage(state, modelA as never, deps);
+  const sentModelId = await sendTelegramModelMenuMessage(
+    state,
+    modelA as never,
+    deps,
+  );
   assert.equal(sentStatusId, 99);
   assert.equal(sentModelId, 99);
   assert.equal(events[0], "edit:1:2:html:<b>Choose a model:</b>");