"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.ProxyOAuthServerProvider = void 0;
const auth_js_1 = require("../../../shared/auth.js");
const errors_js_1 = require("../errors.js");
/**
 * Implements an OAuth server that proxies requests to another OAuth server.
 */
class ProxyOAuthServerProvider {
    constructor(options) {
        this.skipLocalPkceValidation = true;
        this._endpoints = options.endpoints;
        this._verifyAccessToken = options.verifyAccessToken;
        this._getClient = options.getClient;
        this._fetch = options.fetch;
        if (options.endpoints?.revocationUrl) {
            this.revokeToken = async (client, request) => {
                const revocationUrl = this._endpoints.revocationUrl;
                if (!revocationUrl) {
                    throw new Error('No revocation endpoint configured');
                }
                const params = new URLSearchParams();
                params.set('token', request.token);
                params.set('client_id', client.client_id);
                if (client.client_secret) {
                    params.set('client_secret', client.client_secret);
                }
                if (request.token_type_hint) {
                    params.set('token_type_hint', request.token_type_hint);
                }
                const response = await (this._fetch ?? fetch)(revocationUrl, {
                    method: 'POST',
                    headers: {
                        'Content-Type': 'application/x-www-form-urlencoded'
                    },
                    body: params.toString()
                });
                await response.body?.cancel();
                if (!response.ok) {
                    throw new errors_js_1.ServerError(`Token revocation failed: ${response.status}`);
                }
            };
        }
    }
    get clientsStore() {
        const registrationUrl = this._endpoints.registrationUrl;
        return {
            getClient: this._getClient,
            ...(registrationUrl && {
                registerClient: async (client) => {
                    const response = await (this._fetch ?? fetch)(registrationUrl, {
                        method: 'POST',
                        headers: {
                            'Content-Type': 'application/json'
                        },
                        body: JSON.stringify(client)
                    });
                    if (!response.ok) {
                        await response.body?.cancel();
                        throw new errors_js_1.ServerError(`Client registration failed: ${response.status}`);
                    }
                    const data = await response.json();
                    return auth_js_1.OAuthClientInformationFullSchema.parse(data);
                }
            })
        };
    }
    async authorize(client, params, res) {
        // Start with required OAuth parameters
        const targetUrl = new URL(this._endpoints.authorizationUrl);
        const searchParams = new URLSearchParams({
            client_id: client.client_id,
            response_type: 'code',
            redirect_uri: params.redirectUri,
            code_challenge: params.codeChallenge,
            code_challenge_method: 'S256'
        });
        // Add optional standard OAuth parameters
        if (params.state)
            searchParams.set('state', params.state);
        if (params.scopes?.length)
            searchParams.set('scope', params.scopes.join(' '));
        if (params.resource)
            searchParams.set('resource', params.resource.href);
        targetUrl.search = searchParams.toString();
        res.redirect(targetUrl.toString());
    }
    async challengeForAuthorizationCode(_client, _authorizationCode) {
        // In a proxy setup, we don't store the code challenge ourselves
        // Instead, we proxy the token request and let the upstream server validate it
        return '';
    }
    async exchangeAuthorizationCode(client, authorizationCode, codeVerifier, redirectUri, resource) {
        const params = new URLSearchParams({
            grant_type: 'authorization_code',
            client_id: client.client_id,
            code: authorizationCode
        });
        if (client.client_secret) {
            params.append('client_secret', client.client_secret);
        }
        if (codeVerifier) {
            params.append('code_verifier', codeVerifier);
        }
        if (redirectUri) {
            params.append('redirect_uri', redirectUri);
        }
        if (resource) {
            params.append('resource', resource.href);
        }
        const response = await (this._fetch ?? fetch)(this._endpoints.tokenUrl, {
            method: 'POST',
            headers: {
                'Content-Type': 'application/x-www-form-urlencoded'
            },
            body: params.toString()
        });
        if (!response.ok) {
            await response.body?.cancel();
            throw new errors_js_1.ServerError(`Token exchange failed: ${response.status}`);
        }
        const data = await response.json();
        return auth_js_1.OAuthTokensSchema.parse(data);
    }
    async exchangeRefreshToken(client, refreshToken, scopes, resource) {
        const params = new URLSearchParams({
            grant_type: 'refresh_token',
            client_id: client.client_id,
            refresh_token: refreshToken
        });
        if (client.client_secret) {
            params.set('client_secret', client.client_secret);
        }
        if (scopes?.length) {
            params.set('scope', scopes.join(' '));
        }
        if (resource) {
            params.set('resource', resource.href);
        }
        const response = await (this._fetch ?? fetch)(this._endpoints.tokenUrl, {
            method: 'POST',
            headers: {
                'Content-Type': 'application/x-www-form-urlencoded'
            },
            body: params.toString()
        });
        if (!response.ok) {
            await response.body?.cancel();
            throw new errors_js_1.ServerError(`Token refresh failed: ${response.status}`);
        }
        const data = await response.json();
        return auth_js_1.OAuthTokensSchema.parse(data);
    }
    async verifyAccessToken(token) {
        return this._verifyAccessToken(token);
    }
}
exports.ProxyOAuthServerProvider = ProxyOAuthServerProvider;//# sourceMappingURL=http://go/sourcemap/sourcemaps/cd1c87ff9b66021918fb9731605f8d1d5fd2f0b0/node_modules/@modelcontextprotocol/sdk/dist/cjs/server/auth/providers/proxyProvider.js.map