@theokit/sdk 4.52.1 → 4.53.1

package/dist/mcp-auth.cjs

@@ -0,0 +1,299 @@
+'use strict';
+
+var chunkC4H6PB5J_cjs = require('./chunk-C4H6PB5J.cjs');
+var chunk5YKNEMXR_cjs = require('./chunk-5YKNEMXR.cjs');
+var chunkLPVIRJVF_cjs = require('./chunk-LPVIRJVF.cjs');
+var chunkBM6N4KHO_cjs = require('./chunk-BM6N4KHO.cjs');
+var crypto = require('crypto');
+var http = require('http');
+var promises = require('readline/promises');
+var fs = require('fs');
+var module$1 = require('module');
+var os = require('os');
+var path = require('path');
+
+var _documentCurrentScript = typeof document !== 'undefined' ? document.currentScript : null;
+var KEYTAR_SERVICE = "theokit-mcp";
+var FILE_PATH = path.join(os.homedir(), ".theokit", "mcp-tokens.json");
+function ensurePrivateStoreDir() {
+  const dir = path.dirname(FILE_PATH);
+  try {
+    fs.mkdirSync(dir, { recursive: true, mode: 448 });
+    fs.chmodSync(dir, 448);
+  } catch {
+  }
+}
+var keytarWarned = false;
+var cachedKeytar;
+function tryRequireKeytar() {
+  if (cachedKeytar !== void 0) return cachedKeytar;
+  try {
+    const r = module$1.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('mcp-auth.cjs', document.baseURI).href)));
+    cachedKeytar = r("keytar");
+  } catch {
+    if (!keytarWarned) {
+      chunkBM6N4KHO_cjs.diag(
+        "[theokit-sdk] keytar not installed; MCP OAuth tokens will be stored in ~/.theokit/mcp-tokens.json (chmod 600 where POSIX). Install `keytar` for OS keychain storage.\n"
+      );
+      keytarWarned = true;
+    }
+    cachedKeytar = null;
+  }
+  return cachedKeytar;
+}
+var inflightRefresh = /* @__PURE__ */ new Map();
+async function setTokens(serverName, tokens) {
+  const kt = tryRequireKeytar();
+  const payload = JSON.stringify(tokens);
+  if (kt !== null && kt !== void 0) {
+    await kt.setPassword(KEYTAR_SERVICE, serverName, payload);
+    return;
+  }
+  ensurePrivateStoreDir();
+  let allTokens = {};
+  if (fs.existsSync(FILE_PATH)) {
+    try {
+      allTokens = JSON.parse(fs.readFileSync(FILE_PATH, "utf8"));
+    } catch {
+      allTokens = {};
+    }
+  }
+  allTokens[serverName] = tokens;
+  await chunkC4H6PB5J_cjs.atomicWriteJson(FILE_PATH, allTokens);
+  try {
+    fs.chmodSync(FILE_PATH, 384);
+  } catch {
+  }
+}
+async function getTokens(serverName) {
+  const kt = tryRequireKeytar();
+  if (kt !== null && kt !== void 0) {
+    const value = await kt.getPassword(KEYTAR_SERVICE, serverName);
+    if (typeof value !== "string") return void 0;
+    try {
+      return JSON.parse(value);
+    } catch {
+      return void 0;
+    }
+  }
+  if (!fs.existsSync(FILE_PATH)) return void 0;
+  chunk5YKNEMXR_cjs.assertSecureModes(path.dirname(FILE_PATH), FILE_PATH);
+  try {
+    const all = JSON.parse(fs.readFileSync(FILE_PATH, "utf8"));
+    return all[serverName];
+  } catch {
+    return void 0;
+  }
+}
+function lockedRefresh(serverName, refreshFn) {
+  const existing = inflightRefresh.get(serverName);
+  if (existing !== void 0) return existing;
+  const promise = (async () => {
+    try {
+      return await refreshFn();
+    } finally {
+      inflightRefresh.delete(serverName);
+    }
+  })();
+  inflightRefresh.set(serverName, promise);
+  return promise;
+}
+
+// src/internal/mcp/oauth.ts
+function base64url(n) {
+  return crypto.randomBytes(n).toString("base64url");
+}
+function challengeFor(verifier) {
+  return crypto.createHash("sha256").update(verifier).digest("base64url");
+}
+async function refreshAccessToken(serverName, clientId, refreshToken, oauth) {
+  const body = new URLSearchParams({
+    grant_type: "refresh_token",
+    refresh_token: refreshToken,
+    client_id: clientId
+  });
+  const response = await fetch(oauth.tokenEndpoint, {
+    method: "POST",
+    headers: { "Content-Type": "application/x-www-form-urlencoded" },
+    body: body.toString()
+  });
+  if (!response.ok) {
+    throw new chunkLPVIRJVF_cjs.ConfigurationError(`OAuth refresh failed: HTTP ${response.status}`, {
+      code: "oauth_refresh_failed"
+    });
+  }
+  const data = await response.json();
+  const obtainedAt = Date.now();
+  const tokens = {
+    accessToken: data.access_token,
+    refreshToken: data.refresh_token ?? refreshToken,
+    // EC-10: default conservative 3600s if expires_in missing.
+    expiresAt: obtainedAt + (data.expires_in ?? 3600) * 1e3,
+    obtainedAt,
+    ...data.scope !== void 0 ? { scope: data.scope } : {}
+  };
+  await setTokens(serverName, tokens);
+  return tokens;
+}
+async function runPkceFlow(serverName, clientId, scopes, oauth) {
+  const verifier = base64url(48);
+  const challenge = challengeFor(verifier);
+  const state = base64url(16);
+  const timeoutMs = oauth.timeoutMs ?? 3e5;
+  let redirectUri;
+  let codeWaiter;
+  let cleanup = () => {
+  };
+  if (oauth.redirectMode === "localhost") {
+    const result = await openLocalhostCallback(state, timeoutMs);
+    redirectUri = result.redirectUri;
+    codeWaiter = result.waitFor;
+    cleanup = result.close;
+  } else {
+    redirectUri = "urn:ietf:wg:oauth:2.0:oob";
+    codeWaiter = readCodeFromStdin(state, timeoutMs);
+  }
+  try {
+    const authUrl = new URL(oauth.authorizationEndpoint);
+    authUrl.searchParams.set("response_type", "code");
+    authUrl.searchParams.set("client_id", clientId);
+    authUrl.searchParams.set("redirect_uri", redirectUri);
+    if (scopes !== void 0 && scopes.length > 0) {
+      authUrl.searchParams.set("scope", scopes.join(" "));
+    }
+    authUrl.searchParams.set("code_challenge", challenge);
+    authUrl.searchParams.set("code_challenge_method", "S256");
+    authUrl.searchParams.set("state", state);
+    chunkBM6N4KHO_cjs.diag(
+      `
+[theokit-sdk] OAuth required for MCP server "${serverName}".
+Open this URL to authenticate:
+  ${authUrl.toString()}
+
+`
+    );
+    const { code, state: returnedState } = await codeWaiter;
+    if (returnedState !== state) {
+      throw new chunkLPVIRJVF_cjs.ConfigurationError(
+        "OAuth state mismatch \u2014 possible CSRF attempt; aborting token exchange.",
+        { code: "oauth_state_mismatch" }
+      );
+    }
+    const tokenBody = new URLSearchParams({
+      grant_type: "authorization_code",
+      code,
+      client_id: clientId,
+      code_verifier: verifier,
+      redirect_uri: redirectUri
+    });
+    const response = await fetch(oauth.tokenEndpoint, {
+      method: "POST",
+      headers: { "Content-Type": "application/x-www-form-urlencoded" },
+      body: tokenBody.toString()
+    });
+    if (!response.ok) {
+      const errText = await response.text();
+      throw new chunkLPVIRJVF_cjs.ConfigurationError(
+        `OAuth token exchange failed: HTTP ${response.status}: ${errText.slice(0, 200)}`,
+        { code: "oauth_token_exchange_failed" }
+      );
+    }
+    const data = await response.json();
+    const obtainedAt = Date.now();
+    return {
+      accessToken: data.access_token,
+      ...data.refresh_token !== void 0 ? { refreshToken: data.refresh_token } : {},
+      // EC-10: default 3600s when expires_in absent.
+      expiresAt: obtainedAt + (data.expires_in ?? 3600) * 1e3,
+      obtainedAt,
+      ...data.scope !== void 0 ? { scope: data.scope } : {}
+    };
+  } finally {
+    cleanup();
+  }
+}
+async function openLocalhostCallback(expectedState, timeoutMs) {
+  let resolveCode;
+  let rejectCode;
+  const waitFor = new Promise((res, rej) => {
+    resolveCode = res;
+    rejectCode = rej;
+  });
+  const server = http.createServer((req, res) => {
+    const url = new URL(req.url ?? "/", "http://127.0.0.1");
+    const code = url.searchParams.get("code");
+    const state = url.searchParams.get("state");
+    if (typeof code !== "string" || typeof state !== "string") {
+      res.statusCode = 400;
+      res.end("Missing code or state");
+      return;
+    }
+    if (state !== expectedState) {
+      res.statusCode = 400;
+      res.end("State mismatch \u2014 request rejected.");
+      return;
+    }
+    res.statusCode = 200;
+    res.end("Authentication successful. You can close this tab and return to the terminal.");
+    resolveCode?.({ code, state });
+  });
+  server.listen(0, "127.0.0.1");
+  const timeoutHandle = setTimeout(() => {
+    rejectCode?.(new chunkLPVIRJVF_cjs.ConfigurationError("OAuth flow timed out", { code: "oauth_timeout" }));
+  }, timeoutMs);
+  await new Promise((r) => {
+    if (server.listening) r();
+    else server.once("listening", () => r());
+  });
+  const address = server.address();
+  if (typeof address !== "object" || address === null) {
+    server.close();
+    throw new chunkLPVIRJVF_cjs.ConfigurationError("OAuth localhost callback failed to bind", {
+      code: "oauth_bind_failed"
+    });
+  }
+  const port = address.port;
+  return {
+    redirectUri: `http://127.0.0.1:${port}/callback`,
+    waitFor,
+    close: () => {
+      clearTimeout(timeoutHandle);
+      server.close();
+    }
+  };
+}
+async function readCodeFromStdin(expectedState, timeoutMs) {
+  const rl = promises.createInterface({ input: process.stdin, output: process.stderr });
+  const promise = new Promise((resolve, reject) => {
+    rl.question("Paste the full callback URL (or just the code): ").then((answer) => {
+      rl.close();
+      const trimmed = answer.trim();
+      if (trimmed.includes("?")) {
+        try {
+          const parsed = new URL(trimmed);
+          const code = parsed.searchParams.get("code");
+          const state = parsed.searchParams.get("state") ?? "";
+          if (typeof code === "string") {
+            resolve({ code, state });
+            return;
+          }
+        } catch {
+        }
+      }
+      resolve({ code: trimmed, state: expectedState });
+    }).catch(reject);
+    setTimeout(() => {
+      rl.close();
+      reject(new chunkLPVIRJVF_cjs.ConfigurationError("OAuth manual input timed out", { code: "oauth_timeout" }));
+    }, timeoutMs);
+  });
+  return promise;
+}
+
+exports.getTokens = getTokens;
+exports.lockedRefresh = lockedRefresh;
+exports.refreshAccessToken = refreshAccessToken;
+exports.runPkceFlow = runPkceFlow;
+exports.setTokens = setTokens;
+//# sourceMappingURL=mcp-auth.cjs.map
+//# sourceMappingURL=mcp-auth.cjs.map

