From 7e161682f0b0f005dca8ecdef4e82590531e3c0a Mon Sep 17 00:00:00 2001
From: fundon <27926+fundon@users.noreply.github.com>
Date: Sat, 23 Mar 2024 12:29:46 +0000
Subject: [PATCH] fix(core): creating multiple workspaces with consecutive
 clicks (#6259)

Closes #6213
---
 .../affine/create-workspace-modal/index.tsx        | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/packages/frontend/core/src/components/affine/create-workspace-modal/index.tsx b/packages/frontend/core/src/components/affine/create-workspace-modal/index.tsx
index 02d10eb3a..e71c82261 100644
--- a/packages/frontend/core/src/components/affine/create-workspace-modal/index.tsx
+++ b/packages/frontend/core/src/components/affine/create-workspace-modal/index.tsx
@@ -39,6 +39,7 @@ interface ModalProps {
 }
 
 interface NameWorkspaceContentProps extends ConfirmModalProps {
+  loading: boolean;
   onConfirmName: (
     name: string,
     workspaceFlavour: WorkspaceFlavour,
@@ -50,6 +51,7 @@ const shouldEnableCloud =
   !runtimeConfig.allowLocalWorkspace && !environment.isDesktop;
 
 const NameWorkspaceContent = ({
+  loading,
   onConfirmName,
   ...props
 }: NameWorkspaceContentProps) => {
@@ -108,7 +110,7 @@ const NameWorkspaceContent = ({
       cancelText={t['com.affine.nameWorkspace.button.cancel']()}
       confirmButtonOptions={{
         type: 'primary',
-        disabled: !workspaceName,
+        disabled: !workspaceName || loading,
         ['data-testid' as string]: 'create-workspace-create-button',
         children: t['com.affine.nameWorkspace.button.create'](),
       }}
@@ -180,6 +182,7 @@ export const CreateWorkspaceModal = ({
   const [step, setStep] = useState<CreateWorkspaceStep>();
   const t = useAFFiNEI18N();
   const workspaceManager = useService(WorkspaceManager);
+  const [loading, setLoading] = useState(false);
 
   // todo: maybe refactor using xstate?
   useLayoutEffect(() => {
@@ -224,6 +227,9 @@ export const CreateWorkspaceModal = ({
 
   const onConfirmName = useAsyncCallback(
     async (name: string, workspaceFlavour: WorkspaceFlavour) => {
+      if (loading) return;
+      setLoading(true);
+
       // this will be the last step for web for now
       // fix me later
       if (runtimeConfig.enablePreloading) {
@@ -247,8 +253,10 @@ export const CreateWorkspaceModal = ({
         );
         onCreate(id);
       }
+
+      setLoading(false);
     },
-    [onCreate, workspaceManager]
+    [loading, onCreate, workspaceManager]
   );
 
   const onOpenChange = useCallback(
@@ -259,9 +267,11 @@ export const CreateWorkspaceModal = ({
     },
     [onClose]
   );
+
   if (step === 'name-workspace') {
     return (
       <NameWorkspaceContent
+        loading={loading}
         open={mode !== false && !!step}
         onOpenChange={onOpenChange}
         onConfirmName={onConfirmName}