@oh-my-pi/pi-coding-agent 8.0.20 → 8.2.0

package/src/cli/jupyter-cli.ts

@@ -0,0 +1,105 @@
+/**
+ * Jupyter CLI command handlers.
+ *
+ * Handles `omp jupyter` subcommand for managing the shared Python gateway.
+ */
+import chalk from "chalk";
+import { APP_NAME } from "../config";
+import { getGatewayStatus, shutdownSharedGateway } from "../ipy/gateway-coordinator";
+
+export type JupyterAction = "kill" | "status";
+
+export interface JupyterCommandArgs {
+	action: JupyterAction;
+}
+
+export function parseJupyterArgs(args: string[]): JupyterCommandArgs | undefined {
+	if (args.length === 0 || args[0] !== "jupyter") {
+		return undefined;
+	}
+
+	const action = args[1] as JupyterAction | undefined;
+	if (!action || !["kill", "status"].includes(action)) {
+		return { action: "status" };
+	}
+
+	return { action };
+}
+
+export async function runJupyterCommand(cmd: JupyterCommandArgs): Promise<void> {
+	switch (cmd.action) {
+		case "kill":
+			await runKill();
+			break;
+		case "status":
+			await runStatus();
+			break;
+	}
+}
+
+async function runKill(): Promise<void> {
+	const status = await getGatewayStatus();
+
+	if (!status.active) {
+		console.log(chalk.dim("No Jupyter gateway is running"));
+		return;
+	}
+
+	console.log(`Killing Jupyter gateway (PID ${status.pid})...`);
+	await shutdownSharedGateway();
+	console.log(chalk.green("Jupyter gateway stopped"));
+}
+
+async function runStatus(): Promise<void> {
+	const status = await getGatewayStatus();
+
+	if (!status.active) {
+		console.log(chalk.dim("No Jupyter gateway is running"));
+		return;
+	}
+
+	console.log(chalk.bold("Jupyter Gateway Status\n"));
+	console.log(`  ${chalk.green("●")} Running`);
+	console.log(`  PID:    ${status.pid}`);
+	console.log(`  URL:    ${status.url}`);
+	if (status.uptime !== null) {
+		console.log(`  Uptime: ${formatUptime(status.uptime)}`);
+	}
+	if (status.pythonPath) {
+		console.log(`  Python: ${status.pythonPath}`);
+	}
+	if (status.venvPath) {
+		console.log(`  Venv:   ${status.venvPath}`);
+	}
+}
+
+function formatUptime(ms: number): string {
+	const seconds = Math.floor(ms / 1000);
+	const minutes = Math.floor(seconds / 60);
+	const hours = Math.floor(minutes / 60);
+
+	if (hours > 0) {
+		return `${hours}h ${minutes % 60}m`;
+	}
+	if (minutes > 0) {
+		return `${minutes}m ${seconds % 60}s`;
+	}
+	return `${seconds}s`;
+}
+
+export function printJupyterHelp(): void {
+	console.log(`${chalk.bold(`${APP_NAME} jupyter`)} - Manage the shared Jupyter gateway
+
+${chalk.bold("Usage:")}
+  ${APP_NAME} jupyter <command>
+
+${chalk.bold("Commands:")}
+  status    Show gateway status (default)
+  kill      Stop the running gateway
+
+${chalk.bold("Examples:")}
+  ${APP_NAME} jupyter           # Show status
+  ${APP_NAME} jupyter status    # Show status
+  ${APP_NAME} jupyter kill      # Stop the gateway
+`);
+}

package/src/cli/list-models.ts

@@ -1,10 +1,9 @@
 /**
  * List available models with optional fuzzy search
  */
-
 import type { Api, Model } from "@oh-my-pi/pi-ai";
