pi-telegram-plus 0.0.1

package/lib/commands/session.ts

@@ -0,0 +1,281 @@
+import { stat } from "node:fs/promises";
+import { homedir } from "node:os";
+import { resolve } from "node:path";
+import { SessionManager, type SessionEntry } from "@earendil-works/pi-coding-agent";
+import type { SessionNameDeps, CommandRegistry } from "./register.ts";
+import { escapeHtml } from "../html.ts";
+import type { CapturedAgentSession } from "../types.ts";
+
+function truncate(text: string, max: number): string {
+  return text.length <= max ? text : text.slice(0, max - 1) + "…";
+}
+
+function isMessageEntry(entry: SessionEntry): entry is SessionEntry & { type: "message"; message: { role: string; content?: unknown } } {
+  return entry.type === "message";
+}
+
+function getEntryText(entry: { role: string; content?: unknown }): string {
+  if (!entry.content || !Array.isArray(entry.content)) return "(empty)";
+  return (entry.content as Array<{ type?: string; text?: string }>)
+    .filter((c) => c.type === "text" && typeof c.text === "string")
+    .map((c) => c.text!)
+    .join(" ")
+    .replace(/\n/g, " ") || "(empty)";
+}
+
+function stripMatchingQuotes(input: string): string {
+  if ((input.startsWith('"') && input.endsWith('"')) || (input.startsWith("'") && input.endsWith("'"))) {
+    return input.slice(1, -1);
+  }
+  return input;
+}
+
+function resolveTargetCwd(input: string, cwd: string): string {
+  const unquoted = stripMatchingQuotes(input.trim());
+  const expanded = unquoted === "~" || unquoted.startsWith("~/")
+    ? resolve(homedir(), unquoted.slice(2))
+    : unquoted;
+  return resolve(cwd, expanded);
+}
+
+/**
+ * All command handlers capture ctx.ui at entry and use the captured reference.
+ * See model.ts for explanation.
+ */
+export function registerSessionCommands(
+  registry: CommandRegistry,
+  deps: SessionNameDeps,
+): void {
+  // ── /new ──────────────────────────────────────────────────────────────
+  registry.registerCommand("new", {
+    description: "Start a new session",
+    handler: async (_args, ctx) => {
+      const ui = ctx.ui;
+      const confirmed = await ui.confirm("New session", "Start a new session? Current session will be saved.");
+      if (!confirmed) return;
+      const result = await ctx.newSession({
+        withSession: async () => {
+          ui.notify("✅ New session started.", "info");
+        },
+      });
+      if (result.cancelled) ui.notify("Cancelled.", "info");
+    },
+  });
+
+  // ── /fork ────────────────────────────────────────────────────────────
+  registry.registerCommand("fork", {
+    description: "Fork from a previous user message",
+    handler: async (args, ctx) => {
+      const ui = ctx.ui;
+      const session = deps.getSession();
+      if (!session) {
+        ui.notify("No active session", "error");
+        return;
+      }
+
+      if (args) {
+        await ctx.fork(args, { position: "before" });
+        ui.notify("✅ Forked session.", "info");
+        return;
+      }
+
+      const entries = session.sessionManager.getEntries();
+      const userEntries = entries.filter(isMessageEntry).filter((e) => e.message.role === "user");
+      if (userEntries.length === 0) {
+        ui.notify("No user messages to fork from.", "info");
+        return;
+      }
+
+      const shown = userEntries;
+      const labels = shown.map((e) => truncate(getEntryText(e.message), 50));
+      const choice = await ui.select("Fork from message", labels);
+      if (!choice) return;
+      const index = labels.indexOf(choice);
+      if (index < 0) return;
+      await ctx.fork(shown[index].id, { position: "before" });
+      ui.notify("✅ Forked session.", "info");
+    },
+  });
+
+  // ── /clone ────────────────────────────────────────────────────────────
+  registry.registerCommand("clone", {
+    description: "Clone at a previous user message",
+    handler: async (args, ctx) => {
+      const ui = ctx.ui;
+      const session = deps.getSession();
+      if (!session) {
+        ui.notify("No active session", "error");
+        return;
+      }
+
+      if (args) {
+        await ctx.fork(args, { position: "at" });
+        ui.notify("✅ Cloned session.", "info");
+        return;
+      }
+
+      const entries = session.sessionManager.getEntries();
+      const userEntries = entries.filter(isMessageEntry).filter((e) => e.message.role === "user");
+      if (userEntries.length === 0) {
+        ui.notify("No user messages to clone from.", "info");
+        return;
+      }
+
+      const shown = userEntries;
+      const labels = shown.map((e) => truncate(getEntryText(e.message), 50));
+      const choice = await ui.select("Clone at message", labels);
+      if (!choice) return;
+      const index = labels.indexOf(choice);
+      if (index < 0) return;
+      await ctx.fork(shown[index].id, { position: "at" });
+      ui.notify("✅ Cloned session.", "info");
+    },
+  });
+
+  // ── /tree ─────────────────────────────────────────────────────────────
+  registry.registerCommand("tree", {
+    description: "Navigate session tree to a previous message",
+    handler: async (args, ctx) => {
+      const ui = ctx.ui;
+      const session = deps.getSession();
+      if (!session) {
+        ui.notify("No active session", "error");
+        return;
+      }
+
+      if (args) {
+        const wantsSummary = await ui.confirm("Navigate", "Summarize the abandoned branch?");
+        await ctx.navigateTree(args, { summarize: wantsSummary });
+        ui.notify("✅ Navigated to selected point.", "info");
+        return;
+      }
+
+      const entries = session.sessionManager.getEntries();
+      const userEntries = entries.filter(isMessageEntry).filter((e) => e.message.role === "user");
+      if (userEntries.length === 0) {
+        ui.notify("No entries to navigate to.", "info");
+        return;
+      }
+
+      const shown = userEntries;
+      const labels = shown.map((e) => truncate(getEntryText(e.message), 50));
+      const choice = await ui.select("Navigate to message", labels);
+      if (!choice) return;
+      const index = labels.indexOf(choice);
+      if (index < 0) return;
+      const entry = shown[index];
+      const wantsSummary = await ui.confirm("Navigate", "Summarize the abandoned branch?");
+      await ctx.navigateTree(entry.id, { summarize: wantsSummary });
+      ui.notify("✅ Navigated to selected point.", "info");
+    },
+  });
+
+  // ── /cwd ──────────────────────────────────────────────────────────────
+  registry.registerCommand("cwd", {
+    description: "Show current working directory",
+    handler: async (_args, ctx) => {
+      ctx.ui.notify(`cwd: ${escapeHtml(ctx.cwd)}`, "info");
+    },
+  });
+
+  // ── /cd ───────────────────────────────────────────────────────────────
+  registry.registerCommand("cd", {
+    description: "Switch pi working directory",
+    handler: async (args, ctx) => {
+      const ui = ctx.ui;
+      const rawPath = args.trim() || await ui.input("Working directory", ctx.cwd);
+      if (!rawPath) return;
+
+      const targetCwd = resolveTargetCwd(rawPath, ctx.cwd);
+      const info = await stat(targetCwd).catch(() => undefined);
+      if (!info?.isDirectory()) {
+        ui.notify(`Not a directory: ${escapeHtml(targetCwd)}`, "error");
+        return;
+      }
+
+      const currentSessionFile = ctx.sessionManager.getSessionFile();
+      const sessionManager = SessionManager.create(targetCwd, undefined, {
+        ...(currentSessionFile ? { parentSession: currentSessionFile } : {}),
+      });
+      // Ensure the header-only session file exists before switchSession opens it.
+      (sessionManager as unknown as { _rewriteFile?: () => void })._rewriteFile?.();
+      const sessionPath = sessionManager.getSessionFile();
+      if (!sessionPath) {
+        ui.notify("Cannot switch cwd from an ephemeral session.", "error");
+        return;
+      }
+
+      const result = await ctx.switchSession(sessionPath, {
+        withSession: async (nextCtx: { ui: import("@earendil-works/pi-coding-agent").ExtensionUIContext }) => {
+          nextCtx.ui.notify(`✅ Switched cwd:\n${escapeHtml(targetCwd)}`, "info");
+        },
+      });
+      if (result.cancelled) ui.notify("Cwd switch cancelled.", "info");
+    },
+  });
+
+  // ── /resume ───────────────────────────────────────────────────────────
+  registry.registerCommand("resume", {
+    description: "Resume a previous session",
+    handler: async (_args, ctx) => {
+      const ui = ctx.ui;
+      const sessions = await SessionManager.list(ctx.cwd);
+      if (sessions.length === 0) {
+        ui.notify("No sessions found.", "info");
+        return;
+      }
+
+      const shown = sessions;
+      const labels = shown.map((s) => truncate(s.name ?? s.id, 50));
+      const choice = await ui.select("Resume session", labels);
+      if (!choice) return;
+      const index = labels.indexOf(choice);
+      if (index < 0) return;
+      await ctx.switchSession(shown[index].path);
+      ui.notify("✅ Switched session.", "info");
+    },
+  });
+
+  // ── /name ──────────────────────────────────────────────────────────────
+  registry.registerCommand("name", {
+    description: "Set or show session name",
+    handler: async (args, ctx) => {
+      const ui = ctx.ui;
+      if (args) {
+        deps.setSessionName(args);
+        ui.notify(`Session name set: ${args}`, "info");
+        return;
+      }
+      const name = deps.getSessionName();
+      ui.notify(name ? `Session name: ${name}` : "No session name set. Use /name <name> to set one.", "info");
+    },
+  });
+
+  // ── /session ──────────────────────────────────────────────────────────
+  registry.registerCommand("session", {
+    description: "Show session statistics",
+    handler: async (_args, ctx) => {
+      const ui = ctx.ui;
+      const session = deps.getSession();
+      if (!session) {
+        ui.notify("No active session", "error");
+        return;
+      }
+      const stats = session.getSessionStats();
+      const usage = session.getContextUsage();
+      const lines = [
+        `Session Info`,
+        `id: ${escapeHtml(stats.sessionId)}`,
+        `file: ${escapeHtml(stats.sessionFile ?? "ephemeral")}`,
+        `messages: ${stats.userMessages}u / ${stats.assistantMessages}a`,
+        `tool calls: ${stats.toolCalls}`,
+        `tokens: ${stats.tokens.total}`,
+        `cost: ${stats.cost.toFixed(4)}`,
+      ];
+      if (usage) {
+        lines.push(`context: ${usage.tokens ?? "?"} / ${usage.contextWindow}`);
+      }
+      ui.notify(lines.join("\n"), "info");
+    },
+  });
+}

