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>;
|
||||
saveCodeVerifier: ReturnType<typeof vi.fn>;
|
||||
codeVerifier: ReturnType<typeof vi.fn>;
|
||||
saveDiscoveryState: ReturnType<typeof vi.fn>;
|
||||
discoveryState: ReturnType<typeof vi.fn>;
|
||||
}
|
||||
|
||||
function createInner(): FakeInner {
|
||||
|
|
@ -82,6 +84,8 @@ function createInner(): FakeInner {
|
|||
clearCapturedAuthUrl: vi.fn(),
|
||||
saveCodeVerifier: vi.fn(),
|
||||
codeVerifier: vi.fn(() => "verifier-xyz"),
|
||||
saveDiscoveryState: vi.fn(),
|
||||
discoveryState: vi.fn(),
|
||||
};
|
||||
}
|
||||
|
||||
|
|
@ -118,14 +122,54 @@ describe("EmaTransportOAuthProvider", () => {
|
|||
expect(await provider.codeVerifier()).toBe("verifier-xyz");
|
||||
|
||||
await provider.saveClientInformation({ client_id: "new" } as never);
|
||||
expect(inner.saveClientInformation).toHaveBeenCalledWith({
|
||||
client_id: "new",
|
||||
});
|
||||
expect(inner.saveClientInformation).toHaveBeenCalledWith(
|
||||
{ client_id: "new" },
|
||||
undefined,
|
||||
);
|
||||
|
||||
await provider.saveCodeVerifier("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 () => {
|
||||
const stored: OAuthTokens = {
|
||||
access_token: VALID_ACCESS_TOKEN,
|
||||
|
|
|
|||
|
|
@ -1,7 +1,9 @@
|
|||
import type { OAuthClientProvider } from "@modelcontextprotocol/client";
|
||||
import type {
|
||||
OAuthClientInformationContext,
|
||||
OAuthClientInformationMixed,
|
||||
OAuthClientMetadata,
|
||||
OAuthDiscoveryState,
|
||||
OAuthTokens,
|
||||
} from "@modelcontextprotocol/client";
|
||||
import type { BaseOAuthClientProvider } from "../providers.js";
|
||||
|
|
@ -50,17 +52,40 @@ export class EmaTransportOAuthProvider implements OAuthClientProvider {
|
|||
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
|
||||
| undefined
|
||||
| Promise<OAuthClientInformationMixed | undefined> {
|
||||
return this.inner.clientInformation();
|
||||
return this.inner.clientInformation(ctx);
|
||||
}
|
||||
|
||||
saveClientInformation(
|
||||
clientInformation: OAuthClientInformationMixed,
|
||||
ctx?: OAuthClientInformationContext,
|
||||
): 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> {
|
||||
|
|
|
|||
Loading…
Reference in a new issue