fix: Disconnecting should clear oauth state

2025-04-07 15:15:28 -07:00
parent 4053aa122d
commit 6a16e7cd24
2 changed files with 7 additions and 0 deletions
--- a/client/src/lib/auth.ts
+++ b/client/src/lib/auth.ts
@@ -68,6 +68,12 @@ class InspectorOAuthClientProvider implements OAuthClientProvider {

    return verifier;
  }
+
+  clear() {
+    sessionStorage.removeItem(SESSION_KEYS.CLIENT_INFORMATION);
+    sessionStorage.removeItem(SESSION_KEYS.TOKENS);
+    sessionStorage.removeItem(SESSION_KEYS.CODE_VERIFIER);
+  }
 }

 export const authProvider = new InspectorOAuthClientProvider();
--- a/client/src/lib/hooks/useConnection.ts
+++ b/client/src/lib/hooks/useConnection.ts
@@ -361,6 +361,7 @@ export function useConnection({

  const disconnect = async () => {
    await mcpClient?.close();
+    authProvider.clear();
    setMcpClient(null);
    setConnectionStatus("disconnected");
    setCompletionsSupported(false);