[frontend] fixed workbench creation pop-up in case of error (#8265)

Signed-off-by: stefan1anuby <[email protected]>
OpenCTI-Platform · Sep 26, 2024 · 1aa3e4b · 1aa3e4b
1 parent d507f71
commit 1aa3e4b
Showing 1 changed file with 26 additions and 8 deletions.
diff --git a/...form/opencti-front/src/private/components/common/files/workbench/WorkbenchFileCreator.tsx b/...form/opencti-front/src/private/components/common/files/workbench/WorkbenchFileCreator.tsx
@@ -93,6 +93,10 @@ const WorkbenchFileCreator: FunctionComponent<WorkbenchFileCreatorProps> = ({
   const classes = useStyles();
   const [commitWorkbench] = useApiMutation<WorkbenchFileCreatorMutation>(
     workbenchFileCreatorMutation,
+    undefined,
+    {
+      errorMessage: 'An error occurred while creating the workbench.',
+    },
   );
   const entityId = entity?.id;
   const onSubmitCreate: FormikConfig<WorkbenchFileCreatorFormValues>['onSubmit'] = (values, { setSubmitting, resetForm }) => {
@@ -101,6 +105,20 @@ const WorkbenchFileCreator: FunctionComponent<WorkbenchFileCreatorProps> = ({
     if (!name.endsWith('.json')) {
       name += '.json';
     }
+
+    const handleCompleted = () => {
+      setSubmitting(false);
+      resetForm();
+      handleCloseCreate();
+      onCompleted?.();
+    };
+
+    const handleError = () => {
+      setSubmitting(false);
+      resetForm();
+      handleCloseCreate();
+    };
+
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     const objects: any = [];
     if (entityId) {
@@ -125,10 +143,10 @@ const WorkbenchFileCreator: FunctionComponent<WorkbenchFileCreatorProps> = ({
           commitWorkbench({
             variables: { file, labels: finalLabels, entityId },
             onCompleted: () => {
-              setSubmitting(false);
-              resetForm();
-              handleCloseCreate();
-              onCompleted?.();
+              handleCompleted();
+            },
+            onError: () => {
+              handleError();
             },
           });
         });
@@ -142,10 +160,10 @@ const WorkbenchFileCreator: FunctionComponent<WorkbenchFileCreatorProps> = ({
       commitWorkbench({
         variables: { file, labels: finalLabels, entityId },
         onCompleted: () => {
-          setSubmitting(false);
-          resetForm();
-          handleCloseCreate();
-          onCompleted?.();
+          handleCompleted();
+        },
+        onError: () => {
+          handleError();
         },
       });
     }