fix: close unbonding modal after successful request (#522)

babylonlabs-io · Dec 18, 2024 · fd97b55 · fd97b55
1 parent f54c07f
commit fd97b55
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/app/components/Delegations/Delegations.tsx b/src/app/components/Delegations/Delegations.tsx
@@ -98,7 +98,6 @@ export const Delegations = ({}) => {
     });
   };
 
-  // TODO: Hook up with unbonding modal
   const handleUnbond = async (id: string) => {
     try {
       if (selectedDelegation?.stakingTxHashHex != id) {
@@ -107,7 +106,7 @@ export const Delegations = ({}) => {
       // Sign the withdrawal transaction
       const { stakingTx, finalityProviderPkHex, stakingValueSat } =
         selectedDelegation;
-
+      setAwaitingWalletResponse(true);
       await submitUnbondingTx(
         {
           stakingTimelock: stakingTx.timelock,
@@ -129,6 +128,11 @@ export const Delegations = ({}) => {
           errorState: ErrorState.UNBONDING,
         },
       });
+    } finally {
+      setModalOpen(false);
+      setTxID("");
+      setModalMode(undefined);
+      setAwaitingWalletResponse(false);
     }
   };