@llblab/pi-telegram 0.8.1 → 0.8.2

package/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## 0.8.2: Lock-Safe Delivery
+
+- `[Lock Safety]` Active Telegram turns now re-check singleton ownership before preview flushes and final agent-end delivery. Impact: an old π instance stays silent after another instance takes the Telegram bridge lock, even if the old instance finishes a long-running prompt later.
+- `[Inbound Handlers]` The first step of an inbound composition now receives the full configured handler timeout before elapsed-time accounting starts on later steps. Impact: composition timeout behavior is deterministic and avoids one-millisecond test/runtime drift at pipeline start.
+- `[Menu UI]` Model and Thinking submenu headers now include their matching command icons (`🤖` and `🧠`). Impact: submenu headings match the Queue menu's icon-led style.
+- `[Package]` Bumped package metadata to `0.8.2` and kept the lockfile in sync.
+
 ## 0.8.1: Outbound Voice Translation Hotfix
 
 - `[Outbound Voice]` Composed voice handlers now pass the original `telegram_voice` text to the first pipeline step through stdin, then continue piping each step's stdout into the next step. Impact: translate-from-stdin voice pipelines can translate hidden voice text before TTS instead of failing with an empty first-step input.

package/index.ts

@@ -49,6 +49,8 @@ export default function (pi: Pi.ExtensionAPI) {
   const { abort, lifecycle, queue, setup, typing } = bridgeRuntime;
   const configStore = Config.createTelegramConfigStore();
   const lockRuntime = Locks.createTelegramLockRuntime<Pi.ExtensionContext>();
+  const lockOwnershipGuard =
+    Locks.createTelegramLockOwnershipGuard(lockRuntime);
   const activeTurnRuntime = Queue.createTelegramActiveTurnStore();
   const buttonActionStore = OutboundHandlers.createTelegramButtonActionStore();
   const pendingModelSwitchStore =
@@ -198,6 +200,7 @@ export default function (pi: Pi.ExtensionAPI) {
     sendDraft: sendMessageDraft,
     sendMessage,
     editMessageText: editTelegramMessageText,
+    canSend: lockOwnershipGuard.ownsCurrentProcess,
     ...replyTransport,
   });
   const { finalizeMarkdownPreview } =