package/dist/mcp-auth.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/internal/mcp/token-storage.ts","../src/internal/mcp/oauth.ts"],"names":["join","homedir","dirname","mkdirSync","chmodSync","createRequire","diag","existsSync","readFileSync","atomicWriteJson","assertSecureModes","randomBytes","createHash","ConfigurationError","createServer","createInterface"],"mappings":";;;;;;;;;;;;;;;AAqBA,IAAM,cAAA,GAAiB,aAAA;AACvB,IAAM,SAAA,GAAYA,SAAA,CAAKC,UAAA,EAAQ,EAAG,YAAY,iBAAiB,CAAA;AAgB/D,SAAS,qBAAA,GAA8B;AACrC,EAAA,MAAM,GAAA,GAAMC,aAAQ,SAAS,CAAA;AAC7B,EAAA,IAAI;AACF,IAAAC,YAAA,CAAU,KAAK,EAAE,SAAA,EAAW,IAAA,EAAM,IAAA,EAAM,KAAO,CAAA;AAC/C,IAAAC,YAAA,CAAU,KAAK,GAAK,CAAA;AAAA,EACtB,CAAA,CAAA,MAAQ;AAAA,EAGR;AACF;AAQA,IAAI,YAAA,GAAe,KAAA;AACnB,IAAI,YAAA;AACJ,SAAS,gBAAA,GAAsC;AAC7C,EAAA,IAAI,YAAA,KAAiB,QAAW,OAAO,YAAA;AACvC,EAAA,IAAI;AACF,IAAA,MAAM,CAAA,GAAIC,sBAAA,CAAc,8PAAe,CAAA;AACvC,IAAA,YAAA,GAAe,EAAE,QAAQ,CAAA;AAAA,EAC3B,CAAA,CAAA,MAAQ;AACN,IAAA,IAAI,CAAC,YAAA,EAAc;AACjB,MAAAC,sBAAA;AAAA,QACE;AAAA,OACF;AACA,MAAA,YAAA,GAAe,IAAA;AAAA,IACjB;AACA,IAAA,YAAA,GAAe,IAAA;AAAA,EACjB;AACA,EAAA,OAAO,YAAA;AACT;AAIA,IAAM,eAAA,uBAAsB,GAAA,EAAkC;AAO9D,eAAsB,SAAA,CAAU,YAAoB,MAAA,EAAoC;AACtF,EAAA,MAAM,KAAK,gBAAA,EAAiB;AAC5B,EAAA,MAAM,OAAA,GAAU,IAAA,CAAK,SAAA,CAAU,MAAM,CAAA;AACrC,EAAA,IAAI,EAAA,KAAO,IAAA,IAAQ,EAAA,KAAO,MAAA,EAAW;AACnC,IAAA,MAAM,EAAA,CAAG,WAAA,CAAY,cAAA,EAAgB,UAAA,EAAY,OAAO,CAAA;AACxD,IAAA;AAAA,EACF;AAEA,EAAA,qBAAA,EAAsB;AACtB,EAAA,IAAI,YAAyC,EAAC;AAC9C,EAAA,IAAIC,aAAA,CAAW,SAAS,CAAA,EAAG;AACzB,IAAA,IAAI;AACF,MAAA,SAAA,GAAY,IAAA,CAAK,KAAA,CAAMC,eAAA,CAAa,SAAA,EAAW,MAAM,CAAC,CAAA;AAAA,IACxD,CAAA,CAAA,MAAQ;AAEN,MAAA,SAAA,GAAY,EAAC;AAAA,IACf;AAAA,EACF;AACA,EAAA,SAAA,CAAU,UAAU,CAAA,GAAI,MAAA;AACxB,EAAA,MAAMC,iCAAA,CAAgB,WAAW,SAAS,CAAA;AAC1C,EAAA,IAAI;AACF,IAAAL,YAAA,CAAU,WAAW,GAAK,CAAA;AAAA,EAC5B,CAAA,CAAA,MAAQ;AAAA,EAER;AACF;AAMA,eAAsB,UAAU,UAAA,EAAsD;AACpF,EAAA,MAAM,KAAK,gBAAA,EAAiB;AAC5B,EAAA,IAAI,EAAA,KAAO,IAAA,IAAQ,EAAA,KAAO,MAAA,EAAW;AACnC,IAAA,MAAM,KAAA,GAAQ,MAAM,EAAA,CAAG,WAAA,CAAY,gBAAgB,UAAU,CAAA;AAC7D,IAAA,IAAI,OAAO,KAAA,KAAU,QAAA,EAAU,OAAO,MAAA;AACtC,IAAA,IAAI;AACF,MAAA,OAAO,IAAA,CAAK,MAAM,KAAK,CAAA;AAAA,IACzB,CAAA,CAAA,MAAQ;AACN,MAAA,OAAO,MAAA;AAAA,IACT;AAAA,EACF;AACA,EAAA,IAAI,CAACG,aAAA,CAAW,SAAS,CAAA,EAAG,OAAO,MAAA;AAMnC,EAAAG,mCAAA,CAAkBR,YAAA,CAAQ,SAAS,CAAA,EAAG,SAAS,CAAA;AAC/C,EAAA,IAAI;AACF,IAAA,MAAM,MAAM,IAAA,CAAK,KAAA,CAAMM,eAAA,CAAa,SAAA,EAAW,MAAM,CAAC,CAAA;AACtD,IAAA,OAAO,IAAI,UAAU,CAAA;AAAA,EACvB,CAAA,CAAA,MAAQ;AACN,IAAA,OAAO,MAAA;AAAA,EACT;AACF;AAQO,SAAS,aAAA,CACd,YACA,SAAA,EACsB;AACtB,EAAA,MAAM,QAAA,GAAW,eAAA,CAAgB,GAAA,CAAI,UAAU,CAAA;AAC/C,EAAA,IAAI,QAAA,KAAa,QAAW,OAAO,QAAA;AACnC,EAAA,MAAM,WAAW,YAAY;AAC3B,IAAA,IAAI;AACF,MAAA,OAAO,MAAM,SAAA,EAAU;AAAA,IACzB,CAAA,SAAE;AACA,MAAA,eAAA,CAAgB,OAAO,UAAU,CAAA;AAAA,IACnC;AAAA,EACF,CAAA,GAAG;AACH,EAAA,eAAA,CAAgB,GAAA,CAAI,YAAY,OAAO,CAAA;AACvC,EAAA,OAAO,OAAA;AACT;;;AC9IA,SAAS,UAAU,CAAA,EAAmB;AACpC,EAAA,OAAOG,kBAAA,CAAY,CAAC,CAAA,CAAE,QAAA,CAAS,WAAW,CAAA;AAC5C;AAKA,SAAS,aAAa,QAAA,EAA0B;AAC9C,EAAA,OAAOC,kBAAW,QAAQ,CAAA,CAAE,OAAO,QAAQ,CAAA,CAAE,OAAO,WAAW,CAAA;AACjE;AAMA,eAAsB,kBAAA,CACpB,UAAA,EACA,QAAA,EACA,YAAA,EACA,KAAA,EACsB;AACtB,EAAA,MAAM,IAAA,GAAO,IAAI,eAAA,CAAgB;AAAA,IAC/B,UAAA,EAAY,eAAA;AAAA,IACZ,aAAA,EAAe,YAAA;AAAA,IACf,SAAA,EAAW;AAAA,GACZ,CAAA;AACD,EAAA,MAAM,QAAA,GAAW,MAAM,KAAA,CAAM,KAAA,CAAM,aAAA,EAAe;AAAA,IAChD,MAAA,EAAQ,MAAA;AAAA,IACR,OAAA,EAAS,EAAE,cAAA,EAAgB,mCAAA,EAAoC;AAAA,IAC/D,IAAA,EAAM,KAAK,QAAA;AAAS,GACrB,CAAA;AACD,EAAA,IAAI,CAAC,SAAS,EAAA,EAAI;AAChB,IAAA,MAAM,IAAIC,oCAAA,CAAmB,CAAA,2BAAA,EAA8B,QAAA,CAAS,MAAM,CAAA,CAAA,EAAI;AAAA,MAC5E,IAAA,EAAM;AAAA,KACP,CAAA;AAAA,EACH;AACA,EAAA,MAAM,IAAA,GAAQ,MAAM,QAAA,CAAS,IAAA,EAAK;AAMlC,EAAA,MAAM,UAAA,GAAa,KAAK,GAAA,EAAI;AAC5B,EAAA,MAAM,MAAA,GAAsB;AAAA,IAC1B,aAAa,IAAA,CAAK,YAAA;AAAA,IAClB,YAAA,EAAc,KAAK,aAAA,IAAiB,YAAA;AAAA;AAAA,IAEpC,SAAA,EAAW,UAAA,GAAA,CAAc,IAAA,CAAK,UAAA,IAAc,IAAA,IAAQ,GAAA;AAAA,IACpD,UAAA;AAAA,IACA,GAAI,KAAK,KAAA,KAAU,MAAA,GAAY,EAAE,KAAA,EAAO,IAAA,CAAK,KAAA,EAAM,GAAI;AAAC,GAC1D;AACA,EAAA,MAAM,SAAA,CAAU,YAAY,MAAM,CAAA;AAClC,EAAA,OAAO,MAAA;AACT;AASA,eAAsB,WAAA,CACpB,UAAA,EACA,QAAA,EACA,MAAA,EACA,KAAA,EACsB;AACtB,EAAA,MAAM,QAAA,GAAW,UAAU,EAAE,CAAA;AAC7B,EAAA,MAAM,SAAA,GAAY,aAAa,QAAQ,CAAA;AACvC,EAAA,MAAM,KAAA,GAAQ,UAAU,EAAE,CAAA;AAC1B,EAAA,MAAM,SAAA,GAAY,MAAM,SAAA,IAAa,GAAA;AAErC,EAAA,IAAI,WAAA;AACJ,EAAA,IAAI,UAAA;AACJ,EAAA,IAAI,UAAsB,MAAM;AAAA,EAAC,CAAA;AAEjC,EAAA,IAAI,KAAA,CAAM,iBAAiB,WAAA,EAAa;AACtC,IAAA,MAAM,MAAA,GAAS,MAAM,qBAAA,CAAsB,KAAA,EAAO,SAAS,CAAA;AAC3D,IAAA,WAAA,GAAc,MAAA,CAAO,WAAA;AACrB,IAAA,UAAA,GAAa,MAAA,CAAO,OAAA;AACpB,IAAA,OAAA,GAAU,MAAA,CAAO,KAAA;AAAA,EACnB,CAAA,MAAO;AACL,IAAA,WAAA,GAAc,2BAAA;AACd,IAAA,UAAA,GAAa,iBAAA,CAAkB,OAAO,SAAS,CAAA;AAAA,EACjD;AAEA,EAAA,IAAI;AACF,IAAA,MAAM,OAAA,GAAU,IAAI,GAAA,CAAI,KAAA,CAAM,qBAAqB,CAAA;AACnD,IAAA,OAAA,CAAQ,YAAA,CAAa,GAAA,CAAI,eAAA,EAAiB,MAAM,CAAA;AAChD,IAAA,OAAA,CAAQ,YAAA,CAAa,GAAA,CAAI,WAAA,EAAa,QAAQ,CAAA;AAC9C,IAAA,OAAA,CAAQ,YAAA,CAAa,GAAA,CAAI,cAAA,EAAgB,WAAW,CAAA;AACpD,IAAA,IAAI,MAAA,KAAW,KAAA,CAAA,IAAa,MAAA,CAAO,MAAA,GAAS,CAAA,EAAG;AAC7C,MAAA,OAAA,CAAQ,aAAa,GAAA,CAAI,OAAA,EAAS,MAAA,CAAO,IAAA,CAAK,GAAG,CAAC,CAAA;AAAA,IACpD;AACA,IAAA,OAAA,CAAQ,YAAA,CAAa,GAAA,CAAI,gBAAA,EAAkB,SAAS,CAAA;AACpD,IAAA,OAAA,CAAQ,YAAA,CAAa,GAAA,CAAI,uBAAA,EAAyB,MAAM,CAAA;AACxD,IAAA,OAAA,CAAQ,YAAA,CAAa,GAAA,CAAI,OAAA,EAAS,KAAK,CAAA;AAEvC,IAAAP,sBAAA;AAAA,MACE;AAAA,6CAAA,EAAkD,UAAU,CAAA;AAAA;AAAA,EAAA,EACrB,OAAA,CAAQ,UAAU;;AAAA;AAAA,KAC3D;AAEA,IAAA,MAAM,EAAE,IAAA,EAAM,KAAA,EAAO,aAAA,KAAkB,MAAM,UAAA;AAE7C,IAAA,IAAI,kBAAkB,KAAA,EAAO;AAC3B,MAAA,MAAM,IAAIO,oCAAA;AAAA,QACR,6EAAA;AAAA,QACA,EAAE,MAAM,sBAAA;AAAuB,OACjC;AAAA,IACF;AAGA,IAAA,MAAM,SAAA,GAAY,IAAI,eAAA,CAAgB;AAAA,MACpC,UAAA,EAAY,oBAAA;AAAA,MACZ,IAAA;AAAA,MACA,SAAA,EAAW,QAAA;AAAA,MACX,aAAA,EAAe,QAAA;AAAA,MACf,YAAA,EAAc;AAAA,KACf,CAAA;AACD,IAAA,MAAM,QAAA,GAAW,MAAM,KAAA,CAAM,KAAA,CAAM,aAAA,EAAe;AAAA,MAChD,MAAA,EAAQ,MAAA;AAAA,MACR,OAAA,EAAS,EAAE,cAAA,EAAgB,mCAAA,EAAoC;AAAA,MAC/D,IAAA,EAAM,UAAU,QAAA;AAAS,KAC1B,CAAA;AACD,IAAA,IAAI,CAAC,SAAS,EAAA,EAAI;AAChB,MAAA,MAAM,OAAA,GAAU,MAAM,QAAA,CAAS,IAAA,EAAK;AACpC,MAAA,MAAM,IAAIA,oCAAA;AAAA,QACR,CAAA,kCAAA,EAAqC,SAAS,MAAM,CAAA,EAAA,EAAK,QAAQ,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC,CAAA,CAAA;AAAA,QAC9E,EAAE,MAAM,6BAAA;AAA8B,OACxC;AAAA,IACF;AACA,IAAA,MAAM,IAAA,GAAQ,MAAM,QAAA,CAAS,IAAA,EAAK;AAMlC,IAAA,MAAM,UAAA,GAAa,KAAK,GAAA,EAAI;AAC5B,IAAA,OAAO;AAAA,MACL,aAAa,IAAA,CAAK,YAAA;AAAA,MAClB,GAAI,KAAK,aAAA,KAAkB,KAAA,CAAA,GAAY,EAAE,YAAA,EAAc,IAAA,CAAK,aAAA,EAAc,GAAI,EAAC;AAAA;AAAA,MAE/E,SAAA,EAAW,UAAA,GAAA,CAAc,IAAA,CAAK,UAAA,IAAc,IAAA,IAAQ,GAAA;AAAA,MACpD,UAAA;AAAA,MACA,GAAI,KAAK,KAAA,KAAU,KAAA,CAAA,GAAY,EAAE,KAAA,EAAO,IAAA,CAAK,KAAA,EAAM,GAAI;AAAC,KAC1D;AAAA,EACF,CAAA,SAAE;AACA,IAAA,OAAA,EAAQ;AAAA,EACV;AACF;AASA,eAAe,qBAAA,CACb,eACA,SAAA,EAKC;AACD,EAAA,IAAI,WAAA;AACJ,EAAA,IAAI,UAAA;AACJ,EAAA,MAAM,OAAA,GAAU,IAAI,OAAA,CAAyC,CAAC,KAAK,GAAA,KAAQ;AACzE,IAAA,WAAA,GAAc,GAAA;AACd,IAAA,UAAA,GAAa,GAAA;AAAA,EACf,CAAC,CAAA;AACD,EAAA,MAAM,MAAA,GAASC,iBAAA,CAAa,CAAC,GAAA,EAAsB,GAAA,KAAwB;AACzE,IAAA,MAAM,MAAM,IAAI,GAAA,CAAI,GAAA,CAAI,GAAA,IAAO,KAAK,kBAAkB,CAAA;AACtD,IAAA,MAAM,IAAA,GAAO,GAAA,CAAI,YAAA,CAAa,GAAA,CAAI,MAAM,CAAA;AACxC,IAAA,MAAM,KAAA,GAAQ,GAAA,CAAI,YAAA,CAAa,GAAA,CAAI,OAAO,CAAA;AAC1C,IAAA,IAAI,OAAO,IAAA,KAAS,QAAA,IAAY,OAAO,UAAU,QAAA,EAAU;AACzD,MAAA,GAAA,CAAI,UAAA,GAAa,GAAA;AACjB,MAAA,GAAA,CAAI,IAAI,uBAAuB,CAAA;AAC/B,MAAA;AAAA,IACF;AAGA,IAAA,IAAI,UAAU,aAAA,EAAe;AAC3B,MAAA,GAAA,CAAI,UAAA,GAAa,GAAA;AACjB,MAAA,GAAA,CAAI,IAAI,yCAAoC,CAAA;AAC5C,MAAA;AAAA,IACF;AACA,IAAA,GAAA,CAAI,UAAA,GAAa,GAAA;AACjB,IAAA,GAAA,CAAI,IAAI,+EAA+E,CAAA;AACvF,IAAA,WAAA,GAAc,EAAE,IAAA,EAAM,KAAA,EAAO,CAAA;AAAA,EAC/B,CAAC,CAAA;AACD,EAAA,MAAA,CAAO,MAAA,CAAO,GAAG,WAAW,CAAA;AAC5B,EAAA,MAAM,aAAA,GAAgB,WAAW,MAAM;AACrC,IAAA,UAAA,GAAa,IAAID,oCAAA,CAAmB,sBAAA,EAAwB,EAAE,IAAA,EAAM,eAAA,EAAiB,CAAC,CAAA;AAAA,EACxF,GAAG,SAAS,CAAA;AAEZ,EAAA,MAAM,IAAI,OAAA,CAAc,CAAC,CAAA,KAAM;AAC7B,IAAA,IAAI,MAAA,CAAO,WAAW,CAAA,EAAE;AAAA,SACnB,MAAA,CAAO,IAAA,CAAK,WAAA,EAAa,MAAM,GAAG,CAAA;AAAA,EACzC,CAAC,CAAA;AACD,EAAA,MAAM,OAAA,GAAU,OAAO,OAAA,EAAQ;AAC/B,EAAA,IAAI,OAAO,OAAA,KAAY,QAAA,IAAY,OAAA,KAAY,IAAA,EAAM;AACnD,IAAA,MAAA,CAAO,KAAA,EAAM;AACb,IAAA,MAAM,IAAIA,qCAAmB,yCAAA,EAA2C;AAAA,MACtE,IAAA,EAAM;AAAA,KACP,CAAA;AAAA,EACH;AACA,EAAA,MAAM,OAAO,OAAA,CAAQ,IAAA;AACrB,EAAA,OAAO;AAAA,IACL,WAAA,EAAa,oBAAoB,IAAI,CAAA,SAAA,CAAA;AAAA,IACrC,OAAA;AAAA,IACA,OAAO,MAAM;AACX,MAAA,YAAA,CAAa,aAAa,CAAA;AAC1B,MAAA,MAAA,CAAO,KAAA,EAAM;AAAA,IACf;AAAA,GACF;AACF;AAQA,eAAe,iBAAA,CACb,eACA,SAAA,EAC0C;AAC1C,EAAA,MAAM,EAAA,GAAKE,yBAAgB,EAAE,KAAA,EAAO,QAAQ,KAAA,EAAO,MAAA,EAAQ,OAAA,CAAQ,MAAA,EAAQ,CAAA;AAC3E,EAAA,MAAM,OAAA,GAAU,IAAI,OAAA,CAAyC,CAAC,SAAS,MAAA,KAAW;AAChF,IAAA,EAAA,CAAG,QAAA,CAAS,kDAAkD,CAAA,CAE3D,IAAA,CAAK,CAAC,MAAA,KAAW;AAChB,MAAA,EAAA,CAAG,KAAA,EAAM;AACT,MAAA,MAAM,OAAA,GAAU,OAAO,IAAA,EAAK;AAE5B,MAAA,IAAI,OAAA,CAAQ,QAAA,CAAS,GAAG,CAAA,EAAG;AACzB,QAAA,IAAI;AACF,UAAA,MAAM,MAAA,GAAS,IAAI,GAAA,CAAI,OAAO,CAAA;AAC9B,UAAA,MAAM,IAAA,GAAO,MAAA,CAAO,YAAA,CAAa,GAAA,CAAI,MAAM,CAAA;AAC3C,UAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,YAAA,CAAa,GAAA,CAAI,OAAO,CAAA,IAAK,EAAA;AAClD,UAAA,IAAI,OAAO,SAAS,QAAA,EAAU;AAC5B,YAAA,OAAA,CAAQ,EAAE,IAAA,EAAM,KAAA,EAAO,CAAA;AACvB,YAAA;AAAA,UACF;AAAA,QACF,CAAA,CAAA,MAAQ;AAAA,QAER;AAAA,MACF;AAIA,MAAA,OAAA,CAAQ,EAAE,IAAA,EAAM,OAAA,EAAS,KAAA,EAAO,eAAe,CAAA;AAAA,IACjD,CAAC,CAAA,CACA,KAAA,CAAM,MAAM,CAAA;AACf,IAAA,UAAA,CAAW,MAAM;AACf,MAAA,EAAA,CAAG,KAAA,EAAM;AACT,MAAA,MAAA,CAAO,IAAIF,oCAAA,CAAmB,8BAAA,EAAgC,EAAE,IAAA,EAAM,eAAA,EAAiB,CAAC,CAAA;AAAA,IAC1F,GAAG,SAAS,CAAA;AAAA,EACd,CAAC,CAAA;AACD,EAAA,OAAO,OAAA;AACT","file":"mcp-auth.cjs","sourcesContent":["import { chmodSync, existsSync, mkdirSync, readFileSync } from \"node:fs\";\nimport { createRequire } from \"node:module\";\nimport { homedir } from \"node:os\";\nimport { dirname, join } from \"node:path\";\nimport { assertSecureModes } from \"../auth/credential-store.js\";\nimport { diag } from \"../diagnostics.js\";\nimport { atomicWriteJson } from \"../persistence/atomic-write.js\";\n\n/**\n * OAuth token bundle persisted per MCP server. See ADR D41.\n *\n * @internal\n */\nexport interface OAuthTokens {\n  accessToken: string;\n  refreshToken?: string;\n  expiresAt: number; // ms epoch\n  scope?: string;\n  obtainedAt: number; // ms epoch\n}\n\nconst KEYTAR_SERVICE = \"theokit-mcp\";\nconst FILE_PATH = join(homedir(), \".theokit\", \"mcp-tokens.json\");\n\n/**\n * Make the directory private BEFORE anything is written into it.\n *\n * `atomicWriteJson` auto-creates the parent with a bare recursive `mkdir`, so under the common umask\n * 002 `~/.theokit` was born 0775 — and the `chmod 600` applied to the file afterwards protects the\n * wrong thing. Write permission on a DIRECTORY is permission to unlink and recreate its contents, so\n * a 0600 file inside a group-writable directory can be replaced wholesale by another local user. The\n * secret here is a REFRESH TOKEN: replacing it swaps which account the agent authenticates as.\n *\n * The `chmod` is unconditional and not a fallback for the `mkdir` mode: `mkdir`'s mode applies only\n * at CREATION, so every machine that already ran an older build has the loose directory sitting\n * there, and a fix that only covers fresh installs does not reach the population that has the\n * problem.\n */\nfunction ensurePrivateStoreDir(): void {\n  const dir = dirname(FILE_PATH);\n  try {\n    mkdirSync(dir, { recursive: true, mode: 0o700 });\n    chmodSync(dir, 0o700);\n  } catch {\n    // Windows: chmod is a no-op and the mode argument is meaningless. Same posture as the file\n    // chmod below — documented in ADR D41 / EC-14.\n  }\n}\n\ninterface KeytarLike {\n  setPassword: (service: string, account: string, password: string) => Promise<void>;\n  getPassword: (service: string, account: string) => Promise<string | null>;\n  deletePassword: (service: string, account: string) => Promise<boolean>;\n}\n\nlet keytarWarned = false;\nlet cachedKeytar: KeytarLike | undefined | null;\nfunction tryRequireKeytar(): KeytarLike | null {\n  if (cachedKeytar !== undefined) return cachedKeytar;\n  try {\n    const r = createRequire(import.meta.url);\n    cachedKeytar = r(\"keytar\") as KeytarLike;\n  } catch {\n    if (!keytarWarned) {\n      diag(\n        \"[theokit-sdk] keytar not installed; MCP OAuth tokens will be stored in ~/.theokit/mcp-tokens.json (chmod 600 where POSIX). Install `keytar` for OS keychain storage.\\n\",\n      );\n      keytarWarned = true;\n    }\n    cachedKeytar = null;\n  }\n  return cachedKeytar;\n}\n\n// EC-9 (race): serialize concurrent refreshes per server. Each key maps to\n// the in-flight refresh Promise; subsequent callers await the same promise.\nconst inflightRefresh = new Map<string, Promise<OAuthTokens>>();\n\n/**\n * Persist tokens for `serverName`. Tries keychain first, falls back to file\n * with chmod 600 (POSIX). Idempotent.\n *\n */\nexport async function setTokens(serverName: string, tokens: OAuthTokens): Promise<void> {\n  const kt = tryRequireKeytar();\n  const payload = JSON.stringify(tokens);\n  if (kt !== null && kt !== undefined) {\n    await kt.setPassword(KEYTAR_SERVICE, serverName, payload);\n    return;\n  }\n  // File fallback. The directory is locked down first — see `ensurePrivateStoreDir`.\n  ensurePrivateStoreDir();\n  let allTokens: Record<string, OAuthTokens> = {};\n  if (existsSync(FILE_PATH)) {\n    try {\n      allTokens = JSON.parse(readFileSync(FILE_PATH, \"utf8\")) as Record<string, OAuthTokens>;\n    } catch {\n      // corrupt file — start fresh\n      allTokens = {};\n    }\n  }\n  allTokens[serverName] = tokens;\n  await atomicWriteJson(FILE_PATH, allTokens);\n  try {\n    chmodSync(FILE_PATH, 0o600);\n  } catch {\n    // Windows: chmod is a no-op. Documented in ADR D41 / EC-14.\n  }\n}\n\n/**\n * Retrieve tokens for `serverName`. Returns undefined if absent.\n *\n */\nexport async function getTokens(serverName: string): Promise<OAuthTokens | undefined> {\n  const kt = tryRequireKeytar();\n  if (kt !== null && kt !== undefined) {\n    const value = await kt.getPassword(KEYTAR_SERVICE, serverName);\n    if (typeof value !== \"string\") return undefined;\n    try {\n      return JSON.parse(value) as OAuthTokens;\n    } catch {\n      return undefined;\n    }\n  }\n  if (!existsSync(FILE_PATH)) return undefined;\n  // The same gate the credential file gets, and deliberately the same implementation rather than a\n  // second one: a refresh token is a credential, and `assertSecureModes` already carries the attack\n  // it defends against in its docstring. It is called OUTSIDE the try/catch on purpose — the catch\n  // below exists to treat a corrupt file as absent, and letting it swallow this would turn \"someone\n  // else can replace your token\" into a silent `undefined`, which reads as \"not logged in\".\n  assertSecureModes(dirname(FILE_PATH), FILE_PATH);\n  try {\n    const all = JSON.parse(readFileSync(FILE_PATH, \"utf8\")) as Record<string, OAuthTokens>;\n    return all[serverName];\n  } catch {\n    return undefined;\n  }\n}\n\n/**\n * Serialize concurrent refresh attempts per server (EC-9). If a refresh is\n * already in flight for the same `serverName`, the second caller awaits\n * the same Promise.\n *\n */\nexport function lockedRefresh(\n  serverName: string,\n  refreshFn: () => Promise<OAuthTokens>,\n): Promise<OAuthTokens> {\n  const existing = inflightRefresh.get(serverName);\n  if (existing !== undefined) return existing;\n  const promise = (async () => {\n    try {\n      return await refreshFn();\n    } finally {\n      inflightRefresh.delete(serverName);\n    }\n  })();\n  inflightRefresh.set(serverName, promise);\n  return promise;\n}\n\n/**\n * Reset all cached state. Test-only helper.\n *\n * @internal\n */\nexport function _resetForTests(): void {\n  inflightRefresh.clear();\n  cachedKeytar = undefined;\n  keytarWarned = false;\n}\n","import { createHash, randomBytes } from \"node:crypto\";\nimport { createServer, type IncomingMessage, type ServerResponse } from \"node:http\";\nimport { createInterface } from \"node:readline/promises\";\n\nimport { ConfigurationError } from \"../../errors.js\";\nimport type { McpOAuthConfig } from \"../../types/mcp.js\";\nimport { diag } from \"../diagnostics.js\";\nimport { type OAuthTokens, setTokens } from \"./token-storage.js\";\n\n/**\n * OAuth 2.1 PKCE flow runner for MCP HTTP servers. See ADR D41.\n *\n * @internal\n */\n\n/**\n * Generate a random base64url-encoded string. Used for both code_verifier\n * (43-128 chars) and state (CSRF protection).\n */\nfunction base64url(n: number): string {\n  return randomBytes(n).toString(\"base64url\");\n}\n\n/**\n * Compute PKCE code_challenge = base64url(SHA256(code_verifier)).\n */\nfunction challengeFor(verifier: string): string {\n  return createHash(\"sha256\").update(verifier).digest(\"base64url\");\n}\n\n/**\n * Refresh an access token using `refresh_token`. Throws on failure.\n *\n */\nexport async function refreshAccessToken(\n  serverName: string,\n  clientId: string,\n  refreshToken: string,\n  oauth: McpOAuthConfig,\n): Promise<OAuthTokens> {\n  const body = new URLSearchParams({\n    grant_type: \"refresh_token\",\n    refresh_token: refreshToken,\n    client_id: clientId,\n  });\n  const response = await fetch(oauth.tokenEndpoint, {\n    method: \"POST\",\n    headers: { \"Content-Type\": \"application/x-www-form-urlencoded\" },\n    body: body.toString(),\n  });\n  if (!response.ok) {\n    throw new ConfigurationError(`OAuth refresh failed: HTTP ${response.status}`, {\n      code: \"oauth_refresh_failed\",\n    });\n  }\n  const data = (await response.json()) as {\n    access_token: string;\n    refresh_token?: string;\n    expires_in?: number;\n    scope?: string;\n  };\n  const obtainedAt = Date.now();\n  const tokens: OAuthTokens = {\n    accessToken: data.access_token,\n    refreshToken: data.refresh_token ?? refreshToken,\n    // EC-10: default conservative 3600s if expires_in missing.\n    expiresAt: obtainedAt + (data.expires_in ?? 3600) * 1000,\n    obtainedAt,\n    ...(data.scope !== undefined ? { scope: data.scope } : {}),\n  };\n  await setTokens(serverName, tokens);\n  return tokens;\n}\n\n/**\n * Run the full PKCE flow. Generates verifier+challenge+state, opens the\n * authorization URL (browser or manual print), waits for the callback\n * (localhost or stdin), exchanges code for tokens.\n *\n */\n// biome-ignore lint/complexity/noExcessiveCognitiveComplexity: PKCE flow is a linear protocol step (verifier → challenge → state → URL → wait → exchange); refactoring fragments the RFC 7636 narrative.\nexport async function runPkceFlow(\n  serverName: string,\n  clientId: string,\n  scopes: string[] | undefined,\n  oauth: McpOAuthConfig,\n): Promise<OAuthTokens> {\n  const verifier = base64url(48);\n  const challenge = challengeFor(verifier);\n  const state = base64url(16);\n  const timeoutMs = oauth.timeoutMs ?? 300_000;\n\n  let redirectUri: string;\n  let codeWaiter: Promise<{ code: string; state: string }>;\n  let cleanup: () => void = () => {};\n\n  if (oauth.redirectMode === \"localhost\") {\n    const result = await openLocalhostCallback(state, timeoutMs);\n    redirectUri = result.redirectUri;\n    codeWaiter = result.waitFor;\n    cleanup = result.close;\n  } else {\n    redirectUri = \"urn:ietf:wg:oauth:2.0:oob\";\n    codeWaiter = readCodeFromStdin(state, timeoutMs);\n  }\n\n  try {\n    const authUrl = new URL(oauth.authorizationEndpoint);\n    authUrl.searchParams.set(\"response_type\", \"code\");\n    authUrl.searchParams.set(\"client_id\", clientId);\n    authUrl.searchParams.set(\"redirect_uri\", redirectUri);\n    if (scopes !== undefined && scopes.length > 0) {\n      authUrl.searchParams.set(\"scope\", scopes.join(\" \"));\n    }\n    authUrl.searchParams.set(\"code_challenge\", challenge);\n    authUrl.searchParams.set(\"code_challenge_method\", \"S256\");\n    authUrl.searchParams.set(\"state\", state);\n\n    diag(\n      `\\n[theokit-sdk] OAuth required for MCP server \"${serverName}\".\\n` +\n        `Open this URL to authenticate:\\n  ${authUrl.toString()}\\n\\n`,\n    );\n\n    const { code, state: returnedState } = await codeWaiter;\n    // EC-2 MUST FIX: validate state.\n    if (returnedState !== state) {\n      throw new ConfigurationError(\n        \"OAuth state mismatch — possible CSRF attempt; aborting token exchange.\",\n        { code: \"oauth_state_mismatch\" },\n      );\n    }\n\n    // Exchange code for tokens.\n    const tokenBody = new URLSearchParams({\n      grant_type: \"authorization_code\",\n      code,\n      client_id: clientId,\n      code_verifier: verifier,\n      redirect_uri: redirectUri,\n    });\n    const response = await fetch(oauth.tokenEndpoint, {\n      method: \"POST\",\n      headers: { \"Content-Type\": \"application/x-www-form-urlencoded\" },\n      body: tokenBody.toString(),\n    });\n    if (!response.ok) {\n      const errText = await response.text();\n      throw new ConfigurationError(\n        `OAuth token exchange failed: HTTP ${response.status}: ${errText.slice(0, 200)}`,\n        { code: \"oauth_token_exchange_failed\" },\n      );\n    }\n    const data = (await response.json()) as {\n      access_token: string;\n      refresh_token?: string;\n      expires_in?: number;\n      scope?: string;\n    };\n    const obtainedAt = Date.now();\n    return {\n      accessToken: data.access_token,\n      ...(data.refresh_token !== undefined ? { refreshToken: data.refresh_token } : {}),\n      // EC-10: default 3600s when expires_in absent.\n      expiresAt: obtainedAt + (data.expires_in ?? 3600) * 1000,\n      obtainedAt,\n      ...(data.scope !== undefined ? { scope: data.scope } : {}),\n    };\n  } finally {\n    cleanup();\n  }\n}\n\n/**\n * Start a localhost HTTP server on a free port; return the redirect URI,\n * a promise that resolves with the code+state from the first valid request,\n * and a close handle.\n *\n * @internal\n */\nasync function openLocalhostCallback(\n  expectedState: string,\n  timeoutMs: number,\n): Promise<{\n  redirectUri: string;\n  waitFor: Promise<{ code: string; state: string }>;\n  close: () => void;\n}> {\n  let resolveCode: ((value: { code: string; state: string }) => void) | undefined;\n  let rejectCode: ((reason: unknown) => void) | undefined;\n  const waitFor = new Promise<{ code: string; state: string }>((res, rej) => {\n    resolveCode = res;\n    rejectCode = rej;\n  });\n  const server = createServer((req: IncomingMessage, res: ServerResponse) => {\n    const url = new URL(req.url ?? \"/\", \"http://127.0.0.1\");\n    const code = url.searchParams.get(\"code\");\n    const state = url.searchParams.get(\"state\");\n    if (typeof code !== \"string\" || typeof state !== \"string\") {\n      res.statusCode = 400;\n      res.end(\"Missing code or state\");\n      return;\n    }\n    // EC-2: state must match the one we generated. We DO NOT trust any\n    // localhost GET — only the one carrying our exact state.\n    if (state !== expectedState) {\n      res.statusCode = 400;\n      res.end(\"State mismatch — request rejected.\");\n      return;\n    }\n    res.statusCode = 200;\n    res.end(\"Authentication successful. You can close this tab and return to the terminal.\");\n    resolveCode?.({ code, state });\n  });\n  server.listen(0, \"127.0.0.1\");\n  const timeoutHandle = setTimeout(() => {\n    rejectCode?.(new ConfigurationError(\"OAuth flow timed out\", { code: \"oauth_timeout\" }));\n  }, timeoutMs);\n  // Wait for server.listen to bind so we can read the port.\n  await new Promise<void>((r) => {\n    if (server.listening) r();\n    else server.once(\"listening\", () => r());\n  });\n  const address = server.address();\n  if (typeof address !== \"object\" || address === null) {\n    server.close();\n    throw new ConfigurationError(\"OAuth localhost callback failed to bind\", {\n      code: \"oauth_bind_failed\",\n    });\n  }\n  const port = address.port;\n  return {\n    redirectUri: `http://127.0.0.1:${port}/callback`,\n    waitFor,\n    close: () => {\n      clearTimeout(timeoutHandle);\n      server.close();\n    },\n  };\n}\n\n/**\n * Read `code` (and `state`) from stdin (manual mode). User pastes the\n * full callback URL or just the code value.\n *\n * @internal\n */\nasync function readCodeFromStdin(\n  expectedState: string,\n  timeoutMs: number,\n): Promise<{ code: string; state: string }> {\n  const rl = createInterface({ input: process.stdin, output: process.stderr });\n  const promise = new Promise<{ code: string; state: string }>((resolve, reject) => {\n    rl.question(\"Paste the full callback URL (or just the code): \")\n      // biome-ignore lint/complexity/noExcessiveCognitiveComplexity: stdin reader handles 2 input formats (full URL vs raw code) inline; splitting fragments the manual-mode UX.\n      .then((answer) => {\n        rl.close();\n        const trimmed = answer.trim();\n        // Try to parse as URL first.\n        if (trimmed.includes(\"?\")) {\n          try {\n            const parsed = new URL(trimmed);\n            const code = parsed.searchParams.get(\"code\");\n            const state = parsed.searchParams.get(\"state\") ?? \"\";\n            if (typeof code === \"string\") {\n              resolve({ code, state });\n              return;\n            }\n          } catch {\n            // not a URL — fall through\n          }\n        }\n        // Treat the entire input as the code. State validation will fail\n        // unless the user also provided state — but in manual mode they\n        // typically paste the URL with state baked in.\n        resolve({ code: trimmed, state: expectedState });\n      })\n      .catch(reject);\n    setTimeout(() => {\n      rl.close();\n      reject(new ConfigurationError(\"OAuth manual input timed out\", { code: \"oauth_timeout\" }));\n    }, timeoutMs);\n  });\n  return promise;\n}\n"]}