-import type { ModelRegistry } from "@oh-my-pi/pi-coding-agent/config/model-registry";
-import { fuzzyFilter } from "@oh-my-pi/pi-coding-agent/utils/fuzzy";
+import type { ModelRegistry } from "../config/model-registry";
+import { fuzzyFilter } from "../utils/fuzzy";
 
 /**
  * Format a number as human-readable (e.g., 200000 -> "200K", 1000000 -> "1M")
@@ -35,7 +34,7 @@ export async function listModels(modelRegistry: ModelRegistry, searchPattern?: s
 	// Apply fuzzy filter if search pattern provided
 	let filteredModels: Model<Api>[] = models;
 	if (searchPattern) {
-		filteredModels = fuzzyFilter(models, searchPattern, (m) => `${m.provider} ${m.id}`);
+		filteredModels = fuzzyFilter(models, searchPattern, m => `${m.provider} ${m.id}`);
 	}
 
 	if (filteredModels.length === 0) {
@@ -51,7 +50,7 @@ export async function listModels(modelRegistry: ModelRegistry, searchPattern?: s
 	});
 
 	// Calculate column widths
-	const rows = filteredModels.map((m) => ({
+	const rows = filteredModels.map(m => ({
 		provider: m.provider,
 		model: m.id,
 		context: formatTokenCount(m.contextWindow),
@@ -70,12 +69,12 @@ export async function listModels(modelRegistry: ModelRegistry, searchPattern?: s
 	};
 
 	const widths = {
-		provider: Math.max(headers.provider.length, ...rows.map((r) => r.provider.length)),
-		model: Math.max(headers.model.length, ...rows.map((r) => r.model.length)),
-		context: Math.max(headers.context.length, ...rows.map((r) => r.context.length)),
-		maxOut: Math.max(headers.maxOut.length, ...rows.map((r) => r.maxOut.length)),
-		thinking: Math.max(headers.thinking.length, ...rows.map((r) => r.thinking.length)),
-		images: Math.max(headers.images.length, ...rows.map((r) => r.images.length)),
+		provider: Math.max(headers.provider.length, ...rows.map(r => r.provider.length)),
+		model: Math.max(headers.model.length, ...rows.map(r => r.model.length)),
+		context: Math.max(headers.context.length, ...rows.map(r => r.context.length)),
+		maxOut: Math.max(headers.maxOut.length, ...rows.map(r => r.maxOut.length)),
+		thinking: Math.max(headers.thinking.length, ...rows.map(r => r.thinking.length)),
+		images: Math.max(headers.images.length, ...rows.map(r => r.images.length)),
 	};
 
 	// Print header

package/src/cli/plugin-cli.ts

@@ -3,15 +3,10 @@
  *
  * Handles `omp plugin <command>` subcommands for plugin lifecycle management.
  */
