refactor: add debt coverage arg to adjust simulation

OasisDEX · Feb 19, 2024 · 4dbdf5c · 4dbdf5c
1 parent 8599d03
commit 4dbdf5c
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/packages/dma-library/src/strategies/aave-like/multiply/adjust/simulate.ts b/packages/dma-library/src/strategies/aave-like/multiply/adjust/simulate.ts
@@ -32,6 +32,9 @@ export async function simulate(
   )
 
   const currentPosition = await resolveCurrentPositionForProtocol(args, dependencies)
+  if (!args.debtCoverage.isZero()) {
+    currentPosition.debt.amount = currentPosition.debt.amount.plus(args.debtCoverage)
+  }
   const protocolData = await resolveProtocolData(
     {
       collateralTokenAddress,

diff --git a/packages/dma-library/src/strategies/aave-like/multiply/adjust/types.ts b/packages/dma-library/src/strategies/aave-like/multiply/adjust/types.ts
@@ -8,6 +8,7 @@ import BigNumber from 'bignumber.js'
 export type AaveLikeAdjustArgs = StrategyParams.WithAaveLikeMultiplyStrategyArgs &
   StrategyParams.WithMultiple &
   StrategyParams.WithDeposit &
+  StrategyParams.WithDebtCoverage &
   Partial<StrategyParams.WithFlashLoanArgs>
 
 export type ExtendedAaveLikeAdjustArgs = AaveLikeAdjustArgs & StrategyParams.WithFlashLoanArgs

diff --git a/packages/dma-library/src/types/strategy-params.ts b/packages/dma-library/src/types/strategy-params.ts
@@ -154,3 +154,7 @@ export type WithCollateralTokenAddress = {
 export type WithDebtTokenAddress = {
   debtTokenAddress: string
 }
+
+export type WithDebtCoverage = {
+  debtCoverage: BigNumber
+}