feat: add convenience input/output amounts to flash mint quote

IndexCoop · May 27, 2024 · 00855bf · 00855bf
1 parent 9507c2c
commit 00855bf
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 3 deletions.
diff --git a/src/quote/provider/index.test.ts b/src/quote/provider/index.test.ts
@@ -75,6 +75,8 @@ describe('FlashMintQuoteProvider()', () => {
     expect(quote.isMinting).toEqual(request.isMinting)
     expect(quote.inputToken).toEqual(request.inputToken)
     expect(quote.outputToken).toEqual(request.outputToken)
+    expect(quote.inputAmount).toEqual(quote.inputOutputAmount)
+    expect(quote.outputAmount).toEqual(request.indexTokenAmount)
     expect(quote.indexTokenAmount).toEqual(request.indexTokenAmount)
     expect(quote.inputOutputAmount.gt(0)).toBe(true)
     expect(quote.slippage).toEqual(request.slippage)
@@ -110,6 +112,8 @@ describe('FlashMintQuoteProvider()', () => {
     expect(quote.isMinting).toEqual(request.isMinting)
     expect(quote.inputToken).toEqual(request.inputToken)
     expect(quote.outputToken).toEqual(request.outputToken)
+    expect(quote.inputAmount).toEqual(quote.inputOutputAmount)
+    expect(quote.outputAmount).toEqual(request.indexTokenAmount)
     expect(quote.indexTokenAmount).toEqual(request.indexTokenAmount)
     expect(quote.inputOutputAmount.gt(0)).toBe(true)
     expect(quote.slippage).toEqual(request.slippage)
@@ -145,6 +149,8 @@ describe('FlashMintQuoteProvider()', () => {
     expect(quote.isMinting).toEqual(request.isMinting)
     expect(quote.inputToken).toEqual(request.inputToken)
     expect(quote.outputToken).toEqual(request.outputToken)
+    expect(quote.inputAmount).toEqual(request.indexTokenAmount)
+    expect(quote.outputAmount).toEqual(quote.inputOutputAmount)
     expect(quote.indexTokenAmount).toEqual(request.indexTokenAmount)
     expect(quote.inputOutputAmount.gt(0)).toBe(true)
     expect(quote.slippage).toEqual(request.slippage)
@@ -186,6 +192,8 @@ describe('FlashMintQuoteProvider()', () => {
     expect(quote.isMinting).toEqual(request.isMinting)
     expect(quote.inputToken).toEqual(request.inputToken)
     expect(quote.outputToken).toEqual(request.outputToken)
+    expect(quote.outputToken).toEqual(request.outputToken)
+    expect(quote.inputAmount).toEqual(quote.inputOutputAmount)
     expect(quote.indexTokenAmount).toEqual(request.indexTokenAmount)
     expect(quote.inputOutputAmount.gt(0)).toBe(true)
     expect(quote.slippage).toEqual(request.slippage)
@@ -227,6 +235,8 @@ describe('FlashMintQuoteProvider()', () => {
     expect(quote.isMinting).toEqual(request.isMinting)
     expect(quote.inputToken).toEqual(request.inputToken)
     expect(quote.outputToken).toEqual(request.outputToken)
+    expect(quote.inputAmount).toEqual(request.indexTokenAmount)
+    expect(quote.outputAmount).toEqual(quote.inputOutputAmount)
     expect(quote.indexTokenAmount).toEqual(request.indexTokenAmount)
     expect(quote.inputOutputAmount.gt(0)).toBe(true)
     expect(quote.slippage).toEqual(request.slippage)
@@ -263,6 +273,8 @@ describe('FlashMintQuoteProvider()', () => {
     expect(quote.isMinting).toEqual(request.isMinting)
     expect(quote.inputToken).toEqual(request.inputToken)
     expect(quote.outputToken).toEqual(request.outputToken)
+    expect(quote.inputAmount).toEqual(request.indexTokenAmount)
+    expect(quote.outputAmount).toEqual(quote.inputOutputAmount)
     expect(quote.indexTokenAmount).toEqual(request.indexTokenAmount)
     expect(quote.inputOutputAmount.gt(0)).toBe(true)
     expect(quote.slippage).toEqual(request.slippage)

diff --git a/src/quote/provider/index.ts b/src/quote/provider/index.ts
@@ -41,6 +41,8 @@ export interface FlashMintQuote {
   isMinting: boolean
   inputToken: QuoteToken
   outputToken: QuoteToken
+  inputAmount: BigNumber
+  outputAmount: BigNumber
   indexTokenAmount: BigNumber
   inputOutputAmount: BigNumber
   slippage: number
@@ -92,6 +94,7 @@ export class FlashMintQuoteProvider
         }
         const tx = await builder.build(txRequest)
         if (!tx) return null
+        const { inputOutputTokenAmount } = leveragedQuote
         return {
           chainId,
           contractType,
@@ -100,8 +103,10 @@ export class FlashMintQuoteProvider
           isMinting,
           inputToken,
           outputToken,
+          inputAmount: isMinting ? inputOutputTokenAmount : indexTokenAmount,
+          outputAmount: isMinting ? indexTokenAmount : inputOutputTokenAmount,
           indexTokenAmount,
-          inputOutputAmount: leveragedQuote.inputOutputTokenAmount,
+          inputOutputAmount: inputOutputTokenAmount,
           slippage,
           tx,
         }
@@ -130,6 +135,7 @@ export class FlashMintQuoteProvider
         }
         const tx = await builder.build(txRequest)
         if (!tx) return null
+        const { inputOutputTokenAmount } = leveragedExtendedQuote
         return {
           chainId,
           contractType,
@@ -138,8 +144,10 @@ export class FlashMintQuoteProvider
           isMinting,
           inputToken,
           outputToken,
+          inputAmount: isMinting ? inputOutputTokenAmount : indexTokenAmount,
+          outputAmount: isMinting ? indexTokenAmount : inputOutputTokenAmount,
           indexTokenAmount,
-          inputOutputAmount: leveragedExtendedQuote.inputOutputTokenAmount,
+          inputOutputAmount: inputOutputTokenAmount,
           slippage,
           tx,
         }
@@ -164,6 +172,7 @@ export class FlashMintQuoteProvider
         }
         const tx = await builder.build(txRequest)
         if (!tx) return null
+        const { inputOutputTokenAmount } = zeroExQuote
         return {
           chainId,
           contractType,
@@ -172,8 +181,10 @@ export class FlashMintQuoteProvider
           isMinting,
           inputToken,
           outputToken,
+          inputAmount: isMinting ? inputOutputTokenAmount : indexTokenAmount,
+          outputAmount: isMinting ? indexTokenAmount : inputOutputTokenAmount,
           indexTokenAmount,
-          inputOutputAmount: zeroExQuote.inputOutputTokenAmount,
+          inputOutputAmount: inputOutputTokenAmount,
           slippage,
           tx,
         }