add swaptype to response (#163)

balancer · Feb 22, 2024 · 78bf304 · 78bf304
1 parent 7aa3fd5
commit 78bf304
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/modules/sor/sor.gql b/modules/sor/sor.gql
@@ -40,6 +40,7 @@ type GqlSorGetSwapPaths {
     The token address of the tokenOut provided
     """
     tokenOut: String!
+    swapType: GqlSorSwapType!
     swaps: [GqlSorSwap!]! #used by cowswap
     paths: [GqlSorPath!]! #used by b-sdk
     tokenInAmount: AmountHumanReadable!

diff --git a/modules/sor/sorV2/sorV2.service.ts b/modules/sor/sorV2/sorV2.service.ts
@@ -201,6 +201,7 @@ export class SorV2Service implements SwapService {
         return {
             vaultVersion: 2,
             paths: paths,
+            swapType: this.mapSwapKindToSwapType(swap.swapKind),
             swaps: this.mapSwaps(swap.swaps, swap.assets),
             tokenIn: replaceZeroAddressWithEth(inputAmount.token.address),
             tokenOut: replaceZeroAddressWithEth(outputAmount.token.address),
@@ -227,6 +228,10 @@ export class SorV2Service implements SwapService {
         return swapType === 'EXACT_IN' ? SwapKind.GivenIn : SwapKind.GivenOut;
     }
 
+    private mapSwapKindToSwapType(swapKind: SwapKind): GqlSorSwapType {
+        return swapKind === SwapKind.GivenIn ? 'EXACT_IN' : 'EXACT_OUT';
+    }
+
     private mapSwaps(swaps: BatchSwapStep[] | SingleSwap, assets: string[]): GqlSorSwap[] {
         if (Array.isArray(swaps)) {
             return swaps.map((swap) => {

diff --git a/modules/sor/utils.ts b/modules/sor/utils.ts
@@ -41,6 +41,7 @@ export const swapPathsZeroResponse = (tokenIn: string, tokenOut: string): GqlSor
     return {
         swaps: [],
         paths: [],
+        swapType: 'EXACT_IN',
         vaultVersion: 2,
         tokenIn: replaceZeroAddressWithEth(tokenIn),
         tokenOut: replaceZeroAddressWithEth(tokenOut),