@mastra/code-sdk 1.0.3-alpha.0 → 1.1.0-alpha.2

package/dist/mcp/manager.js

@@ -1,487 +1,494 @@
-import { existsSync, mkdirSync, readFileSync, writeFileSync, renameSync } from "node:fs";
-import { dirname, join } from "node:path";
-import { MCPClient, MCPOAuthClientProvider } from "@mastra/mcp";
 import { DEFAULT_CONFIG_DIR } from "../constants.js";
 import { getAppDataDir } from "../utils/project.js";
-import {
-  DEFAULT_OAUTH_REDIRECT_URL,
-  loadMcpConfig,
-  getProjectMcpPath,
-  getGlobalMcpPath,
-  getClaudeSettingsPath,
-  resolveOAuthRedirectUrl
-} from "./config.js";
-const MASTRACODE_MCP_TIMEOUT_MS = 7 * 24 * 60 * 60 * 1e3;
+import { DEFAULT_OAUTH_REDIRECT_URL, getClaudeSettingsPath, getGlobalMcpPath, getProjectMcpPath, loadMcpConfig, resolveOAuthRedirectUrl } from "./config.js";
+import { existsSync, mkdirSync, readFileSync, renameSync, writeFileSync } from "fs";
+import { dirname, join } from "path";
+import { MCPClient, MCPOAuthClientProvider } from "@mastra/mcp";
+//#region src/mcp/manager.ts
+/**
+* MCP manager — orchestrates MCP server connections using MCPClient directly.
+* Created once at startup, provides tools from connected MCP servers.
+*/
+const MASTRACODE_MCP_TIMEOUT_MS = 10080 * 60 * 1e3;
 function getTransport(cfg) {
-  return "url" in cfg ? "http" : "stdio";
-}
-class FileOAuthStorage {
-  constructor(filePath) {
-    this.filePath = filePath;
-  }
-  filePath;
-  get(key) {
-    return this.read()[key];
-  }
-  set(key, value) {
-    const data = this.read();
-    data[key] = value;
-    this.write(data);
-  }
-  delete(key) {
-    const data = this.read();
-    delete data[key];
-    this.write(data);
-  }
-  read() {
-    if (!existsSync(this.filePath)) return {};
-    try {
-      return JSON.parse(readFileSync(this.filePath, "utf-8"));
-    } catch {
-      return {};
-    }
-  }
-  write(data) {
-    const dir = dirname(this.filePath);
-    if (!existsSync(dir)) {
-      mkdirSync(dir, { recursive: true, mode: 448 });
-    }
-    const tmpPath = `${this.filePath}.tmp`;
-    writeFileSync(tmpPath, JSON.stringify(data, null, 2), { encoding: "utf-8", mode: 384 });
-    renameSync(tmpPath, this.filePath);
-  }
+	return "url" in cfg ? "http" : "stdio";
 }
