@triadxyz/triad-protocol 3.5.3-beta → 4.0.1

package/dist/index.d.ts

@@ -2,13 +2,11 @@
 import { Connection, PublicKey } from '@solana/web3.js';
 import { AnchorProvider, Program, Wallet } from '@coral-xyz/anchor';
 import { TriadProtocol as TriadProtocolIDL } from './types/triad_protocol';
-import { OpenOrderArgs, CreateCustomerArgs, MarketBidOrderArgs, CancelBidOrderArgs, CancelAskOrderArgs, PlaceBidOrderArgs, PlaceAskOrderArgs, BookOrder, MarketAskOrderArgs, RpcOptions, CloseOrderArgs, CreateMarketArgs, CreatePoolArgs, UpdateMarketWinningDirectionArgs, OrderDirectionEncoded } from './types';
+import { CreateCustomerArgs, MarketBidOrderArgs, CancelBidOrderArgs, CancelAskOrderArgs, PlaceBidOrderArgs, PlaceAskOrderArgs, BookOrder, MarketAskOrderArgs, RpcOptions, CreateMarketArgs, CreatePoolArgs, UpdateMarketWinningDirectionArgs, OrderDirectionEncoded, UpdateMarketPayoutArgs, UpdateMarketQuestionArgs, UpdateMarketEndArgs } from './types';
 import Stake from './stake';
 import Poseidon from './poseidon';
-import Claim from './claim';
 export * from './types';
 export * from './utils/helpers';
-export * from './utils/merkle';
 export * from './utils/feeCalculator';
 export * from './utils/constants';
 export * from './utils/pda';
