@triadxyz/triad-protocol 0.0.1-beta → 0.0.3-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,33 @@ 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 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;
         });
     }
@@ -300,6 +275,8 @@ class Stake {
             const StakeVault = (0, helpers_1.getStakeVaultAddressSync)(this.program.programId, stakeVault);
             const FromAta = (0, helpers_1.getATASync)(StakeVault, mint);
             const ToAta = (0, helpers_1.getATASync)(wallet, mint);
+            const Stake = (0, helpers_1.getStakeAddressSync)(this.program.programId, nftName);
+            const NFTRewards = (0, helpers_1.getNFTRewardsAddressSync)(this.program.programId, Stake);
             const method = this.program.methods
                 .withdrawNft({
                 nftName,
@@ -309,6 +286,8 @@ class Stake {
                 signer: wallet,
                 fromAta: FromAta,
                 toAta: ToAta,
+                nftRewards: NFTRewards,
+                admin: new web3_js_1.PublicKey('82ppCojm3yrEKgdpH8B5AmBJTU1r1uAWXFWhxvPs9UCR'),
                 mint: mint
             });
             if (options === null || options === void 0 ? void 0 : options.microLamports) {
@@ -321,5 +300,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,
+                commitment: 'confirmed'
+            });
+        });
+    }
+    /**
+     *  Claim Stake Rewards
+     *  @param wallet - User wallet
+     *  @param mint - NFT mint
+     *  @param week - Week rewards
+     *  @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;

package/dist/test.js

@@ -17,13 +17,58 @@ const web3_js_1 = require("@solana/web3.js");
 const index_1 = __importDefault(require("./index"));
 const anchor_1 = require("@coral-xyz/anchor");
 const constants_1 = require("./utils/constants");
-const stakes_json_1 = __importDefault(require("./utils/stakes.json"));
-const file = fs_1.default.readFileSync('/Users/dannpl/.config/solana/triad-man.json');
+const rarity_json_1 = __importDefault(require("./utils/stake-season-1/rarity.json"));
+const users_collections_week_1_json_1 = __importDefault(require("./utils/stake-season-1/users-collections-week-1.json"));
+const helpers_1 = require("./utils/helpers");
+const file = fs_1.default.readFileSync('/Users/dannpl/.config/solana/id.json');
 const rpc_file = fs_1.default.readFileSync('/Users/dannpl/.config/solana/rpc.txt');
 const keypair = web3_js_1.Keypair.fromSecretKey(new Uint8Array(JSON.parse(file.toString())));
 const connection = new web3_js_1.Connection(rpc_file.toString(), 'confirmed');
 const wallet = new anchor_1.Wallet(keypair);
 const triadProtocol = new index_1.default(connection, wallet);
+const populateStakeDay = () => __awaiter(void 0, void 0, void 0, function* () {
+    const { perDay } = yield triadProtocol.stake.getStakeVaultRewards(constants_1.STAKE_SEASON_1);
+    const day = 1719755851;
+    const stakes = yield triadProtocol.stake.getStakesByDay(constants_1.STAKE_SEASON_1, day);
+    const newItems = [];
+    const itemsByAuthority = stakes.reduce((acc, item) => {
+        const itemRarity = rarity_json_1.default.find((rarity) => rarity.onchainId === item.mint);
+        if (!itemRarity) {
+            return acc;
+        }
+        if (!acc[item.authority]) {
+            acc[item.authority] = [];
+        }
+        acc[item.authority].push(Object.assign(Object.assign({}, item), { rarity: itemRarity.rarity, rarityRankHrtt: itemRarity.rarityRankHrtt, collections: {} }));
+        return acc;
+    }, {});
+    Object.keys(itemsByAuthority).forEach((authority) => {
+        const items = itemsByAuthority[authority];
+        items.sort((a, b) => a.rarityRankHrtt - b.rarityRankHrtt);
+        items.forEach((item, index) => {
+            const collections = Object.values(users_collections_week_1_json_1.default.find((collection) => Object.keys(collection)[0] === item.authority))[0];
+            let totalMultiplier = (0, helpers_1.calculateTotalMultiplier)(index === 0
+                ? Object.keys(item.collections).map((x) => x.toUpperCase())
+                : [], {
+                max: 1839,
+                currentPosition: item.rarityRankHrtt
+            });
+            newItems.push(Object.assign(Object.assign({}, item), { totalMultiplier, collections: index === 0 ? collections : {} }));
+        });
+    });
+    const totalMultiplierSum = newItems.reduce((sum, user) => sum + user.totalMultiplier, 0);
+    const rewards = newItems.map((item) => {
+        const rewards = (item.totalMultiplier / totalMultiplierSum) * perDay;
+        return Object.assign(Object.assign({}, item), { rewards, apr: (0, helpers_1.calculateAPR)({
+                rewards,
+                rate: 7,
+                amount: 1,
+                baseRewards: perDay
+            }) });
+    });
+    const orderedRewards = rewards.sort((a, b) => b.rewards - a.rewards);
+    fs_1.default.writeFileSync(`./src/utils/stake-season-1/stakes/2/${day}.json`, JSON.stringify(orderedRewards, null, 2));
+});
 const requestWithdraw = () => __awaiter(void 0, void 0, void 0, function* () {
     const response = yield triadProtocol.stake.requestWithdraw({
         wallet: wallet.publicKey,
@@ -36,6 +81,18 @@ const requestWithdraw = () => __awaiter(void 0, void 0, void 0, function* () {
     });
     console.log(response);
 });
+const updateStakeVaultStatus = () => __awaiter(void 0, void 0, void 0, function* () {
+    const response = yield triadProtocol.stake.updateStakeVaultStatus({
+        wallet: wallet.publicKey,
+        isLocked: true,
+        week: 0,
+        stakeVault: constants_1.STAKE_SEASON_1
+    }, {
+        skipPreflight: true,
+        microLamports: 10000
+    });
+    console.log(response);
+});
 const withdraw = () => __awaiter(void 0, void 0, void 0, function* () {
     const response = yield triadProtocol.stake.withdraw({
         wallet: wallet.publicKey,
@@ -49,11 +106,31 @@ const withdraw = () => __awaiter(void 0, void 0, void 0, function* () {
     console.log(response);
 });
 const getStake = () => __awaiter(void 0, void 0, void 0, function* () {
-    const response = yield triadProtocol.stake.getStakeByWallet(wallet.publicKey);
+    const response = yield triadProtocol.stake.getStakeByWallet(new web3_js_1.PublicKey('E48CKgbZVpDzerQ7DdommgqNobRHLqHy8RUVi8HXkSHE'), constants_1.STAKE_SEASON_1);
     const stakeVaults = yield triadProtocol.stake.getStakeVaults();
     console.log(response);
     console.log(stakeVaults);
 });
+const claimStakeRewards = () => __awaiter(void 0, void 0, void 0, function* () {
+    const response = yield triadProtocol.stake.claimStakeRewards({
+        wallet: wallet.publicKey,
+        mint: new web3_js_1.PublicKey(''),
+        week: 0,
+        stakeVault: constants_1.STAKE_SEASON_1,
+        nftName: ''
+    }, {
+        skipPreflight: true,
+        microLamports: 10000
+    });
+    console.log(response);
+});
+const getStakes = () => __awaiter(void 0, void 0, void 0, function* () {
+    const response = yield triadProtocol.stake.getStakes(constants_1.STAKE_SEASON_1);
+    const users = response
+        .filter((item, index, self) => index === self.findIndex((t) => t.authority === item.authority))
+        .map((user) => user.authority);
+    fs_1.default.writeFileSync('./src/utils/stake-season-1/users.json', JSON.stringify(users, null, 2));
+});
 const stake = () => __awaiter(void 0, void 0, void 0, function* () {
     const response = yield triadProtocol.stake.stake({
         name: 'Triad 0',
@@ -73,38 +150,3 @@ const stake = () => __awaiter(void 0, void 0, void 0, function* () {
     });
     console.log(response);
 });
-stake();
-const getDailyBaseRewards = () => __awaiter(void 0, void 0, void 0, function* () {
-    const stakeVaultRewards = yield triadProtocol.stake.getStakeVaultRewards(constants_1.STAKE_SEASON);
-    const days = {};
-    stakes_json_1.default.forEach((stake) => {
-        const keys = Object.keys(stake.rewards);
-        keys.forEach((key) => {
-            if (!days[key]) {
-                days[key] = 0;
-            }
-            days[key] += Object.values(stake.rewards[key]).length;
-        });
-    });
-    const values = {};
-    Object.keys(days).forEach((key) => {
-        values[key] = stakeVaultRewards.perDay / days[key];
-    });
-    console.log('Daily Rewards:', values);
-});
-const getStakesByWallet = () => __awaiter(void 0, void 0, void 0, function* () {
-    const response = yield triadProtocol.stake.getStakes();
-    const stakeVaultRewards = yield triadProtocol.stake.getStakeVaultRewards(constants_1.STAKE_SEASON);
-    const users = response
-        .map((stake) => stake.authority)
-        .filter((value, index, self) => self.indexOf(value) === index);
-    const data = [];
-    for (const user of users) {
-        const rewardsByWallet = yield triadProtocol.stake.getStakeRewardsByWallet(new web3_js_1.PublicKey(user), stakeVaultRewards);
-        data.push({
-            wallet: user,
-            rewards: rewardsByWallet
-        });
-    }
-    fs_1.default.writeFileSync('./src/utils/stakes.json', JSON.stringify(data, null, 2));
-});