+var FileOAuthStorage = class {
+	filePath;
+	constructor(filePath) {
+		this.filePath = filePath;
+	}
+	get(key) {
+		return this.read()[key];
+	}
+	set(key, value) {
+		const data = this.read();
+		data[key] = value;
+		this.write(data);
+	}
+	delete(key) {
+		const data = this.read();
+		delete data[key];
+		this.write(data);
+	}
+	read() {
+		if (!existsSync(this.filePath)) return {};
+		try {
+			return JSON.parse(readFileSync(this.filePath, "utf-8"));
+		} catch {
+			return {};
+		}
+	}
+	write(data) {
+		const dir = dirname(this.filePath);
+		if (!existsSync(dir)) mkdirSync(dir, {
+			recursive: true,
+			mode: 448
+		});
+		const tmpPath = `${this.filePath}.tmp`;
+		writeFileSync(tmpPath, JSON.stringify(data, null, 2), {
+			encoding: "utf-8",
+			mode: 384
+		});
+		renameSync(tmpPath, this.filePath);
+	}
+};
+/**
+* Zero-config OAuth defaults for servers with a bare `url` entry. Dynamic
+* client registration provisions the client, so no `clientId` is needed.
+*/
 const DEFAULT_OAUTH_CONFIG = { redirectUrl: DEFAULT_OAUTH_REDIRECT_URL };
 function getOAuthStoragePath(projectDir, name, cfg) {
-  const key = JSON.stringify({
-    projectDir,
-    name,
-    url: cfg.url,
-    redirectUrl: resolveOAuthRedirectUrl(cfg.oauth),
-    clientId: cfg.oauth?.clientId,
-    scopes: cfg.oauth?.scopes ?? []
-  });
-  return join(getAppDataDir(), "mcp-oauth", `${getStorageKeyFingerprint(key)}.json`);
+	const key = JSON.stringify({
+		projectDir,
+		name,
+		url: cfg.url,
+		redirectUrl: resolveOAuthRedirectUrl(cfg.oauth),
+		clientId: cfg.oauth?.clientId,
+		scopes: cfg.oauth?.scopes ?? []
+	});
+	return join(getAppDataDir(), "mcp-oauth", `${getStorageKeyFingerprint(key)}.json`);
 }
 function getStorageKeyFingerprint(value) {
-  let fingerprint = 0xcbf29ce484222325n;
-  for (let i = 0; i < value.length; i += 1) {
-    fingerprint ^= BigInt(value.charCodeAt(i));
-    fingerprint = BigInt.asUintN(64, fingerprint * 0x100000001b3n);
-  }
-  return fingerprint.toString(16).padStart(16, "0");
+	let fingerprint = 14695981039346656037n;
+	for (let i = 0; i < value.length; i += 1) {
+		fingerprint ^= BigInt(value.charCodeAt(i));
+		fingerprint = BigInt.asUintN(64, fingerprint * 1099511628211n);
+	}
+	return fingerprint.toString(16).padStart(16, "0");
 }
