@wahack/pi-coding-agent 15.11.0

package/src/cli/ssh-cli.ts

@@ -0,0 +1,179 @@
+/**
+ * SSH CLI command handlers.
+ *
+ * Handles `omp ssh <command>` subcommands for SSH host configuration management.
+ */
+
+import { getSSHConfigPath } from "@oh-my-pi/pi-utils";
+import chalk from "chalk";
+import { addSSHHost, readSSHConfigFile, removeSSHHost, type SSHHostConfig } from "../ssh/config-writer";
+
+// =============================================================================
+// Types
+// =============================================================================
+
+export type SSHAction = "add" | "remove" | "list";
+
+export interface SSHCommandArgs {
+	action: SSHAction;
+	args: string[];
+	flags: {
+		json?: boolean;
+		host?: string;
+		user?: string;
+		port?: string;
+		key?: string;
+		desc?: string;
+		compat?: boolean;
+		scope?: "project" | "user";
+	};
+}
+
+// =============================================================================
+// Main dispatcher
+// =============================================================================
+
+export async function runSSHCommand(cmd: SSHCommandArgs): Promise<void> {
+	switch (cmd.action) {
+		case "add":
+			await handleAdd(cmd);
+			break;
+		case "remove":
+			await handleRemove(cmd);
+			break;
+		case "list":
+			await handleList(cmd);
+			break;
+		default:
+			process.stdout.write(chalk.red(`Unknown action: ${cmd.action}\n`));
+			process.stdout.write(`Valid actions: add, remove, list\n`);
+			process.exitCode = 1;
+	}
+}
+
+// =============================================================================
+// Handlers
+// =============================================================================
+
+async function handleAdd(cmd: SSHCommandArgs): Promise<void> {
+	const name = cmd.args[0];
+	if (!name) {
+		process.stdout.write(chalk.red("Error: Host name required\n"));
+		process.stdout.write(
+			chalk.dim("Usage: omp ssh add <name> --host <address> [--user <user>] [--port <port>] [--key <path>]\n"),
+		);
+		process.exitCode = 1;
+		return;
+	}
+
+	const host = cmd.flags.host;
+	if (!host) {
+		process.stdout.write(chalk.red("Error: --host is required\n"));
+		process.stdout.write(chalk.dim("Usage: omp ssh add <name> --host <address>\n"));
+		process.exitCode = 1;
+		return;
+	}
+
+	// Validate port if provided
+	if (cmd.flags.port !== undefined) {
+		const port = Number.parseInt(cmd.flags.port, 10);
+		if (Number.isNaN(port) || port < 1 || port > 65535) {
+			process.stdout.write(chalk.red("Error: Port must be an integer between 1 and 65535\n"));
+			process.exitCode = 1;
+			return;
+		}
+	}
+
+	const hostConfig: SSHHostConfig = { host };
+	if (cmd.flags.user) hostConfig.username = cmd.flags.user;
+	if (cmd.flags.port) hostConfig.port = Number.parseInt(cmd.flags.port, 10);
+	if (cmd.flags.key) hostConfig.keyPath = cmd.flags.key;
+	if (cmd.flags.desc) hostConfig.description = cmd.flags.desc;
+	if (cmd.flags.compat) hostConfig.compat = true;
+
+	const scope = cmd.flags.scope ?? "project";
+	const filePath = getSSHConfigPath(scope);
+
+	try {
+		await addSSHHost(filePath, name, hostConfig);
+		process.stdout.write(chalk.green(`Added SSH host "${name}" to ${scope} config\n`));
+	} catch (err) {
+		process.stdout.write(chalk.red(`Error: ${err instanceof Error ? err.message : String(err)}\n`));
+		process.exitCode = 1;
+	}
+}
+
+async function handleRemove(cmd: SSHCommandArgs): Promise<void> {
+	const name = cmd.args[0];
+	if (!name) {
+		process.stdout.write(chalk.red("Error: Host name required\n"));
+		process.stdout.write(chalk.dim("Usage: omp ssh remove <name> [--scope project|user]\n"));
+		process.exitCode = 1;
+		return;
+	}
+
+	const scope = cmd.flags.scope ?? "project";
+	const filePath = getSSHConfigPath(scope);
+
+	try {
+		await removeSSHHost(filePath, name);
+		process.stdout.write(chalk.green(`Removed SSH host "${name}" from ${scope} config\n`));
+	} catch (err) {
+		process.stdout.write(chalk.red(`Error: ${err instanceof Error ? err.message : String(err)}\n`));
+		process.exitCode = 1;
+	}
+}
+
+async function handleList(cmd: SSHCommandArgs): Promise<void> {
+	const projectPath = getSSHConfigPath("project");
+	const userPath = getSSHConfigPath("user");
+
+	const [projectConfig, userConfig] = await Promise.all([readSSHConfigFile(projectPath), readSSHConfigFile(userPath)]);
+
+	const projectHosts = projectConfig.hosts ?? {};
+	const userHosts = userConfig.hosts ?? {};
+
+	if (cmd.flags.json) {
+		process.stdout.write(JSON.stringify({ project: projectHosts, user: userHosts }, null, 2));
+		process.stdout.write("\n");
+		return;
+	}
+
+	const hasProject = Object.keys(projectHosts).length > 0;
+	const hasUser = Object.keys(userHosts).length > 0;
+
+	if (!hasProject && !hasUser) {
+		process.stdout.write(chalk.dim("No SSH hosts configured\n"));
+		process.stdout.write(chalk.dim("Add one with: omp ssh add <name> --host <address>\n"));
+		return;
+	}
+
+	if (hasProject) {
+		process.stdout.write(chalk.bold("Project SSH Hosts (.omp/ssh.json):\n"));
+		printHosts(projectHosts);
+	}
+
+	if (hasProject && hasUser) {
+		process.stdout.write("\n");
+	}
+
+	if (hasUser) {
+		process.stdout.write(chalk.bold("User SSH Hosts (~/.omp/agent/ssh.json):\n"));
+		printHosts(userHosts);
+	}
+}
+
+// =============================================================================
+// Helpers
+// =============================================================================
+
+function printHosts(hosts: Record<string, SSHHostConfig>): void {
+	for (const [name, config] of Object.entries(hosts)) {
+		const parts = [chalk.cyan(name), config.host];
+		if (config.username) parts.push(chalk.dim(config.username));
+		if (config.port && config.port !== 22) parts.push(chalk.dim(`port:${config.port}`));
+		if (config.keyPath) parts.push(chalk.dim(config.keyPath));
+		if (config.description) parts.push(chalk.dim(`- ${config.description}`));
+		process.stdout.write(`  ${parts.join("  ")}\n`);
+	}
+}