-
-import { APP_NAME } from "@oh-my-pi/pi-coding-agent/config";
-import {
-	PluginManager,
-	parseSettingValue,
-	validateSetting,
-} from "@oh-my-pi/pi-coding-agent/extensibility/plugins/index";
-import { theme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
 import chalk from "chalk";
+import { APP_NAME } from "../config";
+import { PluginManager, parseSettingValue, validateSetting } from "../extensibility/plugins";
+import { theme } from "../modes/theme/theme";
 
 // =============================================================================
 // Types
@@ -251,7 +246,7 @@ async function handleList(manager: PluginManager, flags: { json?: boolean }): Pr
 			if (availableFeatures.length > 0) {
 				const enabledSet = new Set(plugin.enabledFeatures ?? []);
 				const featureDisplay = availableFeatures
-					.map((f) => (enabledSet.has(f) ? chalk.green(f) : chalk.dim(f)))
+					.map(f => (enabledSet.has(f) ? chalk.green(f) : chalk.dim(f)))
 					.join(", ");
 				console.log(chalk.dim(`  Available: [${featureDisplay}]`));
 			}
@@ -302,10 +297,10 @@ async function handleDoctor(manager: PluginManager, flags: { json?: boolean; fix
 		}
 	}
 
-	const errors = checks.filter((c) => c.status === "error" && !c.fixed).length;
-	const warnings = checks.filter((c) => c.status === "warning" && !c.fixed).length;
-	const ok = checks.filter((c) => c.status === "ok").length;
-	const fixed = checks.filter((c) => c.fixed).length;
+	const errors = checks.filter(c => c.status === "error" && !c.fixed).length;
+	const warnings = checks.filter(c => c.status === "warning" && !c.fixed).length;
+	const ok = checks.filter(c => c.status === "ok").length;
+	const fixed = checks.filter(c => c.fixed).length;
 
 	console.log("");
 	console.log(`Summary: ${ok} ok, ${warnings} warnings, ${errors} errors${fixed > 0 ? `, ${fixed} fixed` : ""}`);
@@ -332,7 +327,7 @@ async function handleFeatures(
 
 	const pluginName = args[0];
 	const plugins = await manager.list();
-	const plugin = plugins.find((p) => p.name === pluginName);
+	const plugin = plugins.find(p => p.name === pluginName);
 
 	if (!plugin) {
 		console.error(chalk.red(`Plugin "${pluginName}" not found`));
@@ -341,21 +336,21 @@ async function handleFeatures(
 
 	// Handle modifications
 	if (flags.enable || flags.disable || flags.set) {
-		let currentFeatures = new Set(manager.getEnabledFeatures(pluginName) ?? []);
+		let currentFeatures = new Set((await manager.getEnabledFeatures(pluginName)) ?? []);
 
 		if (flags.set) {
 			// --set replaces all features
 			currentFeatures = new Set(
 				flags.set
 					.split(",")
-					.map((f) => f.trim())
+					.map(f => f.trim())
 					.filter(Boolean),
 			);
 		} else {
 			if (flags.enable) {
 				for (const f of flags.enable
 					.split(",")
-					.map((f) => f.trim())
+					.map(f => f.trim())
 					.filter(Boolean)) {
 					currentFeatures.add(f);
 				}
@@ -363,7 +358,7 @@ async function handleFeatures(
 			if (flags.disable) {
 				for (const f of flags.disable
 					.split(",")
-					.map((f) => f.trim())
+					.map(f => f.trim())
 					.filter(Boolean)) {
 					currentFeatures.delete(f);
 				}
@@ -375,7 +370,7 @@ async function handleFeatures(
 	}
 
 	// Display current state
-	const updatedFeatures = manager.getEnabledFeatures(pluginName);
+	const updatedFeatures = await manager.getEnabledFeatures(pluginName);
 
 	if (flags.json) {
 		console.log(
@@ -437,7 +432,7 @@ async function handleConfig(
 	}
 
 	const plugins = await manager.list();
-	const plugin = plugins.find((p) => p.name === pluginName);
+	const plugin = plugins.find(p => p.name === pluginName);
 
 	if (!plugin) {
 		console.error(chalk.red(`Plugin "${pluginName}" not found`));
@@ -446,7 +441,7 @@ async function handleConfig(
 
 	switch (subcommand) {
 		case "list": {
-			const settings = manager.getPluginSettings(pluginName);
+			const settings = await manager.getPluginSettings(pluginName);
 			const schema = plugin.manifest.settings || {};
 
 			if (flags.json) {
@@ -481,7 +476,7 @@ async function handleConfig(
 				process.exit(1);
 			}
 
-			const settings = manager.getPluginSettings(pluginName);
+			const settings = await manager.getPluginSettings(pluginName);
 			const schema = plugin.manifest.settings?.[key];
 			const value = settings[key] ?? schema?.default;
 
@@ -516,7 +511,7 @@ async function handleConfig(
 				}
 			}
 
-			manager.setPluginSetting(pluginName, key, value);
+			await manager.setPluginSetting(pluginName, key, value);
 			console.log(chalk.green(`${theme.status.success} Set ${key}`));
 			break;
 		}
@@ -527,7 +522,7 @@ async function handleConfig(
 				process.exit(1);
 			}
 
-			manager.deletePluginSetting(pluginName, key);
+			await manager.deletePluginSetting(pluginName, key);
 			console.log(chalk.green(`${theme.status.success} Deleted ${key}`));
 			break;
 		}
@@ -544,7 +539,7 @@ async function handleConfigValidate(manager: PluginManager, flags: { json?: bool
 	const results: Array<{ plugin: string; key: string; error: string }> = [];
 
 	for (const plugin of plugins) {
-		const settings = manager.getPluginSettings(plugin.name);
+		const settings = await manager.getPluginSettings(plugin.name);
 		const schema = plugin.manifest.settings || {};
 
 		for (const [key, s] of Object.entries(schema)) {

package/src/cli/session-picker.ts

@@ -1,10 +1,9 @@
 /**
  * TUI session selector for --resume flag
  */
-
-import { SessionSelectorComponent } from "@oh-my-pi/pi-coding-agent/modes/components/session-selector";
-import type { SessionInfo } from "@oh-my-pi/pi-coding-agent/session/session-manager";
 import { ProcessTerminal, TUI } from "@oh-my-pi/pi-tui";
+import { SessionSelectorComponent } from "../modes/components/session-selector";
+import type { SessionInfo } from "../session/session-manager";
 
 /** Show TUI session selector and return selected session path or null if cancelled */
 export async function selectSession(sessions: SessionInfo[]): Promise<string | null> {

package/src/cli/setup-cli.ts

@@ -3,11 +3,10 @@
  *
  * Handles `omp setup <component>` to install dependencies for optional features.
  */
-
-import { APP_NAME } from "@oh-my-pi/pi-coding-agent/config";
-import { theme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
 import { $ } from "bun";
 import chalk from "chalk";
+import { APP_NAME } from "../config";
+import { theme } from "../modes/theme/theme";
 
 export type SetupComponent = "python";
 

package/src/cli/stats-cli.ts

@@ -3,9 +3,8 @@
  *
  * Handles `omp stats` subcommand for viewing AI usage statistics.
  */
-
-import { APP_NAME } from "@oh-my-pi/pi-coding-agent/config";
 import chalk from "chalk";
+import { APP_NAME } from "../config";
 
 // =============================================================================
 // Types
@@ -106,7 +105,7 @@ export async function runStatsCommand(cmd: StatsCommandArgs): Promise<void> {
 	}
 
 	// Start the dashboard server
-	const { port } = startServer(cmd.port);
+	const { port } = await startServer(cmd.port);
 	console.log(chalk.green(`Dashboard available at: http://localhost:${port}`));
 	console.log("Press Ctrl+C to stop\n");
 

package/src/cli/update-cli.ts

@@ -4,15 +4,15 @@
  * Handles `omp update` to check for and install updates.
  * Uses bun if available, otherwise downloads binary from GitHub releases.
  */
-
 import { execSync, spawnSync } from "node:child_process";
-import { createWriteStream, existsSync, renameSync, unlinkSync } from "node:fs";
-import { dirname } from "node:path";
+import * as fs from "node:fs";
+import * as path from "node:path";
 import { Readable } from "node:stream";
 import { pipeline } from "node:stream/promises";
-import { APP_NAME, VERSION } from "@oh-my-pi/pi-coding-agent/config";
-import { theme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
+import { isEnoent } from "@oh-my-pi/pi-utils";
 import chalk from "chalk";
+import { APP_NAME, VERSION } from "../config";
+import { theme } from "../modes/theme/theme";
 
 /**
  * Detect if we're running as a Bun compiled binary.
@@ -73,7 +73,7 @@ async function getLatestRelease(): Promise<ReleaseInfo> {
 	return {
 		tag: data.tag_name,
 		version: data.tag_name.replace(/^v/, ""),
-		assets: data.assets.map((a) => ({ name: a.name, url: a.browser_download_url })),
+		assets: data.assets.map(a => ({ name: a.name, url: a.browser_download_url })),
 	};
 }
 
@@ -154,14 +154,14 @@ async function updateViaBun(): Promise<void> {
  */
 async function updateViaBinary(release: ReleaseInfo): Promise<void> {
 	const binaryName = getBinaryName();
-	const asset = release.assets.find((a) => a.name === binaryName);
+	const asset = release.assets.find(a => a.name === binaryName);
 
 	if (!asset) {
 		throw new Error(`No binary found for ${binaryName}`);
 	}
 
 	const execPath = process.execPath;
-	const _execDir = dirname(execPath);
+	const _execDir = path.dirname(execPath);
 	const tempPath = `${execPath}.new`;
 	const backupPath = `${execPath}.bak`;
 
@@ -173,7 +173,7 @@ async function updateViaBinary(release: ReleaseInfo): Promise<void> {
 		throw new Error(`Download failed: ${response.statusText}`);
 	}
 
-	const fileStream = createWriteStream(tempPath, { mode: 0o755 });
+	const fileStream = fs.createWriteStream(tempPath, { mode: 0o755 });
 	const nodeStream = Readable.fromWeb(response.body as import("stream/web").ReadableStream);
 	await pipeline(nodeStream, fileStream);
 
@@ -181,27 +181,30 @@ async function updateViaBinary(release: ReleaseInfo): Promise<void> {
 	console.log(chalk.dim("Installing update..."));
 
 	try {
-		// Backup current binary
-		if (existsSync(backupPath)) {
-			unlinkSync(backupPath);
+		try {
+			await fs.promises.unlink(backupPath);
+		} catch (err) {
+			if (!isEnoent(err)) throw err;
 		}
-		renameSync(execPath, backupPath);
+		await fs.promises.rename(execPath, backupPath);
 
-		// Move new binary into place
-		renameSync(tempPath, execPath);
+		await fs.promises.rename(tempPath, execPath);
 
-		// Clean up backup
-		unlinkSync(backupPath);
+		await fs.promises.unlink(backupPath);
 
 		console.log(chalk.green(`\n${theme.status.success} Updated to ${release.version}`));
 		console.log(chalk.dim(`Restart ${APP_NAME} to use the new version`));
 	} catch (err) {
-		// Restore from backup if possible
-		if (existsSync(backupPath) && !existsSync(execPath)) {
-			renameSync(backupPath, execPath);
+		const [backupExists, execExists, tempExists] = await Promise.all([
+			Bun.file(backupPath).exists(),
+			Bun.file(execPath).exists(),
+			Bun.file(tempPath).exists(),
+		]);
+		if (backupExists && !execExists) {
+			await fs.promises.rename(backupPath, execPath);
 		}
-		if (existsSync(tempPath)) {
-			unlinkSync(tempPath);
+		if (tempExists) {
+			await fs.promises.unlink(tempPath);
 		}
 		throw err;
 	}