@llblab/pi-telegram 0.18.5 → 0.19.0

package/lib/menu-settings.ts

@@ -18,6 +18,7 @@ export type TelegramSettingsMenuReplyMarkup = TelegramInlineKeyboardMarkup;
 
 export interface TelegramSettingsStateDeps {
   isProactivePushEnabled: () => boolean;
+  areRichDraftPreviewsEnabled: () => boolean;
   getTimeInjectionMode: () => TelegramTimeMode;
   getVoiceReplyMode: () => TelegramVoiceReplyMode;
   isVoiceReplyModeConfigured: () => boolean;
@@ -25,6 +26,7 @@ export interface TelegramSettingsStateDeps {
 
 export interface TelegramSettingsMutationDeps extends TelegramSettingsStateDeps {
   setProactivePushEnabled: (enabled: boolean) => Promise<void>;
+  setRichDraftPreviewsEnabled: (enabled: boolean) => Promise<void>;
   setVoiceReplyMode: (
     mode: TelegramVoiceReplyMode | undefined,
   ) => Promise<void>;
@@ -116,6 +118,8 @@ export interface TelegramSettingsMenuRuntimeDeps<
 
 export const SETTINGS_MENU_TITLE = "<b>⚙️ Settings:</b>";
 export const PROACTIVE_PUSH_SETTINGS_TITLE = "<b>📌 Proactive push:</b>";
+export const RICH_DRAFT_PREVIEWS_SETTINGS_TITLE =
+  "<b>📝 Rich draft previews:</b>";
 export const TIME_INJECTION_MODE_SETTINGS_TITLE =
   "<b>🕒 Time injection mode:</b>";
 export const VOICE_REPLY_MODE_SETTINGS_TITLE = "<b>👄 Voice reply mode:</b>";
@@ -151,6 +155,17 @@ export function buildProactivePushSettingsText(
   ].join("\n");
 }
 
+export function buildRichDraftPreviewsSettingsText(enabled: boolean): string {
+  return [
+    `${RICH_DRAFT_PREVIEWS_SETTINGS_TITLE} <code>${enabled ? "on" : "off"}</code>`,
+    "",
+    "Show progressive Rich Markdown draft previews while the model is answering.",
+    "",
+    "<code>-</code> <code>off</code> (default): show native active status, then send the final Rich Markdown answer.",
+    "<code>-</code> <code>on</code>: also stream safe Rich Markdown draft previews before the final answer.",
+  ].join("\n");
+}
+
 export function buildVoiceReplyModeSettingsText(
   mode: TelegramVoiceReplyMode,
   configured = true,
@@ -185,6 +200,7 @@ export function buildTimeInjectionModeSettingsText(
 
 export function buildTelegramSettingsMenuReplyMarkup(
   proactivePushEnabled: boolean,
+  richDraftPreviewsEnabled: boolean,
   voiceReplyMode: TelegramVoiceReplyMode,
   timeInjectionMode: TelegramTimeMode,
   sectionRegistry?: TelegramSectionRegistry,
@@ -217,6 +233,12 @@ export function buildTelegramSettingsMenuReplyMarkup(
         callback_data: "settings:open:time-injection",
       },
     ],
+    [
+      {
+        text: `📝 Rich drafts: ${richDraftPreviewsEnabled ? "on" : "off"}`,
+        callback_data: "settings:open:rich-drafts",
+      },
+    ],
     [
       {
         text: `📌 Proactive push: ${proactivePushEnabled ? "on" : "off"}`,
@@ -239,6 +261,7 @@ export async function openTelegramSettingsMenu<
     buildTelegramSettingsMenuText(),
     buildTelegramSettingsMenuReplyMarkup(
       deps.isProactivePushEnabled(),
+      deps.areRichDraftPreviewsEnabled(),
       deps.getVoiceReplyMode(),
       deps.getTimeInjectionMode(),
       sectionRegistry,
@@ -271,6 +294,26 @@ export function buildProactivePushSettingsReplyMarkup(
   };
 }
 
+export function buildRichDraftPreviewsSettingsReplyMarkup(
+  enabled: boolean,
+): TelegramSettingsMenuReplyMarkup {
+  return {
+    inline_keyboard: [
+      [{ text: "⬆️ Back", callback_data: "settings:list" }],
+      [
+        {
+          text: enabled ? "🟢 On" : "⚫️ On",
+          callback_data: "settings:set:rich-drafts:on",
+        },
+        {
+          text: enabled ? "⚫️ Off" : "🟡 Off",
+          callback_data: "settings:set:rich-drafts:off",
+        },
+      ],
+    ],
+  };
+}
+
 export function buildTimeInjectionModeSettingsReplyMarkup(
   mode: TelegramTimeMode,
 ): TelegramSettingsMenuReplyMarkup {
@@ -320,6 +363,7 @@ export async function updateTelegramSettingsMenuMessage(
     buildTelegramSettingsMenuText(),
     buildTelegramSettingsMenuReplyMarkup(
       deps.isProactivePushEnabled(),
+      deps.areRichDraftPreviewsEnabled(),
       deps.getVoiceReplyMode(),
       deps.getTimeInjectionMode(),
       sectionRegistry,
@@ -338,6 +382,16 @@ export async function updateProactivePushSettingsMessage(
   );
 }
 
+export async function updateRichDraftPreviewsSettingsMessage(
+  deps: TelegramSettingsMenuCallbackDeps,
+): Promise<void> {
+  const enabled = deps.areRichDraftPreviewsEnabled();
+  await deps.updateSettingsMessage(
+    buildRichDraftPreviewsSettingsText(enabled),
+    buildRichDraftPreviewsSettingsReplyMarkup(enabled),
+  );
+}
+
 export async function updateTimeInjectionModeSettingsMessage(
   deps: TelegramSettingsMenuCallbackDeps,
 ): Promise<void> {
@@ -375,6 +429,11 @@ export async function handleTelegramSettingsMenuCallbackAction(
     await deps.answerCallbackQuery(callbackQueryId);
     return true;
   }
+  if (data === "settings:open:rich-drafts") {
+    await updateRichDraftPreviewsSettingsMessage(deps);
+    await deps.answerCallbackQuery(callbackQueryId);
+    return true;
+  }
   if (data === "settings:open:voice-reply") {
     await updateVoiceReplyModeSettingsMessage(deps);
     await deps.answerCallbackQuery(callbackQueryId);
@@ -427,6 +486,19 @@ export async function handleTelegramSettingsMenuCallbackAction(
       return true;
     }
   }
+  if (
+    data === "settings:set:rich-drafts:on" ||
+    data === "settings:set:rich-drafts:off"
+  ) {
+    const enabled = data.endsWith(":on");
+    await deps.setRichDraftPreviewsEnabled(enabled);
+    await updateRichDraftPreviewsSettingsMessage(deps);
+    await deps.answerCallbackQuery(
+      callbackQueryId,
+      `Rich draft previews ${enabled ? "enabled" : "disabled"}`,
+    );
+    return true;
+  }
   if (
     data === "settings:set:proactive:on" ||
     data === "settings:set:proactive:off"
@@ -457,6 +529,7 @@ export function createTelegramSettingsMenuRuntime<
         {
           getModelMenuState: () => deps.getModelMenuState(chatId, ctx),
           isProactivePushEnabled: deps.isProactivePushEnabled,
+          areRichDraftPreviewsEnabled: deps.areRichDraftPreviewsEnabled,
           getVoiceReplyMode: deps.getVoiceReplyMode,
           isVoiceReplyModeConfigured: deps.isVoiceReplyModeConfigured,
           getTimeInjectionMode: deps.getTimeInjectionMode,
@@ -475,6 +548,7 @@ export function createTelegramSettingsMenuRuntime<
       updateTelegramSettingsMenuMessage(
         {
           isProactivePushEnabled: deps.isProactivePushEnabled,
+          areRichDraftPreviewsEnabled: deps.areRichDraftPreviewsEnabled,
           getVoiceReplyMode: deps.getVoiceReplyMode,
           isVoiceReplyModeConfigured: deps.isVoiceReplyModeConfigured,
           getTimeInjectionMode: deps.getTimeInjectionMode,
@@ -513,6 +587,18 @@ export function createTelegramSettingsMenuRuntime<
           );
           return true;
         }
+        if (
+          query.data === "settings:set:rich-drafts:on" ||
+          query.data === "settings:set:rich-drafts:off"
+        ) {
+          const enabled = query.data.endsWith(":on");
+          await deps.setRichDraftPreviewsEnabled(enabled);
+          await deps.answerCallbackQuery(
+            query.id,
+            `Rich draft previews ${enabled ? "enabled" : "disabled"}`,
+          );
+          return true;
+        }
         const hasTimeInjectionPrefix = query.data.startsWith(
           "settings:set:time-injection:",
         );
@@ -543,10 +629,12 @@ export function createTelegramSettingsMenuRuntime<
       }
       return handleTelegramSettingsMenuCallbackAction(query.id, query.data, {
         isProactivePushEnabled: deps.isProactivePushEnabled,
+        areRichDraftPreviewsEnabled: deps.areRichDraftPreviewsEnabled,
         getVoiceReplyMode: deps.getVoiceReplyMode,
         isVoiceReplyModeConfigured: deps.isVoiceReplyModeConfigured,
         getTimeInjectionMode: deps.getTimeInjectionMode,
         setProactivePushEnabled: deps.setProactivePushEnabled,
+        setRichDraftPreviewsEnabled: deps.setRichDraftPreviewsEnabled,
         setVoiceReplyMode: deps.setVoiceReplyMode,
         setTimeInjectionMode: deps.setTimeInjectionMode,
         updateSettingsMessage: (text, replyMarkup) =>

package/lib/routing.ts

@@ -217,21 +217,13 @@ function formatTelegramAllTabMenuChooserText(command: string): string {
 function buildTelegramUnboundRerouteChooserMarkup(
   rerouteId: string,
   records: readonly Threads.TelegramTopicTargetRecord[],
-  options: {
+  _options: {
     currentLeaderProfileKey?: string;
     currentInstanceId?: string;
   } = {},
 ): Menu.TelegramReplyMarkup {
   const activeRecords = records.filter((record) => record.status === "active");
-  const canRestoreCurrentLeader = activeRecords.some(
-    (record) =>
-      typeof record.rerouteConfirmedAtMs === "number" &&
-      isCurrentLeaderTopicRecord(
-        record,
-        options.currentLeaderProfileKey,
-        options.currentInstanceId,
-      ),
-  );
+  const canRestoreAnyLiveThread = activeRecords.length > 0;
   const rows = activeRecords.map((record) => [
     {
       text: getTelegramRouteThreadButtonLabel(record),
@@ -242,7 +234,7 @@ function buildTelegramUnboundRerouteChooserMarkup(
     },
   ]);
   return {
-    inline_keyboard: canRestoreCurrentLeader
+    inline_keyboard: canRestoreAnyLiveThread
       ? [
           ...rows,
           [
@@ -414,6 +406,7 @@ export interface TelegramInboundRouteRuntimeDeps<
   getCurrentInstanceId?: () => string | undefined;
   getMessageOwnership?: Updates.TelegramMessageOwnershipLookup;
   getTargetOwnership?: Updates.TelegramTargetOwnershipLookup;
+  recordMessageOwnership?: Updates.TelegramMessageOwnershipRecorder;
   getLiveThreadTargets?: () => Queue.TelegramQueueTarget[];
   getLocalThreadLabelForTarget?: (
     target: Queue.TelegramQueueTarget,
@@ -1861,6 +1854,7 @@ export function createTelegramInboundRouteRuntime<
     getCurrentInstanceId: deps.getCurrentInstanceId,
     getMessageOwnership: deps.getMessageOwnership,
     getTargetOwnership: deps.getTargetOwnership,
+    recordMessageOwnership: deps.recordMessageOwnership,
     handleTelegramTopicLifecycleUpdate,
     foreignOwnedUpdateForwarder: deps.foreignOwnedUpdateForwarder,
     setAllowedUserId: deps.configStore.setAllowedUserId,

package/lib/runtime.ts

@@ -4,7 +4,7 @@
  * Owns small session-local runtime primitives that are shared by orchestration but are not specific to queueing, rendering, polling, or Telegram transport
  */
 
-const TELEGRAM_TYPING_ACTION_INTERVAL_MS = 1500;
+const TELEGRAM_TYPING_ACTION_INTERVAL_MS = 2500;
 const TELEGRAM_TYPING_IDLE_DRAIN_MAX_MS = 250;
 
 export interface TelegramRuntimeQueueCounters {
@@ -398,13 +398,9 @@ export function createTelegramTypingLoopStarter<TContext>(
       chatId: chatId ?? deps.getDefaultChatId(),
       target: options?.target,
       intervalMs: deps.intervalMs ?? TELEGRAM_TYPING_ACTION_INTERVAL_MS,
-      sendTypingAction: async (targetChatId) => {
+      sendTypingAction: async (targetChatId, actionOptions) => {
         try {
-          const threadParams = getTelegramTypingLoopThreadParams(options?.target);
-          await deps.sendTypingAction(targetChatId, threadParams);
-          if (threadParams?.message_thread_id !== undefined) {
-            await deps.sendAggregateTypingAction?.(targetChatId);
-          }
+          await deps.sendTypingAction(targetChatId, actionOptions);
         } catch (error) {
           const message =
             error instanceof Error ? error.message : String(error);
@@ -419,6 +415,26 @@ export function createTelegramTypingLoopStarter<TContext>(
           });
         }
       },
+      sendAggregateTypingAction: deps.sendAggregateTypingAction
+        ? async (targetChatId) => {
+            try {
+              await deps.sendAggregateTypingAction?.(targetChatId);
+            } catch (error) {
+              const message =
+                error instanceof Error ? error.message : String(error);
+              updateTelegramRuntimeStatusSafely(deps.updateStatus, ctx, {
+                error: message,
+                category: "typing",
+                phase: "status-update",
+                recordRuntimeEvent: deps.recordRuntimeEvent,
+              });
+              deps.recordRuntimeEvent?.("typing", error, {
+                chatId: targetChatId,
+                aggregate: true,
+              });
+            }
+          }
+        : undefined,
     });
   };
 }

package/lib/status.ts

@@ -771,8 +771,6 @@ export function buildTelegramStatusBarText(
     return `${label} ${theme.fg("warning", "electing")}${queued}`;
   if (!state.pollingActive && state.busRole !== "follower")
     return `${label} ${theme.fg("muted", "disconnected")}${queued}`;
-  if (state.busRole === "follower")
-    return `${label} ${theme.fg("success", "follower")}${queued}`;
   if (state.compactionInProgress) {
     return `${label} ${theme.fg("warning", "compacting")}${queued}`;
   }
@@ -784,6 +782,8 @@ export function buildTelegramStatusBarText(
       processingStatus === "active" ? "warning" : "accent";
     return `${label} ${theme.fg(processingToken, processingStatus)}${queued}`;
   }
+  if (state.busRole === "follower")
+    return `${label} ${theme.fg("success", "follower")}${queued}`;
   if (state.busRole === "leader")
     return `${label} ${theme.fg("success", "leader")}`;
   return `${label} ${theme.fg("success", "connected")}`;

package/lib/threads.ts

@@ -1438,7 +1438,7 @@ function getTelegramThreadNameEntropyIndex(
 
 export function getTelegramTopicThreadNameValidationError(
   threadName: string,
-  slot: string | undefined,
+  _slot: string | undefined,
 ): string | undefined {
   const identity = getTelegramTopicIdentityName(threadName);
   const reasons: string[] = [];
@@ -1501,6 +1501,60 @@ function asInteger(value: unknown): number | undefined {
   return Number.isInteger(parsed) ? parsed : undefined;
 }
 
+export interface TelegramPromoteFollowerBindingToLeaderDeps {
+  store: TelegramTopicTargetStore;
+  instanceId: string;
+  cwd?: string;
+  target?: TelegramTarget;
+  slot?: string;
+  threadName?: string;
+  nowMs?: number;
+}
+
+export async function promoteTelegramFollowerBindingToLeader(
+  deps: TelegramPromoteFollowerBindingToLeaderDeps,
+): Promise<TelegramTopicTargetRecord | undefined> {
+  const target = deps.target;
+  if (typeof target?.threadId !== "number") return undefined;
+  await deps.store.load();
+  const nowMs = deps.nowMs ?? Date.now();
+  const existing = deps.store
+    .list()
+    .find(
+      (record) =>
+        record.target.chatId === target.chatId &&
+        record.target.threadId === target.threadId,
+    );
+  const owner: TelegramThreadOwner = {
+    kind: "leader",
+    cwd: deps.cwd,
+    instanceId: deps.instanceId,
+  };
+  const record = deps.store.upsert({
+    profileKey: getTelegramThreadOwnerKey(owner),
+    owner,
+    target: { chatId: target.chatId, threadId: target.threadId },
+    status: "active",
+    createdAtMs: existing?.createdAtMs ?? nowMs,
+    updatedAtMs: nowMs,
+    ...(existing?.threadName ?? deps.threadName
+      ? { threadName: existing?.threadName ?? deps.threadName }
+      : {}),
+    instanceId: deps.instanceId,
+    ...(existing?.slot ?? deps.slot ? { slot: existing?.slot ?? deps.slot } : {}),
+    ...(existing?.syncStatus ? { syncStatus: existing.syncStatus } : {}),
+    ...(existing?.lastSyncObservedAtMs !== undefined
+      ? { lastSyncObservedAtMs: existing.lastSyncObservedAtMs }
+      : {}),
+    lastReconcileAction: "follower-promoted-to-leader",
+    ...(existing?.rerouteConfirmedAtMs !== undefined
+      ? { rerouteConfirmedAtMs: existing.rerouteConfirmedAtMs }
+      : {}),
+  });
+  await deps.store.persist();
+  return record;
+}
+
 export interface TelegramOwnTopicProvisionDeps {
   getAllowedUserId: () => number | undefined;
   instanceId: string;
@@ -2077,7 +2131,7 @@ export function createTelegramTopicTargetProvisioner(
           existing.threadName ?? identityThreadName ?? bakedThreadName,
         instanceId: request.instanceId,
         slot,
-        owner: existing.owner ?? request.owner,
+        owner: request.owner ?? existing.owner,
         lastError: undefined,
       });
       return { target: record.target, reused: true, record };

package/lib/updates.ts

@@ -9,6 +9,7 @@ import {
   createTelegramThreadTarget,
   type TelegramTarget,
 } from "./target.ts";
+import type { TelegramMessageOwnershipStore } from "./ownership.ts";
 import {
   createTelegramUserPairingRuntime,
   getTelegramAuthorizationState,
@@ -538,6 +539,14 @@ export function buildTelegramUpdateExecutionPlanFromUpdate<
 
 // --- Runtime ---
 
+export type TelegramMessageOwnershipRecorderInput = Parameters<
+  TelegramMessageOwnershipStore["record"]
+>[0];
+
+export type TelegramMessageOwnershipRecorder = (
+  input: TelegramMessageOwnershipRecorderInput,
+) => void;
+
 export interface TelegramUpdateRuntimeDeps<
   TContext = unknown,
   TReactionUpdate extends TelegramMessageReactionUpdated =
@@ -549,6 +558,7 @@ export interface TelegramUpdateRuntimeDeps<
   getCurrentInstanceId?: () => string | undefined;
   getMessageOwnership?: TelegramMessageOwnershipLookup;
   getTargetOwnership?: TelegramTargetOwnershipLookup;
+  recordMessageOwnership?: TelegramMessageOwnershipRecorder;
   foreignOwnedUpdateForwarder?: TelegramForeignOwnedUpdateForwarder<
     TContext,
     TReactionUpdate,
@@ -612,6 +622,7 @@ export interface TelegramUpdateRuntimeControllerDeps<
   getCurrentInstanceId?: () => string | undefined;
   getMessageOwnership?: TelegramMessageOwnershipLookup;
   getTargetOwnership?: TelegramTargetOwnershipLookup;
+  recordMessageOwnership?: TelegramMessageOwnershipRecorder;
   foreignOwnedUpdateForwarder?: TelegramForeignOwnedUpdateForwarder<
     TContext,
     TelegramMessageReactionUpdated,
@@ -824,6 +835,7 @@ export function createTelegramPairedUpdateRuntime<
     getCurrentInstanceId: deps.getCurrentInstanceId,
     getMessageOwnership: deps.getMessageOwnership,
     getTargetOwnership: deps.getTargetOwnership,
+    recordMessageOwnership: deps.recordMessageOwnership,
     handleTelegramTopicLifecycleUpdate: deps.handleTelegramTopicLifecycleUpdate,
     foreignOwnedUpdateForwarder: deps.foreignOwnedUpdateForwarder,
     removePendingMediaGroupMessages: deps.removePendingMediaGroupMessages,
@@ -890,6 +902,7 @@ export function createTelegramUpdateRuntime<
         getCurrentInstanceId: deps.getCurrentInstanceId,
         getMessageOwnership: deps.getMessageOwnership,
         getTargetOwnership: deps.getTargetOwnership,
+        recordMessageOwnership: deps.recordMessageOwnership,
         foreignOwnedUpdateForwarder: deps.foreignOwnedUpdateForwarder,
         removePendingMediaGroupMessages: deps.removePendingMediaGroupMessages,
         removeQueuedTelegramTurnsByMessageIds:
@@ -1097,7 +1110,7 @@ export async function executeTelegramUpdatePlan<
       if (plan.shouldDeny) {
         await deps.answerGuestQuery(
           plan.guestMessage.guest_query_id,
-          "Access denied.",
+          "🚫 Access denied.",
         );
         return;
       }
@@ -1109,11 +1122,40 @@ export async function executeTelegramUpdatePlan<
       }
       return;
     }
+    const foreignMessageOwnership = getForeignTelegramMessageOwnership(
+      getTelegramMessageReplyTarget(plan.message),
+      deps,
+    );
+    if (foreignMessageOwnership) {
+      if (plan.kind === "edited-message") {
+        await deps.foreignOwnedUpdateForwarder?.forwardEditedMessage?.({
+          message: plan.message,
+          ownership: foreignMessageOwnership,
+          ctx: deps.ctx,
+        });
+      } else {
+        await deps.foreignOwnedUpdateForwarder?.forwardMessage?.({
+          message: plan.message,
+          ownership: foreignMessageOwnership,
+          ctx: deps.ctx,
+        });
+      }
+      return;
+    }
+    const messageTarget = getTelegramMessageTarget(plan.message);
     const foreignTargetOwnership = getForeignTelegramTargetOwnership(
-      getTelegramMessageTarget(plan.message),
+      messageTarget,
       deps,
     );
     if (foreignTargetOwnership) {
+      if (typeof plan.message.message_id === "number") {
+        deps.recordMessageOwnership?.({
+          chatId: messageTarget!.chatId,
+          messageId: plan.message.message_id,
+          target: messageTarget,
+          instanceId: foreignTargetOwnership.instanceId,
+        });
+      }
       if (plan.kind === "edited-message") {
         await deps.foreignOwnedUpdateForwarder?.forwardEditedMessage?.({
           message: plan.message,
@@ -1129,7 +1171,6 @@ export async function executeTelegramUpdatePlan<
       }
       return;
     }
-    const messageTarget = getTelegramMessageTarget(plan.message);
     if (
       plan.kind === "message" &&
       messageTarget?.threadId != null &&

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llblab/pi-telegram",
-  "version": "0.18.5",
+  "version": "0.19.0",
   "private": false,
   "publishConfig": {
     "access": "public"
@@ -42,7 +42,7 @@
     "BACKLOG.md",
     "CHANGELOG.md",
     "docs/",
-    "screenshot.png"
+    "banner.png"
   ],
   "exports": {
     ".": "./index.ts",