@@ -20,7 +18,6 @@ export default class TriadProtocol {
     provider: AnchorProvider;
     stake: Stake;
     poseidon: Poseidon;
-    claim: Claim;
     constructor(connection: Connection, wallet: Wallet, rpcOptions: RpcOptions);
     /**
      * Get User Orders
@@ -113,41 +110,29 @@ export default class TriadProtocol {
      */
     createPool({ poolId, question, markets, customer, startTime, endTime, feeBps, payoutFee, isFast }: CreatePoolArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
     /**
-     * Resolve Market
+     * Update Market Winning Direction
      * @param args.marketId - The ID of the Market
-     * @param args.poolId - The ID of the Pool
      * @param args.winningDirection - The Winning Direction of the Market
      */
-    updateMarketWinningDirection({ marketId, poolId, winningDirection }: UpdateMarketWinningDirectionArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
+    updateMarketWinningDirection({ marketId, winningDirection }: UpdateMarketWinningDirectionArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
     /**
      * Update Market Payout
-     * @param marketId - The ID of the market
-     * @param poolId - The ID of the pool
-     * @param allowPayout - Whether to allow the market to payout
+     * @param args.marketId - The ID of the market
+     * @param args.allowPayout - Whether to allow the market to payout
      */
-    updateMarketPayout({ marketId, poolId, allowPayout }: {
-        marketId: number;
-        poolId?: number;
-        allowPayout: boolean;
-    }): Promise<string | import("@solana/web3.js").VersionedTransaction>;
+    updateMarketPayout({ marketId, allowPayout }: UpdateMarketPayoutArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
     /**
      * Update Market End
-     * @param marketId - The ID of the market
-     * @param marketEnd - The end time of the market
+     * @param args.marketId - The ID of the market
+     * @param args.marketEnd - The end time of the market
      */
-    updateMarketEnd({ marketId, marketEnd }: {
-        marketId: number;
-        marketEnd: number;
-    }): Promise<string | import("@solana/web3.js").VersionedTransaction>;
+    updateMarketEnd({ marketId, marketEnd }: UpdateMarketEndArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
     /**
      * Update Market Question
-     * @param marketId - The ID of the market
-     * @param question - The question of the market
+     * @param args.marketId - The ID of the market
+     * @param args.question - The question of the market
      */
-    updateMarketQuestion({ marketId, question }: {
-        marketId: number;
-        question: string;
-    }): Promise<string | import("@solana/web3.js").VersionedTransaction>;
+    updateMarketQuestion({ marketId, question }: UpdateMarketQuestionArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
     /**
      * Collect Market Fee
      * @param args.markets - The markets to collect the fee from
@@ -171,31 +156,15 @@ export default class TriadProtocol {
      * @param question - Question
      */
     updatePoolQuestion(poolId: number, question: string): Promise<string | import("@solana/web3.js").VersionedTransaction>;
-    /**
-     * Open Order
-     * @param args.marketId - The ID of the Market
-     * @param args.amount - The amount of the Order
-     * @param args.orderDirection - The direction of the Order
-     * @param args.mint - The mint of the Order
-     * @param args.token - The token to use for the Order
-     */
-    openOrder({ marketId, amount, orderDirection, mint, token }: OpenOrderArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
-    /**
-     * Close Order
-     * @param args.marketId - The ID of the Market
-     * @param args.orderDirection - The direction of the Order
-     */
-    closeOrder({ marketId, orderDirection }: CloseOrderArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
     /**
      * Payout Order
      * @param args.marketId - The ID of the Market
+     * @param args.authority - The authority of the order
      * @param args.orderDirection - The direction of the Order to Payout
-     * @param args.mint - The mint of the market
      */
     payoutOrder(orders: {
         marketId: number;
         orderDirection: OrderDirectionEncoded;
-        mint: PublicKey;
         authority: PublicKey;
     }[]): Promise<string | import("@solana/web3.js").VersionedTransaction>;
     /**
@@ -213,9 +182,8 @@ export default class TriadProtocol {
      * @param args.orders.amount - The amount of the Order
      * @param args.orders.price - The price of the Order
      * @param args.orders.orderDirection - The direction of the Order
-     * @param args.isTrdPayout - Whether to payout in TRD or not
      */
-    placeBidOrder({ orders, isTrdPayout }: PlaceBidOrderArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
+    placeBidOrder({ orders }: PlaceBidOrderArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
     /**
      * Place Ask Order
      * @param args.orders - Array of orders to execute
@@ -245,9 +213,8 @@ export default class TriadProtocol {
      * @param args.marketId - The ID of the Market
      * @param args.amount - The amount of the Order
      * @param args.orderDirection - The direction of the Order
-     * @param args.isTrdPayout - Whether to payout in TRD or not
      */
-    marketBidOrder({ marketId, amount, orderDirection, isTrdPayout }: MarketBidOrderArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
+    marketBidOrder({ marketId, amount, orderDirection }: MarketBidOrderArgs): Promise<string | import("@solana/web3.js").VersionedTransaction>;
     /**
      * Market Ask Order
      * @param args.marketId - The ID of the Market
@@ -279,8 +246,20 @@ export default class TriadProtocol {
      * @param customerId - Customer ID
      * @param feeBps - Fee in basis points
      */
-    updateCustomerFee({ customerId, feeBps }: {
+    updateCustomerFee({ customerId, feeBps, feeRecipient }: {
         customerId: number;
         feeBps: number;
+        feeRecipient: PublicKey | null;
+    }): Promise<string | import("@solana/web3.js").VersionedTransaction>;
+    deposit({ authority, amount, refer, customer }: {
+        authority: PublicKey;
+        amount: number;
+        refer?: PublicKey;
+        customer?: number;
+    }): Promise<string | import("@solana/web3.js").VersionedTransaction>;
+    withdraw({ authority, amount, customer }: {
+        authority: PublicKey;
+        amount: number;
+        customer?: number;
     }): Promise<string | import("@solana/web3.js").VersionedTransaction>;
 }

package/dist/index.js

@@ -38,14 +38,11 @@ const constants_1 = require("./utils/constants");
 const helpers_2 = require("./utils/helpers");
 const pda_1 = require("./utils/pda");
 const sendVersionedTransaction_1 = __importDefault(require("./utils/sendVersionedTransaction"));
-const swap_1 = require("./utils/swap");
 const feeCalculator_1 = require("./utils/feeCalculator");
 const stake_1 = __importDefault(require("./stake"));
 const poseidon_1 = __importDefault(require("./poseidon"));
-const claim_1 = __importDefault(require("./claim"));
 __exportStar(require("./types"), exports);
 __exportStar(require("./utils/helpers"), exports);
-__exportStar(require("./utils/merkle"), exports);
 __exportStar(require("./utils/feeCalculator"), exports);
 __exportStar(require("./utils/constants"), exports);
 __exportStar(require("./utils/pda"), exports);
@@ -63,7 +60,6 @@ class TriadProtocol {
         this.program = new anchor_1.Program(idl_triad_protocol_json_1.default, this.provider);
         this.stake = new stake_1.default(this.program, this.rpcOptions);
         this.poseidon = new poseidon_1.default(this.program, this.rpcOptions);
-        this.claim = new claim_1.default(this.program, this.rpcOptions);
     }
     /**
      * Get User Orders
@@ -275,7 +271,9 @@ class TriadProtocol {
                     marketStart: new bn_js_1.default(market.startTime),
                     marketEnd: new bn_js_1.default(market.endTime),
                     feeBps: market.feeBps,
-                    payoutFee: market.payoutFee
+                    payoutFee: market.payoutFee,
+                    preMarketEndTs: new bn_js_1.default(0),
+                    preOrdersAvailable: new bn_js_1.default(0)
                 })
                     .accounts({
                     signer: this.program.provider.publicKey,
@@ -330,7 +328,9 @@ class TriadProtocol {
                     marketStart: new bn_js_1.default(startTime),
                     marketEnd: new bn_js_1.default(endTime),
                     feeBps,
-                    payoutFee
+                    payoutFee,
+                    preMarketEndTs: new bn_js_1.default(0),
+                    preOrdersAvailable: new bn_js_1.default(0)
                 })
                     .accounts({
                     signer: this.program.provider.publicKey,
@@ -351,24 +351,18 @@ class TriadProtocol {
         });
     }
     /**
-     * Resolve Market
+     * Update Market Winning Direction
      * @param args.marketId - The ID of the Market
-     * @param args.poolId - The ID of the Pool
      * @param args.winningDirection - The Winning Direction of the Market
      */
-    updateMarketWinningDirection({ marketId, poolId, winningDirection }) {
+    updateMarketWinningDirection({ marketId, winningDirection }) {
         return __awaiter(this, void 0, void 0, function* () {
-            let poolPDA = null;
-            if (poolId) {
-                poolPDA = (0, pda_1.getPoolPDA)(this.program.programId, poolId);
-            }
             const ixs = [
                 yield this.program.methods
                     .updateMarketWinningDirection(winningDirection)
                     .accounts({
                     signer: this.program.provider.publicKey,
-                    market: (0, pda_1.getMarketPDA)(this.program.programId, marketId),
-                    pool: poolPDA
+                    market: (0, pda_1.getMarketPDA)(this.program.programId, marketId)
                 })
                     .instruction()
             ];
@@ -377,16 +371,11 @@ class TriadProtocol {
     }
     /**
      * Update Market Payout
-     * @param marketId - The ID of the market
-     * @param poolId - The ID of the pool
-     * @param allowPayout - Whether to allow the market to payout
+     * @param args.marketId - The ID of the market
+     * @param args.allowPayout - Whether to allow the market to payout
      */
-    updateMarketPayout({ marketId, poolId, allowPayout }) {
+    updateMarketPayout({ marketId, allowPayout }) {
         return __awaiter(this, void 0, void 0, function* () {
-            let poolPDA = null;
-            if (poolId) {
-                poolPDA = (0, pda_1.getPoolPDA)(this.program.programId, poolId);
-            }
             const ixs = [
                 yield this.program.methods
                     .updateMarketPayout(allowPayout)
@@ -401,8 +390,8 @@ class TriadProtocol {
     }
     /**
      * Update Market End
-     * @param marketId - The ID of the market
-     * @param marketEnd - The end time of the market
+     * @param args.marketId - The ID of the market
+     * @param args.marketEnd - The end time of the market
      */
     updateMarketEnd({ marketId, marketEnd }) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -420,8 +409,8 @@ class TriadProtocol {
     }
     /**
      * Update Market Question
-     * @param marketId - The ID of the market
-     * @param question - The question of the market
+     * @param args.marketId - The ID of the market
+     * @param args.question - The question of the market
      */
     updateMarketQuestion({ marketId, question }) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -500,80 +489,11 @@ class TriadProtocol {
             return (0, sendVersionedTransaction_1.default)(this.program, ixs, this.rpcOptions);
         });
     }
-    /**
-     * Open Order
-     * @param args.marketId - The ID of the Market
-     * @param args.amount - The amount of the Order
-     * @param args.orderDirection - The direction of the Order
-     * @param args.mint - The mint of the Order
-     * @param args.token - The token to use for the Order
-     */
-    openOrder({ marketId, amount, orderDirection, mint, token }) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const ixs = [];
-            const addressLookupTableAccounts = [];
-            let amountInTRD = amount * Math.pow(10, constants_1.BASE_DECIMALS);
-            if (token !== constants_1.TRD_MINT.toBase58()) {
-                const { setupInstructions, swapIxs, addressLookupTableAccounts, outAmount } = yield (0, swap_1.swap)({
-                    connection: this.program.provider.connection,
-                    wallet: this.program.provider.publicKey.toBase58(),
-                    inToken: token,
-                    outToken: constants_1.TRD_MINT.toString(),
-                    amount,
-                    payer: this.rpcOptions.payer.toBase58()
-                });
-                amountInTRD = outAmount;
-                if (swapIxs.length === 0) {
-                    return;
-                }
-                ixs.push(...setupInstructions);
-                ixs.push(...swapIxs);
-                addressLookupTableAccounts.push(...addressLookupTableAccounts);
-            }
-            ixs.push(yield this.program.methods
-                .openOrder({
-                amount: new bn_js_1.default(amountInTRD),
-                marketId: new bn_js_1.default(marketId),
-                orderDirection
-            })
-                .accounts({
-                signer: this.program.provider.publicKey,
-                payer: this.rpcOptions.payer,
-                market: (0, pda_1.getMarketPDA)(this.program.programId, marketId),
-                mint,
-                order: (0, pda_1.getOrderPDA)(this.program.programId, this.program.provider.publicKey, marketId, (0, helpers_1.getOrderDirection)(orderDirection))
-            })
-                .instruction());
-            return (0, sendVersionedTransaction_1.default)(this.program, ixs, this.rpcOptions, addressLookupTableAccounts);
-        });
-    }
-    /**
-     * Close Order
-     * @param args.marketId - The ID of the Market
-     * @param args.orderDirection - The direction of the Order
-     */
-    closeOrder({ marketId, orderDirection }) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const ixs = [
-                yield this.program.methods
-                    .closeOrder(orderDirection)
-                    .accounts({
-                    signer: this.program.provider.publicKey,
-                    payer: this.rpcOptions.payer,
-                    market: (0, pda_1.getMarketPDA)(this.program.programId, marketId),
-                    mint: constants_1.TRD_MINT,
-                    order: (0, pda_1.getOrderPDA)(this.program.programId, this.program.provider.publicKey, marketId, (0, helpers_1.getOrderDirection)(orderDirection))
-                })
-                    .instruction()
-            ];
-            return (0, sendVersionedTransaction_1.default)(this.program, ixs, this.rpcOptions);
-        });
-    }
     /**
      * Payout Order
      * @param args.marketId - The ID of the Market
+     * @param args.authority - The authority of the order
      * @param args.orderDirection - The direction of the Order to Payout
-     * @param args.mint - The mint of the market
      */
     payoutOrder(orders) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -588,8 +508,7 @@ class TriadProtocol {
                     signer: order.authority,
                     payer: this.rpcOptions.payer,
                     market: (0, pda_1.getMarketPDA)(this.program.programId, order.marketId),
-                    mint: order.mint,
-                    tokenProgram: (0, helpers_2.getTokenProgram)(order.mint),
+                    tokenProgram: (0, helpers_2.getTokenProgram)(constants_1.USDC_MINT),
                     order: (0, pda_1.getOrderPDA)(this.program.programId, order.authority, order.marketId, (0, helpers_1.getOrderDirection)(order.orderDirection))
                 })
                     .instruction());
@@ -624,41 +543,12 @@ class TriadProtocol {
      * @param args.orders.amount - The amount of the Order
      * @param args.orders.price - The price of the Order
      * @param args.orders.orderDirection - The direction of the Order
-     * @param args.isTrdPayout - Whether to payout in TRD or not
      */
-    placeBidOrder({ orders, isTrdPayout }) {
+    placeBidOrder({ orders }) {
         return __awaiter(this, void 0, void 0, function* () {
             const ixs = [];
-            const addressLookupTableAccounts = [];
-            if (orders.length > 5) {
-                throw new Error('Max 5 orders per transaction');
-            }
-            let amountInUSDC = new bn_js_1.default(0);
-            let totalAmount = 0;
-            for (const order of orders) {
-                totalAmount += order.amount;
-            }
-            if (isTrdPayout) {
-                const trdPrice = yield (0, swap_1.getPrice)(constants_1.TRD_MINT.toString());
-                const amountInTRD = (totalAmount / trdPrice) * 1.06;
-                const { swapIxs, addressLookupTableAccounts: swapAddressLookupTableAccounts, outAmount, setupInstructions } = yield (0, swap_1.swap)({
-                    connection: this.program.provider.connection,
-                    wallet: this.program.provider.publicKey.toBase58(),
-                    inToken: constants_1.TRD_MINT.toString(),
-                    outToken: constants_1.USDC_MINT.toString(),
-                    amount: Math.floor(amountInTRD),
-                    payer: this.rpcOptions.payer.toBase58()
-                });
-                if (swapIxs.length === 0) {
-                    return;
-                }
-                amountInUSDC = new bn_js_1.default(outAmount);
-                ixs.push(...setupInstructions);
-                ixs.push(...swapIxs);
-                addressLookupTableAccounts.push(...swapAddressLookupTableAccounts);
-                if (amountInUSDC.lt(new bn_js_1.default(totalAmount * Math.pow(10, constants_1.BASE_DECIMALS)))) {
-                    return;
-                }
+            if (orders.length > 6) {
+                throw new Error('Max 6 orders per transaction');
             }
             for (const order of orders) {
                 ixs.push(yield this.program.methods
@@ -691,8 +581,8 @@ class TriadProtocol {
     placeAskOrder({ orders }) {
         return __awaiter(this, void 0, void 0, function* () {
             const ixs = [];
-            if (orders.length > 5) {
-                throw new Error('Max 5 orders per transaction');
+            if (orders.length > 6) {
+                throw new Error('Max 6 orders per transaction');
             }
             for (const order of orders) {
                 ixs.push(yield this.program.methods
@@ -779,69 +669,18 @@ class TriadProtocol {
      * @param args.marketId - The ID of the Market
      * @param args.amount - The amount of the Order
      * @param args.orderDirection - The direction of the Order
-     * @param args.isTrdPayout - Whether to payout in TRD or not
      */
-    marketBidOrder({ marketId, amount, orderDirection, isTrdPayout }) {
+    marketBidOrder({ marketId, amount, orderDirection }) {
         return __awaiter(this, void 0, void 0, function* () {
             const marketPDA = (0, pda_1.getMarketPDA)(this.program.programId, marketId);
             const ixs = [];
             const addressLookupTableAccounts = [];
             const orderBook = yield this.getOrderBook(marketId);
             let remainingUSDC = new bn_js_1.default(amount * Math.pow(10, constants_1.BASE_DECIMALS));
-            if (isTrdPayout) {
-                const price = yield (0, swap_1.getPrice)(constants_1.TRD_MINT.toString());
-                const trdToUsdc = amount * price;
-                remainingUSDC = new bn_js_1.default(trdToUsdc * Math.pow(10, constants_1.BASE_DECIMALS));
-            }
             const orders = Object.keys(orderDirection)[0] === 'hype'
                 ? orderBook.hype.ask
                 : orderBook.flop.ask;
             const sortedOrders = orders.sort((a, b) => Number(a.price) - Number(b.price));
-            let totalUSDCNeeded = new bn_js_1.default(0);
-            let tempRemainingUSDC = remainingUSDC.clone();
-            if (isTrdPayout) {
-                for (const order of sortedOrders) {
-                    if (tempRemainingUSDC.lte(new bn_js_1.default(0)))
-                        break;
-                    if (order.authority === this.program.provider.publicKey.toBase58()) {
-                        continue;
-                    }
-                    const orderPrice = new bn_js_1.default(order.price);
-                    const availableShares = new bn_js_1.default(order.totalShares).sub(new bn_js_1.default(order.filledShares));
-                    const effectivePriceDecimal = (0, feeCalculator_1.applyBuyFee)(orderPrice.toNumber() / 1000000);
-                    const adjustedPrice = new bn_js_1.default(Math.floor(effectivePriceDecimal * 1000000));
-                    const maxSharesForPrice = tempRemainingUSDC
-                        .mul(new bn_js_1.default(Math.pow(10, constants_1.BASE_DECIMALS)))
-                        .div(adjustedPrice);
-                    const sharesToBuy = bn_js_1.default.min(maxSharesForPrice, availableShares);
-                    if (sharesToBuy.lte(new bn_js_1.default(0)))
-                        continue;
-                    const usdcAmount = sharesToBuy
-                        .mul(adjustedPrice)
-                        .div(new bn_js_1.default(Math.pow(10, constants_1.BASE_DECIMALS)));
-                    if (usdcAmount.lte(new bn_js_1.default(0)))
-                        continue;
-                    totalUSDCNeeded = totalUSDCNeeded.add(usdcAmount);
-                    tempRemainingUSDC = tempRemainingUSDC.sub(usdcAmount);
-                }
-                const price = yield (0, swap_1.getPrice)(constants_1.TRD_MINT.toString());
-                const amountInTRD = totalUSDCNeeded.toNumber() / (price * Math.pow(10, constants_1.BASE_DECIMALS));
-                const { swapIxs, addressLookupTableAccounts: swapAddressLookupTableAccounts, outAmount, setupInstructions } = yield (0, swap_1.swap)({
-                    connection: this.program.provider.connection,
-                    wallet: this.program.provider.publicKey.toBase58(),
-                    inToken: constants_1.TRD_MINT.toString(),
-                    outToken: constants_1.USDC_MINT.toString(),
-                    amount: Math.floor(amountInTRD),
-                    payer: this.rpcOptions.payer.toBase58()
-                });
-                if (swapIxs.length === 0) {
-                    return;
-                }
-                remainingUSDC = new bn_js_1.default(outAmount);
-                ixs.push(...setupInstructions);
-                ixs.push(...swapIxs);
-                addressLookupTableAccounts.push(...swapAddressLookupTableAccounts);
-            }
             for (const order of sortedOrders) {
                 if (remainingUSDC.lte(new bn_js_1.default(0)))
                     break;
@@ -850,7 +689,7 @@ class TriadProtocol {
                 }
                 const orderPrice = new bn_js_1.default(order.price);
                 const availableShares = new bn_js_1.default(order.totalShares).sub(new bn_js_1.default(order.filledShares));
-                const effectivePriceDecimal = (0, feeCalculator_1.applyBuyFee)(orderPrice.toNumber() / 1000000);
+                const effectivePriceDecimal = (0, feeCalculator_1.applyBuyFee)(orderPrice.toNumber() / 1000000, 700);
                 const adjustedPrice = new bn_js_1.default(Math.floor(effectivePriceDecimal * 1000000));
                 const maxSharesForPrice = remainingUSDC
                     .mul(new bn_js_1.default(Math.pow(10, constants_1.BASE_DECIMALS)))
@@ -922,7 +761,7 @@ class TriadProtocol {
                     continue;
                 remainingShares = remainingShares.sub(sharesToSell);
                 const orderPrice = new bn_js_1.default(order.price);
-                const effectivePriceDecimal = (0, feeCalculator_1.applySellFee)(orderPrice.toNumber() / 1000000);
+                const effectivePriceDecimal = (0, feeCalculator_1.applySellFee)(orderPrice.toNumber() / 1000000, 700);
                 const adjustedPrice = new bn_js_1.default(Math.floor(effectivePriceDecimal * 1000000));
                 amountOfUSDC = amountOfUSDC.add(sharesToSell.mul(adjustedPrice).div(new bn_js_1.default(Math.pow(10, constants_1.BASE_DECIMALS))));
                 ixs.push(yield this.program.methods
@@ -990,11 +829,11 @@ class TriadProtocol {
      * @param customerId - Customer ID
      * @param feeBps - Fee in basis points
      */
-    updateCustomerFee({ customerId, feeBps }) {
+    updateCustomerFee({ customerId, feeBps, feeRecipient }) {
         return __awaiter(this, void 0, void 0, function* () {
             const ixs = [
                 yield this.program.methods
-                    .updateCustomerFee({ id: customerId, feeBps })
+                    .updateCustomerFee({ id: customerId, feeBps, feeRecipient })
                     .accounts({
                     signer: this.program.provider.publicKey
                 })
@@ -1003,5 +842,50 @@ class TriadProtocol {
             return (0, sendVersionedTransaction_1.default)(this.program, ixs, this.rpcOptions);
         });
     }
+    deposit({ authority, amount, refer = new web3_js_1.PublicKey('EJL7PeGFTAxMY67XrAJfHM8A2euwcjcBAmK94sew61rG'), customer = 7 }) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const ixs = [];
+            const predictorPDA = (0, pda_1.getPredictorPDA)(this.program.programId, authority, customer);
+            try {
+                yield this.program.account.predictor.fetch(predictorPDA);
+            }
+            catch (e) {
+                ixs.push(yield this.program.methods
+                    .createPredictor({
+                    customer,
+                    refer
+                })
+                    .accounts({
+                    signer: authority,
+                    payer: this.rpcOptions.payer
+                })
+                    .instruction());
+            }
+            ixs.push(yield this.program.methods
+                .deposit(new bn_js_1.default(amount * Math.pow(10, constants_1.BASE_DECIMALS)))
+                .accounts({
+                signer: authority,
+                payer: this.rpcOptions.payer,
+                predictor: predictorPDA
+            })
+                .instruction());
+            return (0, sendVersionedTransaction_1.default)(this.program, ixs, this.rpcOptions);
+        });
+    }
+    withdraw({ authority, amount, customer = 7 }) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const ixs = [];
+            const predictorPDA = (0, pda_1.getPredictorPDA)(this.program.programId, authority, customer);
+            ixs.push(yield this.program.methods
+                .withdraw(new bn_js_1.default(amount * Math.pow(10, constants_1.BASE_DECIMALS)))
+                .accounts({
+                signer: authority,
+                payer: this.rpcOptions.payer,
+                predictor: predictorPDA
+            })
+                .instruction());
+            return (0, sendVersionedTransaction_1.default)(this.program, ixs, this.rpcOptions);
+        });
+    }
 }
 exports.default = TriadProtocol;

package/dist/stake.d.ts

@@ -46,11 +46,6 @@ export default class Stake {
      * @param amount - Reward amount to deposit
      */
     addStakeVaultRewards(amount: number): Promise<string | import("@solana/web3.js").VersionedTransaction>;
-    /**
-     * Update Stake Vault Is Locked
-     * @param isLocked - Whether the stake vault is locked
-     */
-    updateStakeVaultIsLocked(isLocked: boolean): Promise<string | import("@solana/web3.js").VersionedTransaction>;
     /**
      * Request Unstake
      * @param amount - Amount to unstake

package/dist/stake.js

@@ -107,23 +107,6 @@ class Stake {
             return (0, sendVersionedTransaction_1.default)(this.program, ixs, this.rpcOptions);
         });
     }
-    /**
-     * Update Stake Vault Is Locked
-     * @param isLocked - Whether the stake vault is locked
-     */
-    updateStakeVaultIsLocked(isLocked) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const ixs = [
-                yield this.program.methods
-                    .updateStakeVaultIsLocked(isLocked)
-                    .accounts({
-                    signer: this.program.provider.publicKey
-                })
-                    .instruction()
-            ];
-            return (0, sendVersionedTransaction_1.default)(this.program, ixs, this.rpcOptions);
-        });
-    }
     /**
      * Request Unstake
      * @param amount - Amount to unstake