@akms/mcp-ssh 0.1.4 → 0.1.6

package/lib/cli.cjs

@@ -31,11 +31,12 @@ var DEFAULT_MAX_READ_FILE_BYTES = 2e5;
 var KEEPALIVE_INTERVAL_MS = 15e3;
 var SESSION_IDLE_TIMEOUT_MS = 15 * 60 * 1e3;
 var SESSION_SWEEP_INTERVAL_MS = 6e4;
+var SHUTDOWN_DEADLINE_MS = 5e3;
 var CWD_MARKER = "__AKMS_SSH_CWD__";
 var MAX_DIRECTORY_ENTRIES = 1e3;
 var SERVER_NAME = "akms-mcp-ssh";
 var PACKAGE_NAME = "@akms/mcp-ssh";
-var SERVER_VERSION = "0.1.4";
+var SERVER_VERSION = "0.1.6";
 
 // src/_defs/command-rules.ts
 var CRITICAL_PATH_PATTERN = new RegExp(
@@ -498,6 +499,10 @@ var LEVEL_WEIGHTS = {
   error: 4
 };
 var SECRET_META_KEYS = ["password", "passphrase", "privatekey", "privatekeypath", "secret", "token"];
+var stderrBroken = false;
+process.stderr.on("error", () => {
+  stderrBroken = true;
+});
 function resolveThreshold() {
   const configured = process.env[LOG_LEVEL_ENV];
   if (configured == null) {
@@ -521,6 +526,9 @@ function maskMeta(meta) {
   return masked;
 }
 function write(level, message, meta) {
+  if (stderrBroken == true) {
+    return;
+  }
   const threshold = resolveThreshold();
   if (threshold === "silent") {
     return;
@@ -534,8 +542,12 @@ function write(level, message, meta) {
   if (Object.keys(masked).length > 0) {
     line += ` ${JSON.stringify(masked)}`;
   }
-  process.stderr.write(`${line}
+  try {
+    process.stderr.write(`${line}
 `);
+  } catch {
+    stderrBroken = true;
+  }
 }
 function createLogger(boundMeta) {
   return {
@@ -4834,7 +4846,7 @@ async function manageHost(prompter, output, target, entries, name) {
     }
   }
 }
-async function runSetupCommand(options) {
+async function runConfigCommand(options) {
   const { input, output } = options;
   const targets = options.targets ?? listConfigTargets();
   let target = targets[0];
@@ -5644,12 +5656,11 @@ function createSshMcpServer() {
 }
 
 // src/cli.ts
-var SETUP_COMMAND_NAMES = ["setup", "config"];
 function parseArguments(argv) {
-  const parsed = { showHelp: false, showVersion: false, checkOnly: false, setupMode: false };
+  const parsed = { showHelp: false, showVersion: false, checkOnly: false, configMode: false };
   let rest = argv;
-  if (argv[0] != null && SETUP_COMMAND_NAMES.includes(argv[0]) == true) {
-    parsed.setupMode = true;
+  if (argv[0] === "config") {
+    parsed.configMode = true;
     rest = argv.slice(1);
   }
   for (let index = 0; index < rest.length; index += 1) {
@@ -5683,17 +5694,17 @@ var HELP_TEXT = [
   "",
   "Usage:",
   "  akms-mcp-ssh [--check]",
-  "  akms-mcp-ssh setup [--target <client>]",
+  "  akms-mcp-ssh config [--target <client>]",
   "",
   "Commands:",
-  "  setup                Manage the SSH hosts registered in an MCP client's config \u2014",
+  "  config               Manage the SSH hosts registered in an MCP client's config \u2014",
   "                       list, add, edit, remove, check, print. Asks plain questions,",
   "                       so the SSH_* names below never have to be typed.",
   "",
   "Options:",
   "      --check          Connect to the configured host, report the result, and exit.",
   "                       Prints the host key fingerprint so it can be pinned.",
-  "      --target <id>    setup: claude-code | codex | claude-desktop | cursor",
+  "      --target <id>    config: claude-code | codex | claude-desktop | cursor",
   "                       (default: the first one installed)",
   "  -v, --version        Print the version and exit",
   "  -h, --help           Print this help and exit",
@@ -5727,7 +5738,7 @@ var HELP_TEXT = [
 ].join("\n");
 async function runCheckCommand(profile) {
   if (profile == null) {
-    process.stdout.write(`No SSH host is registered \u2014 set ${SINGLE_HOST_ENV.HOST} and ${SINGLE_HOST_ENV.USER}, or run: akms-mcp-ssh setup
+    process.stdout.write(`No SSH host is registered \u2014 set ${SINGLE_HOST_ENV.HOST} and ${SINGLE_HOST_ENV.USER}, or run: akms-mcp-ssh config
 `);
     return 1;
   }
@@ -5739,6 +5750,11 @@ async function runCheckCommand(profile) {
 }
 async function main() {
   const _logger = $logger.child({ context: "cli" });
+  if (process.argv[2] === "setup") {
+    process.stderr.write("The setup command has been removed. Use akms-mcp-ssh config.\n");
+    process.exitCode = 1;
+    return;
+  }
   const cliArguments = parseArguments(process.argv.slice(2));
   if (cliArguments.showHelp == true) {
     process.stderr.write(`${HELP_TEXT}
@@ -5750,11 +5766,11 @@ async function main() {
 `);
     return;
   }
-  if (cliArguments.setupMode == true) {
+  if (cliArguments.configMode == true) {
     if (process.env[LOG_LEVEL_ENV] == null) {
       process.env[LOG_LEVEL_ENV] = "warn";
     }
-    const exitCode = await runSetupCommand({
+    const exitCode = await runConfigCommand({
       input: process.stdin,
       output: process.stdout,
       targetId: cliArguments.targetId
@@ -5770,6 +5786,12 @@ async function main() {
     instance.shutdown();
     process.exit(exitCode);
   }
+  process.on("uncaughtException", (error) => {
+    _logger.error(error, "uncaught exception; the server keeps running");
+  });
+  process.on("unhandledRejection", (reason) => {
+    _logger.error(reason, "unhandled rejection; the server keeps running");
+  });
   let shuttingDown = false;
   const shutdown = (signal) => {
     if (shuttingDown == true) {
@@ -5777,6 +5799,10 @@ async function main() {
     }
     shuttingDown = true;
     _logger.info("shutting down", { signal });
+    setTimeout(() => {
+      _logger.warn("shutdown did not finish in time; exiting anyway", { deadline_ms: SHUTDOWN_DEADLINE_MS });
+      process.exit(0);
+    }, SHUTDOWN_DEADLINE_MS).unref();
     instance.shutdown();
     void instance.server.close().finally(() => {
       process.exit(0);
@@ -5796,7 +5822,8 @@ async function main() {
   _logger.info("mcp server ready on stdio", {
     server: SERVER_NAME,
     version: SERVER_VERSION,
-    profile: instance.profile?.name ?? null
+    profile: instance.profile?.name ?? null,
+    pid: process.pid
   });
 }
 main().catch((error) => {