fix: better handle chain switch rejection

lifinance · Sep 2, 2022 · 48d34ce · 48d34ce
1 parent dc77cf3
commit 48d34ce
Showing 1 changed file with 18 additions and 7 deletions.
diff --git a/packages/wallet-management/src/browserWallets.ts b/packages/wallet-management/src/browserWallets.ts
@@ -1,26 +1,37 @@
 /* eslint-disable radix */
 import type { Token } from '@lifi/sdk';
-import { getChainById, prefixChainId } from '@lifi/sdk';
+import {
+  getChainById,
+  MetaMaskProviderErrorCode,
+  prefixChainId,
+} from '@lifi/sdk';
 
 export const switchChain = async (chainId: number): Promise<boolean> => {
   return new Promise((resolve, reject) => {
     const { ethereum } = window as any;
     if (!ethereum) resolve(false);
 
     try {
-      ethereum.request({
-        method: 'wallet_switchEthereumChain',
-        params: [{ chainId: getChainById(chainId).metamask?.chainId }],
-      });
+      ethereum
+        .request({
+          method: 'wallet_switchEthereumChain',
+          params: [{ chainId: getChainById(chainId).metamask?.chainId }],
+        })
+        .catch((error: any) => {
+          if (error.code !== MetaMaskProviderErrorCode.userRejectedRequest) {
+            addChain(chainId).then((result) => resolve(result));
+          } else {
+            reject(error);
+          }
+        });
       ethereum.once('chainChanged', (id: string) => {
         if (parseInt(id) === chainId) {
           resolve(true);
         }
       });
     } catch (error: any) {
       // const ERROR_CODE_UNKNOWN_CHAIN = 4902
-      const ERROR_CODE_USER_REJECTED = 4001;
-      if (error.code !== ERROR_CODE_USER_REJECTED) {
+      if (error.code !== MetaMaskProviderErrorCode.userRejectedRequest) {
         addChain(chainId).then((result) => resolve(result));
       } else {
         resolve(false);