xtrm-tools 2.1.10 → 2.1.11

package/cli/package.json

@@ -1,6 +1,6 @@
 {
   "name": "xtrm-cli",
-  "version": "2.1.10",
+  "version": "2.1.11",
   "description": "Claude Code tools installer (skills, hooks, MCP servers)",
   "main": "./dist/index.js",
   "type": "module",

package/config/pi/auth.json.template

@@ -0,0 +1,14 @@
+{
+  "zai": {
+    "type": "api_key",
+    "key": "{{ZAI_API_KEY}}"
+  },
+  "dashscope": {
+    "type": "api_key",
+    "key": "{{DASHSCOPE_API_KEY}}"
+  },
+  "anthropic": {},
+  "google-gemini-cli": {},
+  "qwen-cli": {},
+  "google-antigravity": {}
+}

package/config/pi/extensions/auto-session-name.ts

@@ -0,0 +1,29 @@
+/**
+ * oh-pi Auto Session Name Extension
+ *
+ * Automatically names sessions based on the first user message.
+ */
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+
+export default function (pi: ExtensionAPI) {
+  let named = false;
+
+  pi.on("session_start", async (_event, ctx) => {
+    named = !!pi.getSessionName();
+  });
+
+  pi.on("agent_end", async (event) => {
+    if (named) return;
+    const userMsg = event.messages.find((m) => m.role === "user");
+    if (!userMsg) return;
+    const text = typeof userMsg.content === "string"
+      ? userMsg.content
+      : userMsg.content.filter((b) => b.type === "text").map((b) => (b as { text: string }).text).join(" ");
+    if (!text) return;
+    const name = text.slice(0, 60).replace(/\n/g, " ").trim();
+    if (name) {
+      pi.setSessionName(name);
+      named = true;
+    }
+  });
+}

package/config/pi/extensions/auto-update.ts

@@ -0,0 +1,71 @@
+/**
+ * oh-pi Auto Update — check for new oh-pi version on session start
+ */
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { execSync } from "node:child_process";
+import { readFileSync, writeFileSync, existsSync } from "node:fs";
+import { join } from "node:path";
+import { homedir } from "node:os";
+
+const CHECK_INTERVAL = 24 * 60 * 60 * 1000; // 24h
+const STAMP_FILE = join(homedir(), ".pi", "agent", ".update-check");
+
+function readStamp(): number {
+  try { return Number(readFileSync(STAMP_FILE, "utf8").trim()) || 0; } catch { return 0; }
+}
+
+function writeStamp() {
+  try { writeFileSync(STAMP_FILE, String(Date.now())); } catch {}
+}
+
+function getLatestVersion(): string | null {
+  try {
+    return execSync("npm view oh-pi version", { encoding: "utf8", timeout: 8000 }).trim();
+  } catch { return null; }
+}
+
+function getCurrentVersion(): string | null {
+  // Read from the installed package.json
+  try {
+    const pkgPath = join(__dirname, "..", "..", "package.json");
+    if (existsSync(pkgPath)) {
+      return JSON.parse(readFileSync(pkgPath, "utf8")).version;
+    }
+  } catch {}
+  // Fallback: npm list
+  try {
+    const out = JSON.parse(execSync("npm list -g oh-pi --json --depth=0", { encoding: "utf8", timeout: 8000 }));
+    return out.dependencies?.["oh-pi"]?.version ?? null;
+  } catch { return null; }
+}
+
+export function isNewer(latest: string, current: string): boolean {
+  const a = latest.split(".").map(Number);
+  const b = current.split(".").map(Number);
+  for (let i = 0; i < 3; i++) {
+    if ((a[i] ?? 0) > (b[i] ?? 0)) return true;
+    if ((a[i] ?? 0) < (b[i] ?? 0)) return false;
+  }
+  return false;
+}
+
+export default function (pi: ExtensionAPI) {
+  pi.on("session_start", async (_event, ctx) => {
+    // Non-blocking: run check in background
+    setTimeout(async () => {
+      try {
+        if (Date.now() - readStamp() < CHECK_INTERVAL) return;
+        writeStamp();
+
+        const current = getCurrentVersion();
+        const latest = getLatestVersion();
+        if (!current || !latest || !isNewer(latest, current)) return;
+
+        const msg = `oh-pi ${latest} available (current: ${current}). Run: npx oh-pi@latest`;
+        if (ctx.hasUI) {
+          ctx.ui.toast?.(msg) ?? console.log(`\n💡 ${msg}\n`);
+        }
+      } catch {}
+    }, 2000);
+  });
+}

package/config/pi/extensions/bg-process.ts

@@ -0,0 +1,230 @@
+/**
+ * oh-pi Background Process Extension
+ *
+ * 任何 bash 命令超时未完成时，自动送到后台执行。
+ * 进程完成后自动通过 sendMessage 通知 LLM，无需轮询。
+ * 提供 bg_status 工具让 LLM 查看/停止后台进程。
+ */
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { Type } from "@sinclair/typebox";
+import { StringEnum } from "@mariozechner/pi-ai";
+import { spawn, execSync } from "node:child_process";
+import { writeFileSync, readFileSync, appendFileSync, existsSync } from "node:fs";
+
+/** 超时阈值（毫秒），超过此时间自动后台化 */
+const BG_TIMEOUT_MS = 10_000;
+
+interface BgProcess {
+  pid: number;
+  command: string;
+  logFile: string;
+  startedAt: number;
+  finished: boolean;
+  exitCode: number | null;
+}
+
+export default function (pi: ExtensionAPI) {
+  const bgProcesses = new Map<number, BgProcess>();
+
+  // 覆盖内置 bash 工具
+  pi.registerTool({
+    name: "bash",
+    label: "Bash",
+    description: `Execute a bash command. Output is truncated to 2000 lines or 50KB. If a command runs longer than ${BG_TIMEOUT_MS / 1000}s, it is automatically backgrounded and you get the PID + log file path. Use the bg_status tool to check on backgrounded processes.`,
+    parameters: Type.Object({
+      command: Type.String({ description: "Bash command to execute" }),
+      timeout: Type.Optional(Type.Number({ description: "Timeout in seconds (optional)" })),
+    }),
+    async execute(toolCallId, params, signal) {
+      const { command } = params;
+      const userTimeout = params.timeout ? params.timeout * 1000 : undefined;
+      const effectiveTimeout = userTimeout ?? BG_TIMEOUT_MS;
+
+      return new Promise((resolve) => {
+        let stdout = "";
+        let stderr = "";
+        let settled = false;
+        let backgrounded = false;
+
+        const child = spawn("bash", ["-c", command], {
+          cwd: process.cwd(),
+          env: { ...process.env },
+          stdio: ["ignore", "pipe", "pipe"],
+        });
+
+        child.stdout?.on("data", (d: Buffer) => {
+          const chunk = d.toString();
+          stdout += chunk;
+          // 后台化后追加写入日志
+          if (backgrounded) {
+            try { appendFileSync(bgProcesses.get(child.pid!)?.logFile ?? "", chunk); } catch {}
+          }
+        });
+        child.stderr?.on("data", (d: Buffer) => {
+          const chunk = d.toString();
+          stderr += chunk;
+          if (backgrounded) {
+            try { appendFileSync(bgProcesses.get(child.pid!)?.logFile ?? "", chunk); } catch {}
+          }
+        });
+
+        // 超时处理：保持管道，标记为后台
+        const timer = setTimeout(() => {
+          if (settled) return;
+          settled = true;
+          backgrounded = true;
+
+          child.unref();
+
+          const logFile = `/tmp/oh-pi-bg-${Date.now()}.log`;
+          const pid = child.pid!;
+
+          // 把已有输出写入日志
+          writeFileSync(logFile, stdout + stderr);
+
+          const proc: BgProcess = { pid, command, logFile, startedAt: Date.now(), finished: false, exitCode: null };
+          bgProcesses.set(pid, proc);
+
+          // 监听完成事件，自动通知 LLM
+          child.on("close", (code) => {
+            proc.finished = true;
+            proc.exitCode = code;
+            const tail = (stdout + stderr).slice(-3000);
+            const truncated = (stdout + stderr).length > 3000 ? "[...truncated]\n" + tail : tail;
+            // 最终输出写入日志
+            try { writeFileSync(logFile, stdout + stderr); } catch {}
+
+            pi.sendMessage({
+              content: `[BG_PROCESS_DONE] PID ${pid} finished (exit ${code ?? "?"})\nCommand: ${command}\n\nOutput (last 3000 chars):\n${truncated}`,
+              display: true,
+              triggerTurn: true,
+              deliverAs: "followUp",
+            });
+          });
+
+          const preview = (stdout + stderr).slice(0, 500);
+          const text = `Command still running after ${effectiveTimeout / 1000}s, moved to background.\nPID: ${pid}\nLog: ${logFile}\nStop: kill ${pid}\n\nOutput so far:\n${preview}\n\n⏳ You will be notified automatically when it finishes. No need to poll.`;
+
+          resolve({
+            content: [{ type: "text", text }],
+            details: {},
+          });
+        }, effectiveTimeout);
+
+        // 正常结束（超时前）
+        child.on("close", (code) => {
+          if (settled) return;
+          settled = true;
+          clearTimeout(timer);
+
+          const output = (stdout + stderr).trim();
+          const exitInfo = code !== 0 ? `\n[Exit code: ${code}]` : "";
+
+          resolve({
+            content: [{ type: "text", text: output + exitInfo }],
+            details: {},
+          });
+        });
+
+        child.on("error", (err) => {
+          if (settled) return;
+          settled = true;
+          clearTimeout(timer);
+
+          resolve({
+            content: [{ type: "text", text: `Error: ${err.message}` }],
+            details: {},
+            isError: true,
+          });
+        });
+
+        // 处理 abort signal
+        if (signal) {
+          signal.addEventListener("abort", () => {
+            if (settled) return;
+            settled = true;
+            clearTimeout(timer);
+            try { child.kill(); } catch {}
+            resolve({
+              content: [{ type: "text", text: "Command cancelled." }],
+              details: {},
+            });
+          }, { once: true });
+        }
+      });
+    },
+  });
+
+  // bg_status 工具：查看/管理后台进程
+  pi.registerTool({
+    name: "bg_status",
+    label: "Background Process Status",
+    description: "Check status, view output, or stop background processes that were auto-backgrounded.",
+    parameters: Type.Object({
+      action: StringEnum(["list", "log", "stop"] as const, { description: "list=show all, log=view output, stop=kill process" }),
+      pid: Type.Optional(Type.Number({ description: "PID of the process (required for log/stop)" })),
+    }),
+    async execute(toolCallId, params) {
+      const { action, pid } = params;
+
+      if (action === "list") {
+        if (bgProcesses.size === 0) {
+          return { content: [{ type: "text", text: "No background processes." }], details: {} };
+        }
+        const lines = [...bgProcesses.values()].map((p) => {
+          const status = p.finished ? `⚪ stopped (exit ${p.exitCode ?? "?"})` : (isAlive(p.pid) ? "🟢 running" : "⚪ stopped");
+          return `PID: ${p.pid} | ${status} | Log: ${p.logFile}\n  Cmd: ${p.command}`;
+        });
+        return { content: [{ type: "text", text: lines.join("\n\n") }], details: {} };
+      }
+
+      if (!pid) {
+        return { content: [{ type: "text", text: "Error: pid is required for log/stop" }], details: {}, isError: true };
+      }
+
+      const proc = bgProcesses.get(pid);
+
+      if (action === "log") {
+        const logFile = proc?.logFile;
+        if (logFile && existsSync(logFile)) {
+          try {
+            const content = readFileSync(logFile, "utf-8");
+            const tail = content.slice(-5000);
+            const truncated = content.length > 5000 ? `[...truncated, showing last 5000 chars]\n${tail}` : tail;
+            return { content: [{ type: "text", text: truncated || "(empty)" }], details: {} };
+          } catch (e: any) {
+            return { content: [{ type: "text", text: `Error reading log: ${e.message}` }], details: {}, isError: true };
+          }
+        }
+        return { content: [{ type: "text", text: "No log available for this PID." }], details: {} };
+      }
+
+      if (action === "stop") {
+        try {
+          process.kill(pid, "SIGTERM");
+          bgProcesses.delete(pid);
+          return { content: [{ type: "text", text: `Process ${pid} terminated.` }], details: {} };
+        } catch {
+          bgProcesses.delete(pid);
+          return { content: [{ type: "text", text: `Process ${pid} not found (already stopped?).` }], details: {} };
+        }
+      }
+
+      return { content: [{ type: "text", text: `Unknown action: ${action}` }], details: {}, isError: true };
+    },
+  });
+
+  // 清理：退出时杀掉所有后台进程
+  pi.on("session_shutdown", async () => {
+    for (const [pid, proc] of bgProcesses) {
+      if (!proc.finished) {
+        try { process.kill(pid, "SIGTERM"); } catch {}
+      }
+    }
+    bgProcesses.clear();
+  });
+}
+
+function isAlive(pid: number): boolean {
+  try { process.kill(pid, 0); return true; } catch { return false; }
+}

package/config/pi/extensions/compact-header.ts

@@ -0,0 +1,69 @@
+/**
+ * oh-pi Compact Header — table-style startup info with dynamic column widths
+ */
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { VERSION } from "@mariozechner/pi-coding-agent";
+import { truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
+
+export default function (pi: ExtensionAPI) {
+  pi.on("session_start", async (_event, ctx) => {
+    if (!ctx.hasUI) return;
+
+    ctx.ui.setHeader((_tui, theme) => ({
+      render(width: number): string[] {
+        const d = (s: string) => theme.fg("dim", s);
+        const a = (s: string) => theme.fg("accent", s);
+
+        const cmds = pi.getCommands();
+        const prompts = cmds.filter(c => c.source === "prompt").map(c => `/${c.name}`).join("  ");
+        const skills = cmds.filter(c => c.source === "skill").map(c => c.name).join("  ");
+        const model = ctx.model ? `${ctx.model.id}` : "no model";
+        const thinking = pi.getThinkingLevel();
+        const provider = ctx.model?.provider ?? "";
+
+        const pad = (s: string, w: number) => s + " ".repeat(Math.max(0, w - visibleWidth(s)));
+        const t = (s: string) => truncateToWidth(s, width);
+        const sep = d(" │ ");
+
+        // Right two columns are fixed width
+        const rCol = [
+          [d("esc"), a("interrupt"),   d("S-tab"), a("thinking")],
+          [d("^C"),  a("clear/exit"),  d("^O"),    a("expand")],
+          [d("^P"),  a("model"),       d("^G"),    a("editor")],
+          [d("/"),   a("commands"),    d("^V"),    a("paste")],
+          [d("!"),   a("bash"),        d(""),      a("")],
+        ];
+        const k1w = 6, v1w = 13, k2w = 6, v2w = 9;
+        const rightW = k1w + v1w + 3 + k2w + v2w + 3; // 3 for each sep
+
+        // Left column gets remaining space
+        const leftW = Math.max(20, width - rightW);
+        const lk = 9; // label width
+
+        const lCol = [
+          [d("version"), a(`v${VERSION}  ${provider}`)],
+          [d("model"),   a(model)],
+          [d("think"),   a(thinking)],
+          [d(""),        d("")],
+          [d(""),        d("")],
+        ];
+
+        const lines: string[] = [""];
+        for (let i = 0; i < 5; i++) {
+          const [lk0, lv0] = lCol[i];
+          const [rk0, rv0, rk1, rv1] = rCol[i];
+          const left = truncateToWidth(pad(lk0, lk) + lv0, leftW);
+          const right = pad(rk0, k1w) + pad(rv0, v1w) + sep + pad(rk1, k2w) + rv1;
+          lines.push(t(pad(left, leftW) + sep + right));
+        }
+
+        if (prompts) lines.push(t(`${pad(d("prompts"), lk)}${a(prompts)}`));
+        if (skills) lines.push(t(`${pad(d("skills"), lk)}${a(skills)}`));
+        lines.push(d("─".repeat(width)));
+
+        return lines;
+      },
+      invalidate() {},
+    }));
+  });
+}

package/config/pi/extensions/custom-footer.ts

@@ -0,0 +1,95 @@
+/**
+ * Custom Footer Extension — Enhanced status bar
+ *
+ * Displays: in/out/remaining tokens, cost, context%, elapsed, cwd, git branch, model
+ * Color-coded context usage: green <50%, yellow 50-75%, red >75%
+ */
+
+import type { AssistantMessage } from "@mariozechner/pi-ai";
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
+
+export default function (pi: ExtensionAPI) {
+	let sessionStart = Date.now();
+
+	function formatElapsed(ms: number): string {
+		const s = Math.floor(ms / 1000);
+		if (s < 60) return `${s}s`;
+		const m = Math.floor(s / 60);
+		const rs = s % 60;
+		if (m < 60) return `${m}m${rs > 0 ? rs + "s" : ""}`;
+		const h = Math.floor(m / 60);
+		const rm = m % 60;
+		return `${h}h${rm > 0 ? rm + "m" : ""}`;
+	}
+
+	function fmt(n: number): string {
+		if (n < 1000) return `${n}`;
+		return `${(n / 1000).toFixed(1)}k`;
+	}
+
+	pi.on("session_start", async (_event, ctx) => {
+		sessionStart = Date.now();
+
+		ctx.ui.setFooter((tui, theme, footerData) => {
+			const unsub = footerData.onBranchChange(() => tui.requestRender());
+			const timer = setInterval(() => tui.requestRender(), 30000);
+
+			return {
+				dispose() { unsub(); clearInterval(timer); },
+				invalidate() {},
+				render(width: number): string[] {
+					let input = 0, output = 0, cost = 0;
+					for (const e of ctx.sessionManager.getBranch()) {
+						if (e.type === "message" && e.message.role === "assistant") {
+							const m = e.message as AssistantMessage;
+							input += m.usage.input;
+							output += m.usage.output;
+							cost += m.usage.cost.total;
+						}
+					}
+
+					const usage = ctx.getContextUsage();
+					const ctxWindow = usage?.contextWindow ?? 0;
+					const pct = usage?.percent ?? 0;
+					const remaining = Math.max(0, ctxWindow - (usage?.tokens ?? 0));
+
+					const pctColor = pct > 75 ? "error" : pct > 50 ? "warning" : "success";
+
+					const tokenStats = [
+						theme.fg("accent", `${fmt(input)}/${fmt(output)}`),
+						theme.fg("warning", `$${cost.toFixed(2)}`),
+						theme.fg(pctColor, `${pct.toFixed(0)}%`),
+					].join(" ");
+
+					const elapsed = theme.fg("dim", `⏱${formatElapsed(Date.now() - sessionStart)}`);
+
+					const parts = process.cwd().split("/");
+					const short = parts.length > 2 ? parts.slice(-2).join("/") : process.cwd();
+					const cwdStr = theme.fg("muted", `⌂ ${short}`);
+
+					const branch = footerData.getGitBranch();
+					const branchStr = branch ? theme.fg("accent", `⎇ ${branch}`) : "";
+
+					const thinking = pi.getThinkingLevel();
+					const thinkColor = thinking === "high" ? "warning" : thinking === "medium" ? "accent" : thinking === "low" ? "dim" : "muted";
+					const modelId = ctx.model?.id || "no-model";
+					const modelStr = theme.fg(thinkColor, "◆") + " " + theme.fg("accent", modelId);
+
+					const sep = theme.fg("dim", " | ");
+					const leftParts = [modelStr, tokenStats, elapsed, cwdStr];
+					if (branchStr) leftParts.push(branchStr);
+					const left = leftParts.join(sep);
+
+					return [truncateToWidth(left, width)];
+				},
+			};
+		});
+	});
+
+	pi.on("session_switch", async (event, _ctx) => {
+		if (event.reason === "new") {
+			sessionStart = Date.now();
+		}
+	});
+}