package/dist/mcp-auth.d.cts

@@ -0,0 +1,17 @@
+/**
+ * OAuth for remote MCP servers: the PKCE authorization flow, token refresh, and the storage the two
+ * of them need.
+ *
+ * The implementation has existed and been tested at `internal/mcp/` for some time; what it lacked
+ * was a way in. A consumer connecting to an MCP server that requires OAuth had to write RFC 7636
+ * PKCE by hand — not because the package lacked the code, but because nothing exported it.
+ *
+ * `lockedRefresh` is exported alongside deliberately: two callers noticing an expired token at the
+ * same moment will both refresh, and with a rotating refresh token the second one loses. That is the
+ * kind of detail a hand-written flow discovers in production.
+ *
+ * `_resetForTests` is deliberately NOT re-exported — it is module-state surgery for this repo's own
+ * tests, and publishing it invites a consumer to clear a cache the package is responsible for.
+ */
+export { refreshAccessToken, runPkceFlow } from "./internal/mcp/oauth.js";
+export { getTokens, lockedRefresh, type OAuthTokens, setTokens, } from "./internal/mcp/token-storage.js";

package/dist/mcp-auth.d.ts

@@ -0,0 +1,17 @@
+/**
+ * OAuth for remote MCP servers: the PKCE authorization flow, token refresh, and the storage the two
+ * of them need.
+ *
+ * The implementation has existed and been tested at `internal/mcp/` for some time; what it lacked
+ * was a way in. A consumer connecting to an MCP server that requires OAuth had to write RFC 7636
+ * PKCE by hand — not because the package lacked the code, but because nothing exported it.
+ *
+ * `lockedRefresh` is exported alongside deliberately: two callers noticing an expired token at the
+ * same moment will both refresh, and with a rotating refresh token the second one loses. That is the
+ * kind of detail a hand-written flow discovers in production.
+ *
+ * `_resetForTests` is deliberately NOT re-exported — it is module-state surgery for this repo's own
+ * tests, and publishing it invites a consumer to clear a cache the package is responsible for.
+ */
+export { refreshAccessToken, runPkceFlow } from "./internal/mcp/oauth.js";
+export { getTokens, lockedRefresh, type OAuthTokens, setTokens, } from "./internal/mcp/token-storage.js";

