Add network checks to stake, withdraw, and register operator

consensusnetworks · Oct 26, 2023 · 52ffbb5 · 52ffbb5
1 parent fff422b
commit 52ffbb5
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 19 deletions.
diff --git a/apps/web/src/composables/operators.ts b/apps/web/src/composables/operators.ts
@@ -8,6 +8,7 @@ import useEthers from '@/composables/ethers'
 import useLedger from '@/composables/ledger'
 import useTrezor from '@/composables/trezor'
 import useUser from '@/composables/user'
+import useWallets from '@/composables/wallets'
 import useWalletConnectV2 from '@/composables/walletConnectV2'
 import { CasimirManager, CasimirRegistry, CasimirViews } from '@casimir/ethereum/build/@types'
 
@@ -25,6 +26,7 @@ const { browserProvidersList, getEthersBrowserSigner } = useEthers()
 const { getEthersLedgerSigner } = useLedger()
 const { getEthersTrezorSigner } = useTrezor()
 const { user } = useUser()
+const { detectActiveNetwork, switchEthersNetwork } = useWallets()
 const { getWalletConnectSignerV2 } = useWalletConnectV2()
 const loadingInitializeOperators = ref(false)
 const loadingInitializeOperatorsError = ref(false)
@@ -193,6 +195,11 @@ export default function useOperators() {
 
     // TODO: Move this to operators.ts to combine with AddOperator method
     async function registerOperatorWithCasimir({ walletProvider, address, operatorId, collateral, nodeUrl }: RegisterOperatorWithCasimirParams) {
+        const activeNetwork = await detectActiveNetwork(walletProvider)
+        if (activeNetwork !== 5) {
+            await switchEthersNetwork(walletProvider, '0x5')
+            return window.location.reload()
+        }
         loadingRegisteredOperators.value = true
         try {
             let signer

diff --git a/apps/web/src/composables/staking.ts b/apps/web/src/composables/staking.ts
@@ -6,6 +6,7 @@ import useEnvironment from '@/composables/environment'
 import useEthers from '@/composables/ethers'
 import useLedger from '@/composables/ledger'
 import useTrezor from '@/composables/trezor'
+import useWallets from '@/composables/wallets'
 import useWalletConnectV2 from './walletConnectV2'
 import { CasimirManager } from '@casimir/ethereum/build/@types'
 
@@ -14,6 +15,7 @@ const { provider } = useEnvironment()
 const { browserProvidersList, getEthersBrowserSigner } = useEthers()
 const { getEthersLedgerSigner } = useLedger()
 const { getEthersTrezorSigner } = useTrezor()
+const { detectActiveNetwork, switchEthersNetwork } = useWallets()
 const { getWalletConnectSignerV2 } = useWalletConnectV2()
 
 const stakingComposableInitialized = ref(false)
@@ -38,6 +40,12 @@ export default function useStaking() {
 
     async function deposit({ amount, walletProvider, type }: { amount: string, walletProvider: ProviderString, type: 'default' | 'eigen' }) {
         try {
+            const activeNetwork = await detectActiveNetwork(walletProvider)
+            if (activeNetwork !== 5) {
+                await switchEthersNetwork(walletProvider, '0x5')
+                return window.location.reload()
+            }
+
             let signer
             if (browserProvidersList.includes(walletProvider)) {
                 signer = getEthersBrowserSigner(walletProvider)
@@ -111,6 +119,12 @@ export default function useStaking() {
     }
 
     async function withdraw({ amount, walletProvider, type }: { amount: string, walletProvider: ProviderString, type: 'default' | 'eigen' }) {
+        const activeNetwork = await detectActiveNetwork(walletProvider)
+        if (activeNetwork !== 5) {
+            await switchEthersNetwork(walletProvider, '0x5')
+            return window.location.reload()
+        }
+
         let signer
         if (browserProvidersList.includes(walletProvider)) {
             signer = getEthersBrowserSigner(walletProvider)

diff --git a/apps/web/src/composables/wallets.ts b/apps/web/src/composables/wallets.ts
@@ -67,17 +67,17 @@ export default function useWallets() {
         try {
             await provider.request({
                 method:'wallet_switchEthereumChain',
-                params: [{chainId: chainId}]
+                params: [{ chainId }]
             })
         } catch (err: any) {
             console.log(`Error occurred while switching chain to chainId ${chainId}, err: ${err.message} code: ${err.code}`)
-            if (err.code === 4902){
-                if (chainId === '5') {
-                    addEthersNetwork(providerString, goerliNetwork)
-                } else if (chainId === '0x1252') {
-                    addEthersNetwork(providerString, iotexNetwork)
-                }
-            }
+            // if (err.code === 4902){
+            //     if (chainId === '5') {
+            //         addEthersNetwork(providerString, goerliNetwork)
+            //     } else if (chainId === '0x1252') {
+            //         addEthersNetwork(providerString, iotexNetwork)
+            //     }
+            // }
         }
     }
 
@@ -90,14 +90,14 @@ export default function useWallets() {
     }
 }
 
-async function addEthersNetwork (providerString: ProviderString, network: any) {
-    const provider = getBrowserProvider(providerString)
-    try {
-      await provider.request({
-        method: 'wallet_addEthereumChain',
-        params: [network]
-      })
-    } catch(error: any) {
-      console.log(`Error occurred while adding network ${network.chainName}, Message: ${error.message} Code: ${error.code}`)
-    }
-  }
+// async function addEthersNetwork (providerString: ProviderString, network: any) {
+//     const provider = getBrowserProvider(providerString)
+//     try {
+//       await provider.request({
+//         method: 'wallet_addEthereumChain',
+//         params: [network]
+//       })
+//     } catch(error: any) {
+//       console.log(`Error occurred while adding network ${network.chainName}, Message: ${error.message} Code: ${error.code}`)
+//     }
+//   }