package/src/cli/startup-cwd.ts

@@ -0,0 +1,68 @@
+import * as fs from "node:fs/promises";
+import * as os from "node:os";
+import * as path from "node:path";
+import { getProjectDir, normalizePathForComparison, setProjectDir } from "@oh-my-pi/pi-utils";
+import type { Args } from "./args";
+
+async function maybeAutoChdir(parsed: Args): Promise<void> {
+	if (parsed.allowHome || parsed.cwd) {
+		return;
+	}
+
+	const home = os.homedir();
+	if (!home) {
+		return;
+	}
+
+	const normalizePath = normalizePathForComparison;
+
+	const cwd = normalizePath(getProjectDir());
+	const normalizedHome = normalizePath(home);
+	if (cwd !== normalizedHome) {
+		return;
+	}
+
+	const isDirectory = async (p: string) => {
+		try {
+			const s = await fs.stat(p);
+			return s.isDirectory();
+		} catch {
+			return false;
+		}
+	};
+
+	const candidates = [path.join(home, "tmp"), "/tmp", "/var/tmp"];
+	for (const candidate of candidates) {
+		try {
+			if (!(await isDirectory(candidate))) {
+				continue;
+			}
+			setProjectDir(candidate);
+			return;
+		} catch {
+			// Try next candidate.
+		}
+	}
+
+	try {
+		const fallback = os.tmpdir();
+		if (fallback && normalizePath(fallback) !== cwd && (await isDirectory(fallback))) {
+			setProjectDir(fallback);
+		}
+	} catch {
+		// Ignore fallback errors.
+	}
+}
+
+export async function applyStartupCwd(parsed: Args): Promise<void> {
+	if (parsed.cwd) {
+		setProjectDir(parsed.cwd);
+		// setProjectDir resolves the (possibly relative) target against the launch
+		// cwd and chdirs into it. Re-sync parsed.cwd to the resolved absolute path
+		// so downstream consumers (buildSessionOptions, settings/discovery, session
+		// persistence) don't re-resolve a relative string against the new cwd.
+		parsed.cwd = getProjectDir();
+		return;
+	}
+	await maybeAutoChdir(parsed);
+}

