fix loader issue

hats-finance · Jan 30, 2025 · ad9e128 · ad9e128
1 parent 8a3fb61
commit ad9e128
Showing 1 changed file with 34 additions and 2 deletions.
diff --git a/packages/web/src/pages/Submissions/SubmissionFormPage/SubmissionFormPage.tsx b/packages/web/src/pages/Submissions/SubmissionFormPage/SubmissionFormPage.tsx
@@ -77,11 +77,15 @@ export const SubmissionFormPage = () => {
     reset: callReset,
     send: sendVulnerabilityOnChain,
     isLoading: isSigningSubmission,
+    error: contractError
   } = LogClaimContract.hook(vault);
   const { isLoading: isSubmitting } = useWaitForTransaction({
     hash: callData?.hash,
     onSettled(data, error) {
-      if (error) return reset();
+      if (error) {
+        reset();
+        return;
+      }
 
       if (submissionData && data?.transactionHash && chain?.id) {
         const newVulnerabilityData: ISubmissionData = {
@@ -97,7 +101,21 @@ export const SubmissionFormPage = () => {
         };
 
         setSubmissionData(newVulnerabilityData);
-        sendSubmissionToServer(newVulnerabilityData);
+        sendSubmissionToServer(newVulnerabilityData).catch(() => {
+          // If server submission fails, update the UI to show the error
+          setSubmissionData({
+            ...newVulnerabilityData,
+            submissionResult: {
+              ...newVulnerabilityData.submissionResult,
+              verified: true,
+              txStatus: SubmissionOpStatus.Fail,
+              botStatus: SubmissionOpStatus.Fail,
+              transactionHash: data?.transactionHash,
+              chainId: chain?.id,
+              auditCompetitionRepo: undefined,
+            },
+          });
+        });
       }
     },
   });
@@ -109,6 +127,20 @@ export const SubmissionFormPage = () => {
     setAllFormDisabled(false);
   };
 
+  // Handle contract errors
+  useEffect(() => {
+    if (contractError) {
+      reset();
+    }
+  }, [contractError]);
+
+  // Clear submission data if chain changes
+  useEffect(() => {
+    if (submissionData?.submissionResult?.chainId && chain?.id && submissionData.submissionResult.chainId !== chain.id) {
+      reset();
+    }
+  }, [chain?.id]);
+
   // Loads initial state of the vault
   useEffect(() => {
     if (!activeVaults || activeVaults.length === 0) return;