feat(coding-agent): added SSH host management with CLI and slash commands

- Added SSH host management feature with `ssh` CLI command and `/ssh` slash command for add, list, and remove operations. - Added SSH configuration support at project (.omp/ssh.json) and user (~/.omp/agent/ssh.json) scopes with host discovery from project files. - Added SSH host configuration flags: --host, --user, --port, --key, --desc, --compat, and --scope for flexible host setup. - Removed automatic line relocation on stale hash references; now fails with error instead.
2026-02-19 12:52:43 +01:00
parent a9fb9c9791
commit e41ebe61ec
10 changed files with 918 additions and 1 deletions
@@ -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/dirs";
+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`);
+	}
+}