From 8b0a212b78b88d4b5b5291be382fbccbe3b9e2b2 Mon Sep 17 00:00:00 2001
From: RcleydsonR <rafael.cleydson@gmail.com>
Date: Thu, 12 Jan 2023 18:08:08 -0300
Subject: [PATCH] Add listener to listen chain change and update store network
 name state, also start refactor with getProvider function

Co-authored-by: brunoedcf <brest.dallacosta@outlook.com>
---
 src/blockchain/provider.ts | 15 +++++++++++++++
 src/utils/blockchain.ts    | 22 ++++++++++++++++------
 2 files changed, 31 insertions(+), 6 deletions(-)
 create mode 100644 src/blockchain/provider.ts

diff --git a/src/blockchain/provider.ts b/src/blockchain/provider.ts
new file mode 100644
index 0000000..b83143a
--- /dev/null
+++ b/src/blockchain/provider.ts
@@ -0,0 +1,15 @@
+import { ethers } from "ethers";
+import { useEtherStore } from "@/store/ether";
+
+const getProvider = (): ethers.providers.Web3Provider | null => {
+    const etherStore = useEtherStore();
+
+    const window_ = window as any;
+    const connection = window_.ethereum;
+  
+    if (!connection) return null;
+  
+    return new ethers.providers.Web3Provider(connection);
+};
+
+export { getProvider }
\ No newline at end of file
diff --git a/src/utils/blockchain.ts b/src/utils/blockchain.ts
index 9a02fd4..a69057b 100644
--- a/src/utils/blockchain.ts
+++ b/src/utils/blockchain.ts
@@ -7,6 +7,8 @@ import p2pix from "./smart_contract_files/P2PIX.json";
 import addresses from "./smart_contract_files/localhost.json";
 // Mock wallets import
 import { wallets } from "./smart_contract_files/wallets.json";
+import { getProvider } from "../blockchain/provider";
+import { NetworkEnum } from "@/model/NetworkEnum";
 
 // Wallet methods
 // Update wallet state (balance and address)
@@ -272,20 +274,28 @@ const connectProvider = async () => {
   await updateDepositAddedEvents();
   await updateLockAddedEvents();
   await updateLockReleasedEvents();
+  listenToNetworkChange(connection);
 
   connection.on("accountsChanged", async () => {
     await updateWalletStatus();
   });
 };
 
-const getProvider = (): ethers.providers.Web3Provider | null => {
-  const window_ = window as any;
-  const connection = window_.ethereum;
+const listenToNetworkChange = (connection: any) => {
+  const etherStore = useEtherStore();
 
-  if (!connection) return null;
+  const possibleNetworks: {[key: string]: NetworkEnum} = {
+    "0x5": NetworkEnum.ethereum,
+    "0x13881": NetworkEnum.polygon,
+    "0x7a69": NetworkEnum.localhost,
+  }
 
-  return new ethers.providers.Web3Provider(connection);
-};
+  connection.on("chainChanged", (networkChain: string) => {
+    if (Object.keys(possibleNetworks).includes(networkChain)){
+      etherStore.setNetworkName(possibleNetworks[networkChain]);
+    }
+  });
+}
 
 // Deposit methods
 const approveTokens = async (tokenQty: Number) => {