+/**
+* Create an MCP manager that wraps MCPClient with config-file discovery
+* and per-server status tracking.
+*/
 function createMcpManager(projectDir, configDirName = DEFAULT_CONFIG_DIR, extraServers) {
-  const applyExtraServers = (base) => {
-    if (!extraServers || Object.keys(extraServers).length === 0) return base;
-    return { ...base, mcpServers: { ...base.mcpServers, ...extraServers } };
-  };
-  let config = applyExtraServers(loadMcpConfig(projectDir, configDirName));
-  let client = null;
-  let serverDefs = {};
-  let tools = {};
-  let serverStatuses = /* @__PURE__ */ new Map();
-  let stderrLogs = /* @__PURE__ */ new Map();
-  let initialized = false;
-  const authUrlHandlers = /* @__PURE__ */ new Map();
-  const authenticatingServers = /* @__PURE__ */ new Set();
-  const cancelledAuthServers = /* @__PURE__ */ new Set();
-  const withAuthenticating = (status) => authenticatingServers.has(status.name) ? { ...status, authenticating: true } : status;
-  const MAX_STDERR_LINES = 200;
-  function captureStderr(serverName) {
-    if (!client || typeof client.getServerStderr !== "function") return;
-    const stream = client.getServerStderr(serverName);
-    if (!stream) return;
-    let buffer = "";
-    const lines = stderrLogs.get(serverName) ?? [];
-    stderrLogs.set(serverName, lines);
-    stream.on("data", (chunk) => {
-      buffer += chunk.toString();
-      const parts = buffer.split("\n");
-      buffer = parts.pop();
-      for (const line of parts) {
-        if (line.trim()) {
-          lines.push(line);
-          if (lines.length > MAX_STDERR_LINES) {
-            lines.shift();
-          }
-        }
-      }
-    });
-    stream.on("end", () => {
-      if (buffer.trim()) {
-        lines.push(buffer);
-        if (lines.length > MAX_STDERR_LINES) {
-          lines.shift();
-        }
-      }
-    });
-  }
-  function createOAuthProvider(name, cfg) {
-    const oauth = cfg.oauth ?? (existsSync(getOAuthStoragePath(projectDir, name, cfg)) ? DEFAULT_OAUTH_CONFIG : void 0);
-    if (!oauth) return void 0;
-    const redirectUrl = resolveOAuthRedirectUrl(oauth);
-    return new MCPOAuthClientProvider({
-      redirectUrl,
-      clientMetadata: {
-        redirect_uris: [redirectUrl],
-        client_name: oauth.clientName ?? `Mastra Code MCP ${name}`,
-        grant_types: ["authorization_code", "refresh_token"],
-        response_types: ["code"],
-        ...oauth.scopes?.length ? { scope: oauth.scopes.join(" ") } : {}
-      },
-      clientInformation: oauth.clientId ? {
-        client_id: oauth.clientId,
-        ...oauth.clientSecret ? { client_secret: oauth.clientSecret } : {}
-      } : void 0,
-      storage: new FileOAuthStorage(getOAuthStoragePath(projectDir, name, cfg)),
-      onRedirectToAuthorization: (url) => {
-        authUrlHandlers.get(name)?.(url.toString());
-      }
-    });
-  }
-  function buildServerDefs(servers) {
-    const defs = {};
-    for (const [name, cfg] of Object.entries(servers)) {
-      if ("url" in cfg) {
-        const httpCfg = cfg;
-        defs[name] = {
-          url: new URL(httpCfg.url),
-          requestInit: httpCfg.headers ? { headers: httpCfg.headers } : void 0,
-          authProvider: createOAuthProvider(name, httpCfg)
-        };
-      } else {
-        defs[name] = { command: cfg.command, args: cfg.args, env: cfg.env, stderr: "pipe" };
-      }
-    }
-    return defs;
-  }
-  async function connectAndCollectTools() {
-    const servers = config.mcpServers;
-    if (!servers || Object.keys(servers).length === 0) {
-      return;
-    }
-    const serverNames = Object.keys(servers);
-    for (const name of serverNames) {
-      serverStatuses.set(name, {
-        name,
-        connected: false,
-        connecting: true,
-        toolCount: 0,
-        toolNames: [],
-        transport: getTransport(servers[name])
-      });
-    }
-    serverDefs = buildServerDefs(servers);
-    client = new MCPClient({
-      id: "mastra-code-mcp",
-      servers: serverDefs,
-      timeout: MASTRACODE_MCP_TIMEOUT_MS
-    });
-    try {
-      const { toolsets, errors } = await client.listToolsetsWithErrors();
-      const typedToolsets = toolsets;
-      for (const [serverName, serverTools] of Object.entries(typedToolsets)) {
-        for (const [toolName, toolConfig] of Object.entries(serverTools)) {
-          tools[`${serverName}_${toolName}`] = toolConfig;
-        }
-      }
-      for (const name of serverNames) {
-        const serverTools = typedToolsets[name];
-        if (serverTools && Object.keys(serverTools).length > 0) {
-          const toolNames = Object.keys(serverTools).map((t) => `${name}_${t}`);
-          serverStatuses.set(name, {
-            name,
-            connected: true,
-            toolCount: toolNames.length,
-            toolNames,
-            transport: getTransport(servers[name])
-          });
-        } else {
-          const error = errors[name] ?? "Failed to connect";
-          serverStatuses.set(name, {
-            name,
-            connected: false,
-            toolCount: 0,
-            toolNames: [],
-            transport: getTransport(servers[name]),
-            error,
-            ...serverNeedsAuth(name, servers[name], error) ? { needsAuth: true } : {}
-          });
-        }
-      }
-      for (const name of serverNames) {
-        captureStderr(name);
-      }
-    } catch (error) {
-      const errMsg = error instanceof Error ? error.message : String(error);
-      for (const name of serverNames) {
-        serverStatuses.set(name, {
-          name,
-          connected: false,
-          toolCount: 0,
-          toolNames: [],
-          transport: getTransport(servers[name]),
-          error: errMsg,
-          ...serverNeedsAuth(name, servers[name], errMsg) ? { needsAuth: true } : {}
-        });
-      }
-    }
-  }
-  function serverNeedsAuth(name, cfg, error) {
-    if (getTransport(cfg) !== "http") return false;
-    if (client?.getServerAuthState?.(name) === "needs-auth") return true;
-    if (serverDefs[name]?.authProvider) return false;
-    return error !== void 0 && /\b401\b|unauthorized|invalid_token/i.test(error);
-  }
-  async function connectSingleServer(name, cfg, connect) {
-    const transport = getTransport(cfg);
-    const prefix = `${name}_`;
-    for (const key of Object.keys(tools)) {
-      if (key.startsWith(prefix)) {
-        delete tools[key];
-      }
-    }
-    stderrLogs.delete(name);
-    serverStatuses.set(name, {
-      name,
-      connected: false,
-      connecting: true,
-      toolCount: 0,
-      toolNames: [],
-      transport
-    });
-    try {
-      await connect();
-      captureStderr(name);
-      const { toolsets, errors } = await client.listToolsetsWithErrors();
-      const serverTools = toolsets[name];
-      const serverError = errors[name];
-      if (serverError) {
-        const status = {
-          name,
-          connected: false,
-          toolCount: 0,
-          toolNames: [],
-          transport,
-          error: serverError,
-          ...serverNeedsAuth(name, cfg, serverError) ? { needsAuth: true } : {}
-        };
-        serverStatuses.set(name, status);
-        return status;
-      } else if (serverTools && Object.keys(serverTools).length > 0) {
-        const toolNames = Object.keys(serverTools).map((t) => `${name}_${t}`);
-        for (const [toolName, toolConfig] of Object.entries(serverTools)) {
-          tools[`${name}_${toolName}`] = toolConfig;
-        }
-        const status = {
-          name,
-          connected: true,
-          toolCount: toolNames.length,
-          toolNames,
-          transport
-        };
-        serverStatuses.set(name, status);
-        return status;
-      } else {
-        const status = {
-          name,
-          connected: false,
-          toolCount: 0,
-          toolNames: [],
-          transport,
-          error: "Failed to connect"
-        };
-        serverStatuses.set(name, status);
-        return status;
-      }
-    } catch (error) {
-      const errMsg = error instanceof Error ? error.message : String(error);
-      const status = {
-        name,
-        connected: false,
-        toolCount: 0,
-        toolNames: [],
-        transport,
-        error: errMsg,
-        ...serverNeedsAuth(name, cfg, errMsg) ? { needsAuth: true } : {}
-      };
-      serverStatuses.set(name, status);
-      return status;
-    }
-  }
-  async function disconnect() {
-    if (client) {
-      try {
-        await client.disconnect();
-      } catch {
-      }
-      client = null;
-    }
-  }
-  return {
-    async init() {
-      if (initialized) return;
-      await connectAndCollectTools();
-      initialized = true;
-    },
-    async initInBackground() {
-      await this.init();
-      const statuses = Array.from(serverStatuses.values());
-      const connected = statuses.filter((s) => s.connected);
-      const failed = statuses.filter((s) => !s.connected);
-      return {
-        connected,
-        failed,
-        skipped: [...config.skippedServers ?? []],
-        totalTools: connected.reduce((sum, s) => sum + s.toolCount, 0)
-      };
-    },
-    async reload() {
-      await disconnect();
-      config = applyExtraServers(loadMcpConfig(projectDir, configDirName));
-      tools = {};
-      serverStatuses = /* @__PURE__ */ new Map();
-      stderrLogs = /* @__PURE__ */ new Map();
-      initialized = false;
-      await connectAndCollectTools();
-      initialized = true;
-    },
-    async reconnectServer(name) {
-      const cfg = config.mcpServers?.[name];
-      if (!cfg) {
-        return {
-          name,
-          connected: false,
-          toolCount: 0,
-          toolNames: [],
-          transport: "stdio",
-          error: `Server "${name}" not found in config`
-        };
-      }
-      if (!client) {
-        return {
-          name,
-          connected: false,
-          toolCount: 0,
-          toolNames: [],
-          transport: getTransport(cfg),
-          error: "MCP client not initialized"
-        };
-      }
-      return connectSingleServer(name, cfg, () => client.reconnectServer(name));
-    },
-    async authenticateServer(name, options) {
-      const cfg = config.mcpServers?.[name];
-      if (!cfg) {
-        return {
-          name,
-          connected: false,
-          toolCount: 0,
-          toolNames: [],
-          transport: "stdio",
-          error: `Server "${name}" not found in config`
-        };
-      }
-      if (!client) {
-        return {
-          name,
-          connected: false,
-          toolCount: 0,
-          toolNames: [],
-          transport: getTransport(cfg),
-          error: "MCP client not initialized"
-        };
-      }
-      if (getTransport(cfg) !== "http") {
-        return {
-          name,
-          connected: false,
-          toolCount: 0,
-          toolNames: [],
-          transport: "stdio",
-          error: `Server "${name}" uses stdio transport, which does not support OAuth`
-        };
-      }
-      const def = serverDefs[name];
-      if (def?.url && !def.authProvider) {
-        def.authProvider = createOAuthProvider(name, { ...cfg, oauth: DEFAULT_OAUTH_CONFIG });
-      }
-      if (authenticatingServers.has(name)) {
-        const current = serverStatuses.get(name);
-        return {
-          name,
-          connected: current?.connected ?? false,
-          connecting: current?.connecting,
-          needsAuth: current?.needsAuth,
-          toolCount: current?.toolCount ?? 0,
-          toolNames: current?.toolNames ?? [],
-          transport: current?.transport ?? getTransport(cfg),
-          error: `Authentication for "${name}" is already in progress`
-        };
-      }
-      if (options?.onAuthorizationUrl) {
-        authUrlHandlers.set(name, options.onAuthorizationUrl);
-      }
-      authenticatingServers.add(name);
-      cancelledAuthServers.delete(name);
-      try {
-        const result = await connectSingleServer(
-          name,
-          cfg,
-          () => client.authenticate(name, options?.timeoutMs === void 0 ? void 0 : { timeoutMs: options.timeoutMs })
-        );
-        const finalStatus = cancelledAuthServers.has(name) && !result.connected ? { ...result, cancelled: true } : result;
-        serverStatuses.set(name, finalStatus);
-        return finalStatus;
-      } finally {
-        authUrlHandlers.delete(name);
-        authenticatingServers.delete(name);
-        cancelledAuthServers.delete(name);
-      }
-    },
-    async cancelServerAuthentication(name) {
-      if (!client) {
-        return false;
-      }
-      cancelledAuthServers.add(name);
-      const cancelled = await client.cancelAuthentication(name);
-      if (!cancelled) {
-        cancelledAuthServers.delete(name);
-      }
-      return cancelled;
-    },
-    disconnect,
-    getTools() {
-      return { ...tools };
-    },
-    hasServers() {
-      const hasConfigured = config.mcpServers !== void 0 && Object.keys(config.mcpServers).length > 0;
-      const hasSkipped = config.skippedServers !== void 0 && config.skippedServers.length > 0;
-      return hasConfigured || hasSkipped;
-    },
-    getServerStatuses() {
-      return Array.from(serverStatuses.values(), withAuthenticating);
-    },
-    getSkippedServers() {
-      return [...config.skippedServers ?? []];
-    },
-    getConfigPaths() {
-      return {
-        project: getProjectMcpPath(projectDir, configDirName),
-        global: getGlobalMcpPath(configDirName),
-        claude: getClaudeSettingsPath(projectDir)
-      };
-    },
-    getConfig() {
-      return config;
-    },
-    getServerLogs(name) {
-      return [...stderrLogs.get(name) ?? []];
-    }
-  };
+	/** Merge programmatic servers into a base config (highest priority). */
+	const applyExtraServers = (base) => {
+		if (!extraServers || Object.keys(extraServers).length === 0) return base;
+		return {
+			...base,
+			mcpServers: {
+				...base.mcpServers,
+				...extraServers
+			}
+		};
+	};
+	let config = applyExtraServers(loadMcpConfig(projectDir, configDirName));
+	let client = null;
+	let serverDefs = {};
+	let tools = {};
+	let serverStatuses = /* @__PURE__ */ new Map();
+	let stderrLogs = /* @__PURE__ */ new Map();
+	let initialized = false;
+	/** Per-server handlers that receive the OAuth authorization URL during authenticateServer(). */
+	const authUrlHandlers = /* @__PURE__ */ new Map();
+	/**
+	* Servers with an OAuth authorization flow currently in flight. Owned by the
+	* manager (not the TUI) so the state survives a `/mcp` selector being closed
+	* and reopened — the reopened selector reads it back off the server status and
+	* can still offer "Cancel authentication".
+	*/
+	const authenticatingServers = /* @__PURE__ */ new Set();
+	/**
+	* Servers whose in-flight authentication was cancelled by the caller. Set by
+	* cancelServerAuthentication() so the resolving authenticateServer() call can
+	* mark its failed status as a deliberate cancel rather than a genuine failure.
+	* This lives on the manager (not the TUI selector) so the signal survives the
+	* selector being closed and reopened mid-flow.
+	*/
+	const cancelledAuthServers = /* @__PURE__ */ new Set();
+	/** Overlay the manager-owned `authenticating` flag onto a status snapshot. */
+	const withAuthenticating = (status) => authenticatingServers.has(status.name) ? {
+		...status,
+		authenticating: true
+	} : status;
+	const MAX_STDERR_LINES = 200;
+	/** Hook into a server's stderr stream and buffer its output. */
+	function captureStderr(serverName) {
+		if (!client || typeof client.getServerStderr !== "function") return;
+		const stream = client.getServerStderr(serverName);
+		if (!stream) return;
+		let buffer = "";
+		const lines = stderrLogs.get(serverName) ?? [];
+		stderrLogs.set(serverName, lines);
+		stream.on("data", (chunk) => {
+			buffer += chunk.toString();
+			const parts = buffer.split("\n");
+			buffer = parts.pop();
+			for (const line of parts) if (line.trim()) {
+				lines.push(line);
+				if (lines.length > MAX_STDERR_LINES) lines.shift();
+			}
+		});
+		stream.on("end", () => {
+			if (buffer.trim()) {
+				lines.push(buffer);
+				if (lines.length > MAX_STDERR_LINES) lines.shift();
+			}
+		});
+	}
+	function createOAuthProvider(name, cfg) {
+		const oauth = cfg.oauth ?? (existsSync(getOAuthStoragePath(projectDir, name, cfg)) ? DEFAULT_OAUTH_CONFIG : void 0);
+		if (!oauth) return void 0;
+		const redirectUrl = resolveOAuthRedirectUrl(oauth);
+		return new MCPOAuthClientProvider({
+			redirectUrl,
+			clientMetadata: {
+				redirect_uris: [redirectUrl],
+				client_name: oauth.clientName ?? `Mastra Code MCP ${name}`,
+				grant_types: ["authorization_code", "refresh_token"],
+				response_types: ["code"],
+				...oauth.scopes?.length ? { scope: oauth.scopes.join(" ") } : {}
+			},
+			clientInformation: oauth.clientId ? {
+				client_id: oauth.clientId,
+				...oauth.clientSecret ? { client_secret: oauth.clientSecret } : {}
+			} : void 0,
+			storage: new FileOAuthStorage(getOAuthStoragePath(projectDir, name, cfg)),
+			onRedirectToAuthorization: (url) => {
+				authUrlHandlers.get(name)?.(url.toString());
+			}
+		});
+	}
+	function buildServerDefs(servers) {
+		const defs = {};
+		for (const [name, cfg] of Object.entries(servers)) if ("url" in cfg) {
+			const httpCfg = cfg;
+			defs[name] = {
+				url: new URL(httpCfg.url),
+				requestInit: httpCfg.headers ? { headers: httpCfg.headers } : void 0,
+				authProvider: createOAuthProvider(name, httpCfg)
+			};
+		} else defs[name] = {
+			command: cfg.command,
+			args: cfg.args,
+			env: cfg.env,
+			stderr: "pipe"
+		};
+		return defs;
+	}
+	async function connectAndCollectTools() {
+		const servers = config.mcpServers;
+		if (!servers || Object.keys(servers).length === 0) return;
+		const serverNames = Object.keys(servers);
+		for (const name of serverNames) serverStatuses.set(name, {
+			name,
+			connected: false,
+			connecting: true,
+			toolCount: 0,
+			toolNames: [],
+			transport: getTransport(servers[name])
+		});
+		serverDefs = buildServerDefs(servers);
+		client = new MCPClient({
+			id: "mastra-code-mcp",
+			servers: serverDefs,
+			timeout: MASTRACODE_MCP_TIMEOUT_MS
+		});
+		try {
+			const { toolsets, errors } = await client.listToolsetsWithErrors();
+			const typedToolsets = toolsets;
+			for (const [serverName, serverTools] of Object.entries(typedToolsets)) for (const [toolName, toolConfig] of Object.entries(serverTools)) tools[`${serverName}_${toolName}`] = toolConfig;
+			for (const name of serverNames) {
+				const serverTools = typedToolsets[name];
+				if (serverTools && Object.keys(serverTools).length > 0) {
+					const toolNames = Object.keys(serverTools).map((t) => `${name}_${t}`);
+					serverStatuses.set(name, {
+						name,
+						connected: true,
+						toolCount: toolNames.length,
+						toolNames,
+						transport: getTransport(servers[name])
+					});
+				} else {
+					const error = errors[name] ?? "Failed to connect";
+					serverStatuses.set(name, {
+						name,
+						connected: false,
+						toolCount: 0,
+						toolNames: [],
+						transport: getTransport(servers[name]),
+						error,
+						...serverNeedsAuth(name, servers[name], error) ? { needsAuth: true } : {}
+					});
+				}
+			}
+			for (const name of serverNames) captureStderr(name);
+		} catch (error) {
+			const errMsg = error instanceof Error ? error.message : String(error);
+			for (const name of serverNames) serverStatuses.set(name, {
+				name,
+				connected: false,
+				toolCount: 0,
+				toolNames: [],
+				transport: getTransport(servers[name]),
+				error: errMsg,
+				...serverNeedsAuth(name, servers[name], errMsg) ? { needsAuth: true } : {}
+			});
+		}
+	}
+	/**
+	* Whether a failed HTTP server is blocked on OAuth authorization.
+	*
+	* Provider-backed servers report the exact state tracked by `@mastra/mcp`.
+	* Bare `url` entries carry no provider until the user authenticates, so the
+	* signal is a 401 in the connect error — surfaced either as the status text
+	* or as the RFC 6750 `invalid_token` bearer error code in the response body.
+	*/
+	function serverNeedsAuth(name, cfg, error) {
+		if (getTransport(cfg) !== "http") return false;
+		if (client?.getServerAuthState?.(name) === "needs-auth") return true;
+		if (serverDefs[name]?.authProvider) return false;
+		return error !== void 0 && /\b401\b|unauthorized|invalid_token/i.test(error);
+	}
+	/**
+	* Runs a single-server connect action (reconnect or authenticate), then
+	* refreshes that server's tools and status from the client.
+	*/
+	async function connectSingleServer(name, cfg, connect) {
+		const transport = getTransport(cfg);
+		const prefix = `${name}_`;
+		for (const key of Object.keys(tools)) if (key.startsWith(prefix)) delete tools[key];
+		stderrLogs.delete(name);
+		serverStatuses.set(name, {
+			name,
+			connected: false,
+			connecting: true,
+			toolCount: 0,
+			toolNames: [],
+			transport
+		});
+		try {
+			await connect();
+			captureStderr(name);
+			const { toolsets, errors } = await client.listToolsetsWithErrors();
+			const serverTools = toolsets[name];
+			const serverError = errors[name];
+			if (serverError) {
+				const status = {
+					name,
+					connected: false,
+					toolCount: 0,
+					toolNames: [],
+					transport,
+					error: serverError,
+					...serverNeedsAuth(name, cfg, serverError) ? { needsAuth: true } : {}
+				};
+				serverStatuses.set(name, status);
+				return status;
+			} else if (serverTools && Object.keys(serverTools).length > 0) {
+				const toolNames = Object.keys(serverTools).map((t) => `${name}_${t}`);
+				for (const [toolName, toolConfig] of Object.entries(serverTools)) tools[`${name}_${toolName}`] = toolConfig;
+				const status = {
+					name,
+					connected: true,
+					toolCount: toolNames.length,
+					toolNames,
+					transport
+				};
+				serverStatuses.set(name, status);
+				return status;
+			} else {
+				const status = {
+					name,
+					connected: false,
+					toolCount: 0,
+					toolNames: [],
+					transport,
+					error: "Failed to connect"
+				};
+				serverStatuses.set(name, status);
+				return status;
+			}
+		} catch (error) {
+			const errMsg = error instanceof Error ? error.message : String(error);
+			const status = {
+				name,
+				connected: false,
+				toolCount: 0,
+				toolNames: [],
+				transport,
+				error: errMsg,
+				...serverNeedsAuth(name, cfg, errMsg) ? { needsAuth: true } : {}
+			};
+			serverStatuses.set(name, status);
+			return status;
+		}
+	}
+	async function disconnect() {
+		if (client) {
+			try {
+				await client.disconnect();
+			} catch {}
+			client = null;
+		}
+	}
+	return {
+		async init() {
+			if (initialized) return;
+			await connectAndCollectTools();
+			initialized = true;
+		},
+		async initInBackground() {
+			await this.init();
+			const statuses = Array.from(serverStatuses.values());
+			const connected = statuses.filter((s) => s.connected);
+			return {
+				connected,
+				failed: statuses.filter((s) => !s.connected),
+				skipped: [...config.skippedServers ?? []],
+				totalTools: connected.reduce((sum, s) => sum + s.toolCount, 0)
+			};
+		},
+		async reload() {
+			await disconnect();
+			config = applyExtraServers(loadMcpConfig(projectDir, configDirName));
+			tools = {};
+			serverStatuses = /* @__PURE__ */ new Map();
+			stderrLogs = /* @__PURE__ */ new Map();
+			initialized = false;
+			await connectAndCollectTools();
+			initialized = true;
+		},
+		async reconnectServer(name) {
+			const cfg = config.mcpServers?.[name];
+			if (!cfg) return {
+				name,
+				connected: false,
+				toolCount: 0,
+				toolNames: [],
+				transport: "stdio",
+				error: `Server "${name}" not found in config`
+			};
+			if (!client) return {
+				name,
+				connected: false,
+				toolCount: 0,
+				toolNames: [],
+				transport: getTransport(cfg),
+				error: "MCP client not initialized"
+			};
+			return connectSingleServer(name, cfg, () => client.reconnectServer(name));
+		},
+		async authenticateServer(name, options) {
+			const cfg = config.mcpServers?.[name];
+			if (!cfg) return {
+				name,
+				connected: false,
+				toolCount: 0,
+				toolNames: [],
+				transport: "stdio",
+				error: `Server "${name}" not found in config`
+			};
+			if (!client) return {
+				name,
+				connected: false,
+				toolCount: 0,
+				toolNames: [],
+				transport: getTransport(cfg),
+				error: "MCP client not initialized"
+			};
+			if (getTransport(cfg) !== "http") return {
+				name,
+				connected: false,
+				toolCount: 0,
+				toolNames: [],
+				transport: "stdio",
+				error: `Server "${name}" uses stdio transport, which does not support OAuth`
+			};
+			const def = serverDefs[name];
+			if (def?.url && !def.authProvider) def.authProvider = createOAuthProvider(name, {
+				...cfg,
+				oauth: DEFAULT_OAUTH_CONFIG
+			});
+			if (authenticatingServers.has(name)) {
+				const current = serverStatuses.get(name);
+				return {
+					name,
+					connected: current?.connected ?? false,
+					connecting: current?.connecting,
+					needsAuth: current?.needsAuth,
+					toolCount: current?.toolCount ?? 0,
+					toolNames: current?.toolNames ?? [],
+					transport: current?.transport ?? getTransport(cfg),
+					error: `Authentication for "${name}" is already in progress`
+				};
+			}
+			if (options?.onAuthorizationUrl) authUrlHandlers.set(name, options.onAuthorizationUrl);
+			authenticatingServers.add(name);
+			cancelledAuthServers.delete(name);
+			try {
+				const result = await connectSingleServer(name, cfg, () => client.authenticate(name, options?.timeoutMs === void 0 ? void 0 : { timeoutMs: options.timeoutMs }));
+				const finalStatus = cancelledAuthServers.has(name) && !result.connected ? {
+					...result,
+					cancelled: true
+				} : result;
+				serverStatuses.set(name, finalStatus);
+				return finalStatus;
+			} finally {
+				authUrlHandlers.delete(name);
+				authenticatingServers.delete(name);
+				cancelledAuthServers.delete(name);
+			}
+		},
+		async cancelServerAuthentication(name) {
+			if (!client) return false;
+			cancelledAuthServers.add(name);
+			const cancelled = await client.cancelAuthentication(name);
+			if (!cancelled) cancelledAuthServers.delete(name);
+			return cancelled;
+		},
+		disconnect,
+		getTools() {
+			return { ...tools };
+		},
+		hasServers() {
+			const hasConfigured = config.mcpServers !== void 0 && Object.keys(config.mcpServers).length > 0;
+			const hasSkipped = config.skippedServers !== void 0 && config.skippedServers.length > 0;
+			return hasConfigured || hasSkipped;
+		},
+		getServerStatuses() {
+			return Array.from(serverStatuses.values(), withAuthenticating);
+		},
+		getSkippedServers() {
+			return [...config.skippedServers ?? []];
+		},
+		getConfigPaths() {
+			return {
+				project: getProjectMcpPath(projectDir, configDirName),
+				global: getGlobalMcpPath(configDirName),
+				claude: getClaudeSettingsPath(projectDir)
+			};
+		},
+		getConfig() {
+			return config;
+		},
+		getServerLogs(name) {
+			return [...stderrLogs.get(name) ?? []];
+		}
+	};
 }
-export {
-  createMcpManager
-};
+//#endregion
+export { createMcpManager };
+
 //# sourceMappingURL=manager.js.map