pi-ui-extend 0.1.21 → 0.1.24

package/external/pi-tools-suite/src/telegram-mirror/index.ts

@@ -44,7 +44,9 @@ import {
 	IpcSocket,
 	buildInstanceId,
 	tryAcquireLeadership,
+	updateInstanceInfo,
 	type IpcMessage,
+	type InstanceInfo,
 } from "./ipc.js";
 import { Multiplexer, type LocalDispatch } from "./multiplexer.js";
 
@@ -55,19 +57,34 @@ interface MirrorContext {
 	isIdle(): boolean;
 	hasPendingMessages(): boolean;
 	compact(): void;
+	currentDialog(): string | undefined;
+}
+
+interface SessionSnapshot {
+	cwd: string;
+	sessionId?: string;
+	sessionFile?: string;
+	sessionName?: string;
 }
 
 const RECONNECT_DELAY_MS = 2_000;
+const COMMAND_NAME = "telegram-mirror";
+const COMMAND_ALIAS = "tg";
+const COMMAND_OFF = "tg-off";
 
 export default function telegramMirror(pi: ExtensionAPI): void {
 	const cfg = loadTelegramMirrorConfig();
-	if (!cfg || !cfg.enabled) {
-		// Soft-disable: no logging at startup, the user simply hasn't opted in.
+	if (!cfg) {
+		// Soft-disable: no logging at startup, the user hasn't configured
+		// botToken/chatId yet. When the block exists, even with
+		// enabled:false, still register /telegram-mirror and /tg so the
+		// user can activate the bot explicitly by slash command.
 		return;
 	}
 
 	const { botToken: token, chatId } = cfg;
-	const { id: selfId, info: selfInfo } = buildInstanceId();
+	const { id: selfId, info: baseSelfInfo } = buildInstanceId();
+	let selfInfo: InstanceInfo = baseSelfInfo;
 
 	let role: Role = "starting";
 	let bot: TelegramBot | undefined;
@@ -80,6 +97,7 @@ export default function telegramMirror(pi: ExtensionAPI): void {
 	let reconnectTimer: NodeJS.Timeout | undefined;
 	let shutdown = false;
 	let disabled = false;
+	let activationRequested = false;
 
 	const log = (message: string): void => {
 		// eslint-disable-next-line no-console
@@ -87,13 +105,16 @@ export default function telegramMirror(pi: ExtensionAPI): void {
 	};
 
 	// Dispatch the leader uses to execute commands on its own pi session.
-	const localDispatch: LocalDispatch = {
-		sendUserMessage(text) {
-			pi.sendUserMessage(text);
-		},
-		abort() {
-			mirrorCtx?.abort();
-		},
+		const localDispatch: LocalDispatch = {
+			sendUserMessage(text) {
+				pi.sendUserMessage(text);
+			},
+			currentDialog() {
+				return mirrorCtx?.currentDialog();
+			},
+			abort() {
+				mirrorCtx?.abort();
+			},
 		compact() {
 			mirrorCtx?.compact();
 		},
@@ -118,12 +139,18 @@ export default function telegramMirror(pi: ExtensionAPI): void {
 
 	const hooks: PixMirrorHooks = {
 		getRenderer: () => eventSink,
+		describeInstance: (ctx) => describeInstance(ctx),
 		notifyAgentEnd: () => undefined,
 	};
 
 	registerPixEventHandlers(pi, hooks);
+	registerActivationCommand(COMMAND_NAME);
+	registerActivationCommand(COMMAND_ALIAS);
+	registerOffCommand();
 
 	pi.on("session_start", async (_event, ctx) => {
+		refreshCtx(ctx as ExtensionContext | undefined);
+		refreshSelfInfo(ctx as ExtensionContext | undefined);
 		if (!captureHooksInstalled) {
 			captureHooksInstalled = true;
 			captureAbortableContext(ctx as ExtensionContext | undefined, {
@@ -145,11 +172,17 @@ export default function telegramMirror(pi: ExtensionAPI): void {
 				},
 			});
 		}
-		await start();
+		if (activationRequested) await start();
 	});
 
-	pi.on("agent_start", (_e, ctx) => refreshCtx(ctx as ExtensionContext | undefined));
-	pi.on("before_agent_start", (_e, ctx) => refreshCtx(ctx as ExtensionContext | undefined));
+	pi.on("agent_start", (_e, ctx) => {
+		refreshCtx(ctx as ExtensionContext | undefined);
+		refreshSelfInfo(ctx as ExtensionContext | undefined);
+	});
+	pi.on("before_agent_start", (_e, ctx) => {
+		refreshCtx(ctx as ExtensionContext | undefined);
+		refreshSelfInfo(ctx as ExtensionContext | undefined);
+	});
 
 	pi.on("session_shutdown", async (event) => {
 		// On reload/fork the module will be reloaded in the same process —
@@ -160,6 +193,7 @@ export default function telegramMirror(pi: ExtensionAPI): void {
 	});
 
 	async function start(): Promise<void> {
+		activationRequested = true;
 		if (shutdown) return;
 		if (disabled) return;
 		if (role !== "starting") return;
@@ -211,6 +245,17 @@ export default function telegramMirror(pi: ExtensionAPI): void {
 				await stepDown();
 				return;
 			}
+			await created.setMyCommands([
+				{ command: "menu", description: "Choose project/session" },
+				{ command: "list", description: "List pi sessions" },
+				{ command: "use", description: "Follow a session by number" },
+				{ command: "status", description: "Show followed session status" },
+				{ command: "clear", description: "Clear known bot messages" },
+				{ command: "abort", description: "Abort followed session" },
+				{ command: "compact", description: "Compact followed session" },
+				{ command: "disconnect", description: "Stop Telegram mirror" },
+				{ command: "help", description: "Show help" },
+			]).catch((error) => log(`setMyCommands failed: ${errorMessage(error)}`));
 			bot = created;
 			log(`connected as @${me.result.username} (leader) [${selfInfo.label}]`);
 		} catch (error) {
@@ -231,15 +276,16 @@ export default function telegramMirror(pi: ExtensionAPI): void {
 			standDown: clusterStandDown,
 			log,
 		});
-		multiplexer.init();
+			multiplexer.init();
 
-		bot.startPolling(async (update) => {
-			const text = update.message?.text;
-			if (typeof text !== "string") return;
-			if (!update.message?.chat || !bot?.isAllowedChat(update.message.chat.id)) return;
-			await multiplexer?.handleTgText(text);
-		});
-	}
+			bot.startPolling(async (update) => {
+				await multiplexer?.handleTelegramUpdate(update);
+			});
+			await bot.sendMessage(`✅ Telegram mirror active: ${selfInfo.label}`, {
+				replyMarkup: { inline_keyboard: [[{ text: "🧭 Choose project/session", callback_data: "tg:list" }]] },
+			}).catch((error) => log(`startup message failed: ${errorMessage(error)}`));
+			await multiplexer.showActiveDialog();
+		}
 
 	function becomeFollower(socket: IpcSocket): void {
 		role = "follower";
@@ -309,15 +355,53 @@ export default function telegramMirror(pi: ExtensionAPI): void {
 		socket.send({ type: "command_ack", reqId, ok, error });
 	}
 
-	async function handleFollowerQuery(socket: IpcSocket, reqId: string, query: string): Promise<void> {
-		if (query === "status") {
-			const result = mirrorCtx
-				? { idle: mirrorCtx.isIdle(), hasPending: mirrorCtx.hasPendingMessages() }
-				: { idle: true, hasPending: false };
-			socket.send({ type: "query_reply", reqId, ok: true, result });
-			return;
+		async function handleFollowerQuery(socket: IpcSocket, reqId: string, query: string): Promise<void> {
+			if (query === "status") {
+				const result = mirrorCtx
+					? { idle: mirrorCtx.isIdle(), hasPending: mirrorCtx.hasPendingMessages() }
+					: { idle: true, hasPending: false };
+				socket.send({ type: "query_reply", reqId, ok: true, result });
+				return;
+			}
+			if (query === "dialog") {
+				socket.send({ type: "query_reply", reqId, ok: true, result: { text: mirrorCtx?.currentDialog() ?? "" } });
+				return;
+			}
+			socket.send({ type: "query_reply", reqId, ok: false, error: `unknown query: ${query}` });
 		}
-		socket.send({ type: "query_reply", reqId, ok: false, error: `unknown query: ${query}` });
+
+	function registerActivationCommand(name: string): void {
+		pi.registerCommand(name, {
+			description: name === COMMAND_NAME ? "Start Telegram mirror and show connection status" : "Alias for /telegram-mirror",
+			handler: async (args, ctx) => {
+				refreshCtx(ctx as ExtensionContext | undefined);
+				refreshSelfInfo(ctx as ExtensionContext | undefined);
+				const trimmed = args.trim();
+				if (trimmed === "status") {
+					notify(ctx, localStatusText(), "info");
+					return;
+				}
+				if (trimmed === "stop" || trimmed === "disconnect") {
+					await clusterStandDown();
+					notify(ctx, "Telegram mirror stopped. Run /telegram-mirror to start again.", "info");
+					return;
+				}
+				disabled = false;
+				activationRequested = true;
+				await start();
+				notify(ctx, localStatusText(), "info");
+			},
+		});
+	}
+
+	function registerOffCommand(): void {
+		pi.registerCommand(COMMAND_OFF, {
+			description: "Stop Telegram mirror cluster",
+			handler: async (_args, ctx) => {
+				await clusterStandDown();
+				notify(ctx, "Telegram mirror stopped. Run /tg or /telegram-mirror to start again.", "info");
+			},
+		});
 	}
 
 	async function stepDown(): Promise<void> {
@@ -351,6 +435,7 @@ export default function telegramMirror(pi: ExtensionAPI): void {
 	async function clusterStandDown(): Promise<void> {
 		if (disabled) return;
 		disabled = true;
+		activationRequested = false;
 		if (server && role === "leader") {
 			try {
 				server.broadcast({ type: "stand_down" });
@@ -392,19 +477,46 @@ export default function telegramMirror(pi: ExtensionAPI): void {
 	function refreshCtx(ctx: ExtensionContext | undefined): void {
 		if (!ctx) return;
 		if (!mirrorCtx) {
-			mirrorCtx = makeCtx({
-				abort: () => ctx.abort(),
-				isIdle: () => ctx.isIdle(),
-				hasPendingMessages: () => ctx.hasPendingMessages(),
-				compact: () => ctx.compact(),
-			});
-			return;
-		}
+				mirrorCtx = makeCtx({
+					abort: () => ctx.abort(),
+					isIdle: () => ctx.isIdle(),
+					hasPendingMessages: () => ctx.hasPendingMessages(),
+					compact: () => ctx.compact(),
+					currentDialog: () => currentDialogFromContext(ctx),
+				});
+				return;
+			}
 		const m = mirrorCtx as Mutable<MirrorContext>;
 		m.abort = () => ctx.abort();
-		m.isIdle = () => ctx.isIdle();
-		m.hasPendingMessages = () => ctx.hasPendingMessages();
-		m.compact = () => ctx.compact();
+			m.isIdle = () => ctx.isIdle();
+			m.hasPendingMessages = () => ctx.hasPendingMessages();
+			m.compact = () => ctx.compact();
+			m.currentDialog = () => currentDialogFromContext(ctx);
+		}
+
+	function refreshSelfInfo(ctx: ExtensionContext | undefined): void {
+		const snapshot = sessionSnapshot(ctx);
+		if (!snapshot) return;
+		selfInfo = updateInstanceInfo(selfInfo, snapshot);
+		multiplexer?.updateSelfInfo(selfInfo);
+		if (role === "follower" && clientSocket && !clientSocket.isClosed) {
+			clientSocket.send({ type: "instance_update", info: selfInfo });
+		}
+	}
+
+	function describeInstance(ctx: ExtensionContext | undefined) {
+		refreshSelfInfo(ctx);
+		return {
+			label: selfInfo.label,
+			cwd: selfInfo.cwd,
+			...(selfInfo.sessionId ? { sessionId: selfInfo.sessionId } : {}),
+			...(selfInfo.sessionName ? { sessionName: selfInfo.sessionName } : {}),
+		};
+	}
+
+	function localStatusText(): string {
+		const status = role === "leader" ? "leader/polling" : role === "follower" ? "follower/connected" : "not connected";
+		return `Telegram mirror: ${status}\n${selfInfo.label}${selfInfo.sessionName ? ` · ${selfInfo.sessionName}` : ""}`;
 	}
 }
 
@@ -416,9 +528,103 @@ function makeCtx(seed: Partial<MirrorContext>): MirrorContext {
 		isIdle: seed.isIdle ?? (() => true),
 		hasPendingMessages: seed.hasPendingMessages ?? (() => false),
 		compact: seed.compact ?? (() => undefined),
+		currentDialog: seed.currentDialog ?? (() => undefined),
 	};
 }
 
+const TRANSCRIPT_MAX_MESSAGES = 40;
+const TRANSCRIPT_MAX_CHARS = 28_000;
+
+function currentDialogFromContext(ctx: ExtensionContext | undefined): string | undefined {
+	if (!ctx) return undefined;
+	let branch: unknown[];
+	try {
+		const maybeBranch = ctx.sessionManager.getBranch?.();
+		branch = Array.isArray(maybeBranch) ? maybeBranch : [...(maybeBranch ?? [])];
+	} catch {
+		return undefined;
+	}
+
+	const messages: { role: "user" | "assistant"; text: string }[] = [];
+	for (const entry of branch) {
+		if (!isRecord(entry) || entry.type !== "message" || !isRecord(entry.message)) continue;
+		const role = entry.message.role === "user" || entry.message.role === "assistant" ? entry.message.role : undefined;
+		if (!role) continue;
+		const text = stripDcpMarkers(visibleMessageText(entry.message.content, role)).trim();
+		if (!text) continue;
+		messages.push({ role, text });
+	}
+	if (messages.length === 0) return undefined;
+
+	const selected: string[] = [];
+	let used = 0;
+	let omitted = 0;
+	for (let idx = messages.length - 1; idx >= 0; idx -= 1) {
+		const formatted = formatDialogMessage(messages[idx]);
+		if (selected.length >= TRANSCRIPT_MAX_MESSAGES || (used + formatted.length > TRANSCRIPT_MAX_CHARS && selected.length > 0)) {
+			omitted = idx + 1;
+			break;
+		}
+		selected.unshift(formatted);
+		used += formatted.length;
+	}
+	return `${omitted > 0 ? `… ${omitted} earlier message(s) omitted …\n\n` : ""}${selected.join("\n\n")}`;
+}
+
+function formatDialogMessage(message: { role: "user" | "assistant"; text: string }): string {
+	return `${message.role === "user" ? "👤 You" : "🤖 Pi"}\n${message.text}`;
+}
+
+function stripDcpMarkers(text: string): string {
+	return text
+		.split(/\r?\n/u)
+		.map((line) => line.replace(DCP_MARKER_RE, "").trimEnd())
+		.join("\n")
+		.replace(/[ \t]+\n/gu, "\n")
+		.replace(/\n{3,}/gu, "\n\n")
+		.trim();
+}
+
+const DCP_MARKER_RE = /\[dcp(?:-[\w-]+)?\]:\s*#\s*\([^)]*\)/giu;
+
+function visibleMessageText(value: unknown, role: "user" | "assistant"): string {
+	if (typeof value === "string") return value;
+	if (Array.isArray(value)) return value.map((item) => visibleContentBlockText(item, role)).filter(Boolean).join("\n");
+	return visibleContentBlockText(value, role);
+}
+
+function visibleContentBlockText(value: unknown, role: "user" | "assistant"): string {
+	if (typeof value === "string") return value;
+	if (!isRecord(value)) return "";
+	const type = typeof value.type === "string" ? value.type.toLowerCase() : "";
+	if (type.includes("tool") || type.includes("thinking")) return "";
+	if (typeof value.text === "string") return value.text;
+	if (typeof value.content === "string") return value.content;
+	if (Array.isArray(value.content)) return value.content.map((item) => visibleContentBlockText(item, role)).filter(Boolean).join("\n");
+	if (role === "user" && (type.includes("image") || type.includes("file"))) return `[${type || "attachment"}]`;
+	return "";
+}
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+	return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+
+function sessionSnapshot(ctx: ExtensionContext | undefined): SessionSnapshot | undefined {
+	if (!ctx) return undefined;
+	const manager = ctx.sessionManager;
+	return {
+		cwd: manager.getCwd?.() ?? ctx.cwd,
+		...(manager.getSessionId?.() ? { sessionId: manager.getSessionId() } : {}),
+		...(manager.getSessionFile?.() ? { sessionFile: manager.getSessionFile() } : {}),
+		...(manager.getSessionName?.() ? { sessionName: manager.getSessionName() } : {}),
+	};
+}
+
+function notify(ctx: { hasUI?: boolean; ui?: { notify?: (message: string, type?: "info" | "warning" | "error") => void } }, message: string, type: "info" | "warning" | "error" = "info"): void {
+	if (ctx.hasUI) ctx.ui?.notify?.(message, type);
+	else console.error(`[telegram-mirror] ${message}`);
+}
+
 function errorMessage(error: unknown): string {
 	return error instanceof Error ? error.message : String(error);
 }

package/external/pi-tools-suite/src/telegram-mirror/ipc.ts

@@ -57,12 +57,19 @@ export interface InstanceInfo {
 	cwd: string;
 	/** Human-friendly label, e.g. basename of cwd + short pid suffix. */
 	label: string;
+	/** Current pi session id, when available. */
+	sessionId?: string;
+	/** Current pi session file, when available. */
+	sessionFile?: string;
+	/** Human-friendly pi session name, when set. */
+	sessionName?: string;
 	/** ms-since-epoch when this instance started. */
 	started: number;
 }
 
 export type IpcMessage =
 	| { type: "register"; info: InstanceInfo }
+	| { type: "instance_update"; info: InstanceInfo }
 	| { type: "registered"; leader: InstanceInfo; activeId: string | null }
 	| { type: "ping"; t: number }
 	| { type: "pong"; t: number }
@@ -414,6 +421,19 @@ export function buildInstanceId(): { id: string; info: InstanceInfo } {
 	};
 }
 
+export function updateInstanceInfo(base: InstanceInfo, patch: Partial<Pick<InstanceInfo, "cwd" | "sessionId" | "sessionFile" | "sessionName">>): InstanceInfo {
+	const cwd = patch.cwd ?? base.cwd;
+	const cwdBase = path.basename(cwd) || cwd;
+	return {
+		...base,
+		cwd,
+		label: `${cwdBase} (#${base.pid})`,
+		...(patch.sessionId !== undefined ? { sessionId: patch.sessionId } : base.sessionId ? { sessionId: base.sessionId } : {}),
+		...(patch.sessionFile !== undefined ? { sessionFile: patch.sessionFile } : base.sessionFile ? { sessionFile: base.sessionFile } : {}),
+		...(patch.sessionName !== undefined ? { sessionName: patch.sessionName } : base.sessionName ? { sessionName: base.sessionName } : {}),
+	};
+}
+
 /** Generate a unique request id for command/query correlation. */
 export function generateReqId(): string {
 	return `${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;