@@ -459,6 +462,7 @@ export default function (pi: Pi.ExtensionAPI) {
     sendQueuedAttachments: queuedAttachmentSender,
     planOutboundReply: outboundReplyPlanner,
     sendOutboundReplyArtifacts: outboundReplyArtifactSender,
+    isCurrentOwner: lockOwnershipGuard.ownsContext,
     getActiveToolExecutions: lifecycle.getActiveToolExecutions,
     setActiveToolExecutions: lifecycle.setActiveToolExecutions,
     triggerPendingModelSwitchAbort: modelSwitchController.triggerPendingAbort,

package/lib/inbound-handlers.ts

@@ -258,6 +258,15 @@ function getRemainingTelegramInboundTimeout(
   return Math.max(1, timeout - (Date.now() - startedAt));
 }
 
+function getTelegramInboundInitialCompositionStepTimeout(
+  handler: TelegramInboundHandlerConfig,
+  step: TelegramInboundCommandTemplateConfig,
+): number {
+  const timeout = getTelegramInboundHandlerTimeout(handler);
+  const stepTimeout = getTelegramInboundHandlerConfiguredTimeout(step);
+  return stepTimeout === undefined ? timeout : Math.min(stepTimeout, timeout);
+}
+
 function getTelegramInboundCompositionStepTimeout(
   handler: TelegramInboundHandlerConfig,
   step: TelegramInboundCommandTemplateConfig,
@@ -421,7 +430,9 @@ async function executeTelegramTextHandler(
         output,
         cwd,
         deps,
-        getTelegramInboundCompositionStepTimeout(handler, step, startedAt),
+        index === 0
+          ? getTelegramInboundInitialCompositionStepTimeout(handler, step)
+          : getTelegramInboundCompositionStepTimeout(handler, step, startedAt),
       );
     } catch (error) {
       if (typeof step === "object" && step.critical) throw error;
@@ -498,7 +509,9 @@ async function executeTelegramInboundHandler(
         cwd,
         deps,
         false,
-        getTelegramInboundCompositionStepTimeout(handler, step, startedAt),
+        index === 0
+          ? getTelegramInboundInitialCompositionStepTimeout(handler, step)
+          : getTelegramInboundCompositionStepTimeout(handler, step, startedAt),
         index === 0 ? undefined : output,
       );
     } catch (error) {

package/lib/locks.ts

@@ -61,6 +61,11 @@ export interface TelegramLockRuntime<TContext extends TelegramLockContext> {
   owns: (ctx?: TelegramLockContext) => boolean;
 }
 
+export interface TelegramLockOwnershipGuard<TContext extends TelegramLockContext> {
+  ownsCurrentProcess: () => boolean;
+  ownsContext: (ctx: TContext) => boolean;
+}
+
 export interface TelegramLockRuntimeOptions {
   key?: string;
   locksPath?: string;
@@ -234,6 +239,17 @@ export function createTelegramLockRuntime<TContext extends TelegramLockContext>(
   };
 }
 
+export function createTelegramLockOwnershipGuard<
+  TContext extends TelegramLockContext,
+>(
+  lock: TelegramLockRuntime<TContext>,
+): TelegramLockOwnershipGuard<TContext> {
+  return {
+    ownsCurrentProcess: () => lock.owns(),
+    ownsContext: (ctx) => lock.owns(ctx),
+  };
+}
+
 export function createTelegramLockedPollingRuntime<
   TContext extends TelegramLockContext,
 >(

package/lib/menu-model.ts

@@ -236,7 +236,7 @@ export interface TelegramModelMenuRuntime<
 
 export const TELEGRAM_MODEL_PAGE_SIZE = 6;
 const TELEGRAM_MODEL_PAGE_PICKER_ROW_SIZE = 4;
-export const MODEL_MENU_TITLE = "<b>Choose a model:</b>";
+export const MODEL_MENU_TITLE = "<b>🤖 Choose a model:</b>";
 export const MODEL_PAGE_MENU_TITLE = "<b>Choose a page:</b>";
 
 function truncateTelegramButtonLabel(label: string, maxLength = 56): string {

package/lib/menu-thinking.ts

@@ -109,7 +109,7 @@ export async function handleTelegramThinkingMenuCallbackAction(
 }
 
 export function buildThinkingMenuText(): string {
-  return "<b>Choose a thinking level:</b>";
+  return "<b>🧠 Choose a thinking level:</b>";
 }
 
 export function buildThinkingMenuReplyMarkup(

package/lib/preview.ts

@@ -89,6 +89,7 @@ export interface TelegramPreviewRuntimeDeps<
     chunks: TelegramRenderedChunk[],
     options?: { replyMarkup?: TReplyMarkup },
   ) => Promise<number | undefined>;
+  canSend?: () => boolean;
 }
 
 export interface TelegramPreviewActiveTurn {
@@ -182,6 +183,7 @@ export interface TelegramPreviewControllerDeps<
     chunks: TelegramRenderedChunk[],
     options?: { replyMarkup?: TReplyMarkup },
   ) => Promise<number | undefined>;
+  canSend?: () => boolean;
   throttleMs?: number;
   maxDraftId?: number;
   setTimer?: (
@@ -418,6 +420,7 @@ export function createTelegramPreviewController<
         options,
       ),
     editRenderedMessage: deps.editRenderedMessage,
+    canSend: deps.canSend,
   });
   return {
     getState: () => state,
@@ -574,6 +577,10 @@ async function performTelegramPreviewFlush<
   state: TelegramPreviewRuntimeState,
   deps: TelegramPreviewRuntimeDeps<TReplyMarkup>,
 ): Promise<void> {
+  if (deps.canSend && !deps.canSend()) {
+    await clearTelegramPreview(chatId, deps);
+    return;
+  }
   const snapshot = buildTelegramPreviewSnapshot({
     state,
     maxMessageLength: deps.maxMessageLength,
@@ -658,6 +665,10 @@ export async function finalizeTelegramPreview<
 ): Promise<boolean> {
   const state = deps.getState();
   if (!state) return false;
+  if (deps.canSend && !deps.canSend()) {
+    await clearTelegramPreview(chatId, deps);
+    return false;
+  }
   await flushTelegramPreview(chatId, deps);
   const finalText = buildTelegramPreviewFinalText(state);
   if (!finalText) {
@@ -683,6 +694,10 @@ export async function finalizeTelegramMarkdownPreview<
 ): Promise<boolean> {
   const state = deps.getState();
   if (!state) return false;
+  if (deps.canSend && !deps.canSend()) {
+    await clearTelegramPreview(chatId, deps);
+    return false;
+  }
   await flushTelegramPreview(chatId, deps);
   const chunks = deps.renderTelegramMessage(markdown, { mode: "markdown" });
   if (chunks.length === 0) {

package/lib/queue.ts

@@ -771,6 +771,7 @@ export interface TelegramAgentEndRuntimeDeps<
   preserveQueuedTurnsAsHistory: boolean;
   resetRuntimeState: () => void;
   updateStatus: () => void;
+  isCurrentOwner?: () => boolean;
   dispatchNextQueuedTelegramTurn: () => void;
   clearPreview: (chatId: number) => Promise<void>;
   setPreviewPendingText: (text: string) => void;
@@ -815,6 +816,7 @@ export interface TelegramAgentEndHookRuntimeDeps<
   getPreserveQueuedTurnsAsHistory: () => boolean;
   resetRuntimeState: () => void;
   updateStatus: (ctx: TContext) => void;
+  isCurrentOwner?: (ctx: TContext) => boolean;
   dispatchNextQueuedTelegramTurn: (ctx: TContext) => void;
   requestDeferredDispatchNextQueuedTelegramTurn: (
     dispatch: (ctx: TContext) => void,
@@ -932,6 +934,9 @@ export function createTelegramAgentEndHook<
       preserveQueuedTurnsAsHistory: deps.getPreserveQueuedTurnsAsHistory(),
       resetRuntimeState: deps.resetRuntimeState,
       updateStatus: () => deps.updateStatus(ctx),
+      isCurrentOwner: deps.isCurrentOwner
+        ? () => deps.isCurrentOwner?.(ctx) ?? false
+        : undefined,
       dispatchNextQueuedTelegramTurn: () => {
         deps.requestDeferredDispatchNextQueuedTelegramTurn(
           deps.dispatchNextQueuedTelegramTurn,
@@ -964,6 +969,10 @@ export async function handleTelegramAgentEndRuntime<
   const replyMarkup = outboundReply?.replyMarkup;
   deps.resetRuntimeState();
   deps.updateStatus();
+  if (turn && deps.isCurrentOwner && !deps.isCurrentOwner()) {
+    await deps.clearPreview(turn.chatId);
+    return;
+  }
   const endPlan = buildTelegramAgentEndPlan({
     hasTurn: !!turn,
     stopReason: assistant.stopReason,

package/package.json

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