fix(workspace): check affine login auth (#2070)

apps/web/src/hooks/affine/use-affine-refresh-auth-token.ts

@@ -15,29 +15,37 @@ const logger = new DebugLogger('auth-token');
 const revalidate = async () => {
   const storage = getLoginStorage();
   if (storage) {
-    const tokenMessage = parseIdToken(storage.token);
-    logger.debug('revalidate affine user');
-    if (isExpired(tokenMessage)) {
-      logger.debug('need to refresh token');
-      const response = await affineAuth.refreshToken(storage);
-      if (response) {
-        setLoginStorage(response);
-        storageChangeSlot.emit();
+    try {
+      const tokenMessage = parseIdToken(storage.token);
+      logger.debug('revalidate affine user');
+      if (isExpired(tokenMessage)) {
+        logger.debug('need to refresh token');
+        const response = await affineAuth.refreshToken(storage);
+        if (response) {
+          setLoginStorage(response);
+          storageChangeSlot.emit();
+        }
       }
+    } catch (e) {
+      return false;
     }
+    return true;
+  } else {
+    return false;
   }
-  return true;
 };
 
 export function useAffineRefreshAuthToken(
   // every 30 seconds, check if the token is expired
   refreshInterval = 30 * 1000
 ) {
-  useSWR('autoRefreshToken', {
+  const { data } = useSWR<boolean>('autoRefreshToken', {
+    suspense: true,
     fetcher: revalidate,
     refreshInterval,
     revalidateOnFocus: true,
     revalidateOnReconnect: true,
     revalidateOnMount: true,
   });
+  return data;
 }

apps/web/src/hooks/use-sync-router-with-current-workspace-id.ts

@@ -20,6 +20,19 @@ export function useSyncRouterWithCurrentWorkspaceId(router: NextRouter) {
       return;
     }
     if (currentWorkspaceId) {
+      if (currentWorkspaceId !== workspaceId) {
+        const target = metadata.find(workspace => workspace.id === workspaceId);
+        if (!target) {
+          // workspaceId is invalid, redirect to currentWorkspaceId
+          void router.push({
+            pathname: router.pathname,
+            query: {
+              ...router.query,
+              workspaceId: currentWorkspaceId,
+            },
+          });
+        }
+      }
       return;
     }
     const targetWorkspace = metadata.find(