@hheei/ssh-exec-mcp 0.1.0

package/plugins/ssh-exec/scripts/ssh-mount.ts

@@ -0,0 +1,27 @@
+import { SessionManager, type ProcessRunner, type SshMountResult } from "./session-manager";
+import { validateHost } from "./ssh-exec";
+
+export interface SshMountArgs {
+	host: string;
+}
+
+export function validateSshMountArgs(value: unknown): SshMountArgs {
+	if (!value || typeof value !== "object" || Array.isArray(value)) {
+		throw new Error("ssh_mount arguments must be an object");
+	}
+
+	const record = value as Record<string, unknown>;
+	if (typeof record.host !== "string" || record.host.trim() === "") {
+		throw new Error("ssh_mount.host must be a non-empty string");
+	}
+
+	return { host: validateHost(record.host) };
+}
+
+export async function executeSshMount(
+	manager: SessionManager,
+	args: SshMountArgs,
+	runner: ProcessRunner,
+): Promise<SshMountResult> {
+	return await manager.ensureMounted(validateHost(args.host), runner);
+}

package/ssh-exec-mcp.ts

@@ -0,0 +1,43 @@
+#!/usr/bin/env bun
+
+import {
+	createMcpServer,
+	runCleanupSshProcess,
+	runStdio,
+} from "./plugins/ssh-exec/scripts/ssh-exec-mcp";
+
+const server = createMcpServer();
+const runner = (args: string[], timeoutMs?: number) =>
+	runCleanupSshProcess(server.manager.sshBin, args, timeoutMs, server.manager.sensitiveValues());
+
+let cleanupPromise: Promise<void> | undefined;
+
+async function cleanup(budgetMs = 1_500): Promise<void> {
+	cleanupPromise ??= Promise.race([
+		server.manager.closeAll(runner, Math.min(1_000, budgetMs)),
+		new Promise<void>((resolve) => setTimeout(resolve, budgetMs)),
+	]).catch(() => undefined);
+	await cleanupPromise;
+}
+
+process.once("beforeExit", () => {
+	void cleanup();
+});
+
+process.on("SIGINT", () => {
+	void cleanup().finally(() => process.exit(130));
+});
+
+process.on("SIGTERM", () => {
+	void cleanup().finally(() => process.exit(143));
+});
+
+runStdio(server)
+	.then(async () => {
+		await cleanup();
+		process.exit(0);
+	})
+	.catch((error) => {
+		process.stderr.write(`${error instanceof Error ? error.stack ?? error.message : String(error)}\n`);
+		process.exit(1);
+	});