From 0d2a1a593d268662e6b3d47ed355818443e53f95 Mon Sep 17 00:00:00 2001
From: Willian Mitsuda <wmitsuda@gmail.com>
Date: Mon, 27 Dec 2021 19:20:13 -0300
Subject: [PATCH] Add next/last nonce navigation support

---
 src/transaction/Details.tsx  |  6 +-----
 src/transaction/NavNonce.tsx | 23 +++++++++++++----------
 src/useErigonHooks.ts        | 24 ++++++++++++++++++++++++
 3 files changed, 38 insertions(+), 15 deletions(-)

diff --git a/src/transaction/Details.tsx b/src/transaction/Details.tsx
index 8c0d779..261e651 100644
--- a/src/transaction/Details.tsx
+++ b/src/transaction/Details.tsx
@@ -254,11 +254,7 @@ const Details: React.FC<DetailsProps> = ({
           </div>
           <div className="flex items-baseline pl-3">
             <Nonce value={txData.nonce} />
-            <NavNonce
-              sender={txData.from}
-              nonce={txData.nonce}
-              latestBlockNumber={undefined}
-            />
+            <NavNonce sender={txData.from} nonce={txData.nonce} />
           </div>
         </div>
       </InfoRow>
diff --git a/src/transaction/NavNonce.tsx b/src/transaction/NavNonce.tsx
index f17c5cc..236b30f 100644
--- a/src/transaction/NavNonce.tsx
+++ b/src/transaction/NavNonce.tsx
@@ -5,19 +5,17 @@ import { faChevronRight } from "@fortawesome/free-solid-svg-icons/faChevronRight
 import NavButton from "./NavButton";
 import { ChecksummedAddress } from "../types";
 import { RuntimeContext } from "../useRuntime";
-import { useTransactionBySenderAndNonce } from "../useErigonHooks";
+import {
+  useTransactionBySenderAndNonce,
+  useTransactionCount,
+} from "../useErigonHooks";
 
 type NavNonceProps = {
   sender: ChecksummedAddress;
   nonce: number;
-  latestBlockNumber: number | undefined;
 };
 
-const NavNonce: React.FC<NavNonceProps> = ({
-  sender,
-  nonce,
-  latestBlockNumber,
-}) => {
+const NavNonce: React.FC<NavNonceProps> = ({ sender, nonce }) => {
   const { provider } = useContext(RuntimeContext);
   const prevTxHash = useTransactionBySenderAndNonce(
     provider,
@@ -29,7 +27,12 @@ const NavNonce: React.FC<NavNonceProps> = ({
     sender,
     nonce + 1
   );
-  const lastTxHash = nextTxHash;
+  const count = useTransactionCount(provider, sender);
+  const lastTxHash = useTransactionBySenderAndNonce(
+    provider,
+    sender,
+    count !== undefined ? count - 1 : undefined
+  );
 
   return (
     <div className="pl-2 self-center flex space-x-1">
@@ -38,13 +41,13 @@ const NavNonce: React.FC<NavNonceProps> = ({
       </NavButton>
       <NavButton
         txHash={nextTxHash}
-        disabled={latestBlockNumber === undefined || nonce >= latestBlockNumber}
+        disabled={count === undefined || nonce >= count - 1}
       >
         <FontAwesomeIcon icon={faChevronRight} />
       </NavButton>
       <NavButton
         txHash={lastTxHash}
-        disabled={latestBlockNumber === undefined || nonce >= latestBlockNumber}
+        disabled={count === undefined || nonce >= count - 1}
       >
         <FontAwesomeIcon icon={faChevronRight} />
         <FontAwesomeIcon icon={faChevronRight} />
diff --git a/src/useErigonHooks.ts b/src/useErigonHooks.ts
index d26fcd0..71b95cc 100644
--- a/src/useErigonHooks.ts
+++ b/src/useErigonHooks.ts
@@ -503,6 +503,30 @@ export const useTransactionError = (
   return [errorMsg, data, isCustomError];
 };
 
+export const useTransactionCount = (
+  provider: JsonRpcProvider | undefined,
+  sender: ChecksummedAddress | undefined
+): number | undefined => {
+  const [count, setCount] = useState<number | undefined>();
+
+  useEffect(() => {
+    // Reset
+    setCount(undefined);
+
+    if (provider === undefined || sender === undefined) {
+      return;
+    }
+
+    const readCount = async () => {
+      const _count = await provider.getTransactionCount(sender);
+      setCount(_count);
+    };
+    readCount();
+  }, [provider, sender]);
+
+  return count;
+};
+
 export const useTransactionBySenderAndNonce = (
   provider: JsonRpcProvider | undefined,
   sender: ChecksummedAddress | undefined,