@0dotxyz/p0-ts-sdk 1.2.0-alpha.2 → 1.2.1

package/dist/index.cjs

@@ -43353,8 +43353,22 @@ async function buildSwapCollateralFlashloanTx({
   overrideInferAccounts,
   blockhash
 }) {
-  const { withdrawBank, tokenProgram: withdrawTokenProgram, totalPositionAmount } = withdrawOpts;
+  const {
+    withdrawBank,
+    tokenProgram: withdrawTokenProgram,
+    totalPositionAmount,
+    withdrawAmount
+  } = withdrawOpts;
   const { depositBank, tokenProgram: depositTokenProgram } = depositOpts;
+  if (withdrawAmount !== void 0 && withdrawAmount <= 0) {
+    throw new Error("withdrawAmount must be greater than 0");
+  }
+  const actualWithdrawAmount = Math.min(withdrawAmount ?? totalPositionAmount, totalPositionAmount);
+  const isFullWithdraw = isWholePosition(
+    { amount: totalPositionAmount, isLending: true },
+    actualWithdrawAmount,
+    withdrawBank.mintDecimals
+  );
   const swapResult = [];
   const cuRequestIxs = [
     web3_js.ComputeBudgetProgram.setComputeUnitLimit({ units: 12e5 }),
@@ -43371,7 +43385,7 @@ async function buildSwapCollateralFlashloanTx({
           withdrawOpts.withdrawBank.tokenSymbol
         );
       }
-      const adjustedAmount = new BigNumber10.BigNumber(totalPositionAmount).div(withdrawOpts.withdrawBank.assetShareValue).times(1.0001).toNumber();
+      const adjustedAmount = new BigNumber10.BigNumber(actualWithdrawAmount).div(withdrawOpts.withdrawBank.assetShareValue).times(1.0001).toNumber();
       withdrawIxs = await makeKaminoWithdrawIx3({
         program,
         bank: withdrawBank,
@@ -43381,7 +43395,7 @@ async function buildSwapCollateralFlashloanTx({
         marginfiAccount,
         authority: marginfiAccount.authority,
         reserve,
-        withdrawAll: true,
+        withdrawAll: isFullWithdraw,
         isSync: true,
         opts: {
           createAtas: false,
@@ -43405,12 +43419,12 @@ async function buildSwapCollateralFlashloanTx({
         bank: withdrawOpts.withdrawBank,
         bankMap,
         tokenProgram: withdrawOpts.tokenProgram,
-        amount: totalPositionAmount,
+        amount: actualWithdrawAmount,
         marginfiAccount,
         authority: marginfiAccount.authority,
         driftSpotMarket: driftState.spotMarketState,
         userRewards: driftState.userRewards,
-        withdrawAll: true,
+        withdrawAll: isFullWithdraw,
         isSync: false,
         opts: {
           createAtas: false,
@@ -43426,10 +43440,10 @@ async function buildSwapCollateralFlashloanTx({
         bank: withdrawBank,
         bankMap,
         tokenProgram: withdrawTokenProgram,
-        amount: totalPositionAmount,
+        amount: actualWithdrawAmount,
         marginfiAccount,
         authority: marginfiAccount.authority,
-        withdrawAll: true,
+        withdrawAll: isFullWithdraw,
         isSync: true,
         opts: {
           createAtas: false,
@@ -43442,7 +43456,7 @@ async function buildSwapCollateralFlashloanTx({
   }
   if (depositBank.mint.equals(withdrawBank.mint)) {
     swapResult.push({
-      amountToDeposit: totalPositionAmount,
+      amountToDeposit: actualWithdrawAmount,
       swapInstructions: [],
       setupInstructions: [],
       swapLookupTables: []
@@ -43458,7 +43472,7 @@ async function buildSwapCollateralFlashloanTx({
       quoteParams: {
         inputMint: withdrawBank.mint.toBase58(),
         outputMint: depositBank.mint.toBase58(),
-        amount: uiToNative(totalPositionAmount, withdrawBank.mintDecimals).toNumber(),
+        amount: uiToNative(actualWithdrawAmount, withdrawBank.mintDecimals).toNumber(),
         dynamicSlippage: swapOpts.jupiterOptions ? swapOpts.jupiterOptions.slippageMode === "DYNAMIC" : true,
         slippageBps: swapOpts.jupiterOptions?.slippageBps,
         swapMode: "ExactIn",
@@ -43725,56 +43739,74 @@ async function buildSwapDebtFlashloanTx({
   overrideInferAccounts,
   blockhash
 }) {
-  const { repayBank, tokenProgram: repayTokenProgram, totalPositionAmount } = repayOpts;
+  const {
+    repayBank,
+    tokenProgram: repayTokenProgram,
+    totalPositionAmount,
+    repayAmount
+  } = repayOpts;
   const { borrowBank, tokenProgram: borrowTokenProgram } = borrowOpts;
+  if (repayAmount !== void 0 && repayAmount <= 0) {
+    throw new Error("repayAmount must be greater than 0");
+  }
+  const actualRepayAmount = Math.min(repayAmount ?? totalPositionAmount, totalPositionAmount);
   const swapResult = [];
   const cuRequestIxs = [
     web3_js.ComputeBudgetProgram.setComputeUnitLimit({ units: 12e5 }),
     web3_js.ComputeBudgetProgram.setComputeUnitPrice({ microLamports: 1 })
   ];
-  if (borrowBank.mint.equals(repayBank.mint)) {
-    swapResult.push({
-      amountToRepay: totalPositionAmount,
-      borrowAmount: totalPositionAmount,
-      swapInstructions: [],
-      setupInstructions: [],
-      swapLookupTables: []
-    });
-  } else {
-    const destinationTokenAccount = getAssociatedTokenAddressSync(
-      repayBank.mint,
-      marginfiAccount.authority,
-      true,
-      repayTokenProgram.equals(TOKEN_2022_PROGRAM_ID) ? TOKEN_2022_PROGRAM_ID : void 0
+  const destinationTokenAccount = getAssociatedTokenAddressSync(
+    repayBank.mint,
+    marginfiAccount.authority,
+    true,
+    repayTokenProgram.equals(TOKEN_2022_PROGRAM_ID) ? TOKEN_2022_PROGRAM_ID : void 0
+  );
+  const jupiterApiClient = swapOpts.jupiterOptions?.configParams?.basePath ? new api.SwapApi(new api.Configuration(swapOpts.jupiterOptions.configParams)) : api.createJupiterApiClient(swapOpts.jupiterOptions?.configParams);
+  const estimateQuote = await jupiterApiClient.quoteGet({
+    inputMint: borrowBank.mint.toBase58(),
+    outputMint: repayBank.mint.toBase58(),
+    amount: uiToNative(actualRepayAmount, repayBank.mintDecimals).toNumber(),
+    swapMode: "ExactOut",
+    dynamicSlippage: swapOpts.jupiterOptions ? swapOpts.jupiterOptions.slippageMode === "DYNAMIC" : true,
+    slippageBps: swapOpts.jupiterOptions?.slippageBps
+  });
+  const estimatedBorrowAmount = nativeToUi(
+    estimateQuote.otherAmountThreshold,
+    borrowBank.mintDecimals
+  );
+  const swapResponses = await getJupiterSwapIxsForFlashloan({
+    quoteParams: {
+      inputMint: borrowBank.mint.toBase58(),
+      outputMint: repayBank.mint.toBase58(),
+      amount: uiToNative(estimatedBorrowAmount, borrowBank.mintDecimals).toNumber(),
+      dynamicSlippage: swapOpts.jupiterOptions ? swapOpts.jupiterOptions.slippageMode === "DYNAMIC" : true,
+      slippageBps: swapOpts.jupiterOptions?.slippageBps,
+      swapMode: "ExactIn",
+      platformFeeBps: swapOpts.jupiterOptions?.platformFeeBps,
+      onlyDirectRoutes: swapOpts.jupiterOptions?.directRoutesOnly ?? false
+    },
+    authority: marginfiAccount.authority,
+    connection,
+    destinationTokenAccount,
+    configParams: swapOpts.jupiterOptions?.configParams
+  });
+  swapResponses.forEach((response) => {
+    const outAmount = nativeToUi(response.quoteResponse.outAmount, repayBank.mintDecimals);
+    const outAmountThreshold = nativeToUi(
+      response.quoteResponse.otherAmountThreshold,
+      repayBank.mintDecimals
     );
-    const swapResponses = await getJupiterSwapIxsForFlashloan({
-      quoteParams: {
-        inputMint: borrowBank.mint.toBase58(),
-        outputMint: repayBank.mint.toBase58(),
-        amount: uiToNative(totalPositionAmount, repayBank.mintDecimals).toNumber(),
-        dynamicSlippage: swapOpts.jupiterOptions ? swapOpts.jupiterOptions.slippageMode === "DYNAMIC" : true,
-        slippageBps: swapOpts.jupiterOptions?.slippageBps,
-        swapMode: "ExactOut",
-        platformFeeBps: swapOpts.jupiterOptions?.platformFeeBps,
-        onlyDirectRoutes: swapOpts.jupiterOptions?.directRoutesOnly ?? false
-      },
-      authority: marginfiAccount.authority,
-      connection,
-      destinationTokenAccount,
-      configParams: swapOpts.jupiterOptions?.configParams
-    });
-    swapResponses.forEach((response) => {
-      const borrowAmount = nativeToUi(response.quoteResponse.inAmount, borrowBank.mintDecimals);
-      swapResult.push({
-        amountToRepay: totalPositionAmount,
-        borrowAmount,
-        swapInstructions: [response.swapInstruction],
-        setupInstructions: response.setupInstructions,
-        swapLookupTables: response.addressLookupTableAddresses,
-        quoteResponse: response.quoteResponse
-      });
+    const amountToRepay = outAmount > totalPositionAmount ? totalPositionAmount : outAmountThreshold;
+    const borrowAmount = nativeToUi(response.quoteResponse.inAmount, borrowBank.mintDecimals);
+    swapResult.push({
+      amountToRepay,
+      borrowAmount,
+      swapInstructions: [response.swapInstruction],
+      setupInstructions: response.setupInstructions,
+      swapLookupTables: response.addressLookupTableAddresses,
+      quoteResponse: response.quoteResponse
     });
-  }
+  });
   if (swapResult.length === 0) {
     throw new Error(
       `No swap routes found for ${borrowBank.mint.toBase58()} -> ${repayBank.mint.toBase58()}`