@triadxyz/triad-protocol 0.3.5-beta → 0.3.6-beta

package/dist/stake.d.ts

@@ -30,9 +30,26 @@ export default class Stake {
      * Get Stake by wallet
      * @param wallet - User wallet
      * @param stakeVault - Stake Vault name
+     * @param collections - NFT collections
      *
      */
-    getStakeByWallet(wallet: PublicKey, stakeVault: string): Promise<StakeResponse[]>;
+    getStakeByWallet(wallet: PublicKey, stakeVault: string, collections: number, ranks: {
+        onchainId: string;
+        name: string;
+        rarityRankHrtt: number;
+    }[]): Promise<{
+        available: number;
+        name: string;
+        stakeVault: string;
+        authority: string;
+        initTs: number;
+        withdrawTs: number;
+        mint: string;
+        claimedTs: number;
+        boost: boolean;
+        claimed: number;
+        amount: number;
+    }[]>;
     /**
      * Get Stakes by day
      * @param stakeVault - Stake Vault name

package/dist/stake.js

@@ -14,6 +14,7 @@ const web3_js_1 = require("@solana/web3.js");
 const helpers_1 = require("./utils/helpers");
 const constants_1 = require("./utils/constants");
 const base64_js_1 = require("base64-js");
+const getRarity_1 = require("./utils/getRarity");
 class Stake {
     constructor(program, provider) {
         this.provider = provider;
@@ -85,13 +86,26 @@ class Stake {
      * Get Stake by wallet
      * @param wallet - User wallet
      * @param stakeVault - Stake Vault name
+     * @param collections - NFT collections
      *
      */
-    getStakeByWallet(wallet, stakeVault) {
+    getStakeByWallet(wallet, stakeVault, collections, ranks) {
         return __awaiter(this, void 0, void 0, function* () {
             const response = yield this.getStakes(stakeVault);
-            const myStakes = response.filter((item) => item.authority === wallet.toBase58());
-            return myStakes;
+            const stakePromises = response
+                .filter((item) => item.authority === wallet.toBase58())
+                .map((nft) => __awaiter(this, void 0, void 0, function* () {
+                const getRank = (0, getRarity_1.getRarityRank)(ranks, nft.mint, nft.name);
+                const available = yield this.getStakeRewards({
+                    wallet,
+                    nftName: nft.name,
+                    stakeVault,
+                    rank: getRank,
+                    collections
+                });
+                return Object.assign(Object.assign({}, nft), { available });
+            }));
+            return Promise.all(stakePromises);
         });
     }
     /**

package/dist/test.js

@@ -36,22 +36,7 @@ const updateStakeVaultStatus = () => __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(new web3_js_1.PublicKey('BCTdjdcjMiECGFbF5Ps15yjLRPzy5YZGJNa4VdGRbhjB'), constants_1.STAKE_SEASON);
+    const response = yield triadProtocol.stake.getStakeByWallet(new web3_js_1.PublicKey('HjJQdfTHgC3EBX3471w4st8BXbBmtbaMyCAXNgcUb7dq'), constants_1.STAKE_SEASON, 5, []);
     console.log(response);
 });
-const getRewards = () => __awaiter(void 0, void 0, void 0, function* () {
-    let sum = 0;
-    const a = yield triadProtocol.stake.getStakeByWallet(new web3_js_1.PublicKey('HjJQdfTHgC3EBX3471w4st8BXbBmtbaMyCAXNgcUb7dq'), constants_1.STAKE_SEASON);
-    for (let i = 0; i < a.length; i++) {
-        const stake = a[i];
-        const response = yield triadProtocol.stake.getStakeRewards({
-            wallet: new web3_js_1.PublicKey('HjJQdfTHgC3EBX3471w4st8BXbBmtbaMyCAXNgcUb7dq'),
-            nftName: stake.name,
-            stakeVault: constants_1.STAKE_SEASON,
-            collections: 5,
-            rank: i
-        });
-        sum += response;
-    }
-    console.log(sum);
-});
+getStake();

package/dist/utils/getRarity.d.ts

@@ -0,0 +1,5 @@
+export declare const getRarityRank: (ranks: {
+    onchainId: string;
+    name: string;
+    rarityRankHrtt: number;
+}[], onchainId: string, name: string) => number;

package/dist/utils/getRarity.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getRarityRank = void 0;
+const getRarityRank = (ranks, onchainId, name) => {
+    const item = ranks.find((entry) => entry.onchainId === onchainId || entry.name === name);
+    return item ? item.rarityRankHrtt : 0;
+};
+exports.getRarityRank = getRarityRank;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@triadxyz/triad-protocol",
-  "version": "0.3.5-beta",
+  "version": "0.3.6-beta",
   "main": "./dist/index.js",
   "module": "./dist/index.js",
   "types": "./dist/index.d.ts",