zo-sdk 0.1.31 → 0.1.33

package/src/implementations/ZLPAPI.ts

@@ -223,6 +223,87 @@ export class ZLPAPI extends BaseAPI implements IZLPAPI {
     return tx
   }
 
+  /**
+   * Deposits collateral into ZLP vault and returns a coin
+   */
+  public async depositPtb(
+    coin: string,
+    coinObjects: string[],
+    amount: number,
+    minAmountOut = 0,
+    referralAddress?: string,
+    sender?: string,
+    tx?: Transaction,
+    sponsoredTx?: boolean,
+    suiCoinObjectsForPythUpdate?: string[],
+  ): Promise<TransactionObjectArgument> {
+    if (!tx) {
+      tx = new Transaction()
+    }
+
+    // Add referral if needed
+    if (referralAddress && !(await this.dataAPI.hasReferral(sender || ''))) {
+      tx = await this.addReferral(referralAddress, tx)
+    }
+
+    // Initialize oracle transaction
+    const pythFeederKeys = Object.keys(this.consts.pythFeeder.feeder)
+
+    // Handle sponsored transaction case
+    if (sponsoredTx) {
+      const suiCoinObject = this.processCoins(tx, 'sui', suiCoinObjectsForPythUpdate || [], true)
+      tx = await this.initOracleTxb(pythFeederKeys, tx, true, suiCoinObject)
+
+      // Process deposit coins
+      const depositObject = coin === 'sui'
+        ? tx.splitCoins(suiCoinObject, [tx.pure.u64(amount)])[0]
+        : tx.splitCoins(this.processCoins(tx, coin, coinObjects, true), [tx.pure.u64(amount)])[0]
+
+      const { vaultsValuation, symbolsValuation } = this.dataAPI.valuate(tx)
+
+      const [mintedCoin] = tx.moveCall({
+        target: `${this.consts.zoCore.upgradedPackage}::market::deposit_ptb`,
+        typeArguments: [`${this.consts.zoCore.package}::zlp::ZLP`, this.consts.coins[coin].module],
+        arguments: [
+          tx.object(this.consts.zoCore.market),
+          tx.object(this.consts.zoCore.rebaseFeeModel),
+          depositObject,
+          tx.pure.u64(minAmountOut),
+          vaultsValuation,
+          symbolsValuation,
+        ],
+      })
+      return mintedCoin
+    }
+
+    // Handle non-sponsored transaction case
+    tx = await this.initOracleTxb(pythFeederKeys, tx)
+    const depositObject = tx.splitCoins(
+      this.processCoins(tx, coin, coinObjects, false),
+      [tx.pure.u64(amount)],
+    )[0]
+
+    const { vaultsValuation, symbolsValuation } = this.dataAPI.valuate(tx)
+
+    const [mintedCoin] = tx.moveCall({
+      target: `${this.consts.zoCore.upgradedPackage}::market::deposit_ptb`,
+      typeArguments: [
+        `${this.consts.zoCore.package}::zlp::ZLP`,
+        this.consts.coins[coin].module,
+      ],
+      arguments: [
+        tx.object(this.consts.zoCore.market),
+        tx.object(this.consts.zoCore.rebaseFeeModel),
+        depositObject,
+        tx.pure.u64(minAmountOut),
+        vaultsValuation,
+        symbolsValuation,
+      ],
+    })
+
+    return mintedCoin
+  }
+
   public async depositWithPtb(
     coin: string,
     depositObject: TransactionObjectArgument,
@@ -397,7 +478,7 @@ export class ZLPAPI extends BaseAPI implements IZLPAPI {
     const [depositObject] = tx.splitCoins(coinObject, [tx.pure.u64(amount)])
 
     tx.moveCall({
-      target: `${this.sharedConfig.zoStaking.package}::pool::deposit`,
+      target: `${this.sharedConfig.zoStaking.upgradedPackage}::pool::deposit`,
       typeArguments: [
         `${this.consts.zoCore.package}::zlp::ZLP`,
         `${this.consts.coins.sui.module}`,
@@ -424,7 +505,7 @@ export class ZLPAPI extends BaseAPI implements IZLPAPI {
     }
 
     tx.moveCall({
-      target: `${this.sharedConfig.zoStaking.package}::pool::deposit`,
+      target: `${this.sharedConfig.zoStaking.upgradedPackage}::pool::deposit`,
       typeArguments: [
         `${this.consts.zoCore.package}::zlp::ZLP`,
         `${this.consts.coins.sui.module}`,
@@ -458,7 +539,7 @@ export class ZLPAPI extends BaseAPI implements IZLPAPI {
       })()
       unstakeAmount -= withdrawAmount
       tx.moveCall({
-        target: `${this.sharedConfig.zoStaking.package}::pool::withdraw`,
+        target: `${this.sharedConfig.zoStaking.upgradedPackage}::pool::withdraw`,
         typeArguments: [
           `${this.consts.zoCore.package}::zlp::ZLP`,
           `${this.consts.coins.sui.module}`,
@@ -472,7 +553,7 @@ export class ZLPAPI extends BaseAPI implements IZLPAPI {
       })
       if (credential.amount === BigInt(0)) {
         tx.moveCall({
-          target: `${this.sharedConfig.zoStaking.package}::pool::clear_empty_credential`,
+          target: `${this.sharedConfig.zoStaking.upgradedPackage}::pool::clear_empty_credential`,
           typeArguments: [
             `${this.consts.zoCore.package}::zlp::ZLP`,
             `${this.consts.coins.sui.module}`,
@@ -484,6 +565,56 @@ export class ZLPAPI extends BaseAPI implements IZLPAPI {
     return tx
   }
 
+  /**
+   * Withdraws ZLP tokens from a staking pool credential (PTB)
+   * Returns the withdrawn coin object for use in programmable transaction blocks
+   */
+  public withdrawFromVaultPtb(
+    pool: string,
+    credential: string,
+    withdrawAmount: bigint,
+    tx: Transaction,
+  ): TransactionObjectArgument {
+    const [withdrawnCoin] = tx.moveCall({
+      target: `${this.sharedConfig.zoStaking.upgradedPackage}::pool::withdraw_ptb`,
+      typeArguments: [
+        `${this.consts.zoCore.package}::zlp::ZLP`,
+        `${this.consts.coins.sui.module}`,
+      ],
+      arguments: [
+        tx.object(pool),
+        tx.object(SUI_CLOCK_OBJECT_ID),
+        tx.object(credential),
+        tx.pure.u64(withdrawAmount),
+      ],
+    })
+    return withdrawnCoin
+  }
+
+  /**
+   * Claims rewards from a staking pool credential (PTB)
+   * Returns the reward coin object for use in programmable transaction blocks
+   */
+  public claimRewardsPtb(
+    pool: string,
+    credential: string,
+    tx: Transaction,
+  ): TransactionObjectArgument {
+    const [rewardCoin] = tx.moveCall({
+      target: `${this.sharedConfig.zoStaking.upgradedPackage}::pool::claim_rewards_ptb`,
+      typeArguments: [
+        `${this.consts.zoCore.package}::zlp::ZLP`,
+        `${this.consts.coins.sui.module}`,
+      ],
+      arguments: [
+        tx.object(pool),
+        tx.object(SUI_CLOCK_OBJECT_ID),
+        tx.object(credential),
+      ],
+    })
+    return rewardCoin
+  }
+
   public async swap(
     fromToken: string,
     toToken: string,

package/src/interfaces/slp.ts

@@ -4,7 +4,7 @@
  */
 
 import type { KioskClient, KioskOwnerCap } from '@mysten/kiosk'
-import type { Transaction } from '@mysten/sui/transactions'
+import type { Transaction, TransactionObjectArgument } from '@mysten/sui/transactions'
 
 import type {
   IBaseAPI,
@@ -182,4 +182,34 @@ export interface ISLPAPI extends IBaseAPI {
     kioskCap: KioskOwnerCap,
     scard: string
   ) => Transaction
+  depositPtb: (
+    coin: string,
+    coinObjects: string[],
+    amount: number,
+    minAmountOut?: number,
+    referralAddress?: string,
+    sender?: string,
+    tx?: Transaction,
+    sponsoredTx?: boolean,
+    suiCoinObjectsForPythUpdate?: string[],
+  ) => Promise<TransactionObjectArgument>
+
+  stakeCoinObject: (
+    coinObject: TransactionObjectArgument,
+    pool: string,
+    tx?: Transaction,
+  ) => Transaction
+
+  withdrawFromVaultPtb: (
+    pool: string,
+    credential: string,
+    withdrawAmount: bigint,
+    tx: Transaction,
+  ) => TransactionObjectArgument
+
+  claimRewardsPtb: (
+    pool: string,
+    credential: string,
+    tx: Transaction,
+  ) => TransactionObjectArgument
 }

package/src/interfaces/usdz.ts

@@ -3,7 +3,7 @@
  * Extends base interfaces with USDZ-specific implementations
  */
 
-import type { Transaction } from '@mysten/sui/transactions'
+import type { Transaction, TransactionObjectArgument } from '@mysten/sui/transactions'
 
 import type {
   IBaseAPI,
@@ -106,10 +106,24 @@ export interface IUSDZAPI extends IBaseAPI {
     tx?: Transaction,
     sponsoredTx?: boolean,
     suiCoinObjectsForPythUpdate?: string[],
-  ) => Promise<any>
+  ) => Promise<TransactionObjectArgument>
+
   stakeCoinObject: (
-    coinObject: any,
+    coinObject: TransactionObjectArgument,
     pool: string,
     tx?: Transaction,
-  ) => any
+  ) => Transaction
+
+  withdrawFromVaultPtb: (
+    pool: string,
+    credential: string,
+    withdrawAmount: bigint,
+    tx: Transaction,
+  ) => TransactionObjectArgument
+
+  claimRewardsPtb: (
+    pool: string,
+    credential: string,
+    tx: Transaction,
+  ) => TransactionObjectArgument
 }

package/src/interfaces/zbtcvc.ts

@@ -3,7 +3,7 @@
  * Extends base interfaces with USDZ-specific implementations
  */
 
-import type { Transaction } from '@mysten/sui/transactions'
+import type { Transaction, TransactionObjectArgument } from '@mysten/sui/transactions'
 
 import type {
   IBaseAPI,
@@ -106,10 +106,11 @@ export interface IZBTCVCAPI extends IBaseAPI {
     tx?: Transaction,
     sponsoredTx?: boolean,
     suiCoinObjectsForPythUpdate?: string[],
-  ) => Promise<any>
+  ) => Promise<TransactionObjectArgument>
+
   stakeCoinObject: (
-    coinObject: any,
+    coinObject: TransactionObjectArgument,
     pool: string,
     tx?: Transaction,
-  ) => any
+  ) => Transaction
 }

package/src/interfaces/zlp.ts

@@ -3,7 +3,7 @@
  * Extends base interfaces with ZLP-specific implementations
  */
 
-import type { Transaction } from '@mysten/sui/transactions'
+import type { Transaction, TransactionObjectArgument } from '@mysten/sui/transactions'
 
 import type {
   IBaseAPI,
@@ -134,4 +134,35 @@ export interface IZLPAPI extends IBaseAPI {
     coin: string,
     lootbox: string
   ) => Transaction
+
+  depositPtb: (
+    coin: string,
+    coinObjects: string[],
+    amount: number,
+    minAmountOut?: number,
+    referralAddress?: string,
+    sender?: string,
+    tx?: Transaction,
+    sponsoredTx?: boolean,
+    suiCoinObjectsForPythUpdate?: string[],
+  ) => Promise<TransactionObjectArgument>
+
+  stakeCoinObject: (
+    coinObject: TransactionObjectArgument,
+    pool: string,
+    tx?: Transaction,
+  ) => Transaction
+
+  withdrawFromVaultPtb: (
+    pool: string,
+    credential: string,
+    withdrawAmount: bigint,
+    tx: Transaction,
+  ) => TransactionObjectArgument
+
+  claimRewardsPtb: (
+    pool: string,
+    credential: string,
+    tx: Transaction,
+  ) => TransactionObjectArgument
 }