fix: reuse function for offset

aave · Apr 2, 2021 · 19da691 · 19da691
1 parent 7fb80bc
commit 19da691
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 27 deletions.
diff --git a/src/tx-builder/services/LiquiditySwapAdapterParaswap.ts b/src/tx-builder/services/LiquiditySwapAdapterParaswap.ts
@@ -16,7 +16,7 @@ import { LiquiditySwapValidator } from '../validators/methodValidators';
 import { IsEthAddress, IsPositiveAmount } from '../validators/paramValidators';
 import BaseService from './BaseService';
 
-function augustusFromAmountOffsetFromCalldata(calldata: string) {
+export function augustusFromAmountOffsetFromCalldata(calldata: string) {
   switch (calldata.slice(0, 10)) {
     case '0xda8567c8': // Augustus V3 multiSwap
       return 4 + 32 + 2 * 32;

diff --git a/src/tx-builder/services/v2/LendingPool.ts b/src/tx-builder/services/v2/LendingPool.ts
@@ -46,6 +46,7 @@ import {
 import LiquiditySwapAdapterInterface from '../../interfaces/LiquiditySwapAdapterParaswap';
 import RepayWithCollateralAdapterInterface from '../../interfaces/RepayWithCollateralAdapter';
 import BaseService from '../BaseService';
+import { augustusFromAmountOffsetFromCalldata } from '../LiquiditySwapAdapterParaswap';
 
 const buildParaSwapLiquiditySwapParams = (
   assetToSwapTo: tEthereumAddress,
@@ -79,15 +80,6 @@ const buildParaSwapLiquiditySwapParams = (
   );
 };
 
-function augustusFromAmountOffsetFromCalldata(calldata: string) {
-  switch (calldata.slice(0, 10)) {
-    case '0xda8567c8': // Augustus V3 multiSwap
-      return 4 + 32 + 2 * 32;
-    default:
-      throw new Error('Unrecognized function selector for Augustus');
-  }
-}
-
 export default class LendingPool
   extends BaseService<ILendingPool>
   implements LendingPoolInterface {
@@ -593,22 +585,6 @@ export default class LendingPool
       swapCallData,
     }: LPSwapCollateral
   ): Promise<EthereumTransactionTypeExtended[]> {
-    console.log('inParams', {
-      user,
-      flash,
-      fromAsset,
-      fromAToken,
-      toAsset,
-      fromAmount,
-      toAmount,
-      maxSlippage,
-      permitSignature,
-      swapAll,
-      onBehalfOf,
-      referralCode,
-      augustus,
-      swapCallData,
-    });
     const txs: EthereumTransactionTypeExtended[] = [];
 
     const permitParams = permitSignature || {
@@ -664,7 +640,9 @@ export default class LendingPool
     const params = buildParaSwapLiquiditySwapParams(
       toAsset,
       amountSlippageConverted,
-      swapAll ? 4 + 2 * 32 : 0,
+      swapAll
+        ? augustusFromAmountOffsetFromCalldata(swapCallData as string)
+        : 0,
       swapCallData,
       augustus,
       permitParams.amount,