From 25ac0f700b37ba0c4e55912aa8ee6ff706f6ab0e Mon Sep 17 00:00:00 2001
From: Himself65 <himself65@outlook.com>
Date: Fri, 17 Feb 2023 00:33:36 -0600
Subject: [PATCH] fix: jump to 404 when be removed from page (#1090)

---
 .../src/pages/workspace/[workspaceId]/[pageId].tsx  | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/apps/web/src/pages/workspace/[workspaceId]/[pageId].tsx b/apps/web/src/pages/workspace/[workspaceId]/[pageId].tsx
index dd566cf921..abd6feb489 100644
--- a/apps/web/src/pages/workspace/[workspaceId]/[pageId].tsx
+++ b/apps/web/src/pages/workspace/[workspaceId]/[pageId].tsx
@@ -14,7 +14,7 @@ import { usePageHelper } from '@/hooks/use-page-helper';
 import dynamic from 'next/dynamic';
 import Head from 'next/head';
 import { useTranslation } from '@affine/i18n';
-import { useGlobalState } from '@/store/app';
+import { useGlobalState, useGlobalStateApi } from '@/store/app';
 import exampleMarkdown from '@/templates/Welcome-to-AFFiNE-Alpha-Downhills.md';
 import { assertEquals } from '@blocksuite/store';
 
@@ -107,6 +107,7 @@ const PageDefender = ({ children }: PropsWithChildren) => {
   const currentWorkspace = useGlobalState(
     useCallback(store => store.currentDataCenterWorkspace, [])
   );
+  const dataCenter = useGlobalState(store => store.dataCenter);
   const { createPage } = usePageHelper();
 
   useEffect(() => {
@@ -124,6 +125,16 @@ const PageDefender = ({ children }: PropsWithChildren) => {
     };
     initPage();
   }, [createPage, currentWorkspace, loadPage, router.query.pageId]);
+  const api = useGlobalStateApi();
+  useEffect(
+    () =>
+      dataCenter.onWorkspacesChange(({ deleted }) => {
+        if (deleted?.some(workspace => workspace.id === currentWorkspace?.id)) {
+          router.replace('/404?code=kicked');
+        }
+      }),
+    [api, currentWorkspace?.id, dataCenter, router]
+  );
 
   return <>{pageLoaded ? children : null}</>;
 };