fix: cleanup data when logout (#1443)

apps/web/src/providers/ModalProvider.tsx

@@ -1,10 +1,11 @@
-import { useAtom, useAtomValue } from 'jotai';
+import { useAtom, useAtomValue, useSetAtom } from 'jotai';
 import dynamic from 'next/dynamic';
 import { useRouter } from 'next/router';
 import React, { useCallback } from 'react';
 
 import {
   currentWorkspaceIdAtom,
+  jotaiWorkspacesAtom,
   openCreateWorkspaceModalAtom,
   openWorkspacesModalAtom,
 } from '../atoms';
@@ -12,7 +13,8 @@ import { useCurrentUser } from '../hooks/current/use-current-user';
 import { useCurrentWorkspace } from '../hooks/current/use-current-workspace';
 import { useRouterHelper } from '../hooks/use-router-helper';
 import { useWorkspaces, useWorkspacesHelper } from '../hooks/use-workspaces';
-import { WorkspaceSubPath } from '../shared';
+import { WorkspacePlugins } from '../plugins';
+import { RemWorkspaceFlavour, WorkspaceSubPath } from '../shared';
 import { apis } from '../shared/apis';
 
 const WorkspaceListModal = dynamic(
@@ -40,6 +42,7 @@ export function Modals() {
   const currentWorkspaceId = useAtomValue(currentWorkspaceIdAtom);
   const [, setCurrentWorkspace] = useCurrentWorkspace();
   const { createLocalWorkspace } = useWorkspacesHelper();
+  const set = useSetAtom(jotaiWorkspacesAtom);
 
   return (
     <>
@@ -66,8 +69,14 @@ export function Modals() {
         }, [router])}
         onClickLogout={useCallback(() => {
           apis.auth.clear();
+          set(workspaces =>
+            workspaces.filter(
+              workspace => workspace.flavour !== RemWorkspaceFlavour.AFFINE
+            )
+          );
+          WorkspacePlugins[RemWorkspaceFlavour.AFFINE].cleanup?.();
           router.reload();
-        }, [router])}
+        }, [router, set])}
         onCreateWorkspace={useCallback(() => {
           setOpenCreateWorkspaceModal(true);
         }, [setOpenCreateWorkspaceModal])}