add claimable reward

MeteoraAg · Dec 21, 2023 · 0ca54cb · 0ca54cb
1 parent 5375974
commit 0ca54cb
Show file tree

Hide file tree

Showing 4 changed files with 1,822 additions and 1,329 deletions.
diff --git a/ts-client/src/farm.ts b/ts-client/src/farm.ts
@@ -7,9 +7,11 @@ import {
   PublicKey,
   Transaction,
   TransactionInstruction,
+  SYSVAR_CLOCK_PUBKEY,
+  ParsedAccountData,
 } from "@solana/web3.js";
 
-import { FarmProgram, Opt, PoolState, UserState } from "./types";
+import { FarmProgram, Opt, PoolState, UserState, ParsedClockState } from "./types";
 import {
   chunks,
   getFarmInfo,
@@ -19,6 +21,7 @@ import {
 } from "./utils";
 import { FARM_PROGRAM_ID, SIMULATION_USER } from "./constant";
 
+
 const chunkedFetchMultipleUserAccount = async (
   program: FarmProgram,
   pks: PublicKey[],
@@ -63,6 +66,18 @@ const getAllPoolState = async (
   return poolStates;
 };
 
+const getAllUserState = async (
+  farmMints: Array<PublicKey>,
+  program: FarmProgram
+) => {
+  const poolStates = (await chunkedFetchMultipleUserAccount(
+    program,
+    farmMints
+  )) as Array<UserState>;
+
+  return poolStates;
+};
+
 const MAX_CLAIM_ALL_ALLOWED = 2;
 
 export class PoolFarmImpl {
@@ -351,25 +366,25 @@ export class PoolFarmImpl {
       await Promise.all(
         isDual
           ? [
-              getOrCreateATAInstruction(
-                this.poolState.rewardAMint,
-                owner,
-                this.program.provider.connection
-              ),
-              getOrCreateATAInstruction(
-                this.poolState.rewardBMint,
-                owner,
-                this.program.provider.connection
-              ),
-            ]
+            getOrCreateATAInstruction(
+              this.poolState.rewardAMint,
+              owner,
+              this.program.provider.connection
+            ),
+            getOrCreateATAInstruction(
+              this.poolState.rewardBMint,
+              owner,
+              this.program.provider.connection
+            ),
+          ]
           : [
-              getOrCreateATAInstruction(
-                this.poolState.rewardAMint,
-                owner,
-                this.program.provider.connection
-              ),
-              [undefined, undefined],
-            ]
+            getOrCreateATAInstruction(
+              this.poolState.rewardAMint,
+              owner,
+              this.program.provider.connection
+            ),
+            [undefined, undefined],
+          ]
       );
     userRewardAIx && preInstructions.push(userRewardAIx);
     userRewardBIx && preInstructions.push(userRewardBIx);
@@ -431,3 +446,39 @@ export class PoolFarmImpl {
     return simulatedReward;
   }
 }
+
+export const getOnchainTime = async (connection: Connection) => {
+  const parsedClock = await connection.getParsedAccountInfo(SYSVAR_CLOCK_PUBKEY);
+
+  const parsedClockAccount = (parsedClock.value!.data as ParsedAccountData).parsed as ParsedClockState;
+
+  const currentTime = parsedClockAccount.info.unixTimestamp;
+  return currentTime;
+};
+
+function rewardPerToken(pool: PoolState, lastTimeRewardApplicable: number) {
+  let totalStake = pool.totalStaked;
+  if (totalStake.isZero()) {
+    return {
+      a: pool.rewardAPerTokenStored,
+      b: pool.rewardBPerTokenStored,
+    }
+  }
+  let timePeriod = new BN(lastTimeRewardApplicable - pool.lastUpdateTime.toNumber());
+  return {
+    a: pool.rewardAPerTokenStored.add(timePeriod.mul(pool.rewardARateU128).div(totalStake)),
+    b: pool.rewardAPerTokenStored.add(timePeriod.mul(pool.rewardARateU128).div(totalStake))
+  }
+}
+
+export function getClaimableRewardSync(onchainTIme: number, userState: UserState, poolState: PoolState) {
+  // update reward
+  let rewardDurationEnd = poolState.rewardDurationEnd.toNumber();
+  let lastTimeRewardApplicable = ((onchainTIme < rewardDurationEnd) ? onchainTIme : rewardDurationEnd);
+  let { a, b } = rewardPerToken(poolState, lastTimeRewardApplicable);
+
+  return {
+    a: (userState.balanceStaked.mul(a.sub(userState.rewardAPerTokenComplete)).div(new BN(1_000_000_000))).add(userState.rewardAPerTokenPending),
+    b: (userState.balanceStaked.mul(b.sub(userState.rewardBPerTokenComplete)).div(new BN(1_000_000_000))).add(userState.rewardBPerTokenPending),
+  }
+}