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

packages/workspace/src/affine/api/index.ts

@@ -403,7 +403,7 @@ export function createWorkspaceApis(prefixUrl = '/') {
           },
         })
           .then(r =>
-            r.status === 403
+            !r.ok
               ? Promise.reject(new RequestError(MessageCode.noPermission))
               : r
           )

packages/workspace/src/affine/channel.ts

@@ -4,6 +4,8 @@ import {
   isExpired,
   parseIdToken,
 } from '@affine/workspace/affine/login';
+import { WorkspaceFlavour } from '@affine/workspace/type';
+import { cleanupWorkspace } from '@affine/workspace/utils';
 import { assertExists } from '@blocksuite/global/utils';
 import * as url from 'lib0/url';
 import * as websocket from 'lib0/websocket';
@@ -28,6 +30,10 @@ export class WebsocketClient {
 
   public connect(callback: (message: any) => void) {
     const loginResponse = getLoginStorage();
+    if (!loginResponse || isExpired(parseIdToken(loginResponse.token))) {
+      cleanupWorkspace(WorkspaceFlavour.AFFINE);
+      return;
+    }
     assertExists(loginResponse, 'loginResponse is null');
     const encodedParams = url.encodeQueryParams({
       token: loginResponse.token,

packages/workspace/src/utils.ts

@@ -1,4 +1,5 @@
 import type { createWorkspaceApis } from '@affine/workspace/affine/api';
+import { rootStore, rootWorkspacesMetadataAtom } from '@affine/workspace/atom';
 import { createAffineBlobStorage } from '@affine/workspace/blob';
 import { __unstableSchemas, AffineSchemas } from '@blocksuite/blocks/models';
 import type { Generator, StoreOptions } from '@blocksuite/store';
@@ -7,6 +8,12 @@ import { createIndexeddbStorage, Workspace } from '@blocksuite/store';
 import { createSQLiteStorage } from './blob/sqlite-blob-storage';
 import { WorkspaceFlavour } from './type';
 
+export function cleanupWorkspace(flavour: WorkspaceFlavour) {
+  rootStore.set(rootWorkspacesMetadataAtom, metas =>
+    metas.filter(meta => meta.flavour !== flavour)
+  );
+}
+
 const hashMap = new Map<string, Workspace>();
 
 export function createEmptyBlockSuiteWorkspace(