refac

src/lib/apis/configs/index.ts

@@ -572,9 +572,13 @@ export const initiateOAuthRedirect = (tool: {
 }) => {
 	sessionStorage.setItem('pendingOAuthToolId', tool.id);
 	sessionStorage.setItem('oauthRedirectInProgressToolId', tool.id);
+	localStorage.setItem('pendingOAuthToolId', tool.id);
+	localStorage.setItem('oauthRedirectInProgressToolId', tool.id);
 	const authUrl = getOAuthClientAuthorizationUrl(tool.serverId, tool.authType ?? 'mcp');
 	const tab = window.open(authUrl, '_blank');
 	if (!tab) {
+		localStorage.removeItem('pendingOAuthToolId');
+		localStorage.removeItem('oauthRedirectInProgressToolId');
 		window.open(authUrl, '_self', 'noopener');
 	}
 };

src/lib/components/chat/Chat.svelte

@@ -1419,12 +1419,15 @@
 
 			if (event.data.error) {
 				sessionStorage.removeItem('oauthRedirectInProgressToolId');
+				localStorage.removeItem('oauthRedirectInProgressToolId');
 				toast.error(event.data.error);
 				return;
 			}
 
 			sessionStorage.removeItem('pendingOAuthToolId');
 			sessionStorage.removeItem('oauthRedirectInProgressToolId');
+			localStorage.removeItem('pendingOAuthToolId');
+			localStorage.removeItem('oauthRedirectInProgressToolId');
 			const updatedTools = await getTools(localStorage.token).catch(() => null);
 			if (updatedTools) {
 				tools.set(updatedTools);

src/routes/+layout.svelte

@@ -1069,7 +1069,8 @@
 
 		window.addEventListener('message', windowMessageEventHandler);
 
-		const pendingOAuthToolId = sessionStorage.getItem('pendingOAuthToolId');
+		const pendingOAuthToolId =
+			sessionStorage.getItem('pendingOAuthToolId') ?? localStorage.getItem('pendingOAuthToolId');
 		if (window.opener && window.location.pathname === '/' && pendingOAuthToolId) {
 			window.opener.postMessage(
 				{
@@ -1081,6 +1082,8 @@
 			);
 			sessionStorage.removeItem('pendingOAuthToolId');
 			sessionStorage.removeItem('oauthRedirectInProgressToolId');
+			localStorage.removeItem('pendingOAuthToolId');
+			localStorage.removeItem('oauthRedirectInProgressToolId');
 			window.close();
 		}