package/src/cli/stats-cli.ts

@@ -0,0 +1,238 @@
+/**
+ * Stats CLI command handlers.
+ *
+ * Handles `omp stats` subcommand for viewing AI usage statistics.
+ */
+
+import { APP_NAME, formatDuration, formatNumber, formatPercent } from "@oh-my-pi/pi-utils";
+import chalk from "chalk";
+import { openPath } from "../utils/open";
+
+/**
+ * Single-line TTY progress bar. On a non-TTY stream we just stay quiet -
+ * the final "Synced ..." summary still prints either way.
+ */
+function createSyncProgressReporter(): {
+	onProgress: (event: { current: number; total: number; sessionFile: string }) => void;
+	finish: () => void;
+} {
+	const stream = process.stderr;
+	const isTty = stream.isTTY === true;
+	let lastWidth = 0;
+	let lastRender = 0;
+	return {
+		onProgress(event) {
+			if (!isTty) return;
+			const now = Date.now();
+			// Throttle to ~30 fps and always force a render for the last file.
+			if (event.current < event.total && now - lastRender < 33) return;
+			lastRender = now;
+			const label = chalk.dim(shortenSessionFile(event.sessionFile));
+			const pct = ((event.current / event.total) * 100).toFixed(0).padStart(3, " ");
+			const counter = chalk.cyan(`[${event.current}/${event.total}]`);
+			const line = `${counter} ${pct}%  ${label}`;
+			const columns = stream.columns ?? 120;
+			const trimmed = truncateToColumns(line, columns - 1);
+			stream.write(`\r${trimmed.padEnd(lastWidth)}`);
+			lastWidth = trimmed.length;
+		},
+		finish() {
+			if (!isTty || lastWidth === 0) return;
+			stream.write(`\r${" ".repeat(lastWidth)}\r`);
+			lastWidth = 0;
+		},
+	};
+}
+
+function shortenSessionFile(p: string): string {
+	const marker = "/sessions/";
+	const idx = p.indexOf(marker);
+	return idx >= 0 ? p.slice(idx + marker.length) : p;
+}
+
+function truncateToColumns(s: string, max: number): string {
+	if (max <= 0) return "";
+	const width = Bun.stringWidth(s, { countAnsiEscapeCodes: false });
+	if (width <= max) return s;
+	// Cheap right-trim with an ellipsis - we don't need ANSI-aware slicing
+	// because the colored prefix is short and the truncated tail is the
+	// dim filename, where dropping bytes is fine.
+	return `${s.slice(0, Math.max(0, max - 1))}\u2026`;
+}
+
+// =============================================================================
+// Types
+// =============================================================================
+
+export interface StatsCommandArgs {
+	port: number;
+	json: boolean;
+	summary: boolean;
+}
+
+// =============================================================================
+// Argument Parser
+// =============================================================================
+
+/**
+ * Parse stats subcommand arguments.
+ * Returns undefined if not a stats command.
+ */
+export function parseStatsArgs(args: string[]): StatsCommandArgs | undefined {
+	if (args.length === 0 || args[0] !== "stats") {
+		return undefined;
+	}
+
+	const result: StatsCommandArgs = {
+		port: 3847,
+		json: false,
+		summary: false,
+	};
+
+	for (let i = 1; i < args.length; i++) {
+		const arg = args[i];
+		if (arg === "--json" || arg === "-j") {
+			result.json = true;
+		} else if (arg === "--summary" || arg === "-s") {
+			result.summary = true;
+		} else if ((arg === "--port" || arg === "-p") && i + 1 < args.length) {
+			result.port = parseInt(args[++i], 10);
+		} else if (arg.startsWith("--port=")) {
+			result.port = parseInt(arg.split("=")[1], 10);
+		}
+	}
+
+	return result;
+}
+
+function formatCost(n: number): string {
+	if (n < 0.01) return `$${n.toFixed(4)}`;
+	if (n < 1) return `$${n.toFixed(3)}`;
+	return `$${n.toFixed(2)}`;
+}
+
+function normalizePremiumRequests(n: number): number {
+	return Math.round((n + Number.EPSILON) * 100) / 100;
+}
+
+// =============================================================================
+// Command Handler
+// =============================================================================
+
+export async function runStatsCommand(cmd: StatsCommandArgs): Promise<void> {
+	// Lazy import to avoid loading stats module when not needed
+	const { getDashboardStats, syncAllSessions, getTotalMessageCount, startServer, closeDb } = await import(
+		"@oh-my-pi/omp-stats"
+	);
+
+	// Sync session files first
+	const progress = createSyncProgressReporter();
+	process.stderr.write("Syncing session files...\n");
+	const { processed, files } = await syncAllSessions({ onProgress: progress.onProgress });
+	progress.finish();
+	const total = await getTotalMessageCount();
+	console.log(`Synced ${processed} new entries from ${files} files (${total} total)\n`);
+
+	if (cmd.json) {
+		const stats = await getDashboardStats();
+		console.log(JSON.stringify(stats, null, 2));
+		return;
+	}
+
+	if (cmd.summary) {
+		await printStatsSummary();
+		return;
+	}
+
+	// Start the dashboard server
+	const { port } = await startServer(cmd.port);
+	console.log(chalk.green(`Dashboard available at: http://localhost:${port}`));
+
+	// Open browser
+	const url = `http://localhost:${port}`;
+	openPath(url);
+
+	console.log("Press Ctrl+C to stop\n");
+
+	// Keep process running
+	process.on("SIGINT", () => {
+		console.log("\nShutting down...");
+		closeDb();
+		process.exit(0);
+	});
+
+	// Keep the process alive
+	await new Promise(() => {});
+}
+
+async function printStatsSummary(): Promise<void> {
+	const { getDashboardStats } = await import("@oh-my-pi/omp-stats");
+	const stats = await getDashboardStats();
+	const { overall, byModel, byFolder } = stats;
+
+	console.log(chalk.bold("\n=== AI Usage Statistics ===\n"));
+
+	console.log(chalk.bold("Overall:"));
+	console.log(`  Requests: ${formatNumber(overall.totalRequests)} (${formatNumber(overall.failedRequests)} errors)`);
+	console.log(`  Error Rate: ${formatPercent(overall.errorRate)}`);
+	console.log(`  Total Tokens: ${formatNumber(overall.totalInputTokens + overall.totalOutputTokens)}`);
+	console.log(`  Input Tokens: ${formatNumber(overall.totalInputTokens)}`);
+	console.log(`  Output Tokens: ${formatNumber(overall.totalOutputTokens)}`);
+	console.log(`  Cache Rate: ${formatPercent(overall.cacheRate)}`);
+	console.log(`  Total Cost: ${formatCost(overall.totalCost)}`);
+	console.log(`  Premium Requests: ${formatNumber(normalizePremiumRequests(overall.totalPremiumRequests ?? 0))}`);
+	console.log(`  Avg Duration: ${overall.avgDuration !== null ? formatDuration(overall.avgDuration) : "-"}`);
+	console.log(`  Avg TTFT: ${overall.avgTtft !== null ? formatDuration(overall.avgTtft) : "-"}`);
+	if (overall.avgTokensPerSecond !== null) {
+		console.log(`  Avg Tokens/s: ${overall.avgTokensPerSecond.toFixed(1)}`);
+	}
+
+	if (byModel.length > 0) {
+		console.log(chalk.bold("\nBy Model:"));
+		for (const m of byModel.slice(0, 10)) {
+			console.log(
+				`  ${m.model}: ${formatNumber(m.totalRequests)} reqs, ${formatCost(m.totalCost)}, ${formatPercent(m.cacheRate)} cache`,
+			);
+		}
+	}
+
+	if (byFolder.length > 0) {
+		console.log(chalk.bold("\nBy Folder:"));
+		for (const f of byFolder.slice(0, 10)) {
+			console.log(`  ${f.folder}: ${formatNumber(f.totalRequests)} reqs, ${formatCost(f.totalCost)}`);
+		}
+	}
+
+	console.log("");
+}
+
+// =============================================================================
+// Help
+// =============================================================================
+
+export function printStatsHelp(): void {
+	console.log(`${chalk.bold(`${APP_NAME} stats`)} - AI Usage Statistics Dashboard
+
+${chalk.bold("Usage:")}
+  ${APP_NAME} stats [options]
+
+${chalk.bold("Options:")}
+  -p, --port <port>  Port for the dashboard server (default: 3847)
+  -j, --json         Output stats as JSON and exit
+  -s, --summary      Print summary to console and exit
+  -h, --help         Show this help message
+
+${chalk.bold("Examples:")}
+  ${APP_NAME} stats              # Start dashboard server
+  ${APP_NAME} stats --json       # Print stats as JSON
+  ${APP_NAME} stats --summary    # Print summary to console
+  ${APP_NAME} stats --port 8080  # Start on custom port
+
+${chalk.bold("Metrics:")}
+  - Total requests and error rate
+  - Token usage (input, output, cache)
+  - Cost breakdown
+  - Average duration and time to first token (TTFT)
+  - Tokens per second throughput
+`);
+}

