From e0dd94ae41f5119b0e61cc3270ef8ab829613b6c Mon Sep 17 00:00:00 2001
From: Himself65 <himself65@outlook.com>
Date: Mon, 6 Mar 2023 22:01:57 -0600
Subject: [PATCH] fix: public workspace state (#1366)

---
 apps/web/src/hooks/affine/use-toggle-workspace-publish.ts | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/apps/web/src/hooks/affine/use-toggle-workspace-publish.ts b/apps/web/src/hooks/affine/use-toggle-workspace-publish.ts
index c7c3998ed2..b297725de6 100644
--- a/apps/web/src/hooks/affine/use-toggle-workspace-publish.ts
+++ b/apps/web/src/hooks/affine/use-toggle-workspace-publish.ts
@@ -1,5 +1,5 @@
 import { useCallback } from 'react';
-import { mutate } from 'swr';
+import useSWR from 'swr';
 
 import { jotaiStore, jotaiWorkspacesAtom } from '../../atoms';
 import { QueryKey } from '../../plugins/affine/fetcher';
@@ -7,6 +7,7 @@ import { AffineWorkspace } from '../../shared';
 import { apis } from '../../shared/apis';
 
 export function useToggleWorkspacePublish(workspace: AffineWorkspace) {
+  const { mutate } = useSWR(QueryKey.getWorkspaces);
   return useCallback(
     async (isPublish: boolean) => {
       await apis.updateWorkspace({
@@ -19,6 +20,6 @@ export function useToggleWorkspacePublish(workspace: AffineWorkspace) {
         ...jotaiStore.get(jotaiWorkspacesAtom),
       ]);
     },
-    [workspace]
+    [mutate, workspace.id]
   );
 }