package/dist/mcp-auth.js

@@ -0,0 +1,292 @@
+import { atomicWriteJson } from './chunk-6AZDM6LK.js';
+import { assertSecureModes } from './chunk-3V7UJLQT.js';
+import { ConfigurationError } from './chunk-U4UJHOSH.js';
+import { diag } from './chunk-5F5NWKPN.js';
+import { randomBytes, createHash } from 'crypto';
+import { createServer } from 'http';
+import { createInterface } from 'readline/promises';
+import { existsSync, readFileSync, chmodSync, mkdirSync } from 'fs';
+import { createRequire } from 'module';
+import { homedir } from 'os';
+import { join, dirname } from 'path';
+
+var KEYTAR_SERVICE = "theokit-mcp";
+var FILE_PATH = join(homedir(), ".theokit", "mcp-tokens.json");
+function ensurePrivateStoreDir() {
+  const dir = dirname(FILE_PATH);
+  try {
+    mkdirSync(dir, { recursive: true, mode: 448 });
+    chmodSync(dir, 448);
+  } catch {
+  }
+}
+var keytarWarned = false;
+var cachedKeytar;
+function tryRequireKeytar() {
+  if (cachedKeytar !== void 0) return cachedKeytar;
+  try {
+    const r = createRequire(import.meta.url);
+    cachedKeytar = r("keytar");
+  } catch {
+    if (!keytarWarned) {
+      diag(
+        "[theokit-sdk] keytar not installed; MCP OAuth tokens will be stored in ~/.theokit/mcp-tokens.json (chmod 600 where POSIX). Install `keytar` for OS keychain storage.\n"
+      );
+      keytarWarned = true;
+    }
+    cachedKeytar = null;
+  }
+  return cachedKeytar;
+}
+var inflightRefresh = /* @__PURE__ */ new Map();
+async function setTokens(serverName, tokens) {
+  const kt = tryRequireKeytar();
+  const payload = JSON.stringify(tokens);
+  if (kt !== null && kt !== void 0) {
+    await kt.setPassword(KEYTAR_SERVICE, serverName, payload);
+    return;
+  }
+  ensurePrivateStoreDir();
+  let allTokens = {};
+  if (existsSync(FILE_PATH)) {
+    try {
+      allTokens = JSON.parse(readFileSync(FILE_PATH, "utf8"));
+    } catch {
+      allTokens = {};
+    }
+  }
+  allTokens[serverName] = tokens;
+  await atomicWriteJson(FILE_PATH, allTokens);
+  try {
+    chmodSync(FILE_PATH, 384);
+  } catch {
+  }
+}
+async function getTokens(serverName) {
+  const kt = tryRequireKeytar();
+  if (kt !== null && kt !== void 0) {
+    const value = await kt.getPassword(KEYTAR_SERVICE, serverName);
+    if (typeof value !== "string") return void 0;
+    try {
+      return JSON.parse(value);
+    } catch {
+      return void 0;
+    }
+  }
+  if (!existsSync(FILE_PATH)) return void 0;
+  assertSecureModes(dirname(FILE_PATH), FILE_PATH);
+  try {
+    const all = JSON.parse(readFileSync(FILE_PATH, "utf8"));
+    return all[serverName];
+  } catch {
+    return void 0;
+  }
+}
+function lockedRefresh(serverName, refreshFn) {
+  const existing = inflightRefresh.get(serverName);
+  if (existing !== void 0) return existing;
+  const promise = (async () => {
+    try {
+      return await refreshFn();
+    } finally {
+      inflightRefresh.delete(serverName);
+    }
+  })();
+  inflightRefresh.set(serverName, promise);
+  return promise;
+}
+
+// src/internal/mcp/oauth.ts
+function base64url(n) {
+  return randomBytes(n).toString("base64url");
+}
+function challengeFor(verifier) {
+  return createHash("sha256").update(verifier).digest("base64url");
+}
+async function refreshAccessToken(serverName, clientId, refreshToken, oauth) {
+  const body = new URLSearchParams({
+    grant_type: "refresh_token",
+    refresh_token: refreshToken,
+    client_id: clientId
+  });
+  const response = await fetch(oauth.tokenEndpoint, {
+    method: "POST",
+    headers: { "Content-Type": "application/x-www-form-urlencoded" },
+    body: body.toString()
+  });
+  if (!response.ok) {
+    throw new ConfigurationError(`OAuth refresh failed: HTTP ${response.status}`, {
+      code: "oauth_refresh_failed"
+    });
+  }
+  const data = await response.json();
+  const obtainedAt = Date.now();
+  const tokens = {
+    accessToken: data.access_token,
+    refreshToken: data.refresh_token ?? refreshToken,
+    // EC-10: default conservative 3600s if expires_in missing.
+    expiresAt: obtainedAt + (data.expires_in ?? 3600) * 1e3,
+    obtainedAt,
+    ...data.scope !== void 0 ? { scope: data.scope } : {}
+  };
+  await setTokens(serverName, tokens);
+  return tokens;
+}
+async function runPkceFlow(serverName, clientId, scopes, oauth) {
+  const verifier = base64url(48);
+  const challenge = challengeFor(verifier);
+  const state = base64url(16);
+  const timeoutMs = oauth.timeoutMs ?? 3e5;
+  let redirectUri;
+  let codeWaiter;
+  let cleanup = () => {
+  };
+  if (oauth.redirectMode === "localhost") {
+    const result = await openLocalhostCallback(state, timeoutMs);
+    redirectUri = result.redirectUri;
+    codeWaiter = result.waitFor;
+    cleanup = result.close;
+  } else {
+    redirectUri = "urn:ietf:wg:oauth:2.0:oob";
+    codeWaiter = readCodeFromStdin(state, timeoutMs);
+  }
+  try {
+    const authUrl = new URL(oauth.authorizationEndpoint);
+    authUrl.searchParams.set("response_type", "code");
+    authUrl.searchParams.set("client_id", clientId);
+    authUrl.searchParams.set("redirect_uri", redirectUri);
+    if (scopes !== void 0 && scopes.length > 0) {
+      authUrl.searchParams.set("scope", scopes.join(" "));
+    }
+    authUrl.searchParams.set("code_challenge", challenge);
+    authUrl.searchParams.set("code_challenge_method", "S256");
+    authUrl.searchParams.set("state", state);
+    diag(
+      `
+[theokit-sdk] OAuth required for MCP server "${serverName}".
+Open this URL to authenticate:
+  ${authUrl.toString()}
+
+`
+    );
+    const { code, state: returnedState } = await codeWaiter;
+    if (returnedState !== state) {
+      throw new ConfigurationError(
+        "OAuth state mismatch \u2014 possible CSRF attempt; aborting token exchange.",
+        { code: "oauth_state_mismatch" }
+      );
+    }
+    const tokenBody = new URLSearchParams({
+      grant_type: "authorization_code",
+      code,
+      client_id: clientId,
+      code_verifier: verifier,
+      redirect_uri: redirectUri
+    });
+    const response = await fetch(oauth.tokenEndpoint, {
+      method: "POST",
+      headers: { "Content-Type": "application/x-www-form-urlencoded" },
+      body: tokenBody.toString()
+    });
+    if (!response.ok) {
+      const errText = await response.text();
+      throw new ConfigurationError(
+        `OAuth token exchange failed: HTTP ${response.status}: ${errText.slice(0, 200)}`,
+        { code: "oauth_token_exchange_failed" }
+      );
+    }
+    const data = await response.json();
+    const obtainedAt = Date.now();
+    return {
+      accessToken: data.access_token,
+      ...data.refresh_token !== void 0 ? { refreshToken: data.refresh_token } : {},
+      // EC-10: default 3600s when expires_in absent.
+      expiresAt: obtainedAt + (data.expires_in ?? 3600) * 1e3,
+      obtainedAt,
+      ...data.scope !== void 0 ? { scope: data.scope } : {}
+    };
+  } finally {
+    cleanup();
+  }
+}
+async function openLocalhostCallback(expectedState, timeoutMs) {
+  let resolveCode;
+  let rejectCode;
+  const waitFor = new Promise((res, rej) => {
+    resolveCode = res;
+    rejectCode = rej;
+  });
+  const server = createServer((req, res) => {
+    const url = new URL(req.url ?? "/", "http://127.0.0.1");
+    const code = url.searchParams.get("code");
+    const state = url.searchParams.get("state");
+    if (typeof code !== "string" || typeof state !== "string") {
+      res.statusCode = 400;
+      res.end("Missing code or state");
+      return;
+    }
+    if (state !== expectedState) {
+      res.statusCode = 400;
+      res.end("State mismatch \u2014 request rejected.");
+      return;
+    }
+    res.statusCode = 200;
+    res.end("Authentication successful. You can close this tab and return to the terminal.");
+    resolveCode?.({ code, state });
+  });
+  server.listen(0, "127.0.0.1");
+  const timeoutHandle = setTimeout(() => {
+    rejectCode?.(new ConfigurationError("OAuth flow timed out", { code: "oauth_timeout" }));
+  }, timeoutMs);
+  await new Promise((r) => {
+    if (server.listening) r();
+    else server.once("listening", () => r());
+  });
+  const address = server.address();
+  if (typeof address !== "object" || address === null) {
+    server.close();
+    throw new ConfigurationError("OAuth localhost callback failed to bind", {
+      code: "oauth_bind_failed"
+    });
+  }
+  const port = address.port;
+  return {
+    redirectUri: `http://127.0.0.1:${port}/callback`,
+    waitFor,
+    close: () => {
+      clearTimeout(timeoutHandle);
+      server.close();
+    }
+  };
+}
+async function readCodeFromStdin(expectedState, timeoutMs) {
+  const rl = createInterface({ input: process.stdin, output: process.stderr });
+  const promise = new Promise((resolve, reject) => {
+    rl.question("Paste the full callback URL (or just the code): ").then((answer) => {
+      rl.close();
+      const trimmed = answer.trim();
+      if (trimmed.includes("?")) {
+        try {
+          const parsed = new URL(trimmed);
+          const code = parsed.searchParams.get("code");
+          const state = parsed.searchParams.get("state") ?? "";
+          if (typeof code === "string") {
+            resolve({ code, state });
+            return;
+          }
+        } catch {
+        }
+      }
+      resolve({ code: trimmed, state: expectedState });
+    }).catch(reject);
+    setTimeout(() => {
+      rl.close();
+      reject(new ConfigurationError("OAuth manual input timed out", { code: "oauth_timeout" }));
+    }, timeoutMs);
+  });
+  return promise;
+}
+
+export { getTokens, lockedRefresh, refreshAccessToken, runPkceFlow, setTokens };
+//# sourceMappingURL=mcp-auth.js.map
+//# sourceMappingURL=mcp-auth.js.map