hzhang/inspector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update OAuth callback code

Browse Source
This commit is contained in:
Justin Spahr-Summers
2025-02-11 17:42:49 +00:00
parent eb6af47b21
commit e9a50adde7
3 changed files with 11 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
client/src/lib/auth.ts
View File

@@ -1,7 +1,7 @@
import { OAuthClientInformation, OAuthClientInformationSchema, OAuthClientProvider, OAuthTokens, OAuthTokensSchema } from "@modelcontextprotocol/sdk/client/auth.js";
import { SESSION_KEYS } from "./constants";
export class InspectorOAuthClientProvider implements OAuthClientProvider {
class InspectorOAuthClientProvider implements OAuthClientProvider {
get redirectUrl() {
return window.location.origin + "/oauth/callback";
}
@@ -69,3 +69,5 @@ export class InspectorOAuthClientProvider implements OAuthClientProvider {
return verifier;
}
}
export const authProvider = new InspectorOAuthClientProvider();

5
client/src/lib/hooks/useConnection.ts
View File

@@ -19,7 +19,7 @@ import { z } from "zod";
import { SESSION_KEYS } from "../constants";
import { Notification, StdErrNotificationSchema } from "../notificationTypes";
import { auth } from "@modelcontextprotocol/sdk/client/auth.js";
import { InspectorOAuthClientProvider } from "../auth";
import { authProvider } from "../auth";
const DEFAULT_REQUEST_TIMEOUT_MSEC = 10000;
@@ -122,12 +122,11 @@ export function useConnection({
}
};
const authProvider = new InspectorOAuthClientProvider();
const handleAuthError = async (error: unknown) => {
if (error instanceof SseError && error.code === 401) {
sessionStorage.setItem(SESSION_KEYS.SERVER_URL, sseUrl);
const result = await auth(authProvider, { serverUrl: sseUrl })
const result = await auth(authProvider, { serverUrl: sseUrl });
return result === "AUTHORIZED";
}