Function handleMCPOauthCode

core/context/mcp/MCPOauth.ts:276–344

* handle the authentication code received from the oauth redirect

(authorizationCode: string, state?: string)

 * handle the authentication code received from the oauth redirect
 */
async function handleMCPOauthCode(authorizationCode: string, state?: string) {
  let serverUrl: string | undefined;
  let context: MCPOauthContext | undefined;

  if (state) {
    // Use state parameter to find the correct server
    serverUrl = stateToServerUrl.get(state);
    if (serverUrl) {
      context = authenticatingContexts.get(serverUrl);
    }
  } else {
    // Fallback: if no state or single context, use the first one
    const contexts = Array.from(authenticatingContexts.entries());
    if (contexts.length === 1) {
      [serverUrl, context] = contexts[0];
    }
  }

  if (!context || !serverUrl) {
    console.error("No matching authenticating context found for state:", state);
    return;
  }

  const { ide, serverId } = context;

  try {
    if (!serverUrl) {
      throw new Error("No MCP server url found for authentication");
    }
    if (!authorizationCode) {
      throw new Error(`No MCP authorization code found for ${serverUrl}`);
    }

    // Close the server before processing auth
    if (serverInstance) {
      await new Promise<void>((resolve) => {
        serverInstance!.close(() => {
          console.debug("Server for MCP Oauth process was closed");
          serverInstance = null;
          resolve();
        });
      });
    }

    const authProvider = new MCPConnectionOauthProvider(serverUrl, ide);
    await authProvider.ensureRedirectUrl();
    const authStatus = await auth(authProvider, {
      serverUrl,
      authorizationCode,
    });

    if (authStatus === "AUTHORIZED") {
      const { MCPManagerSingleton } = await import("./MCPManagerSingleton"); // put dynamic import to avoid cyclic imports
      await MCPManagerSingleton.getInstance().refreshConnection(serverId);
    }
  } catch (error) {
    const errorMessage = error instanceof Error ? error.message : String(error);
    console.error("OAuth authorization failed:", errorMessage);