package/lib/commands/settings.ts

@@ -0,0 +1,129 @@
+import type { CommandRegistry } from "./register.ts";
+import type { CapturedAgentSession } from "../types.ts";
+
+export function registerSettingsCommands(
+  registry: CommandRegistry,
+  deps: { getSession: () => CapturedAgentSession | undefined },
+): void {
+  registry.registerCommand("settings", {
+    description: "View or change common settings",
+    handler: async (_args, ctx) => {
+      const ui = ctx.ui;
+      const session = deps.getSession();
+      if (!session) {
+        ui.notify("No active session", "error");
+        return;
+      }
+
+      const s = session.settingsManager;
+      const options = [
+        `Hide thinking: ${s.getHideThinkingBlock() ? "on" : "off"}`,
+        `Compaction: ${s.getCompactionEnabled() ? "on" : "off"}`,
+        `Retry: ${s.getRetryEnabled() ? "on" : "off"}`,
+        `Show images: ${s.getShowImages() ? "on" : "off"}`,
+        `Terminal progress: ${s.getShowTerminalProgress() ? "on" : "off"}`,
+        `Quiet startup: ${s.getQuietStartup() ? "on" : "off"}`,
+        `Theme: ${s.getTheme() ?? "default"}`,
+        `Default thinking: ${s.getDefaultThinkingLevel() ?? "unset"}`,
+        `Block images: ${s.getBlockImages() ? "on" : "off"}`,
+        `Image auto resize: ${s.getImageAutoResize() ? "on" : "off"}`,
+        `Clear on shrink: ${s.getClearOnShrink() ? "on" : "off"}`,
+        `Double escape: ${s.getDoubleEscapeAction()}`,
+        `Tree filter: ${s.getTreeFilterMode()}`,
+        `Shell prefix: ${s.getShellCommandPrefix() ?? "unset"}`,
+      ];
+
+      const choice = await ui.select("Settings", options);
+      if (!choice) return;
+
+      if (choice.startsWith("Hide thinking")) {
+        const next = !s.getHideThinkingBlock();
+        s.setHideThinkingBlock(next);
+        ui.notify(`Hide thinking: ${next ? "on" : "off"}`, "info");
+        return;
+      }
+      if (choice.startsWith("Compaction")) {
+        const next = !s.getCompactionEnabled();
+        s.setCompactionEnabled(next);
+        ui.notify(`Compaction: ${next ? "on" : "off"}`, "info");
+        return;
+      }
+      if (choice.startsWith("Retry")) {
+        const next = !s.getRetryEnabled();
+        s.setRetryEnabled(next);
+        ui.notify(`Retry: ${next ? "on" : "off"}`, "info");
+        return;
+      }
+      if (choice.startsWith("Show images")) {
+        const next = !s.getShowImages();
+        s.setShowImages(next);
+        ui.notify(`Show images: ${next ? "on" : "off"}`, "info");
+        return;
+      }
+      if (choice.startsWith("Terminal progress")) {
+        const next = !s.getShowTerminalProgress();
+        s.setShowTerminalProgress(next);
+        ui.notify(`Terminal progress: ${next ? "on" : "off"}`, "info");
+        return;
+      }
+      if (choice.startsWith("Quiet startup")) {
+        const next = !s.getQuietStartup();
+        s.setQuietStartup(next);
+        ui.notify(`Quiet startup: ${next ? "on" : "off"}`, "info");
+        return;
+      }
+      if (choice.startsWith("Theme")) {
+        const theme = await ui.input("Theme name", s.getTheme() ?? "dark");
+        if (!theme) return;
+        s.setTheme(theme.trim());
+        ui.notify(`Theme set: ${theme.trim()}. Use /reload to apply.`, "info");
+        return;
+      }
+      if (choice.startsWith("Default thinking")) {
+        const levels = ["off", "minimal", "low", "medium", "high", "xhigh"];
+        const level = await ui.select("Default thinking", levels);
+        if (!level) return;
+        s.setDefaultThinkingLevel(level as any);
+        ui.notify(`Default thinking: ${level}`, "info");
+        return;
+      }
+      if (choice.startsWith("Block images")) {
+        const next = !s.getBlockImages();
+        s.setBlockImages(next);
+        ui.notify(`Block images: ${next ? "on" : "off"}`, "info");
+        return;
+      }
+      if (choice.startsWith("Image auto resize")) {
+        const next = !s.getImageAutoResize();
+        s.setImageAutoResize(next);
+        ui.notify(`Image auto resize: ${next ? "on" : "off"}`, "info");
+        return;
+      }
+      if (choice.startsWith("Clear on shrink")) {
+        const next = !s.getClearOnShrink();
+        s.setClearOnShrink(next);
+        ui.notify(`Clear on shrink: ${next ? "on" : "off"}`, "info");
+        return;
+      }
+      if (choice.startsWith("Double escape")) {
+        const value = await ui.select("Double escape action", ["fork", "tree", "none"]);
+        if (!value) return;
+        s.setDoubleEscapeAction(value as any);
+        ui.notify(`Double escape: ${value}`, "info");
+        return;
+      }
+      if (choice.startsWith("Tree filter")) {
+        const value = await ui.select("Tree filter", ["default", "no-tools", "user-only", "labeled-only", "all"]);
+        if (!value) return;
+        s.setTreeFilterMode(value as any);
+        ui.notify(`Tree filter: ${value}`, "info");
+        return;
+      }
+      if (choice.startsWith("Shell prefix")) {
+        const value = await ui.input("Shell command prefix", s.getShellCommandPrefix() ?? "!");
+        s.setShellCommandPrefix(value?.trim() || undefined);
+        ui.notify(`Shell prefix: ${value?.trim() || "unset"}`, "info");
+      }
+    },
+  });
+}

