zo-sdk 0.1.8 → 0.1.9

package/src/implementations/USDZAPI.ts

@@ -216,17 +216,13 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
     relayerFee = BigInt(0.5),
     referralAddress?: string,
     sender?: string,
+    sponsoredTx?: boolean,
+    suiCoinObjectsForPythUpdate?: string[],
   ): Promise<Transaction> {
     let tx = new Transaction()
     if (referralAddress && !(await this.dataAPI.hasReferral(sender || ''))) {
       tx = await this.addReferral(referralAddress, tx)
     }
-    tx = await this.initOracleTxb([collateralToken, indexToken], tx)
-    const coinObject = this.processCoins(tx, collateralToken, coinObjects)
-    const [depositObject, feeObject] = tx.splitCoins(coinObject, [
-      tx.pure.u64(collateralAmount),
-      tx.pure.u64(relayerFee),
-    ])
 
     const symbol = joinSymbol(long ? 'long' : 'short', indexToken)
     const adjustPrice = this.processSlippage(indexPrice, long, isLimitOrder ? 0 : pricesSlippage)
@@ -248,6 +244,26 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
       itemId: scard,
     })
 
+    // Handle oracle initialization and coin processing
+    let suiCoinObject
+    if (sponsoredTx) {
+      suiCoinObject = this.processCoins(tx, 'sui', suiCoinObjectsForPythUpdate || [], true)
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx, true, suiCoinObject)
+    }
+    else {
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx)
+    }
+
+    // Process coin splitting
+    const [depositObject, feeObject] = this.processCoinSplitting(
+      tx,
+      collateralToken,
+      coinObjects,
+      [tx.pure.u64(collateralAmount), tx.pure.u64(relayerFee)],
+      sponsoredTx,
+      suiCoinObject,
+    )
+
     tx.moveCall({
       target: `${this.consts.zoCore.package}::market::open_position_with_scard`,
       typeArguments: [
@@ -304,11 +320,14 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
     collateralSlippage = 0.5,
     relayerFee = BigInt(0.5),
     coinObjects?: string[],
+    sponsoredTx?: boolean,
+    suiCoinObjectsForPythUpdate?: string[],
   ): Promise<Transaction> {
     if (!coinObjects) {
       throw new Error(`${this.constructor.name}: coinObjects is required`)
     }
-    const tx = await this.initOracleTxb([collateralToken, indexToken])
+
+    let tx = new Transaction()
 
     const kioskTx = new KioskTransaction({
       transaction: tx,
@@ -322,8 +341,6 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
     })
 
     const symbol = joinSymbol(long ? 'long' : 'short', indexToken)
-    const coinObject = this.processCoins(tx, collateralToken, coinObjects || [])
-    const feeObject = tx.splitCoins(coinObject, [tx.pure.u64(relayerFee)])
 
     const adjustPrice = this.processSlippage(indexPrice, !long, isTriggerOrder ? 0 : pricesSlippage)
     const adjustCollateralPrice = this.processSlippage(collateralPrice, false, collateralSlippage)
@@ -336,6 +353,26 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
       isTakeProfitOrder = true
     }
 
+    // Handle oracle initialization and coin processing
+    let suiCoinObject
+    if (sponsoredTx) {
+      suiCoinObject = this.processCoins(tx, 'sui', suiCoinObjectsForPythUpdate || [], true)
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx, true, suiCoinObject)
+    }
+    else {
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx)
+    }
+
+    // Process coin splitting
+    const [feeObject] = this.processCoinSplitting(
+      tx,
+      collateralToken,
+      coinObjects,
+      [tx.pure.u64(relayerFee)],
+      sponsoredTx,
+      suiCoinObject,
+    )
+
     tx.moveCall({
       target: `${this.consts.zoCore.package}::market::decrease_position_with_scard`,
       typeArguments: [
@@ -394,11 +431,22 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
     kioskCap: KioskOwnerCap,
     scard: string,
     tx?: Transaction,
+    sponsoredTx?: boolean,
+    suiCoinObjectsForPythUpdate?: string[],
   ): Promise<Transaction> {
     if (!tx) {
       tx = new Transaction()
     }
-    tx = await this.initOracleTxb(positions.flatMap(position => [position.collateralToken, position.indexToken]), tx)
+
+    // Handle oracle initialization and coin processing
+    let suiCoinObject
+    if (sponsoredTx) {
+      suiCoinObject = this.processCoins(tx, 'sui', suiCoinObjectsForPythUpdate || [], true)
+      tx = await this.initOracleTxb(positions.flatMap(position => [position.collateralToken, position.indexToken]), tx, true, suiCoinObject)
+    }
+    else {
+      tx = await this.initOracleTxb(positions.flatMap(position => [position.collateralToken, position.indexToken]), tx)
+    }
 
     const kioskTx = new KioskTransaction({
       transaction: tx,
@@ -430,8 +478,6 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
       } = position
       let innerIsTakeProfitOrder = isTakeProfitOrder
       const symbol = joinSymbol(long ? 'long' : 'short', indexToken)
-      const coinObject = this.processCoins(tx, collateralToken, coinObjects)
-      const feeObject = tx.splitCoins(coinObject, [tx.pure.u64(relayerFee)])
 
       const adjustPrice = this.processSlippage(indexPrice, !long, isTriggerOrder ? 0 : pricesSlippage)
       const adjustCollateralPrice = this.processSlippage(collateralPrice, false, collateralSlippage)
@@ -444,6 +490,16 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
         innerIsTakeProfitOrder = true
       }
 
+      // Process coin splitting
+      const [feeObject] = this.processCoinSplitting(
+        tx,
+        collateralToken,
+        coinObjects,
+        [tx.pure.u64(relayerFee)],
+        sponsoredTx,
+        suiCoinObject,
+      )
+
       tx.moveCall({
         target: `${this.consts.zoCore.package}::market::decrease_position_with_scard`,
         typeArguments: [
@@ -494,7 +550,6 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
     sender?: string,
     sponsoredTx?: boolean,
     suiCoinObjectsForPythUpdate?: string[],
-    isDepositingSui?: boolean,
   ): Promise<Transaction> {
     let tx = new Transaction()
 
@@ -512,7 +567,7 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
       tx = await this.initOracleTxb(pythFeederKeys, tx, true, suiCoinObject)
 
       // Process deposit coins
-      const depositObject = isDepositingSui
+      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]
 
@@ -650,32 +705,48 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
     collateralPrice: number,
     isLimitOrder?: boolean,
     isIocOrder?: boolean,
-    pricesSlippage?: number,
-    collateralSlippage?: number,
-    relayerFee?: bigint,
+    pricesSlippage = 0.003,
+    collateralSlippage = 0.5,
+    relayerFee = BigInt(0.5),
     referralAddress?: string,
     sender?: string,
+    sponsoredTx?: boolean,
+    suiCoinObjectsForPythUpdate?: string[],
   ): Promise<Transaction> {
     let tx = new Transaction()
     if (referralAddress && !(await this.dataAPI.hasReferral(sender || ''))) {
       tx = await this.addReferral(referralAddress, tx)
     }
-    tx = await this.initOracleTxb([collateralToken, indexToken], tx)
-    const coinObject = this.processCoins(tx, collateralToken, coinObjects)
-    const [depositObject, feeObject] = tx.splitCoins(coinObject, [
-      tx.pure.u64(collateralAmount),
-      tx.pure.u64(relayerFee || BigInt(0.5)),
-    ])
 
     const symbol = joinSymbol(long ? 'long' : 'short', indexToken)
-    const adjustPrice = this.processSlippage(indexPrice, long, isLimitOrder ? 0 : pricesSlippage || 0.003)
-    const adjustCollateralPrice = this.processSlippage(collateralPrice, false, collateralSlippage || 0.5)
+    const adjustPrice = this.processSlippage(indexPrice, long, isLimitOrder ? 0 : pricesSlippage)
+    const adjustCollateralPrice = this.processSlippage(collateralPrice, false, collateralSlippage)
 
     let allowTrade = ALLOW_TRADE_MUST_TRADE
     if (isLimitOrder) {
       allowTrade = isIocOrder ? ALLOW_TRADE_NO_TRADE : ALLOW_TRADE_CAN_TRADE
     }
 
+    // Handle oracle initialization and coin processing
+    let suiCoinObject
+    if (sponsoredTx) {
+      suiCoinObject = this.processCoins(tx, 'sui', suiCoinObjectsForPythUpdate || [], true)
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx, true, suiCoinObject)
+    }
+    else {
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx)
+    }
+
+    // Process coin splitting
+    const [depositObject, feeObject] = this.processCoinSplitting(
+      tx,
+      collateralToken,
+      coinObjects,
+      [tx.pure.u64(collateralAmount), tx.pure.u64(relayerFee)],
+      sponsoredTx,
+      suiCoinObject,
+    )
+
     tx.moveCall({
       target: `${this.consts.zoCore.package}::market::open_position`,
       typeArguments: [
@@ -723,14 +794,14 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
     collateralSlippage = 0.5,
     relayerFee = BigInt(0.5),
     coinObjects?: string[],
+    sponsoredTx?: boolean,
+    suiCoinObjectsForPythUpdate?: string[],
   ): Promise<Transaction> {
     if (!coinObjects) {
       throw new Error(`${this.constructor.name}: coinObjects is required`)
     }
-    const tx = await this.initOracleTxb([collateralToken, indexToken])
+    let tx = new Transaction()
     const symbol = joinSymbol(long ? 'long' : 'short', indexToken)
-    const coinObject = this.processCoins(tx, collateralToken, coinObjects || [])
-    const feeObject = tx.splitCoins(coinObject, [tx.pure.u64(relayerFee)])
 
     const adjustPrice = this.processSlippage(indexPrice, !long, isTriggerOrder ? 0 : pricesSlippage)
     const adjustCollateralPrice = this.processSlippage(collateralPrice, false, collateralSlippage)
@@ -743,6 +814,26 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
       isTakeProfitOrder = true
     }
 
+    // Handle oracle initialization and coin processing
+    let suiCoinObject
+    if (sponsoredTx) {
+      suiCoinObject = this.processCoins(tx, 'sui', suiCoinObjectsForPythUpdate || [], true)
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx, true, suiCoinObject)
+    }
+    else {
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx)
+    }
+
+    // Process coin splitting
+    const [feeObject] = this.processCoinSplitting(
+      tx,
+      collateralToken,
+      coinObjects,
+      [tx.pure.u64(relayerFee)],
+      sponsoredTx,
+      suiCoinObject,
+    )
+
     tx.moveCall({
       target: `${this.consts.zoCore.package}::market::decrease_position`,
       typeArguments: [
@@ -787,11 +878,20 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
     collateralSlippage?: number
     relayerFee?: bigint
     coinObjects?: string[]
-  }>, tx?: Transaction): Promise<Transaction> {
+  }>, tx?: Transaction, sponsoredTx?: boolean, suiCoinObjectsForPythUpdate?: string[]): Promise<Transaction> {
     if (!tx) {
       tx = new Transaction()
     }
-    tx = await this.initOracleTxb(positions.flatMap(position => [position.collateralToken, position.indexToken]), tx)
+
+    // Handle oracle initialization and coin processing
+    let suiCoinObject
+    if (sponsoredTx) {
+      suiCoinObject = this.processCoins(tx, 'sui', suiCoinObjectsForPythUpdate || [], true)
+      tx = await this.initOracleTxb(positions.flatMap(position => [position.collateralToken, position.indexToken]), tx, true, suiCoinObject)
+    }
+    else {
+      tx = await this.initOracleTxb(positions.flatMap(position => [position.collateralToken, position.indexToken]), tx)
+    }
 
     for (const position of positions) {
       const {
@@ -812,8 +912,6 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
       } = position
       let innerIsTakeProfitOrder = isTakeProfitOrder
       const symbol = joinSymbol(long ? 'long' : 'short', indexToken)
-      const coinObject = this.processCoins(tx, collateralToken, coinObjects)
-      const feeObject = tx.splitCoins(coinObject, [tx.pure.u64(relayerFee)])
 
       const adjustPrice = this.processSlippage(indexPrice, !long, isTriggerOrder ? 0 : pricesSlippage)
       const adjustCollateralPrice = this.processSlippage(collateralPrice, false, collateralSlippage)
@@ -826,6 +924,16 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
         innerIsTakeProfitOrder = true
       }
 
+      // Process coin splitting
+      const [feeObject] = this.processCoinSplitting(
+        tx,
+        collateralToken,
+        coinObjects,
+        [tx.pure.u64(relayerFee)],
+        sponsoredTx,
+        suiCoinObject,
+      )
+
       tx.moveCall({
         target: `${this.consts.zoCore.package}::market::decrease_position`,
         typeArguments: [
@@ -865,10 +973,29 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
     amount: number,
     coinObjects: string[],
     long: boolean,
+    sponsoredTx?: boolean,
+    suiCoinObjectsForPythUpdate?: string[],
   ): Promise<Transaction> {
-    const tx = await this.initOracleTxb([collateralToken, indexToken])
-    const coinObject = this.processCoins(tx, collateralToken, coinObjects)
-    const [depositObject] = tx.splitCoins(coinObject, [tx.pure.u64(amount)])
+    let tx = new Transaction()
+    // Handle oracle initialization and coin processing
+    let suiCoinObject
+    if (sponsoredTx) {
+      suiCoinObject = this.processCoins(tx, 'sui', suiCoinObjectsForPythUpdate || [], true)
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx, true, suiCoinObject)
+    }
+    else {
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx)
+    }
+
+    // Process coin splitting
+    const [depositObject] = this.processCoinSplitting(
+      tx,
+      collateralToken,
+      coinObjects,
+      [tx.pure.u64(amount)],
+      sponsoredTx,
+      suiCoinObject,
+    )
 
     tx.moveCall({
       target: `${this.consts.zoCore.package}::market::pledge_in_position`,
@@ -889,8 +1016,19 @@ export class USDZAPI extends BaseAPI implements IUSDZAPI {
     indexToken: string,
     amount: number,
     long: boolean,
+    sponsoredTx?: boolean,
+    suiCoinObjectsForPythUpdate?: string[],
   ): Promise<Transaction> {
-    const tx = await this.initOracleTxb([collateralToken, indexToken])
+    let tx = new Transaction()
+    // Handle oracle initialization and coin processing
+    let suiCoinObject
+    if (sponsoredTx) {
+      suiCoinObject = this.processCoins(tx, 'sui', suiCoinObjectsForPythUpdate || [], true)
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx, true, suiCoinObject)
+    }
+    else {
+      tx = await this.initOracleTxb([collateralToken, indexToken], tx)
+    }
     const symbol = joinSymbol(long ? 'long' : 'short', indexToken)
 
     tx.moveCall({