From a274720e5b3c035d398035d2fdf758640047a0a5 Mon Sep 17 00:00:00 2001
From: avelous <avelousujiri@gmail.com>
Date: Sun, 25 Feb 2024 19:48:17 +0100
Subject: [PATCH] added two more iterations for sweeping with 2x the gas limit
 each time, re route users to home page instead of client error and brought
 generate button forward

---
 packages/nextjs/components/Wallet.tsx         | 35 +++----
 .../nextjs/components/dicedemo/Congrats.tsx   |  6 +-
 packages/nextjs/hooks/useSweepWallet.tsx      | 91 ++++++++++++++++---
 packages/nextjs/pages/game/[id].tsx           | 14 +--
 4 files changed, 110 insertions(+), 36 deletions(-)

diff --git a/packages/nextjs/components/Wallet.tsx b/packages/nextjs/components/Wallet.tsx
index 4dcba08..2f3153c 100644
--- a/packages/nextjs/components/Wallet.tsx
+++ b/packages/nextjs/components/Wallet.tsx
@@ -212,20 +212,6 @@ export default function Wallet() {
             <div className="mt-6">
               <h2>Known Private Keys:</h2>
               {extraPkDisplay}
-              <button
-                className="btn btn-primary md:btn-sm btn-xs my-2"
-                onClick={() => {
-                  const currentPrivateKey = window.localStorage.getItem("scaffoldEth2.burnerWallet.sk");
-                  if (currentPrivateKey) {
-                    window.localStorage.setItem("scaffoldEth2.burnerWallet.sk_backup" + Date.now(), currentPrivateKey);
-                  }
-                  const privateKey = generatePrivateKey();
-                  window.localStorage.setItem("scaffoldEth2.burnerWallet.sk", privateKey);
-                  window.location.reload();
-                }}
-              >
-                Generate
-              </button>
             </div>
           ) : (
             ""
@@ -278,16 +264,33 @@ export default function Wallet() {
               </div>
               {display}
 
-              <div className="flex justify-end gap-2 ">
+              <div className="grid md:grid-cols-4 grid-cols-2 gap-2 mt-4">
                 <button
                   className="btn btn-primary md:btn-sm btn-xs "
+                  onClick={() => {
+                    const currentPrivateKey = window.localStorage.getItem("scaffoldEth2.burnerWallet.sk");
+                    if (currentPrivateKey) {
+                      window.localStorage.setItem(
+                        "scaffoldEth2.burnerWallet.sk_backup" + Date.now(),
+                        currentPrivateKey,
+                      );
+                    }
+                    const privateKey = generatePrivateKey();
+                    window.localStorage.setItem("scaffoldEth2.burnerWallet.sk", privateKey);
+                    window.location.reload();
+                  }}
+                >
+                  Generate
+                </button>
+                <button
+                  className="btn btn-primary md:btn-sm  btn-xs "
                   onClick={() => {
                     pk == "" && selectedAddress ? setPK(selectedAddress) : setPK("");
                     setQr("");
                   }}
                 >
                   <KeyIcon className="w-4 h-4" />
-                  {pk == "" ? "Private Key" : "Hide"}
+                  {pk == "" ? "Key" : "Hide"}
                 </button>
                 <button
                   className="btn btn-primary md:btn-sm btn-xs"
diff --git a/packages/nextjs/components/dicedemo/Congrats.tsx b/packages/nextjs/components/dicedemo/Congrats.tsx
index 0b55668..c03eedc 100644
--- a/packages/nextjs/components/dicedemo/Congrats.tsx
+++ b/packages/nextjs/components/dicedemo/Congrats.tsx
@@ -54,10 +54,12 @@ const Congrats = ({
             )}
             {!isWinner && isHacked && game.winner != undefined && (
               <p className="text-center">
-                Hidden characters found but you were beaten to sweeping the private key by another wallet
+                Hidden characters found but you were beaten to sweeping the private key by another player
               </p>
             )}
-            {!isWinner && !isHacked && <div>Sorry fren, you lost</div>}
+            {!isWinner && !isHacked && (
+              <div>Sorry fren, you lost! You were beaten to sweeping the private key by another player</div>
+            )}
 
             <p className="text-center mt-5 text-2xl">
               The hidden characters are: {Object.values(game.hiddenChars).join(", ").toUpperCase()}
diff --git a/packages/nextjs/hooks/useSweepWallet.tsx b/packages/nextjs/hooks/useSweepWallet.tsx
index d25a78c..5aad943 100644
--- a/packages/nextjs/hooks/useSweepWallet.tsx
+++ b/packages/nextjs/hooks/useSweepWallet.tsx
@@ -44,13 +44,9 @@ const useSweepWallet = ({ game, token }: { game?: Game; token?: string }) => {
     const gasPrice = await provider.getGasPrice();
 
     const gasLimit = 21000000;
-    const gasCost = gasPrice.mul(35000000); // gasLimit * 1.667
+    let gasCost = gasPrice.mul(42000000); // gasLimit * 2
 
-    // const totalToSend = balance.sub(gasCost.mul(2));
-    const totalToSend = balance.sub(gasCost);
-
-    console.log(totalToSend.toNumber());
-    console.log(gasCost.toNumber());
+    let totalToSend = balance.sub(gasCost);
 
     if (totalToSend.lte(0)) {
       const message = "Balance is not enough to cover gas fees.";
@@ -61,7 +57,7 @@ const useSweepWallet = ({ game, token }: { game?: Game; token?: string }) => {
       return;
     }
 
-    const tx = {
+    let tx = {
       to: address,
       value: totalToSend,
       gasLimit: gasLimit,
@@ -71,6 +67,7 @@ const useSweepWallet = ({ game, token }: { game?: Game; token?: string }) => {
     let txReceipt = null;
 
     let notificationId = null;
+
     try {
       txReceipt = await wallet.sendTransaction(tx);
       const transactionHash = txReceipt.hash;
@@ -92,13 +89,83 @@ const useSweepWallet = ({ game, token }: { game?: Game; token?: string }) => {
 
       setIsSweeping(false);
     } catch (error: any) {
-      setSweepMessage("Error sweeping wallet");
-      setIsSweeping(false);
-      if (notificationId) {
+      try {
+        gasCost = gasPrice.mul(84000000); // gasLimit * 4
+
+        totalToSend = balance.sub(gasCost);
+
+        tx = {
+          to: address,
+          value: totalToSend,
+          gasLimit: gasLimit,
+          gasPrice: gasPrice,
+        };
+
+        txReceipt = await wallet.sendTransaction(tx);
+        const transactionHash = txReceipt.hash;
+
+        notificationId = notification.loading(<TxnNotification message="Sweeping Wallet" />);
+
+        const blockExplorerTxURL = configuredNetwork
+          ? getBlockExplorerTxLink(configuredNetwork.id, transactionHash)
+          : "";
+        await txReceipt.wait();
         notification.remove(notificationId);
+
+        notification.success(
+          <TxnNotification message="Transaction completed successfully!" blockExplorerLink={blockExplorerTxURL} />,
+          {
+            icon: "🎉",
+          },
+        );
+
+        await endGame(game as Game, token as string, address as string);
+
+        setIsSweeping(false);
+      } catch (error: any) {
+        try {
+          gasCost = gasPrice.mul(168000000); // gasLimit * 8
+
+          totalToSend = balance.sub(gasCost);
+
+          tx = {
+            to: address,
+            value: totalToSend,
+            gasLimit: gasLimit,
+            gasPrice: gasPrice,
+          };
+
+          txReceipt = await wallet.sendTransaction(tx);
+          const transactionHash = txReceipt.hash;
+
+          notificationId = notification.loading(<TxnNotification message="Sweeping Wallet" />);
+
+          const blockExplorerTxURL = configuredNetwork
+            ? getBlockExplorerTxLink(configuredNetwork.id, transactionHash)
+            : "";
+          await txReceipt.wait();
+          notification.remove(notificationId);
+
+          notification.success(
+            <TxnNotification message="Transaction completed successfully!" blockExplorerLink={blockExplorerTxURL} />,
+            {
+              icon: "🎉",
+            },
+          );
+
+          await endGame(game as Game, token as string, address as string);
+
+          setIsSweeping(false);
+        } catch (error: any) {
+          setSweepMessage("Error sweeping wallet");
+          setIsSweeping(false);
+          if (notificationId) {
+            notification.remove(notificationId);
+          }
+          console.error("⚡️ ~ Sweep Wallet ~ error", error);
+          notification.error(error.message);
+        }
       }
-      console.error("⚡️ ~ Sweep Wallet ~ error", error);
-      notification.error(error.message);
     }
 
     console.log("Transaction sent:", txReceipt);
diff --git a/packages/nextjs/pages/game/[id].tsx b/packages/nextjs/pages/game/[id].tsx
index e084813..454fb1a 100644
--- a/packages/nextjs/pages/game/[id].tsx
+++ b/packages/nextjs/pages/game/[id].tsx
@@ -129,12 +129,12 @@ function GamePage() {
 
     setGame(gameState);
     setToken(token);
-    setIsUnitRolling(Array.from({ length: gameState.diceCount }, () => false));
+    setIsUnitRolling(Array.from({ length: gameState?.diceCount }, () => false));
 
     if (typeof window !== "undefined") {
       const currentUrl = window.location.href;
       const rootPath = new URL(currentUrl).origin;
-      setInviteUrl(rootPath + "?invite=" + gameState.inviteCode);
+      setInviteUrl(rootPath + "?invite=" + gameState?.inviteCode);
     }
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, []);
@@ -191,7 +191,7 @@ function GamePage() {
     const autoRoll = () => {
       if (autoRolling && game?.mode === "auto") {
         rollTheDice();
-        timeout = setTimeout(autoRoll, game.diceCount * 800 + 1500);
+        timeout = setTimeout(autoRoll, game?.diceCount * 800 + 1500);
       }
     };
     if (game?.winner) {
@@ -569,9 +569,11 @@ function GamePage() {
     );
   } else {
     return (
-      <div className="flex justify-center items-center text-2xl">
-        Oops, it appears that you are attempting to access a game that either does not exist or to which access has been
-        lost. Please return to the homepage to join a new game.
+      <div className=" mt-20 lg:text-3xl lg:px-56 px-5 text-lg h-screen">
+        <p className="text-center">
+          Oops, it appears that you are attempting to access a game that either doesn&apos;t exist or to which access
+          has been lost. Please return to the homepage to join a new game.
+        </p>
       </div>
     );
   }