@triadxyz/triad-protocol 0.0.1-beta → 0.0.2-beta

package/dist/stake.d.ts

@@ -1,7 +1,8 @@
 import { AnchorProvider, Program } from '@coral-xyz/anchor';
 import { PublicKey } from '@solana/web3.js';
 import { TriadProtocol } from './types/triad_protocol';
-import { DepositStakeRewardsArgs, InitializeStakeArgs, RpcOptions, StakeArgs, RequestWithdrawArgs, WithdrawArgs } from './types';
+import { RpcOptions } from './types';
+import { DepositStakeRewardsArgs, InitializeStakeArgs, StakeArgs, RequestWithdrawArgs, WithdrawArgs, StakeResponse, UpdateStakeVaultStatusArgs, UpdateStakeRewardsArgs, ClaimStakeRewardsArgs } from './types/stake';
 export default class Stake {
     program: Program<TriadProtocol>;
     provider: AnchorProvider;
@@ -9,115 +10,29 @@ export default class Stake {
     /**
      * Get all Stake Vaults
      */
-    getStakeVaults(): Promise<{
-        name: string;
-        collection: string;
-        slots: number;
-        amount: number;
-        isLocked: boolean;
-        usersPaid: PublicKey;
-        amountPaid: number;
-        amountUsers: number;
-        apr: number;
-        initTs: number;
-        endTs: number;
-    }[]>;
+    getStakeVaults(): Promise<import("./types/stake").StakeVaultResponse[]>;
     /**
      * Get Stake Vault by name
      * @param stakeVault - Stake Vault name
      */
-    getStakeVaultByName(stakeVault: string): Promise<{
-        name: string;
-        collection: string;
-        slots: number;
-        amount: number;
-        isLocked: boolean;
-        usersPaid: PublicKey;
-        amountPaid: number;
-        amountUsers: number;
-        apr: number;
-        initTs: number;
-        endTs: number;
-    }>;
+    getStakeVaultByName(stakeVault: string): Promise<import("./types/stake").StakeVaultResponse>;
     /**
-     * Get all stakes
+     * Get all stakes by vault
+     * @param stakeVault - Stake Vault name
+     *
      */
-    getStakes(): Promise<{
-        name: string;
-        collections: (({
-            undead?: never;
-            alligators?: never;
-            pyth?: never;
-        } & {
-            coleta: Record<string, never>;
-        }) | ({
-            coleta?: never;
-            alligators?: never;
-            pyth?: never;
-        } & {
-            undead: Record<string, never>;
-        }) | ({
-            coleta?: never;
-            undead?: never;
-            pyth?: never;
-        } & {
-            alligators: Record<string, never>;
-        }) | ({
-            coleta?: never;
-            undead?: never;
-            alligators?: never;
-        } & {
-            pyth: Record<string, never>;
-        }))[];
-        rarity: string;
-        stakeVault: string;
-        authority: string;
-        initTs: number;
-        isLocked: boolean;
-        withdrawTs: number;
-        mint: string;
-        stakeRewards: string;
-    }[]>;
+    getStakes(stakeVault: string): Promise<StakeResponse[]>;
     /**
      * Get Stake by wallet
      * @param wallet - User wallet
+     * @param stakeVault - Stake Vault name
+     *
+     */
+    getStakeByWallet(wallet: PublicKey, stakeVault: string): Promise<StakeResponse[]>;
+    /**
+     * Get Stake Vault Rewards details
+     * @param stakeVault - Stake Vault name
      */
-    getStakeByWallet(wallet: PublicKey): Promise<{
-        name: string;
-        collections: (({
-            undead?: never;
-            alligators?: never;
-            pyth?: never;
-        } & {
-            coleta: Record<string, never>;
-        }) | ({
-            coleta?: never;
-            alligators?: never;
-            pyth?: never;
-        } & {
-            undead: Record<string, never>;
-        }) | ({
-            coleta?: never;
-            undead?: never;
-            pyth?: never;
-        } & {
-            alligators: Record<string, never>;
-        }) | ({
-            coleta?: never;
-            undead?: never;
-            alligators?: never;
-        } & {
-            pyth: Record<string, never>;
-        }))[];
-        rarity: string;
-        stakeVault: string;
-        authority: string;
-        initTs: number;
-        isLocked: boolean;
-        withdrawTs: number;
-        mint: string;
-        stakeRewards: string;
-    }[]>;
     getStakeVaultRewards(stakeVault: string): Promise<{
         amount: number;
         perDay: number;
@@ -126,14 +41,12 @@ export default class Stake {
         period: number;
         days: number[];
     }>;
-    getStakeRewardsByWallet(wallet: PublicKey, stakeVaultRewards: {
-        amount: number;
-        perDay: number;
-        perWeek: number;
-        perMonth: number;
-        period: number;
-        days: number[];
-    }): Promise<{}>;
+    /**
+     * Get Stakes by day
+     * @param stakeVault - Stake Vault name
+     * @param day - Day timestamp
+     */
+    getStakesByDay(stakeVault: string, day: number): Promise<StakeResponse[]>;
     /**
      *  Stake NFT
      *  @param name - NFT name
@@ -177,4 +90,34 @@ export default class Stake {
      *
      */
     withdraw({ wallet, nftName, mint, stakeVault }: WithdrawArgs, options?: RpcOptions): Promise<string>;
+    /**
+     *  Update Stake Vault Status
+     *  @param wallet - User wallet
+     *  @param stakeVault - Name of the stake vault
+     *  @param isLocked - Status of the stake vault
+     *  @param week - Current week rewards (Starts from 0)
+     *
+     */
+    updateStakeVaultStatus({ wallet, isLocked, week, stakeVault }: UpdateStakeVaultStatusArgs, options?: RpcOptions): Promise<string>;
+    /**
+     *  Update Stake Rewards
+     *  @param wallet - User wallet
+     *  @param nft_name - Name of the nft
+     *  @param apr - APR based in the current day
+     *  @param day - Day for update rewards (Starts from 0)
+     *  @param rewards - Rewards for the day
+     *
+     */
+    updateStakeRewards({ wallet, day, items }: UpdateStakeRewardsArgs, options?: RpcOptions): Promise<string>;
+    /**
+     *  Claim Stake Rewards
+     *  @param wallet - User wallet
+     *  @param mint - NFT mint
+     *  @param week - Week rewards
+     *  @param amount - Reward amount
+     *  @param stakeVault - Name of the stake vault
+     *  @param nftName - Name of the nft
+     *
+     */
+    claimStakeRewards({ wallet, mint, week, stakeVault, nftName }: ClaimStakeRewardsArgs, options?: RpcOptions): Promise<string>;
 }

package/dist/stake.js

@@ -23,19 +23,7 @@ class Stake {
     getStakeVaults() {
         return __awaiter(this, void 0, void 0, function* () {
             const response = yield this.program.account.stakeVault.all();
-            return response.map((stakeVault) => ({
-                name: stakeVault.account.name,
-                collection: stakeVault.account.collection,
-                slots: stakeVault.account.slots.toNumber(),
-                amount: stakeVault.account.amount.toNumber(),
-                isLocked: stakeVault.account.isLocked,
-                usersPaid: stakeVault.account.usersPaid,
-                amountPaid: stakeVault.account.amountPaid.toNumber(),
-                amountUsers: stakeVault.account.amountUsers.toNumber(),
-                apr: stakeVault.account.apr,
-                initTs: stakeVault.account.initTs.toNumber(),
-                endTs: stakeVault.account.endTs.toNumber()
-            }));
+            return response.map((stakeVault) => (0, helpers_1.formatStakeVault)(stakeVault.account));
         });
     }
     /**
@@ -45,112 +33,99 @@ class Stake {
     getStakeVaultByName(stakeVault) {
         return __awaiter(this, void 0, void 0, function* () {
             const StakeVault = (0, helpers_1.getStakeVaultAddressSync)(this.program.programId, stakeVault);
-            const response = yield this.program.account.stakeVault.fetch(StakeVault);
-            return {
-                name: response.name,
-                collection: response.collection,
-                slots: response.slots.toNumber(),
-                amount: response.amount.toNumber(),
-                isLocked: response.isLocked,
-                usersPaid: response.usersPaid,
-                amountPaid: response.amountPaid.toNumber(),
-                amountUsers: response.amountUsers.toNumber(),
-                apr: response.apr,
-                initTs: response.initTs.toNumber(),
-                endTs: response.endTs.toNumber()
-            };
+            return (0, helpers_1.formatStakeVault)(yield this.program.account.stakeVault.fetch(StakeVault));
         });
     }
     /**
-     * Get all stakes
+     * Get all stakes by vault
+     * @param stakeVault - Stake Vault name
+     *
      */
-    getStakes() {
+    getStakes(stakeVault) {
         return __awaiter(this, void 0, void 0, function* () {
             const response = yield this.program.account.stake.all();
-            return response.map((stake) => ({
-                name: stake.account.name,
-                collections: stake.account.collections,
-                rarity: Object.keys(stake.account.rarity)[0],
-                stakeVault: stake.account.stakeVault.toBase58(),
-                authority: stake.account.authority.toBase58(),
-                initTs: stake.account.initTs.toNumber(),
-                isLocked: stake.account.isLocked,
-                withdrawTs: stake.account.withdrawTs.toNumber(),
-                mint: stake.account.mint.toBase58(),
-                stakeRewards: stake.account.stakeRewards.toBase58()
-            }));
+            const StakeVault = (0, helpers_1.getStakeVaultAddressSync)(this.program.programId, stakeVault);
+            return response
+                .filter((item) => item.account.stakeVault.toBase58() === StakeVault.toBase58())
+                .map((stake) => (0, helpers_1.formatStake)(stake.account));
         });
     }
     /**
      * Get Stake by wallet
      * @param wallet - User wallet
+     * @param stakeVault - Stake Vault name
+     *
      */
-    getStakeByWallet(wallet) {
+    getStakeByWallet(wallet, stakeVault) {
         return __awaiter(this, void 0, void 0, function* () {
-            const response = yield this.program.account.stake.all();
-            return response
-                .filter((stake) => stake.account.authority.equals(wallet))
-                .map((stake) => ({
-                name: stake.account.name,
-                collections: stake.account.collections,
-                rarity: Object.keys(stake.account.rarity)[0],
-                stakeVault: stake.account.stakeVault.toBase58(),
-                authority: stake.account.authority.toBase58(),
-                initTs: stake.account.initTs.toNumber(),
-                isLocked: stake.account.isLocked,
-                withdrawTs: stake.account.withdrawTs.toNumber(),
-                mint: stake.account.mint.toBase58(),
-                stakeRewards: stake.account.stakeRewards.toBase58()
-            }));
+            const response = yield this.getStakes(stakeVault);
+            const stakeVaultByName = yield this.getStakeVaultByName(stakeVault);
+            const myStakes = response.filter((item) => item.authority === wallet.toBase58());
+            for (const stake of myStakes) {
+                try {
+                    const stakeRewards = yield this.program.account.nftRewards.fetch(new web3_js_1.PublicKey(stake.stakeRewards));
+                    let start = stakeVaultByName.week * 7;
+                    let end = stakeVaultByName.week == 4 ? 30 : start + 7;
+                    stake.apr = stakeRewards.apr;
+                    stake.dailyRewards = stakeRewards.dailyRewards.map((reward) => reward.toNumber() / Math.pow(10, constants_1.TTRIAD_DECIMALS));
+                    stake.weeklyRewardsPaid = stakeRewards.weeklyRewardsPaid;
+                    let rewards = stake.dailyRewards
+                        .slice(start, end)
+                        .reduce((a, b) => a + b, 0);
+                    stake.weeklyRewards = rewards;
+                }
+                catch (error) {
+                    stake.apr = 0;
+                    stake.dailyRewards = [];
+                    stake.weeklyRewardsPaid = [];
+                    stake.weeklyRewards = 0;
+                }
+            }
+            return myStakes;
         });
     }
+    /**
+     * Get Stake Vault Rewards details
+     * @param stakeVault - Stake Vault name
+     */
     getStakeVaultRewards(stakeVault) {
         return __awaiter(this, void 0, void 0, function* () {
             const StakeVault = (0, helpers_1.getStakeVaultAddressSync)(this.program.programId, stakeVault);
             const response = yield this.program.account.stakeVault.fetch(StakeVault);
+            const amount = response.amount.toNumber() / Math.pow(10, constants_1.TTRIAD_DECIMALS);
+            const period = (response.endTs.toNumber() - response.initTs.toNumber()) / (60 * 60 * 24);
+            const netAmount = amount - (amount * constants_1.TTRIAD_FEE) / 100;
             const data = {
-                amount: 0,
-                perDay: 0,
-                perWeek: 0,
-                perMonth: 0,
-                period: 0,
+                amount: netAmount,
+                perDay: netAmount / period,
+                perWeek: (netAmount / period) * 7,
+                perMonth: (netAmount / period) * 30,
+                period,
                 days: []
             };
-            const amount = response.amount.toNumber() / Math.pow(10, constants_1.TTRIAD_DECIMALS);
-            data.period =
-                (response.endTs.toNumber() * 1000 - response.initTs.toNumber() * 1000) /
-                    (1000 * 60 * 60 * 24);
-            data.amount = amount - (amount * constants_1.TTRIAD_FEE) / 100;
-            data.perDay = data.amount / data.period;
-            data.perWeek = data.perDay * 7;
-            data.perMonth = data.perDay * 30;
-            const endTsInMs = response.endTs.toNumber() * 1000;
-            const initTsInMs = response.initTs.toNumber() * 1000;
-            let currentTs = initTsInMs;
-            while (currentTs <= endTsInMs) {
-                data.days.push(currentTs);
-                currentTs = currentTs + 1000 * 60 * 60 * 24;
+            for (let ts = response.initTs.toNumber(); ts <= response.endTs.toNumber(); ts += 60 * 60 * 24) {
+                data.days.push(ts);
             }
             return data;
         });
     }
-    getStakeRewardsByWallet(wallet, stakeVaultRewards) {
+    /**
+     * Get Stakes by day
+     * @param stakeVault - Stake Vault name
+     * @param day - Day timestamp
+     */
+    getStakesByDay(stakeVault, day) {
         return __awaiter(this, void 0, void 0, function* () {
-            const stakes = yield this.getStakeByWallet(wallet);
-            const rewards = {};
-            for (const day of stakeVaultRewards.days) {
-                stakes.forEach((stake) => {
-                    const date = stake.initTs * 1000;
-                    const currentDate = new Date().getTime();
-                    if (date <= day && day <= currentDate) {
-                        const key = new Date(day).toISOString().split('T')[0];
-                        if (!rewards[key]) {
-                            rewards[key] = [];
-                        }
-                        rewards[key].push(stake);
-                    }
-                });
-            }
+            const stakes = yield this.getStakes(stakeVault);
+            const rewards = [];
+            stakes.forEach((stake) => {
+                const date = stake.initTs * 1000;
+                const stakeDay = day * 1000;
+                const currentDate = new Date().getTime();
+                if (date <= stakeDay && stakeDay <= currentDate) {
+                    rewards.push(stake);
+                }
+            });
             return rewards;
         });
     }
@@ -309,6 +284,7 @@ class Stake {
                 signer: wallet,
                 fromAta: FromAta,
                 toAta: ToAta,
+                admin: new web3_js_1.PublicKey('82ppCojm3yrEKgdpH8B5AmBJTU1r1uAWXFWhxvPs9UCR'),
                 mint: mint
             });
             if (options === null || options === void 0 ? void 0 : options.microLamports) {
@@ -321,5 +297,120 @@ class Stake {
             return method.rpc({ skipPreflight: options === null || options === void 0 ? void 0 : options.skipPreflight });
         });
     }
+    /**
+     *  Update Stake Vault Status
+     *  @param wallet - User wallet
+     *  @param stakeVault - Name of the stake vault
+     *  @param isLocked - Status of the stake vault
+     *  @param week - Current week rewards (Starts from 0)
+     *
+     */
+    updateStakeVaultStatus({ wallet, isLocked, week, stakeVault }, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const StakeVault = (0, helpers_1.getStakeVaultAddressSync)(this.program.programId, stakeVault);
+            const method = this.program.methods
+                .updateStakeVaultStatus({
+                isLocked,
+                week
+            })
+                .accounts({
+                signer: wallet,
+                stakeVault: StakeVault
+            });
+            if (options === null || options === void 0 ? void 0 : options.microLamports) {
+                method.postInstructions([
+                    web3_js_1.ComputeBudgetProgram.setComputeUnitPrice({
+                        microLamports: options.microLamports
+                    })
+                ]);
+            }
+            return method.rpc({ skipPreflight: options === null || options === void 0 ? void 0 : options.skipPreflight });
+        });
+    }
+    /**
+     *  Update Stake Rewards
+     *  @param wallet - User wallet
+     *  @param nft_name - Name of the nft
+     *  @param apr - APR based in the current day
+     *  @param day - Day for update rewards (Starts from 0)
+     *  @param rewards - Rewards for the day
+     *
+     */
+    updateStakeRewards({ wallet, day, items }, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const ixs = [
+                web3_js_1.ComputeBudgetProgram.setComputeUnitPrice({
+                    microLamports: options.microLamports
+                }),
+                web3_js_1.ComputeBudgetProgram.setComputeUnitLimit({
+                    units: 600000
+                })
+            ];
+            for (const item of items) {
+                const Stake = (0, helpers_1.getStakeAddressSync)(this.program.programId, item.nftName);
+                ixs.push(yield this.program.methods
+                    .updateStakeRewards({
+                    rewards: item.rewards,
+                    apr: item.apr,
+                    day
+                })
+                    .accounts({
+                    signer: wallet,
+                    stake: Stake
+                })
+                    .instruction());
+            }
+            const { blockhash } = yield this.provider.connection.getLatestBlockhash();
+            const messageV0 = new web3_js_1.TransactionMessage({
+                instructions: ixs,
+                recentBlockhash: blockhash,
+                payerKey: wallet
+            }).compileToV0Message();
+            const tx = new web3_js_1.VersionedTransaction(messageV0);
+            return this.provider.sendAndConfirm(tx, [], {
+                skipPreflight: options === null || options === void 0 ? void 0 : options.skipPreflight
+            });
+        });
+    }
+    /**
+     *  Claim Stake Rewards
+     *  @param wallet - User wallet
+     *  @param mint - NFT mint
+     *  @param week - Week rewards
+     *  @param amount - Reward amount
+     *  @param stakeVault - Name of the stake vault
+     *  @param nftName - Name of the nft
+     *
+     */
+    claimStakeRewards({ wallet, mint, week, stakeVault, nftName }, options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const StakeVault = (0, helpers_1.getStakeVaultAddressSync)(this.program.programId, stakeVault);
+            const Stake = (0, helpers_1.getStakeAddressSync)(this.program.programId, nftName);
+            const NFTRewards = (0, helpers_1.getNFTRewardsAddressSync)(this.program.programId, Stake);
+            const FromAta = (0, helpers_1.getATASync)(StakeVault, mint);
+            const ToAta = (0, helpers_1.getATASync)(wallet, mint);
+            const method = this.program.methods
+                .claimStakeRewards({
+                week
+            })
+                .accounts({
+                signer: wallet,
+                fromAta: FromAta,
+                toAta: ToAta,
+                mint: mint,
+                nftRewards: NFTRewards,
+                stake: Stake,
+                stakeVault: StakeVault
+            });
+            if (options === null || options === void 0 ? void 0 : options.microLamports) {
+                method.postInstructions([
+                    web3_js_1.ComputeBudgetProgram.setComputeUnitPrice({
+                        microLamports: options.microLamports
+                    })
+                ]);
+            }
+            return method.rpc({ skipPreflight: options === null || options === void 0 ? void 0 : options.skipPreflight });
+        });
+    }
 }
 exports.default = Stake;