From 8f8b1164fe62857bbe3a7cb3100769ab9c0a60c8 Mon Sep 17 00:00:00 2001
From: D <ddisdead462@gmail.com>
Date: Thu, 8 Aug 2024 16:17:05 +0500
Subject: [PATCH] approve stake

---
 src/handler/index.ts | 42 ++++++++++++++++++++++++++++++------------
 1 file changed, 30 insertions(+), 12 deletions(-)

diff --git a/src/handler/index.ts b/src/handler/index.ts
index b3bf982..0f53bb5 100644
--- a/src/handler/index.ts
+++ b/src/handler/index.ts
@@ -120,13 +120,11 @@ export async function listenEvents(
         if (!tx?.status) throw new Error("Approve failed");
         return tx;
       } catch (err) {
-        if (
-          (err as { shortMessage: string }).shortMessage ===
-          'execution reverted: "Signature already used"'
-        ) {
+        const err_ = err as unknown as { shortMessage: string };
+        if (err_.shortMessage.includes("Signature already used")) {
           return null;
         }
-        log.error(err, "Error while approving");
+        log.error(err_.shortMessage, "Error while approving");
         throw err;
       }
     };
@@ -214,12 +212,30 @@ export async function listenStakeEvents(
         });
 
         const approvalFn = async () => {
-          const tx = await deps.storage.approveStake(
-            currentValidatorAddress,
-            signatures,
-          );
-          if (!(await tx.wait())?.status) throw new Error("TxFailed");
-          return tx;
+          try {
+            const tx = Promise.race([
+              (
+                await deps.storage.approveStake(
+                  currentValidatorAddress,
+                  signatures,
+                )
+              ).wait(),
+              setTimeout(10 * 1000),
+            ]);
+
+            // @ts-ignore
+            if (!tx?.status) throw new Error("TxFailed");
+            return tx;
+          } catch (err) {
+            const err_ = err as unknown as { shortMessage: string };
+            if (
+              err_.shortMessage.includes("Already voted for this validator")
+            ) {
+              return null;
+            }
+            log.error(err_.shortMessage, "Error while approving");
+            throw err;
+          }
         };
         const approved = await retry(
           approvalFn,
@@ -228,7 +244,9 @@ export async function listenStakeEvents(
           6,
         );
         log.info(
-          `Approved and Signed Data for Staking Chain at TX: ${approved.hash}`,
+          approved
+            ? `Approved and Signed Data for Staking Chain at TX: ${approved.hash}`
+            : `Already approved for ${currentValidatorAddress}`,
         );
       }
     });