From ba564f64c1979218475c2432bfcc7dfd5f8efc29 Mon Sep 17 00:00:00 2001
From: Heikrana <49754221+Heikrana@users.noreply.github.com>
Date: Tue, 8 Nov 2022 23:20:51 +0530
Subject: [PATCH] Add admin functionality

---
 components/AdminControls.tsx | 151 +++++++++++++++++++++++++++++++++++
 pages/index.tsx              |  13 ++-
 styles/globals.css           |   4 +
 3 files changed, 165 insertions(+), 3 deletions(-)
 create mode 100644 components/AdminControls.tsx

diff --git a/components/AdminControls.tsx b/components/AdminControls.tsx
new file mode 100644
index 0000000..db6f925
--- /dev/null
+++ b/components/AdminControls.tsx
@@ -0,0 +1,151 @@
+import React from "react";
+import {
+	StarIcon,
+	CurrencyDollarIcon,
+	ArrowPathIcon,
+	ArrowUturnDownIcon,
+} from "@heroicons/react/24/solid";
+import {
+	useContract,
+	useContractRead,
+	useContractWrite,
+} from "@thirdweb-dev/react";
+import { ethers } from "ethers";
+import { currency } from "../constants";
+import toast from "react-hot-toast";
+
+function AdminControls() {
+	const { contract, isLoading } = useContract(
+		process.env.NEXT_PUBLIC_WALLET_ADDRESS
+	);
+	const { data: totalCommission } = useContractRead(
+		contract,
+		"operatorTotalCommission"
+	);
+	const { mutateAsync: DrawWinnerTicket } = useContractWrite(
+		contract,
+		"DrawWinnerTicket"
+	);
+	const { mutateAsync: RefundAll } = useContractWrite(contract, "RefundAll");
+	const { mutateAsync: WithdrawCommission } = useContractWrite(
+		contract,
+		"WithdrawCommission"
+	);
+	const { mutateAsync: restartDraw } = useContractWrite(
+		contract,
+		"restartDraw"
+	);
+
+	const drawWinner = async () => {
+		const notification = toast.loading("Picking a Lucky Winner...");
+
+		try {
+			const data = await DrawWinnerTicket([{}]);
+
+			toast.success("A Winner has been selected!", {
+				id: notification,
+			});
+
+			console.info("Contract call success", data);
+		} catch (err) {
+			toast.error("Whoops, something went worng!", {
+				id: notification,
+			});
+
+			console.error("Contract call failure", err);
+		}
+	};
+
+	const onWithdrawCommission = async () => {
+		const notification = toast.loading("Withdrawing commission...");
+
+		try {
+			const data = await WithdrawCommission([{}]);
+
+			toast.success("Your commission has been withdrawn successfully!", {
+				id: notification,
+			});
+
+			console.info("Contract call success", data);
+		} catch (err) {
+			toast.error("Whoops, something went worng!", {
+				id: notification,
+			});
+
+			console.error("Contract call failure", err);
+		}
+	};
+
+	const onRefundAll = async () => {
+		const notification = toast.loading("Refunding all...");
+
+		try {
+			const data = await RefundAll([{}]);
+
+			toast.success("All refunded successfully", {
+				id: notification,
+			});
+
+			console.info("Contract call success", data);
+		} catch (err) {
+			toast.error("Whoops, something went worng!", {
+				id: notification,
+			});
+
+			console.error("Contract call failure", err);
+		}
+	};
+
+	const onRestartDraw = async () => {
+		const notification = toast.loading("Restarting draw...");
+
+		try {
+			const data = await restartDraw([{}]);
+
+			toast.success("Draw restarted successfully!", {
+				id: notification,
+			});
+
+			console.info("Contract call success", data);
+		} catch (err) {
+			toast.error("Whoops, something went worng!", {
+				id: notification,
+			});
+
+			console.error("Contract call failure", err);
+		}
+	};
+
+	return (
+		<div className="text-white text-center px-5 py-3 rounded-md border-[#0a396f]/70 border">
+			<h2 className="font-bold">Admin Controls</h2>
+			<p className="mb-5">
+				Total Commission to be withdrawn:{" "}
+				{totalCommission &&
+					ethers.utils.formatEther(totalCommission?.toString())}{" "}
+				{currency}
+			</p>
+
+			<div className="flex flex-col space-y-2 md:flex-row md:space-y-0 md:space-x-2">
+				<button onClick={drawWinner} className="admin-button">
+					<StarIcon className="h-6 mx-auto mb-2" />
+					Draw Winner
+				</button>
+				<button onClick={onWithdrawCommission} className="admin-button">
+					<CurrencyDollarIcon className="h-6 mx-auto mb-2" />
+					Withdraw Commission
+				</button>
+				<button onClick={onRefundAll} className="admin-button">
+					<ArrowUturnDownIcon className="h-6 mx-auto mb-2" />
+					Refund All
+				</button>
+				<button onClick={onRestartDraw} className="admin-button">
+					<ArrowPathIcon className="h-6 mx-auto mb-2" />
+					Restart the Draw
+				</button>
+			</div>
+		</div>
+	);
+}
+
+export default AdminControls;
diff --git a/pages/index.tsx b/pages/index.tsx
index a28bee6..351adb0 100644
--- a/pages/index.tsx
+++ b/pages/index.tsx
@@ -18,6 +18,7 @@ import Login from "../components/Login";
 import Loading from "../components/Loading";
 import CountdownTimer from "../components/CountdownTimer";
 import { ClimbingBoxLoader } from "react-spinners";
+import AdminControls from "../components/AdminControls";
 
 const Home: NextPage = () => {
 	const address = useAddress();
@@ -41,7 +42,7 @@ const Home: NextPage = () => {
 	);
 	const { data: expiration } = useContractRead(contract, "expiration");
 	const { data: ticketPrice } = useContractRead(contract, "ticketPrice");
-	const { data: operatorTotalCommission } = useContractRead(
+	const { data: totalCommission } = useContractRead(
 		contract,
 		"operatorTotalCommission"
 	);
@@ -159,6 +160,12 @@ const Home: NextPage = () => {
 					</div>
 				</Marquee>
 
+				{isLotteryOperator === address && (
+					<div className="flex justify-center">
+						<AdminControls />
+					</div>
+				)}
+
 				{winnings > 0 && (
 					<div className="max-w-md md:max-w-2xl lg:max-w-4xl mx-auto mt-5">
 						<button
@@ -266,9 +273,9 @@ const Home: NextPage = () => {
 								<div className="flex items-center justify-between text-cyan-300 text-xs italic">
 									<p>+ Network Fees</p>
 									<p>
-										{operatorTotalCommission &&
+										{totalCommission &&
 											ethers.utils.formatEther(
-												operatorTotalCommission.toString()
+												totalCommission.toString()
 											)}{" "}
 										{currency}
 									</p>
diff --git a/styles/globals.css b/styles/globals.css
index 8fc07d7..f0bb9b2 100644
--- a/styles/globals.css
+++ b/styles/globals.css
@@ -18,4 +18,8 @@
 	.countdown-label {
 		@apply text-center text-white uppercase text-sm pt-4;
 	}
+
+	.admin-button {
+		@apply bg-[#041f3e] p-2 flex-1 rounded-md border-[#0a396f] border-2 hover:bg-[#0a396f]/70;
+	}
 }