fix: delegate issuer context and discovery state through the EMA wrapper
Copilot review (#2287): `EmaTransportOAuthProvider` forwarded `clientInformation` / `saveClientInformation` without the SDK's `ctx`, and implemented neither `discoveryState` nor `saveDiscoveryState`. Since the wrapper does expose `clientMetadataUrl`, an EMA connection can still take SDK `auth()`'s URL-based-client-ID branch — and the inner provider then saw `issuer === undefined` with no discovery state to read back, so the CIMD write was recorded as DCR. Forward `ctx` on both, and delegate the two discovery-state methods to the inner provider. Both were pre-existing SEP-2352 gaps in their own right: dropping the issuer put every EMA read and write on the unkeyed slot, and the missing discovery state meant the SDK re-discovered on every call and warned that it could not run its callback-leg authorization-server binding check. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01Lmj6X2epWpPE7ScdSKQmtA Signed-off-by: cliffhall <cliff@futurescale.com>
This commit is contained in:
parent
be08f6d37d
commit
aca522de6a
2 changed files with 75 additions and 6 deletions
|
|
@ -65,6 +65,8 @@ interface FakeInner {
|
||||||
clearCapturedAuthUrl: ReturnType<typeof vi.fn>;
|
clearCapturedAuthUrl: ReturnType<typeof vi.fn>;
|
||||||
saveCodeVerifier: ReturnType<typeof vi.fn>;
|
saveCodeVerifier: ReturnType<typeof vi.fn>;
|
||||||
codeVerifier: ReturnType<typeof vi.fn>;
|
codeVerifier: ReturnType<typeof vi.fn>;
|
||||||
|
saveDiscoveryState: ReturnType<typeof vi.fn>;
|
||||||
|
discoveryState: ReturnType<typeof vi.fn>;
|
||||||
}
|
}
|
||||||
|
|
||||||
function createInner(): FakeInner {
|
function createInner(): FakeInner {
|
||||||
|
|
@ -82,6 +84,8 @@ function createInner(): FakeInner {
|
||||||
clearCapturedAuthUrl: vi.fn(),
|
clearCapturedAuthUrl: vi.fn(),
|
||||||
saveCodeVerifier: vi.fn(),
|
saveCodeVerifier: vi.fn(),
|
||||||
codeVerifier: vi.fn(() => "verifier-xyz"),
|
codeVerifier: vi.fn(() => "verifier-xyz"),
|
||||||
|
saveDiscoveryState: vi.fn(),
|
||||||
|
discoveryState: vi.fn(),
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -118,14 +122,54 @@ describe("EmaTransportOAuthProvider", () => {
|
||||||
expect(await provider.codeVerifier()).toBe("verifier-xyz");
|
expect(await provider.codeVerifier()).toBe("verifier-xyz");
|
||||||
|
|
||||||
await provider.saveClientInformation({ client_id: "new" } as never);
|
await provider.saveClientInformation({ client_id: "new" } as never);
|
||||||
expect(inner.saveClientInformation).toHaveBeenCalledWith({
|
expect(inner.saveClientInformation).toHaveBeenCalledWith(
|
||||||
client_id: "new",
|
{ client_id: "new" },
|
||||||
});
|
undefined,
|
||||||
|
);
|
||||||
|
|
||||||
await provider.saveCodeVerifier("cv");
|
await provider.saveCodeVerifier("cv");
|
||||||
expect(inner.saveCodeVerifier).toHaveBeenCalledWith("cv");
|
expect(inner.saveCodeVerifier).toHaveBeenCalledWith("cv");
|
||||||
});
|
});
|
||||||
|
|
||||||
|
// SEP-2352: the wrapper used to drop the SDK's `ctx`, so every EMA read and
|
||||||
|
// write landed on the unkeyed slot — and, since #2242, the registration-kind
|
||||||
|
// resolver had no issuer to check and recorded a CIMD registration made over
|
||||||
|
// an EMA connection as DCR (Copilot).
|
||||||
|
it("forwards the SDK issuer context on client-information reads and writes", async () => {
|
||||||
|
const ctx = { issuer: "https://as.example.com" };
|
||||||
|
|
||||||
|
await provider.clientInformation(ctx);
|
||||||
|
expect(inner.clientInformation).toHaveBeenCalledWith(ctx);
|
||||||
|
|
||||||
|
await provider.saveClientInformation({ client_id: "new" } as never, ctx);
|
||||||
|
expect(inner.saveClientInformation).toHaveBeenCalledWith(
|
||||||
|
{ client_id: "new" },
|
||||||
|
ctx,
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
// Without these the SDK persists no discovery state for an EMA connection, so
|
||||||
|
// it re-discovers every call, cannot run its callback-leg AS binding check,
|
||||||
|
// and leaves the registration-kind resolver nothing to read back.
|
||||||
|
it("delegates discovery state to the inner provider", async () => {
|
||||||
|
const state = {
|
||||||
|
authorizationServerUrl: "https://as.example.com",
|
||||||
|
authorizationServerMetadata: {
|
||||||
|
issuer: "https://as.example.com",
|
||||||
|
authorization_endpoint: "https://as.example.com/authorize",
|
||||||
|
token_endpoint: "https://as.example.com/token",
|
||||||
|
response_types_supported: ["code"],
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
await provider.saveDiscoveryState(state);
|
||||||
|
expect(inner.saveDiscoveryState).toHaveBeenCalledWith(state);
|
||||||
|
|
||||||
|
inner.discoveryState.mockReturnValue(state);
|
||||||
|
expect(await provider.discoveryState()).toEqual(state);
|
||||||
|
expect(inner.discoveryState).toHaveBeenCalled();
|
||||||
|
});
|
||||||
|
|
||||||
it("tokens() returns stored tokens when the access token is still usable", async () => {
|
it("tokens() returns stored tokens when the access token is still usable", async () => {
|
||||||
const stored: OAuthTokens = {
|
const stored: OAuthTokens = {
|
||||||
access_token: VALID_ACCESS_TOKEN,
|
access_token: VALID_ACCESS_TOKEN,
|
||||||
|
|
|
||||||
|
|
@ -1,7 +1,9 @@
|
||||||
import type { OAuthClientProvider } from "@modelcontextprotocol/client";
|
import type { OAuthClientProvider } from "@modelcontextprotocol/client";
|
||||||
import type {
|
import type {
|
||||||
|
OAuthClientInformationContext,
|
||||||
OAuthClientInformationMixed,
|
OAuthClientInformationMixed,
|
||||||
OAuthClientMetadata,
|
OAuthClientMetadata,
|
||||||
|
OAuthDiscoveryState,
|
||||||
OAuthTokens,
|
OAuthTokens,
|
||||||
} from "@modelcontextprotocol/client";
|
} from "@modelcontextprotocol/client";
|
||||||
import type { BaseOAuthClientProvider } from "../providers.js";
|
import type { BaseOAuthClientProvider } from "../providers.js";
|
||||||
|
|
@ -50,17 +52,40 @@ export class EmaTransportOAuthProvider implements OAuthClientProvider {
|
||||||
return this.inner.state();
|
return this.inner.state();
|
||||||
}
|
}
|
||||||
|
|
||||||
clientInformation():
|
// SEP-2352: `ctx` carries the authorization-server `issuer` the SDK resolved,
|
||||||
|
// and the inner provider keys registrations by it. Dropping it here made every
|
||||||
|
// EMA read and write land on the unkeyed slot — and, since #2242, left
|
||||||
|
// `resolveSdkRegistrationKind` with no issuer to check, so a CIMD registration
|
||||||
|
// made over an EMA connection was recorded as DCR (Copilot).
|
||||||
|
clientInformation(
|
||||||
|
ctx?: OAuthClientInformationContext,
|
||||||
|
):
|
||||||
| OAuthClientInformationMixed
|
| OAuthClientInformationMixed
|
||||||
| undefined
|
| undefined
|
||||||
| Promise<OAuthClientInformationMixed | undefined> {
|
| Promise<OAuthClientInformationMixed | undefined> {
|
||||||
return this.inner.clientInformation();
|
return this.inner.clientInformation(ctx);
|
||||||
}
|
}
|
||||||
|
|
||||||
saveClientInformation(
|
saveClientInformation(
|
||||||
clientInformation: OAuthClientInformationMixed,
|
clientInformation: OAuthClientInformationMixed,
|
||||||
|
ctx?: OAuthClientInformationContext,
|
||||||
): void | Promise<void> {
|
): void | Promise<void> {
|
||||||
return this.inner.saveClientInformation(clientInformation);
|
return this.inner.saveClientInformation(clientInformation, ctx);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Without these the SDK persists no discovery state for an EMA connection, so
|
||||||
|
// it re-discovers on every call, cannot perform its SEP-2352 callback-leg
|
||||||
|
// authorization-server binding check (it warns as much), and — since #2242 —
|
||||||
|
// leaves the registration-kind resolver nothing to read back.
|
||||||
|
saveDiscoveryState(state: OAuthDiscoveryState): void | Promise<void> {
|
||||||
|
return this.inner.saveDiscoveryState(state);
|
||||||
|
}
|
||||||
|
|
||||||
|
discoveryState():
|
||||||
|
| OAuthDiscoveryState
|
||||||
|
| undefined
|
||||||
|
| Promise<OAuthDiscoveryState | undefined> {
|
||||||
|
return this.inner.discoveryState();
|
||||||
}
|
}
|
||||||
|
|
||||||
async tokens(): Promise<OAuthTokens | undefined> {
|
async tokens(): Promise<OAuthTokens | undefined> {
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue