From 15960f5aa484f0bb656abe17d877fba9ad03d21d Mon Sep 17 00:00:00 2001
From: Maxwell Gerber <max@stytch.com>
Date: Wed, 16 Apr 2025 16:24:49 -0700
Subject: [PATCH] refactor: Use new serverspecifickey API

---
 client/src/lib/auth.ts                | 6 +++---
 client/src/lib/hooks/useConnection.ts | 1 +
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/client/src/lib/auth.ts b/client/src/lib/auth.ts
index 39b489d..65650fd 100644
--- a/client/src/lib/auth.ts
+++ b/client/src/lib/auth.ts
@@ -90,8 +90,8 @@ export class InspectorOAuthClientProvider implements OAuthClientProvider {
   }
 
   clear() {
-    sessionStorage.removeItem(SESSION_KEYS.CLIENT_INFORMATION);
-    sessionStorage.removeItem(SESSION_KEYS.TOKENS);
-    sessionStorage.removeItem(SESSION_KEYS.CODE_VERIFIER);
+    sessionStorage.removeItem(getServerSpecificKey(SESSION_KEYS.CLIENT_INFORMATION, this.serverUrl));
+    sessionStorage.removeItem(getServerSpecificKey(SESSION_KEYS.TOKENS, this.serverUrl));
+    sessionStorage.removeItem(getServerSpecificKey(SESSION_KEYS.CODE_VERIFIER, this.serverUrl));
   }
 }
diff --git a/client/src/lib/hooks/useConnection.ts b/client/src/lib/hooks/useConnection.ts
index 8c5c009..73f2e1c 100644
--- a/client/src/lib/hooks/useConnection.ts
+++ b/client/src/lib/hooks/useConnection.ts
@@ -396,6 +396,7 @@ export function useConnection({
 
   const disconnect = async () => {
     await mcpClient?.close();
+    const authProvider = new InspectorOAuthClientProvider(sseUrl);
     authProvider.clear();
     setMcpClient(null);
     setConnectionStatus("disconnected");