@mgsoftwarebv/mg-dashboard-mcp 3.4.0 → 3.4.2

package/dist/index.js

@@ -83,7 +83,7 @@ function deriveAuthBaseUrl(proxyUrl2) {
   const u = new URL(proxyUrl2);
   return `${u.protocol}//${u.host}`;
 }
-async function performHandshake(proxyUrl2, pubPath, pubText) {
+async function performHandshake(proxyUrl2, apiKey2, pubPath, pubText) {
   const base = deriveAuthBaseUrl(proxyUrl2);
   const challengeRes = await fetch(`${base}/v1/auth/ssh/challenge`, {
     method: "POST",
@@ -110,6 +110,7 @@ async function performHandshake(proxyUrl2, pubPath, pubText) {
     headers: { "Content-Type": "application/json" },
     body: JSON.stringify({
       challenge_id: challenge.challenge_id,
+      api_key: apiKey2,
       pubkey: pubText,
       signature: sign.stdout
     })
@@ -127,10 +128,11 @@ async function performHandshake(proxyUrl2, pubPath, pubText) {
 }
 async function runBridge(handshake, proxyUrl2, refresh) {
   let currentToken = handshake.token;
+  const requestHeaders = {
+    Authorization: `Bearer ${currentToken}`
+  };
   const upstreamTransport = new StreamableHTTPClientTransport(new URL(proxyUrl2), {
-    requestInit: () => ({
-      headers: { Authorization: `Bearer ${currentToken}` }
-    })
+    requestInit: { headers: requestHeaders }
   });
   const upstream = new Client$1({ name: "mg-dashboard-mcp-proxy-bridge", version: "1.0.0" });
   const scheduleRefresh = (result) => {
@@ -139,6 +141,7 @@ async function runBridge(handshake, proxyUrl2, refresh) {
       try {
         const fresh = await refresh();
         currentToken = fresh.token;
+        requestHeaders.Authorization = `Bearer ${currentToken}`;
         console.error(`[Proxy] Token refreshed (key: ${fresh.proxyKeyName})`);
         scheduleRefresh(fresh);
       } catch (err) {
@@ -169,14 +172,30 @@ async function runBridge(handshake, proxyUrl2, refresh) {
   };
   await stdioServer.connect(new StdioServerTransport());
   console.error(`[Proxy] Bridge ready. Forwarding stdio \u2192 ${proxyUrl2}`);
+  await new Promise((resolve) => {
+    const keepAlive = setInterval(() => void 0, 2147483647);
+    const done = () => {
+      clearInterval(keepAlive);
+      resolve();
+    };
+    process.stdin.once("close", done);
+    process.stdin.once("end", done);
+    process.once("SIGINT", done);
+    process.once("SIGTERM", done);
+  });
 }
 async function runProxyMode(args2) {
   const proxyUrl2 = getArg(args2, "proxy-url") || process.env.MG_DASHBOARD_PROXY_URL;
+  const apiKey2 = getArg(args2, "api-key") || process.env.MG_DASHBOARD_API_KEY;
   const sshKey = getArg(args2, "ssh-key") || process.env.MG_DASHBOARD_SSH_KEY;
   if (!proxyUrl2) {
     console.error("--proxy-url is required (or set MG_DASHBOARD_PROXY_URL)");
     process.exit(1);
   }
+  if (!apiKey2) {
+    console.error("--api-key is required in proxy mode (or set MG_DASHBOARD_API_KEY)");
+    process.exit(1);
+  }
   if (!sshKey) {
     console.error("--ssh-key is required in proxy mode (or set MG_DASHBOARD_SSH_KEY)");
     process.exit(1);
@@ -189,7 +208,7 @@ async function runProxyMode(args2) {
     console.error(`[Proxy] ${err instanceof Error ? err.message : String(err)}`);
     process.exit(1);
   }
-  const refresh = () => performHandshake(proxyUrl2, resolved.pubPath, resolved.pubText);
+  const refresh = () => performHandshake(proxyUrl2, apiKey2, resolved.pubPath, resolved.pubText);
   let handshake;
   try {
     handshake = await refresh();