@mathismeadows/roamer-device-auth 1.1.2 → 1.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mathismeadows/roamer-device-auth",
-  "version": "1.1.2",
+  "version": "1.1.3",
   "private": false,
   "type": "module",
   "description": "AUTH-25: server-mediated device authorization flow + stdio<->HTTP MCP proxy for Roamer MCP. Used by roamer-bridge.sh for every OS and every default browser (no more Safari-specific branching, AUTH-14) — Cloudflare Access's Managed OAuth has no device-code grant of its own, so RoamerMcp's server implements it and this script talks to that instead of Entra directly.",

package/roamer-device-auth.mjs

@@ -25,6 +25,7 @@ import { join } from "node:path";
 import { setTimeout as sleep } from "node:timers/promises";
 import { execFile } from "node:child_process";
 import { promisify } from "node:util";
+import { pathToFileURL } from "node:url";
 import qrcode from "qrcode-terminal";
 
 const execFileAsync = promisify(execFile);
@@ -258,10 +259,103 @@ async function doGetValidTokens(forceRefresh) {
   }
 }
 
+// AUTH-28: extracted so it can run both for live traffic (post-sign-in) and for messages
+// queued while sign-in was still in progress, with identical forwarding/retry behavior.
+async function forwardLine(transport, line, getTokens, setTokens) {
+  try {
+    let tokens = getTokens();
+    if (expiresSoon(tokens)) {
+      tokens = await getValidTokens();
+      setTokens(tokens);
+    }
+    try {
+      await transport.send(JSON.parse(line));
+    } catch (err) {
+      // Reactive invalidation: proactive expiry math can't catch everything (server-side
+      // revocation, clock skew, or an incompatible cache — see CACHE_VERSION). A real
+      // auth failure from the server is the ground truth; when we see one, invalidate
+      // whatever we're holding, force a genuinely fresh token, and retry once.
+      if (!isAuthError(err)) throw err;
+      log(`Send failed with an auth error (${err.message}) — invalidating cached token and retrying once.`);
+      tokens = await getValidTokens(true);
+      setTokens(tokens);
+      await transport.send(JSON.parse(line));
+    }
+  } catch (err) {
+    log(`Send failed: ${err.message}`);
+  }
+}
+
+// AUTH-28: a hard sign-in failure (denied/expired device code) must tell the host exactly
+// that, per request, rather than leaving requests unanswered for the host to time out on.
+function respondWithSignInError(line, err) {
+  let id = null;
+  try {
+    id = JSON.parse(line)?.id ?? null;
+  } catch {
+    // Malformed input never had a usable id anyway — respond with null per JSON-RPC convention.
+  }
+  const response = {
+    jsonrpc: "2.0",
+    id,
+    error: { code: -32001, message: `Roamer MCP sign-in failed: ${err.message}` },
+  };
+  process.stdout.write(`${JSON.stringify(response)}\n`);
+}
+
 async function main() {
-  let tokens = await getValidTokens();
+  // AUTH-28: stdin is read (and, until sign-in completes, queued) from the very first tick —
+  // a cold-cache device-code flow can take minutes, and the MCP host must never see this
+  // process as unresponsive/silent during that window. That silence, not the auth flow
+  // itself, is what triggers the host's connect-timeout kill/respawn (see readPendingFlow's
+  // comment above) — this fix targets the silence, not the flow's real, unavoidable duration.
+  let tokens = null;
+  let transport = null;
+  let ready = false;
+  const pendingLines = [];
+
+  const getTokens = () => tokens;
+  const setTokens = (fresh) => {
+    tokens = fresh;
+  };
 
-  const transport = new StreamableHTTPClientTransport(new URL(ROAMER_MCP_URL), {
+  let buffer = "";
+  process.stdin.setEncoding("utf8");
+  process.stdin.on("data", (chunk) => {
+    buffer += chunk;
+    let newlineIndex;
+    while ((newlineIndex = buffer.indexOf("\n")) !== -1) {
+      const line = buffer.slice(0, newlineIndex);
+      buffer = buffer.slice(newlineIndex + 1);
+      if (!line.trim()) continue;
+      if (ready) {
+        forwardLine(transport, line, getTokens, setTokens);
+      } else {
+        pendingLines.push(line);
+      }
+    }
+  });
+
+  process.stdin.on("end", async () => {
+    log("stdin closed, shutting down.");
+    if (transport) await transport.close();
+    process.exit(0);
+  });
+
+  log("Local STDIO proxy running. Press Ctrl+C to exit.");
+
+  try {
+    tokens = await getValidTokens();
+  } catch (err) {
+    // Sign-in genuinely failed (denied/expired), not just slow — every message queued while
+    // we waited, starting with `initialize`, gets a real JSON-RPC error so the host sees an
+    // explicit failure instead of a process it has to silently time out on.
+    log(`Sign-in failed: ${err.message}`);
+    for (const line of pendingLines) respondWithSignInError(line, err);
+    process.exit(1);
+  }
+
+  transport = new StreamableHTTPClientTransport(new URL(ROAMER_MCP_URL), {
     requestInit: {
       get headers() {
         return { Authorization: `Bearer ${tokens.access_token}` };
@@ -271,14 +365,11 @@ async function main() {
 
   transport.onerror = (err) => {
     log(`Transport error: ${err.message}`);
-    // Same reasoning as the send-path retry below: an auth error means whatever's cached is
+    // Same reasoning as the send-path retry above: an auth error means whatever's cached is
     // known-bad, so drop it now rather than let the next proactive expiresSoon() check
     // (which only reasons about calendar time) keep handing it out.
     if (isAuthError(err)) clearCachedTokens();
   };
-  await transport.start();
-  log("Connected to remote server using StreamableHTTPClientTransport.");
-
   // stdin/stdout <-> transport pass-through. Each stdin line is one JSON-RPC message;
   // transport.send() delivers it, and transport.onmessage delivers whatever comes back
   // (including server-initiated messages over the SSE half of the streamable-HTTP transport).
@@ -286,49 +377,23 @@ async function main() {
     process.stdout.write(`${JSON.stringify(message)}\n`);
   };
 
-  let buffer = "";
-  process.stdin.setEncoding("utf8");
-  process.stdin.on("data", (chunk) => {
-    buffer += chunk;
-    let newlineIndex;
-    while ((newlineIndex = buffer.indexOf("\n")) !== -1) {
-      const line = buffer.slice(0, newlineIndex);
-      buffer = buffer.slice(newlineIndex + 1);
-      if (!line.trim()) continue;
-      (async () => {
-        try {
-          if (expiresSoon(tokens)) {
-            tokens = await getValidTokens();
-          }
-          try {
-            await transport.send(JSON.parse(line));
-          } catch (err) {
-            // Reactive invalidation: proactive expiry math can't catch everything (server-side
-            // revocation, clock skew, or an incompatible cache — see CACHE_VERSION). A real
-            // auth failure from the server is the ground truth; when we see one, invalidate
-            // whatever we're holding, force a genuinely fresh token, and retry once.
-            if (!isAuthError(err)) throw err;
-            log(`Send failed with an auth error (${err.message}) — invalidating cached token and retrying once.`);
-            tokens = await getValidTokens(true);
-            await transport.send(JSON.parse(line));
-          }
-        } catch (err) {
-          log(`Send failed: ${err.message}`);
-        }
-      })();
-    }
-  });
+  await transport.start();
+  log("Connected to remote server using StreamableHTTPClientTransport.");
 
-  process.stdin.on("end", async () => {
-    log("stdin closed, shutting down.");
-    await transport.close();
-    process.exit(0);
-  });
+  ready = true;
+  for (const line of pendingLines) {
+    forwardLine(transport, line, getTokens, setTokens);
+  }
+}
 
-  log("Local STDIO proxy running. Press Ctrl+C to exit.");
+// Only auto-run when invoked directly (npx/CLI) — importing this module from a test file
+// must not trigger a live device-auth flow and stdio takeover as a side effect.
+const isMainModule = process.argv[1] && import.meta.url === pathToFileURL(process.argv[1]).href;
+if (isMainModule) {
+  main().catch((err) => {
+    log(`Fatal error: ${err.stack ?? err.message}`);
+    process.exit(1);
+  });
 }
 
-main().catch((err) => {
-  log(`Fatal error: ${err.stack ?? err.message}`);
-  process.exit(1);
-});
+export { respondWithSignInError, forwardLine };