Cleanup AccountPresenter

Author: skibitsky

Packages/com.walletconnect.web3modal/Runtime/Presenters/AccountPresenter.cs

@@ -1,7 +1,5 @@
 using System.Collections.Generic;
 using System.ComponentModel;
-using System.Threading.Tasks;
-using Unity.VectorGraphics;
 using UnityEngine;
 using UnityEngine.UIElements;
 using WalletConnect.UI;
@@ -38,8 +36,7 @@ public AccountPresenter(RouterController router, VisualElement parent) : base(ro
             CreateButtons(View.Buttons);
 
             Web3Modal.AccountController.PropertyChanged += AccountPropertyChangedHandler;
-            // Web3Modal.NetworkController.ChainChanged += ChainChangedHandler;
-            // Web3Modal.AccountChanged += AccountChangedHandler;
+            Web3Modal.NetworkController.ChainChanged += ChainChangedHandler;
         }
 
         private void CreateButtons(VisualElement view)
@@ -82,6 +79,11 @@ private void AccountPropertyChangedHandler(object sender, PropertyChangedEventAr
             }
         }
 
+        private void ChainChangedHandler(object sender, NetworkController.ChainChangedEventArgs e)
+        {
+            UpdateNetworkButton(e.Chain);
+        }
+
         private void UpdateProfileName()
         {
             var profileName = Web3Modal.AccountController.ProfileName;
@@ -116,12 +118,6 @@ protected override void OnVisibleCore()
             base.OnVisibleCore();
             UpdateNetworkButton(Web3Modal.NetworkController.ActiveChain);
         }
-
-        private void UpdateNetworkButton(string chainId)
-        {
-            var chain = Web3Modal.NetworkController.Chains[chainId];
-            UpdateNetworkButton(chain);
-        }
 
         private void UpdateNetworkButton(Chain chain)
         {