package/lib/commands/telegram-commands.ts

@@ -0,0 +1,162 @@
+import { resolve } from "node:path";
+import { bindWorkspaceTelegramConfig, readTelegramConfigStore, unbindWorkspaceTelegramConfig } from "../config.ts";
+import { escapeHtml } from "../html.ts";
+import { getTelegramBotUsername } from "../telegram-api.ts";
+import type { ResolvedTelegramConfig, TelegramConfig, TelegramTransport } from "../types.ts";
+import type { TelegramPollingRuntime } from "../polling.ts";
+
+export type TelegramCommandDeps = {
+  getConfig: () => TelegramConfig;
+  setConfig: (c: TelegramConfig) => void;
+  persistConfig: (c: TelegramConfig) => Promise<void>;
+  getResolvedConfig: () => ResolvedTelegramConfig | undefined;
+  switchResolvedConfig: (next: ResolvedTelegramConfig) => void;
+  isTelegramEnabled: () => boolean;
+  transport: TelegramTransport;
+  getPolling: () => TelegramPollingRuntime;
+  refreshStatus: () => void;
+  syncTelegramCommands: () => Promise<void>;
+  startStatusHeartbeat: () => void;
+  clearStatusError: () => void;
+};
+
+/** Shared "connect and start" sequence used by both /tg-setup and /tg-connect. */
+async function connectAndStart(
+  deps: TelegramCommandDeps,
+  token: string,
+  botUsername: string | undefined,
+): Promise<void> {
+  const config = deps.getConfig();
+  deps.setConfig({ ...config, botToken: token, botUsername, telegramEnabled: true });
+  await deps.persistConfig(deps.getConfig());
+  deps.getPolling().start();
+  await deps.syncTelegramCommands();
+  deps.refreshStatus();
+}
+
+export function configureTelegramToken(
+  ui: { input: (title: string, placeholder?: string) => Promise<string | undefined>; inputSecret?: (title: string, placeholder?: string) => Promise<string | undefined> },
+  deps: TelegramCommandDeps,
+): Promise<boolean> {
+  return (async () => {
+    const token = await (ui.inputSecret?.("Telegram bot token") ?? ui.input("Telegram bot token"));
+    if (!token) return false;
+    const botUsername = await getTelegramBotUsername(token).catch(() => undefined);
+    if (!deps.getResolvedConfig()) {
+      deps.switchResolvedConfig({ store: { version: 2, global: {}, workspaces: [] }, scope: "global", config: {} });
+    }
+    await connectAndStart(deps, token, botUsername);
+    return true;
+  })();
+}
+
+export function registerTelegramCommands(
+  registry: { registerCommand: (name: string, options: { description?: string; handler: (args: string, ctx: any) => Promise<void> }) => void },
+  deps: TelegramCommandDeps,
+): void {
+  // ── /tg-setup ─────────────────────────────────────────────────────────
+  registry.registerCommand("tg-setup", {
+    description: "Configure Telegram bot token",
+    handler: async (_args, ctx) => {
+      const ui = ctx.ui as typeof ctx.ui & { inputSecret?: (title: string, placeholder?: string) => Promise<string | undefined> };
+      if (!(await configureTelegramToken(ui, deps))) return;
+      deps.clearStatusError();
+      deps.startStatusHeartbeat();
+      ui.notify("Telegram bot token saved and connected.", "info");
+    },
+  });
+
+  // ── /tg-connect ───────────────────────────────────────────────────────
+  registry.registerCommand("tg-connect", {
+    description: "Enable/start Telegram connection",
+    handler: async (_args, ctx) => {
+      const ui = ctx.ui as typeof ctx.ui & { inputSecret?: (title: string, placeholder?: string) => Promise<string | undefined> };
+      try {
+        if (!deps.getConfig().botToken) {
+          if (!(await configureTelegramToken(ui, deps))) return;
+        } else {
+          const botUsername = deps.getConfig().botUsername ?? await getTelegramBotUsername(deps.getConfig().botToken!).catch(() => undefined);
+          await connectAndStart(deps, deps.getConfig().botToken!, botUsername);
+        }
+      } catch { /* connection errors reported via polling onError */ }
+      deps.clearStatusError();
+      deps.startStatusHeartbeat();
+      ui.notify("Telegram connected.", "info");
+    },
+  });
+
+  // ── /tg-disconnect ────────────────────────────────────────────────────
+  registry.registerCommand("tg-disconnect", {
+    description: "Disable/stop Telegram connection without deleting the token",
+    handler: async (_args, ctx) => {
+      await deps.getPolling().stop();
+      deps.setConfig({ ...deps.getConfig(), telegramEnabled: false });
+      await deps.persistConfig(deps.getConfig());
+      deps.clearStatusError();
+      deps.refreshStatus();
+      ctx.ui.notify("Telegram disconnected. Token is kept; use /tg-connect to reconnect.", "info");
+    },
+  });
+
+  // ── /tg-bind-cwd ──────────────────────────────────────────────────────
+  registry.registerCommand("tg-bind-cwd", {
+    description: "Bind current directory to a Telegram bot",
+    handler: async (args, ctx) => {
+      const ui = ctx.ui as typeof ctx.ui & { inputSecret?: (title: string, placeholder?: string) => Promise<string | undefined> };
+      const config = deps.getConfig();
+      const workspacePath = resolve(args.trim() || ctx.cwd || process.cwd());
+      const token = await (ui.inputSecret?.(`Telegram bot token for ${workspacePath}`) ?? ui.input(`Telegram bot token for ${workspacePath}`));
+      if (!token) return;
+      const botUsername = await getTelegramBotUsername(token).catch(() => undefined);
+      await deps.getPolling().stop();
+      deps.switchResolvedConfig(await bindWorkspaceTelegramConfig(workspacePath, {
+        botToken: token,
+        botUsername,
+        telegramEnabled: true,
+        tool: config.tool,
+        thinking: config.thinking,
+        messageMode: config.messageMode,
+      }));
+      deps.getPolling().start();
+      await deps.syncTelegramCommands();
+      deps.startStatusHeartbeat();
+      deps.refreshStatus();
+      ui.notify(`Telegram workspace bot bound:\n${escapeHtml(workspacePath)}\n${botUsername ? `@${botUsername}` : "bot username unknown"}`, "info");
+    },
+  });
+
+  // ── /tg-unbind-cwd ────────────────────────────────────────────────────
+  registry.registerCommand("tg-unbind-cwd", {
+    description: "Remove current directory Telegram bot binding",
+    handler: async (_args, ctx) => {
+      const previous = deps.getResolvedConfig();
+      if (previous?.scope !== "workspace") {
+        ctx.ui.notify("Current directory is using the global Telegram bot; no workspace binding to remove.", "info");
+        return;
+      }
+      await deps.getPolling().stop();
+      deps.switchResolvedConfig(await unbindWorkspaceTelegramConfig(ctx.cwd || process.cwd()));
+      if (deps.isTelegramEnabled()) deps.getPolling().start();
+      await deps.syncTelegramCommands();
+      deps.refreshStatus();
+      ctx.ui.notify(`Removed Telegram workspace binding:\n${escapeHtml(previous.workspacePath ?? "")}`, "info");
+    },
+  });
+
+  // ── /tg-list ───────────────────────────────────────────────────────────
+  registry.registerCommand("tg-list", {
+    description: "List Telegram bot bindings",
+    handler: async (_args, ctx) => {
+      const store = await readTelegramConfigStore();
+      const lines = [
+        `global: ${store.global?.botUsername ? `@${store.global.botUsername}` : store.global?.botToken ? "configured" : "not configured"}`,
+        "",
+        "workspaces:",
+        ...((store.workspaces ?? []).length
+          ? (store.workspaces ?? []).map((workspace) => `- ${escapeHtml(workspace.path)}\n  ${workspace.config.botUsername ? `@${workspace.config.botUsername}` : workspace.config.botToken ? "configured" : "not configured"}`)
+          : ["none"]),
+      ];
+      ctx.ui.notify(lines.join("\n"), "info");
+    },
+  });
+}