From a0c9f9a49c25668cc85df6e7cd1de7232b0cad59 Mon Sep 17 00:00:00 2001
From: DiamondThree <857159145@qq.com>
Date: Thu, 12 Jan 2023 17:33:12 +0800
Subject: [PATCH] feat: add backhome

---
 packages/app/src/components/404/index.tsx | 16 +++++++++++++++-
 packages/app/src/pages/404.tsx            |  4 ----
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/packages/app/src/components/404/index.tsx b/packages/app/src/components/404/index.tsx
index 8ed423280..36f1b90a3 100644
--- a/packages/app/src/components/404/index.tsx
+++ b/packages/app/src/components/404/index.tsx
@@ -1,10 +1,24 @@
 import { NotFoundTitle, PageContainer } from './styles';
 import { useTranslation } from '@affine/i18n';
+import { Button } from '@/ui/button';
+import { useRouter } from 'next/router';
 export const NotfoundPage = () => {
   const { t } = useTranslation();
+  const router = useRouter();
   return (
     <PageContainer>
-      <NotFoundTitle>{t('404 - Page Not Found')}</NotFoundTitle>
+      <NotFoundTitle>
+        {t('404 - Page Not Found')}
+        <p>
+          <Button
+            onClick={() => {
+              router.push('/workspace');
+            }}
+          >
+            Back Home
+          </Button>
+        </p>
+      </NotFoundTitle>
     </PageContainer>
   );
 };
diff --git a/packages/app/src/pages/404.tsx b/packages/app/src/pages/404.tsx
index 74f25f07e..0eef5261e 100644
--- a/packages/app/src/pages/404.tsx
+++ b/packages/app/src/pages/404.tsx
@@ -1,9 +1,5 @@
 import NotfoundPage from '@/components/404';
-import { useAppState } from '@/providers/app-state-provider';
 
 export default function Custom404() {
-  const { workspaceList } = useAppState();
-  console.log('workspaceList: ', workspaceList);
-
   return <NotfoundPage></NotfoundPage>;
 }