fix: switch workspace

2026-07-26 14:58:55 +08:00 · 2022-12-22 23:19:53 +08:00
parent e1ca69aaf9
commit 2665d0c5cd
4 changed files with 51 additions and 47 deletions
@@ -0,0 +1,35 @@
+import { useRouter } from 'next/router';
+import { useAppState } from '@/providers/app-state-provider/context';
+import { useEffect, useState } from 'react';
+
+export const useInitWorkspace = (disabled?: boolean) => {
+  const [loading, setLoading] = useState(true);
+  const router = useRouter();
+  const {
+    workspacesMeta,
+    loadWorkspace,
+    currentWorkspace,
+    currentWorkspaceId,
+  } = useAppState();
+  const workspaceId =
+    (router.query.workspaceId as string) ||
+    workspacesMeta?.[0]?.id ||
+    new Date().getTime().toString();
+
+  useEffect(() => {
+    if (disabled) {
+      setLoading(false);
+      return;
+    }
+    setLoading(true);
+    loadWorkspace(workspaceId).finally(() => {
+      setLoading(false);
+    });
+  }, [workspaceId, disabled]);
+
+  return {
+    workspaceId,
+    workspace: workspaceId === currentWorkspaceId ? currentWorkspace : null,
+    loading,
+  };
+};