package/src/cli/tiny-models-cli.ts

@@ -0,0 +1,127 @@
+import { formatBytes } from "@oh-my-pi/pi-utils";
+import chalk from "chalk";
+import {
+	DEFAULT_TINY_TITLE_LOCAL_MODEL_KEY,
+	getTinyLocalModelSpec,
+	isTinyLocalModelKey,
+	TINY_LOCAL_MODELS,
+	type TinyLocalModelKey,
+} from "../tiny/models";
+import { shutdownTinyTitleClient, tinyTitleClient } from "../tiny/title-client";
+import type { TinyTitleProgressEvent } from "../tiny/title-protocol";
+
+export type TinyModelsAction = "download" | "list";
+
+export interface TinyModelsCommandArgs {
+	action: TinyModelsAction;
+	model?: string;
+	flags: {
+		json?: boolean;
+	};
+}
+
+interface ProgressReporter {
+	onProgress(event: TinyTitleProgressEvent): void;
+	finish(ok: boolean): void;
+}
+
+interface DownloadResult {
+	model: TinyLocalModelKey;
+	ok: boolean;
+}
+
+function writeLine(text = ""): void {
+	process.stdout.write(`${text}\n`);
+}
+
+function resolveModels(model: string | undefined): TinyLocalModelKey[] {
+	if (!model) return [DEFAULT_TINY_TITLE_LOCAL_MODEL_KEY];
+	if (model === "all") return TINY_LOCAL_MODELS.map(spec => spec.key);
+	if (!isTinyLocalModelKey(model)) {
+		const values = TINY_LOCAL_MODELS.map(spec => spec.key).join(", ");
+		throw new Error(`Unknown tiny local model: ${model}. Expected one of: ${values}, all`);
+	}
+	return [model];
+}
+
+function listModels(json: boolean | undefined): void {
+	if (json) {
+		writeLine(JSON.stringify({ models: TINY_LOCAL_MODELS }));
+		return;
+	}
+	writeLine(chalk.bold("Tiny local models"));
+	for (const spec of TINY_LOCAL_MODELS) {
+		const defaultMark = spec.key === DEFAULT_TINY_TITLE_LOCAL_MODEL_KEY ? chalk.cyan(" default") : "";
+		writeLine(`${chalk.cyan(spec.key)}${defaultMark}`);
+		writeLine(`  ${spec.label} — ${spec.description}`);
+	}
+}
+
+function makeProgressReporter(modelKey: TinyLocalModelKey, json: boolean | undefined): ProgressReporter {
+	if (json || !process.stdout.isTTY) {
+		return { onProgress: () => undefined, finish: () => undefined };
+	}
+	const label = getTinyLocalModelSpec(modelKey)?.label ?? modelKey;
+	let lastWidth = 0;
+	let lastProgress = -1;
+	const render = (event: TinyTitleProgressEvent): void => {
+		const progress = event.progress ?? lastProgress;
+		if (progress >= 0 && progress < lastProgress + 1 && event.status !== "ready") return;
+		if (progress >= 0) lastProgress = progress;
+		const ratio = progress >= 0 ? Math.max(0, Math.min(1, progress / 100)) : 0;
+		const barWidth = 30;
+		const filled = Math.round(ratio * barWidth);
+		const bar = `${"█".repeat(filled)}${"░".repeat(barWidth - filled)}`;
+		const pct = progress >= 0 ? `${Math.floor(progress).toString().padStart(3, " ")}%` : " --%";
+		const bytes = event.loaded && event.total ? ` ${formatBytes(event.loaded)}/${formatBytes(event.total)}` : "";
+		const file = event.file ? ` ${event.file.split("/").at(-1) ?? event.file}` : "";
+		const statusLabel = event.status === "ready" ? "Ready" : "Downloading";
+		const line = `${chalk.cyan(statusLabel)} ${label} [${bar}] ${pct}${bytes}${file}`;
+		process.stdout.write(`\r${line.padEnd(lastWidth)}`);
+		lastWidth = line.length;
+	};
+	return {
+		onProgress(event) {
+			if (event.modelKey !== modelKey) return;
+			render(event);
+		},
+		finish(ok) {
+			const suffix = ok ? chalk.green("done") : chalk.red("failed");
+			process.stdout.write(`\r${`${label}: ${suffix}`.padEnd(lastWidth)}\n`);
+		},
+	};
+}
+
+async function downloadOne(modelKey: TinyLocalModelKey, json: boolean | undefined): Promise<DownloadResult> {
+	const label = getTinyLocalModelSpec(modelKey)?.label ?? modelKey;
+	if (!json && !process.stdout.isTTY) writeLine(`Downloading ${label} (${modelKey})...`);
+	const progress = makeProgressReporter(modelKey, json);
+	const ok = await tinyTitleClient.downloadModel(modelKey, { onProgress: progress.onProgress });
+	progress.finish(ok);
+	if (!json && !process.stdout.isTTY) writeLine(ok ? `Downloaded ${label}.` : `Failed to download ${label}.`);
+	return { model: modelKey, ok };
+}
+
+export async function runTinyModelsCommand(command: TinyModelsCommandArgs): Promise<void> {
+	if (command.action === "list") {
+		listModels(command.flags.json);
+		return;
+	}
+
+	const models = resolveModels(command.model);
+	const results: DownloadResult[] = [];
+	try {
+		for (const model of models) {
+			results.push(await downloadOne(model, command.flags.json));
+		}
+	} finally {
+		await shutdownTinyTitleClient();
+	}
+
+	if (command.flags.json) {
+		writeLine(JSON.stringify({ results }));
+	}
+	if (results.some(result => !result.ok)) {
+		throw new Error("One or more tiny title models failed to download");
+	}
+}