network switching tweaks part 2

Uniswap · Dec 10, 2024 · df23201 · df23201
1 parent aac4b00
commit df23201
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/frontend/src/hooks/useBalanceDisplay.ts b/frontend/src/hooks/useBalanceDisplay.ts
@@ -75,6 +75,7 @@ export function useBalanceDisplay() {
             title: 'Switching Network',
             message: `Please confirm the network switch in your wallet...`,
             txHash: tempTxId,
+            stage: 'initiated',
             autoHide: false,
           });
 
@@ -95,6 +96,7 @@ export function useBalanceDisplay() {
             title: 'Network Switched',
             message: `Successfully switched to ${chainNames[chainId] || `Chain ${chainId}`}`,
             txHash: tempTxId,
+            stage: 'confirmed',
             autoHide: true,
           });
         } catch (switchError) {
@@ -104,6 +106,7 @@ export function useBalanceDisplay() {
               title: 'Network Not Found',
               message: 'Please add this network to your wallet first.',
               txHash: tempTxId,
+              stage: 'confirmed',
               autoHide: true,
             });
           } else {
@@ -116,6 +119,7 @@ export function useBalanceDisplay() {
                   ? switchError.message
                   : 'Failed to switch network. Please switch manually.',
               txHash: tempTxId,
+              stage: 'confirmed',
               autoHide: true,
             });
           }
@@ -150,7 +154,6 @@ export function useBalanceDisplay() {
   );
 
   const handleInitiateWithdrawal = useCallback((lockId: string) => {
-    // Remove manual network switching since wagmi will handle it
     setSelectedLockId(lockId);
     setIsWithdrawalDialogOpen(true);
   }, []);
@@ -173,6 +176,7 @@ export function useBalanceDisplay() {
             title: 'Switching Network',
             message: `Please confirm the network switch in your wallet...`,
             txHash: tempTxId,
+            stage: 'initiated',
             autoHide: false,
           });
 
@@ -193,6 +197,7 @@ export function useBalanceDisplay() {
             title: 'Network Switched',
             message: `Successfully switched to ${chainNames[chainId] || `Chain ${chainId}`}`,
             txHash: tempTxId,
+            stage: 'confirmed',
             autoHide: true,
           });
         } catch (switchError) {
@@ -202,6 +207,7 @@ export function useBalanceDisplay() {
               title: 'Network Not Found',
               message: 'Please add this network to your wallet first.',
               txHash: tempTxId,
+              stage: 'confirmed',
               autoHide: true,
             });
           } else {
@@ -214,6 +220,7 @@ export function useBalanceDisplay() {
                   ? switchError.message
                   : 'Failed to switch network. Please switch manually.',
               txHash: tempTxId,
+              stage: 'confirmed',
               autoHide: true,
             });
           }