Switch/Add Networks Metamask SDK

Author: 0xFirekeeper

Assets/Thirdweb/Core/Scripts/Wallet.cs

@@ -18,6 +18,7 @@
 using Nethereum.Hex.HexTypes;
 using Nethereum.ABI.EIP712;
 using Nethereum.Signer.EIP712;
+using Newtonsoft.Json;
 
 //using WalletConnectSharp.NEthereum;
 
@@ -80,16 +81,13 @@ public async Task<string> Connect(WalletConnection walletConnection)
                             oldSession.siweSession
                         );
 
-                        // Switch to chain
                         try
                         {
                             await WalletConnect.Instance.WalletSwitchEthChain(new EthChain() { chainId = ThirdwebManager.Instance.SDK.currentChainData.chainId });
                         }
                         catch (System.Exception e)
                         {
                             Debug.LogWarning("Switching chain error, attempting to add chain: " + e.Message);
-
-                            // Add chain
                             try
                             {
                                 await WalletConnect.Instance.WalletAddEthChain(ThirdwebManager.Instance.SDK.currentChainData);
@@ -152,6 +150,24 @@ public async Task<string> Connect(WalletConnection walletConnection)
 
                         await new WaitUntil(() => connected || !Application.isPlaying);
 
+                        try
+                        {
+                            await MetaMaskUnity.Instance.WalletSwitchEthChain(new EthChain() { chainId = ThirdwebManager.Instance.SDK.currentChainData.chainId });
+                        }
+                        catch (System.Exception e)
+                        {
+                            Debug.LogWarning("Switching chain error, attempting to add chain: " + e.Message);
+                            try
+                            {
+                                await MetaMaskUnity.Instance.WalletAddEthChain(ThirdwebManager.Instance.SDK.currentChainData);
+                                await MetaMaskUnity.Instance.WalletSwitchEthChain(new EthChain() { chainId = ThirdwebManager.Instance.SDK.currentChainData.chainId });
+                            }
+                            catch (System.Exception f)
+                            {
+                                Debug.LogWarning("Adding chain error: " + f.Message);
+                            }
+                        }
+
                         break;
                     default:
                         throw new UnityException("This wallet connection method is not supported on this platform!");

Assets/Thirdweb/Plugins/MetaMask/Plugins/Libraries/SocketIOUnity/Runtime/WebSocketDispatcher.cs

@@ -14,7 +14,7 @@ public static WebSocketDispatcher Instance
         {
             get
             {
-                if (instance == null)
+                if (instance == null && Application.isPlaying)
                 {
                     instance = new GameObject("WebSocket Dispatcher").AddComponent<WebSocketDispatcher>();
                 }

Assets/Thirdweb/Plugins/MetaMask/Scripts/MetaMaskUnity.cs

@@ -9,6 +9,7 @@
 using MetaMask.Transports.Unity.UI;
 using Nethereum.ABI.EIP712;
 using UnityEngine;
+using Newtonsoft.Json;
 
 namespace MetaMask.Unity
 {
@@ -259,6 +260,20 @@ public async Task<string> SignTypedDataV4<T, TDomain>(T data, TypedData<TDomain>
             return result.GetString();
         }
 
+        public async Task<string> WalletAddEthChain(object ethChainData)
+        {
+            var request = new MetaMaskEthereumRequest { Method = "wallet_addEthereumChain", Parameters = new object[] { ethChainData } };
+            var result = await MetaMaskUnity.Instance.Wallet.Request(request);
+            return result.GetString();
+        }
+
+        public async Task<string> WalletSwitchEthChain(object ethChain)
+        {
+            var request = new MetaMaskEthereumRequest { Method = "wallet_switchEthereumChain", Parameters = new object[] { ethChain } };
+            var result = await MetaMaskUnity.Instance.Wallet.Request(request);
+            return result.GetString();
+        }
+
         #endregion
     }
 }