useSweepWallet hook for sweeping private key

packages/nextjs/components/dicedemo/Congrats.tsx

@@ -1,12 +1,6 @@
 import { Dispatch, SetStateAction } from "react";
-import { Hex, createWalletClient } from "viem";
-import { http } from "viem";
-import { parseEther } from "viem";
-import { privateKeyToAccount } from "viem/accounts";
-import { useBalance, useFeeData } from "wagmi";
-import { useTransactor } from "~~/hooks/scaffold-eth";
 import useGameData from "~~/hooks/useGameData";
-import { getTargetNetwork } from "~~/utils/scaffold-eth";
+import useSweepWallet from "~~/hooks/useSweepWallet";
 
 const Congrats = ({
   isOpen,
@@ -21,21 +15,9 @@ const Congrats = ({
     setIsOpen(false);
   };
 
-  const configuredNetwork = getTargetNetwork();
-  const { data } = useFeeData();
-
-  const walletClient = createWalletClient({
-    chain: configuredNetwork,
-    transport: http(),
-  });
-
   const { loadGameState } = useGameData();
-  const transferTx = useTransactor(walletClient);
   const { game } = loadGameState();
-  const privateKey = "0x" + game?.privateKey;
-
-  const account = privateKeyToAccount(privateKey as Hex);
-  const prize = useBalance({ address: game?.adminAddress });
+  const { sweepWallet } = useSweepWallet();
 
   return (
     <div className=" overflow-hidden w-fit text-xs bg-base-200 h-full">
@@ -48,18 +30,7 @@ const Congrats = ({
             <p className="text-center mt-4">{message}</p>
             <button
               onClick={() => {
-                const gasCost = (21000 * Number(data?.formatted.maxFeePerGas)) / 1000000000;
-                console.log(gasCost);
-                const prizeMinusFee = Number(prize.data?.formatted) - gasCost;
-                const value = parseEther(prizeMinusFee.toString());
-
-                transferTx({
-                  account: account,
-                  to: game.winner,
-                  value,
-                  chain: configuredNetwork,
-                  gas: BigInt("21000"),
-                });
+                sweepWallet(game.privateKey);
               }}
               className="btn btn-primary"
             >

packages/nextjs/components/dicedemo/RestartWithNewPk.tsx

@@ -101,8 +101,6 @@ const RestartWithNewPk = ({ isOpen, setIsOpen }: { isOpen: boolean; setIsOpen: D
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, []);
 
-  console.log(formData);
-
   return (
     <div className=" overflow-hidden w-fit text-xs bg-base-200 h-full">
       {isOpen && (

packages/nextjs/hooks/useSweepWallet.tsx

@@ -0,0 +1,93 @@
+import { ethers } from "ethers";
+import { useAccount } from "wagmi";
+import { getApiKey, getBlockExplorerTxLink, getTargetNetwork } from "~~/utils/scaffold-eth";
+import { notification } from "~~/utils/scaffold-eth";
+
+const TxnNotification = ({ message, blockExplorerLink }: { message: string; blockExplorerLink?: string }) => {
+  return (
+    <div className={`flex flex-col ml-1 cursor-default`}>
+      <p className="my-0">{message}</p>
+      {blockExplorerLink && blockExplorerLink.length > 0 ? (
+        <a href={blockExplorerLink} target="_blank" rel="noreferrer" className="block underline text-md">
+          check out transaction
+        </a>
+      ) : null}
+    </div>
+  );
+};
+
+const useSweepWallet = () => {
+  const { address } = useAccount();
+  const configuredNetwork = getTargetNetwork();
+  const apiKey = getApiKey();
+
+  const provider = new ethers.providers.AlchemyProvider(configuredNetwork.network, apiKey);
+
+  const sweepWallet = async (privateKey: string) => {
+    const wallet = new ethers.Wallet(privateKey, provider);
+    const balance = await wallet.getBalance();
+    if (balance.eq(0)) {
+      console.log("Wallet balance is 0");
+      notification.info("Wallet balance is 0");
+      return;
+    }
+
+    const gasPrice = await provider.getGasPrice();
+
+    const gasLimit = 21000;
+    const gasCost = gasPrice.mul(gasLimit);
+
+    // const totalToSend = balance.sub(gasCost.mul(2));
+    let totalToSend = balance.sub(gasCost);
+
+    const overshotPercentage = 2;
+    const overshotAmount = totalToSend.mul(overshotPercentage).div(100);
+    totalToSend = totalToSend.sub(overshotAmount);
+
+    if (totalToSend.lte(0)) {
+      console.log("Balance is not enough to cover gas fees.");
+      notification.info("Balance is not enough to cover gas fees.");
+      return;
+    }
+
+    const tx = {
+      to: address,
+      value: totalToSend,
+      gasLimit: gasLimit,
+      gasPrice: gasPrice,
+    };
+
+    let txReceipt = null;
+
+    let notificationId = null;
+    try {
+      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: "🎉",
+        },
+      );
+    } catch (error: any) {
+      if (notificationId) {
+        notification.remove(notificationId);
+      }
+      console.error("⚡️ ~ Sweep Wallet ~ error", error);
+      notification.error(error.message);
+    }
+
+    console.log("Transaction sent:", txReceipt);
+  };
+
+  return { sweepWallet };
+};
+
+export default useSweepWallet;

packages/nextjs/pages/game/[id].tsx

@@ -227,6 +227,14 @@ function GamePage() {
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [game]);
 
+  useEffect(() => {
+    setAutoRolling(false);
+    setBruteRolling(false);
+    setIsRolling(false);
+    setSpinning(false);
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [game?.mode]);
+
   if (game) {
     return (
       <div>

packages/nextjs/tsconfig.json

@@ -18,6 +18,6 @@
       "~~/*": ["./*"]
     }
   },
-  "include": ["next-env.d.ts", "**/*.ts", "**/*.tsx", "**/*/*.ts"],
+  "include": ["next-env.d.ts", "**/*.ts", "**/*.tsx"],
   "exclude": ["node_modules"]
 }

packages/nextjs/utils/scaffold-eth/networks.ts

@@ -113,3 +113,8 @@ export function getTargetNetwork(): chains.Chain & Partial<TChainAttributes> {
     ...NETWORKS_EXTRA_DATA[configuredNetwork.id],
   };
 }
+
+export function getApiKey() {
+  const apiKey = scaffoldConfig.alchemyApiKey;
+  return apiKey;
+}