From 22a0bc58883729c026bbb1719b620e9d49cb002d Mon Sep 17 00:00:00 2001
From: Willian Mitsuda <wmitsuda@gmail.com>
Date: Sat, 19 Feb 2022 08:01:34 -0300
Subject: [PATCH] Add current balance

---
 src/address/AddressTransactionResults.tsx | 10 ++++++++--
 src/useErigonHooks.ts                     | 21 +++++++++++++++++++++
 2 files changed, 29 insertions(+), 2 deletions(-)

diff --git a/src/address/AddressTransactionResults.tsx b/src/address/AddressTransactionResults.tsx
index 4fd7b25..48fb2d9 100644
--- a/src/address/AddressTransactionResults.tsx
+++ b/src/address/AddressTransactionResults.tsx
@@ -2,6 +2,7 @@ import React, { useContext, useEffect, useMemo, useState } from "react";
 import { BlockTag } from "@ethersproject/providers";
 import ContentFrame from "../ContentFrame";
 import InfoRow from "../components/InfoRow";
+import FormattedBalance from "../components/FormattedBalance";
 import TransactionAddress from "../components/TransactionAddress";
 import Copy from "../components/Copy";
 import TransactionLink from "../components/TransactionLink";
@@ -17,7 +18,7 @@ import { RuntimeContext } from "../useRuntime";
 import { useParams, useSearchParams } from "react-router-dom";
 import { ChecksummedAddress, ProcessedTransaction } from "../types";
 import { useContractsMetadata } from "../hooks";
-import { useContractCreator } from "../useErigonHooks";
+import { useAddressBalance, useContractCreator } from "../useErigonHooks";
 
 type AddressTransactionResultsProps = {
   address: ChecksummedAddress;
@@ -122,12 +123,17 @@ const AddressTransactionResults: React.FC<AddressTransactionResultsProps> = ({
     return _addresses;
   }, [address, page]);
   const metadatas = useContractsMetadata(addresses, provider);
+  const balance = useAddressBalance(provider, address);
   const creator = useContractCreator(provider, address);
 
   return (
     <ContentFrame tabs>
       <SelectionContext.Provider value={selectionCtx}>
-        <InfoRow title="Balance"></InfoRow>
+        {balance && (
+          <InfoRow title="Balance">
+            <FormattedBalance value={balance} /> Ether{" "}
+          </InfoRow>
+        )}
         {creator && (
           <InfoRow title="Contract creator">
             <div className="flex items-baseline space-x-2 -ml-1">
diff --git a/src/useErigonHooks.ts b/src/useErigonHooks.ts
index 0039963..5db5ee1 100644
--- a/src/useErigonHooks.ts
+++ b/src/useErigonHooks.ts
@@ -658,3 +658,24 @@ const getContractCreatorFetcher =
     }
     return result;
   };
+
+export const useAddressBalance = (
+  provider: JsonRpcProvider | undefined,
+  address: ChecksummedAddress | undefined
+): BigNumber | null | undefined => {
+  const [balance, setBalance] = useState<BigNumber | undefined>();
+
+  useEffect(() => {
+    if (!provider || !address) {
+      return undefined;
+    }
+
+    const readBalance = async () => {
+      const _balance = await provider.getBalance(address);
+      setBalance(_balance);
+    };
+    readBalance();
+  }, [provider, address]);
+
+  return balance;
+};