chore: complete restake logic

Abstracted-Labs · Dec 18, 2023 · 93f1fdf · 93f1fdf
1 parent 6028b51
commit 93f1fdf
Show file tree

Hide file tree

Showing 5 changed files with 65 additions and 45 deletions.
diff --git a/src/components/LoadingSpinner.tsx b/src/components/LoadingSpinner.tsx
@@ -1,6 +1,6 @@
 const LoadingSpinner = () => {
   return (
-    <div className="flex h-full w-full items-center justify-center">
+    <div className="flex flex-col items-center justify-center">
       <svg
         className="-ml-1 mr-3 h-6 w-6 animate-spin text-white"
         xmlns="http://www.w3.org/2000/svg"

diff --git a/src/routes/claim.tsx b/src/routes/claim.tsx
@@ -238,8 +238,9 @@ const Claim = () => {
   return (
     <div className="overflow-hidden mx-auto w-full flex max-w-7xl flex-col justify-between p-4 sm:px-6 lg:px-8 mt-14 md:mt-0">
       <div className="z-10 w-full sm:max-w-3xl">
-        <h2 className="lg:text-xl font-bold my-3">
+        <h2 className="lg:text-xl font-bold flex flex-row items-center gap-4 my-3">
           <span>Claim Vesting</span>
+          <span>{isBalanceLoading ? <LoadingSpinner /> : null}</span>
         </h2>
 
         {!selectedAccount ? (
@@ -253,8 +254,6 @@ const Claim = () => {
           </div>
         ) : null}
 
-        {isBalanceLoading ? <LoadingSpinner /> : null}
-
         {!isBalanceLoading && selectedAccount && vestingSummary ? (
           <div className="overflow-hidden rounded-md border border-gray-50 backdrop-blur-sm shadow">
             <div className="p-4 sm:grid sm:w-full sm:grid-cols-2 sm:px-6">

diff --git a/src/routes/overview.tsx b/src/routes/overview.tsx
@@ -20,6 +20,7 @@ import { loadProjectCores, loadStakedDaos } from '../utils/stakingServices';
 import { StakingCore, CoreEraStakedInfoType, UserStakedInfoType, ChainPropertiesType, TotalUserStakedData, StakedType, BalanceType, LockedType, CoreEraStakeType, LedgerType, TotalRewardsClaimedSubscription, TotalRewardsClaimedQuery, RewardQueryType, getCoreInfo, getTotalUserStaked } from "./staking";
 import { calculateVestingData, fetchSystemData } from "../utils/vestingServices";
 import OnOffSwitch from "../components/Switch";
+import { formatBalance } from "@polkadot/util";
 
 export type StakedDaoType = StakingCore & { members?: AnyJson; };
 
@@ -337,7 +338,7 @@ const Overview = () => {
     });
   };
 
-  const handleAutoRestake = (bool: boolean) => {
+  const handleAutoRestakeSwitch = (bool: boolean) => {
     // use toasts to show if auto-restake is enabled or disabled
     if (bool) {
       toast.success("Auto-restake enabled");
@@ -348,6 +349,33 @@ const Overview = () => {
     localStorage.setItem("autoRestake", JSON.stringify(bool));
   };
 
+  const handleRestakingLogic = () => {
+    // grab the total unclaimed rewards and account for the existential deposit
+    const unclaimedMinusED = new BigNumber(totalUnclaimed);
+
+    // Check if unclaimedMinusED is a valid number
+    if (isNaN(unclaimedMinusED.toNumber())) {
+      console.error("Invalid unclaimedMinusED");
+      return;
+    }
+
+    if (unclaimedMinusED.toNumber() <= 0) {
+      console.error("unclaimedMinusED must be greater than 0");
+      return;
+    }
+
+    // Check if stakedDaos.length is a valid number and not zero to avoid division by zero
+    if (isNaN(stakedDaos.length) || stakedDaos.length === 0) {
+      console.error("Invalid stakedDaos.length");
+      return;
+    }
+
+    // divide unclaimedMinusED by the number of stakedDaos the user is part of
+    const unclaimedPerCore = unclaimedMinusED.div(stakedDaos.length);
+
+    return unclaimedPerCore;
+  };
+
   const handleClaimAll = async () => {
     if (!selectedAccount) return;
 
@@ -367,28 +395,38 @@ const Overview = () => {
       const uniqueCores = [
         ...new Map(unclaimedEras.cores.map((x) => [x.coreId, x])).values(),
       ];
-
+      console.log("uniqueCores", uniqueCores);
       for (const core of uniqueCores) {
         if (!core?.earliestEra) continue;
 
-        if (typeof currentStakingEra === 'number' && core && typeof core.earliestEra === 'number') {
-          if (core.earliestEra <= currentStakingEra) {
-            const iterations = currentStakingEra - core.earliestEra;
-            for (let i = 0; i < iterations; i++) {
-              batch.push(api.tx.ocifStaking.stakerClaimRewards(core.coreId));
-              console.log("pushed core: ", core.coreId);
-            }
-          } else {
-            console.error("core.earliestEra is greater than currentStakingEra");
+        const localEarliestEra = core.earliestEra; // Create a local copy of core.earliestEra
+
+        if (typeof currentStakingEra === 'number' && core && typeof localEarliestEra === 'number') {
+          for (let i = 0; i < currentStakingEra - localEarliestEra; i++) {
+            batch.push(api.tx.ocifStaking.stakerClaimRewards(core.coreId));
           }
         } else {
-          console.error("currentStakingEra, core, or core.earliestEra is undefined or not a number");
+          console.error("currentStakingEra, core, or localEarliestEra is undefined or not a number");
         }
+
+        console.log("utility.batch", batch);
       }
 
-      console.log("batch: ", batch);
-      console.log("uniqueCores: ", uniqueCores);
-      console.log("unclaimedEras: ", unclaimedEras);
+      if (enableAutoRestake) {
+        for (const core of uniqueCores) {
+          if (!core?.earliestEra) continue;
+
+          // using the restaking logic, calculate the amount to restake
+          const restakeAmount = handleRestakingLogic();
+          // Check if restakeAmount is not zero
+          if (restakeAmount && !restakeAmount.isZero()) {
+            // Convert restakeAmount to an integer string
+            const restakeAmountInteger = restakeAmount.integerValue().toString();
+            // push restake tx to the batch
+            batch.push(api.tx.ocifStaking.stake(core.coreId, restakeAmountInteger));
+          }
+        }
+      }
 
       await api.tx.utility.batch(batch).signAndSend(
         selectedAccount.address,
@@ -581,20 +619,14 @@ const Overview = () => {
     setTotalUnclaimed(totalUnclaimed);
   }, [selectedAccount, rewardsClaimedQuery]);
 
-  if (isLoading) {
-    return (
-      <div className="flex items-center justify-center">
-        <LoadingSpinner />
-      </div>
-    );
-  }
-
   return (
     <div className="overflow-y-scroll mx-auto w-full flex max-w-7xl flex-col justify-between p-4 sm:px-6 lg:px-8 mt-14 md:mt-0">
       <div className="flex flex-col md:flex-row md:justify-between items-start">
-        <h2 className="lg:text-xl font-bold leading-none mt-3">
+        <h2 className="lg:text-xl font-bold leading-none mt-3 flex flex-row items-center gap-4">
           <span>Account Overview</span>
+          <span>{isLoading || !isDataLoaded ? <LoadingSpinner /> : null}</span>
         </h2>
+
         {selectedAccount && <div className="flex flex-row w-full md:w-auto gap-2 items-center justify-start mb-4 z-1">
           <Button
             mini
@@ -612,7 +644,7 @@ const Overview = () => {
           </Button>
           <div className="flex flex-col items-center justify-around relative border border-tinkerYellow border-opacity-50 bg-tinkerGrey rounded-lg scale-70 lg:scale-90">
             <div className="flex-grow">
-              <OnOffSwitch defaultEnabled={enableAutoRestake} onChange={(bool) => handleAutoRestake(bool)} />
+              <OnOffSwitch defaultEnabled={enableAutoRestake} onChange={(bool) => handleAutoRestakeSwitch(bool)} />
             </div>
             <span className="text-xxs text-gray-300 relative bottom-1">Auto-Restake</span>
           </div>

diff --git a/src/routes/staking.tsx b/src/routes/staking.tsx
@@ -635,21 +635,15 @@ const Staking = () => {
     setTotalClaimed(rewardsClaimed);
   }, [selectedAccount, rewardsClaimedQuery]);
 
-  if (isLoading) {
-    return (
-      <div className="flex items-center justify-center">
-        <LoadingSpinner />
-      </div>
-    );
-  }
-
   return (
     <div className="overflow-y-scroll mx-auto w-full flex max-w-7xl flex-col justify-between p-4 sm:px-6 lg:px-8 mt-14 md:mt-0">
       <div className="flex flex-col md:flex-row justify-between items-start md:items-center">
-        <h2 className="lg:text-xl font-bold my-3">
+        <h2 className="lg:text-xl font-bold my-3 flex flex-row items-center gap-4">
           <span>DAO Staking</span>
+          <span>{isLoading || !isDataLoaded ? <LoadingSpinner /> : null}</span>
         </h2>
       </div>
+
       {selectedAccount ? (
         <>
           <MetricDashboard

diff --git a/src/routes/xtransfer.tsx b/src/routes/xtransfer.tsx
@@ -415,8 +415,9 @@ const Transfer = () => {
   return (
     <div className="overflow-hidden mx-auto w-full flex max-w-7xl flex-col justify-between p-4 sm:px-6 lg:px-8 mt-14 md:mt-0">
       <div className="z-10 w-full sm:max-w-2xl">
-        <h2 className="lg:text-xl font-bold my-3">
+        <h2 className="lg:text-xl font-bold my-3 flex flex-row items-center gap-4">
           <span>Asset Transfers</span>
+          <span>{isLoading ? <LoadingSpinner /> : null}</span>
         </h2>
 
         {!selectedAccount ? (
@@ -430,12 +431,6 @@ const Transfer = () => {
           </div>
         ) : null}
 
-        {isLoading ? (
-          <div className="flex items-center justify-center">
-            <LoadingSpinner />
-          </div>
-        ) : null}
-
         {!isLoading && selectedAccount ? (
           <div className="overflow-hidden w-full rounded-md border border-neutral-50 backdrop-blur-sm shadow">
             <div className="border-b border-neutral-50 p-4">