sanity-plugin-workflow 1.0.5 → 1.0.6

package/lib/index.js

@@ -295,6 +295,21 @@ function BeginWorkflow(props) {
     }
   };
 }
+const handleDeleteMetadata = async (client, toast, id) => {
+  try {
+    await client.delete("workflow-metadata.".concat(id));
+    toast.push({
+      status: "success",
+      title: "Workflow completed"
+    });
+  } catch (error) {
+    console.error(error);
+    toast.push({
+      status: "error",
+      title: "Could not complete Workflow"
+    });
+  }
+};
 function CompleteWorkflow(props) {
   const {
     id
@@ -308,12 +323,13 @@ function CompleteWorkflow(props) {
   const client = sanity.useClient({
     apiVersion: API_VERSION
   });
+  const toast = ui.useToast();
   if (error) {
     console.error(error);
   }
-  const handle = React.useCallback(() => {
-    client.delete("workflow-metadata.".concat(id));
-  }, [id, client]);
+  const handle = React.useCallback(async () => {
+    await handleDeleteMetadata(client, toast, id);
+  }, [client, toast, id]);
   if (!metadata) {
     return null;
   }
@@ -325,8 +341,9 @@ function CompleteWorkflow(props) {
     disabled: loading || error || !isLastState,
     label: "Complete Workflow",
     title: isLastState ? "Removes the document from the Workflow process" : "Cannot remove from workflow until in the last state",
-    onHandle: () => {
-      handle();
+    onHandle: async () => {
+      await handle();
+      props.onComplete();
     },
     color: "positive"
   };
@@ -889,17 +906,7 @@ function CompleteButton(props) {
     if (!id) {
       return;
     }
-    client.delete("workflow-metadata.".concat(id)).then(() => {
-      toast.push({
-        status: "success",
-        title: "Workflow completed"
-      });
-    }).catch(() => {
-      toast.push({
-        status: "error",
-        title: "Could not complete Workflow"
-      });
-    });
+    handleDeleteMetadata(client, toast, id);
   }, [client, toast]);
   return /* @__PURE__ */jsxRuntime.jsx(ui.Tooltip, {
     portal: true,