@gearbox-protocol/sdk 14.12.0-next.9 → 14.12.0-test.1

package/dist/esm/plugins/adapters/abi/targetContractAbi.js

@@ -3042,99 +3042,6 @@ const iInfinifiGatewayAbi = [
     stateMutability: "nonpayable"
   }
 ];
-const iInfinifiUnwindingGatewayAbi = [
-  {
-    type: "function",
-    name: "contractType",
-    inputs: [],
-    outputs: [{ name: "", type: "bytes32", internalType: "bytes32" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "getPendingAssets",
-    inputs: [{ name: "user", type: "address", internalType: "address" }],
-    outputs: [{ name: "", type: "uint256", internalType: "uint256" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "getUserUnwindingData",
-    inputs: [{ name: "user", type: "address", internalType: "address" }],
-    outputs: [
-      {
-        name: "",
-        type: "tuple",
-        internalType: "struct UserUnwindingData",
-        components: [
-          { name: "shares", type: "uint256", internalType: "uint256" },
-          { name: "unwindingEpochs", type: "uint32", internalType: "uint32" },
-          {
-            name: "unwindingTimestamp",
-            type: "uint256",
-            internalType: "uint256"
-          },
-          { name: "unclaimedAssets", type: "uint256", internalType: "uint256" },
-          { name: "isWithdrawn", type: "bool", internalType: "bool" }
-        ]
-      }
-    ],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "iUSD",
-    inputs: [],
-    outputs: [{ name: "", type: "address", internalType: "address" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "lockingController",
-    inputs: [],
-    outputs: [{ name: "", type: "address", internalType: "address" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "startUnwinding",
-    inputs: [
-      { name: "shares", type: "uint256", internalType: "uint256" },
-      { name: "unwindingEpochs", type: "uint32", internalType: "uint32" }
-    ],
-    outputs: [],
-    stateMutability: "nonpayable"
-  },
-  {
-    type: "function",
-    name: "unwindingModule",
-    inputs: [],
-    outputs: [{ name: "", type: "address", internalType: "address" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "version",
-    inputs: [],
-    outputs: [{ name: "", type: "uint256", internalType: "uint256" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "withdraw",
-    inputs: [
-      { name: "unwindingTimestamp", type: "uint256", internalType: "uint256" }
-    ],
-    outputs: [],
-    stateMutability: "nonpayable"
-  },
-  { type: "error", name: "InsufficientPendingAssetsException", inputs: [] },
-  { type: "error", name: "InsufficientSharesException", inputs: [] },
-  { type: "error", name: "MoreThanOneUnwindingPerBlockException", inputs: [] },
-  { type: "error", name: "UnwindingNotClaimableException", inputs: [] },
-  { type: "error", name: "UserAlreadyUnwindingException", inputs: [] },
-  { type: "error", name: "UserNotUnwindingException", inputs: [] }
-];
 const iKelpLRTDepositPoolGatewayAbi = [
   {
     type: "function",
@@ -3368,7 +3275,7 @@ const iMellowWrapperAbi = [
     stateMutability: "nonpayable"
   }
 ];
-const iMidasIssuanceVaultAbi = [
+const iMidasIssuanceVaultV310Abi = [
   {
     type: "function",
     name: "depositInstant",
@@ -3389,106 +3296,6 @@ const iMidasIssuanceVaultAbi = [
     stateMutability: "view"
   }
 ];
-const iMidasRedemptionVaultAdapterAbi = [
-  {
-    type: "function",
-    name: "contractType",
-    inputs: [],
-    outputs: [{ name: "", type: "bytes32", internalType: "bytes32" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "getCurrentRequestTokenOut",
-    inputs: [{ name: "account", type: "address", internalType: "address" }],
-    outputs: [{ name: "", type: "address", internalType: "address" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "mToken",
-    inputs: [],
-    outputs: [{ name: "", type: "address", internalType: "address" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "midasRedemptionVault",
-    inputs: [],
-    outputs: [{ name: "", type: "address", internalType: "address" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "pendingRedemptions",
-    inputs: [{ name: "user", type: "address", internalType: "address" }],
-    outputs: [
-      { name: "isActive", type: "bool", internalType: "bool" },
-      { name: "isManuallyCleared", type: "bool", internalType: "bool" },
-      { name: "requestId", type: "uint256", internalType: "uint256" },
-      { name: "timestamp", type: "uint256", internalType: "uint256" },
-      { name: "remainder", type: "uint256", internalType: "uint256" }
-    ],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "pendingTokenOutAmount",
-    inputs: [
-      { name: "user", type: "address", internalType: "address" },
-      { name: "tokenOut", type: "address", internalType: "address" }
-    ],
-    outputs: [{ name: "", type: "uint256", internalType: "uint256" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "redeemInstant",
-    inputs: [
-      { name: "tokenOut", type: "address", internalType: "address" },
-      { name: "amountMTokenIn", type: "uint256", internalType: "uint256" },
-      { name: "minReceiveAmount", type: "uint256", internalType: "uint256" }
-    ],
-    outputs: [],
-    stateMutability: "nonpayable"
-  },
-  {
-    type: "function",
-    name: "requestRedeem",
-    inputs: [
-      { name: "tokenOut", type: "address", internalType: "address" },
-      { name: "amountMTokenIn", type: "uint256", internalType: "uint256" }
-    ],
-    outputs: [],
-    stateMutability: "nonpayable"
-  },
-  {
-    type: "function",
-    name: "version",
-    inputs: [],
-    outputs: [{ name: "", type: "uint256", internalType: "uint256" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "withdraw",
-    inputs: [{ name: "amount", type: "uint256", internalType: "uint256" }],
-    outputs: [],
-    stateMutability: "nonpayable"
-  },
-  { type: "error", name: "AmountExceedsAvailableException", inputs: [] },
-  { type: "error", name: "AmountIsLessThanRequiredException", inputs: [] },
-  { type: "error", name: "HasPendingRedemptionException", inputs: [] },
-  { type: "error", name: "InvalidRequestException", inputs: [] },
-  { type: "error", name: "NoPendingRedemptionException", inputs: [] },
-  { type: "error", name: "RedemptionNotFulfilledException", inputs: [] },
-  {
-    type: "error",
-    name: "RequestNotCancelledOrManuallyClearedException",
-    inputs: []
-  },
-  { type: "error", name: "ZeroAmountException", inputs: [] }
-];
 const iPendleRouterAbi = [
   {
     type: "function",
@@ -4747,98 +4554,6 @@ const iUniswapV3Abi = [
     stateMutability: "payable"
   }
 ];
-const iUniswapV4GatewayAbi = [
-  {
-    type: "function",
-    name: "permit2",
-    inputs: [],
-    outputs: [{ name: "", type: "address", internalType: "address" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "swapExactInputSingle",
-    inputs: [
-      {
-        name: "poolKey",
-        type: "tuple",
-        internalType: "struct PoolKey",
-        components: [
-          { name: "token0", type: "address", internalType: "address" },
-          { name: "token1", type: "address", internalType: "address" },
-          { name: "fee", type: "uint24", internalType: "uint24" },
-          { name: "tickSpacing", type: "int24", internalType: "int24" },
-          { name: "hooks", type: "address", internalType: "address" }
-        ]
-      },
-      { name: "zeroForOne", type: "bool", internalType: "bool" },
-      { name: "amountIn", type: "uint128", internalType: "uint128" },
-      { name: "amountOutMinimum", type: "uint128", internalType: "uint128" },
-      { name: "hookData", type: "bytes", internalType: "bytes" }
-    ],
-    outputs: [{ name: "amount", type: "uint256", internalType: "uint256" }],
-    stateMutability: "nonpayable"
-  },
-  {
-    type: "function",
-    name: "universalRouter",
-    inputs: [],
-    outputs: [{ name: "", type: "address", internalType: "address" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "weth",
-    inputs: [],
-    outputs: [{ name: "", type: "address", internalType: "address" }],
-    stateMutability: "view"
-  },
-  { type: "error", name: "UnexpectedETHTransferException", inputs: [] }
-];
-const iUpshiftVaultGatewayAbi = [
-  {
-    type: "function",
-    name: "claim",
-    inputs: [{ name: "amount", type: "uint256", internalType: "uint256" }],
-    outputs: [],
-    stateMutability: "nonpayable"
-  },
-  {
-    type: "function",
-    name: "contractType",
-    inputs: [],
-    outputs: [{ name: "", type: "bytes32", internalType: "bytes32" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "pendingAssetsOf",
-    inputs: [{ name: "account", type: "address", internalType: "address" }],
-    outputs: [{ name: "", type: "uint256", internalType: "uint256" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "requestRedeem",
-    inputs: [{ name: "shares", type: "uint256", internalType: "uint256" }],
-    outputs: [],
-    stateMutability: "nonpayable"
-  },
-  {
-    type: "function",
-    name: "upshiftVault",
-    inputs: [],
-    outputs: [{ name: "", type: "address", internalType: "address" }],
-    stateMutability: "view"
-  },
-  {
-    type: "function",
-    name: "version",
-    inputs: [],
-    outputs: [{ name: "", type: "uint256", internalType: "uint256" }],
-    stateMutability: "view"
-  }
-];
 const iVelodromeV2RouterAbi = [
   {
     type: "function",
@@ -5093,20 +4808,16 @@ export {
   iERC4626ReferralAbi,
   iFluidDexAbi,
   iInfinifiGatewayAbi,
-  iInfinifiUnwindingGatewayAbi,
   iKelpLRTDepositPoolGatewayAbi,
   iKelpLRTWithdrawalManagerGatewayAbi,
   iMellowClaimerAbi,
   iMellowWrapperAbi,
-  iMidasIssuanceVaultAbi,
-  iMidasRedemptionVaultAdapterAbi,
+  iMidasIssuanceVaultV310Abi,
   iPendleRouterAbi,
   iStakingRewardsAbi,
   iTraderJoeRouterAbi,
   iUniswapV2Router02Abi,
   iUniswapV3Abi,
-  iUniswapV4GatewayAbi,
-  iUpshiftVaultGatewayAbi,
   iVelodromeV2RouterAbi,
   iwstETHAbi,
   lidoV1_WETHGatewayAbi

package/dist/esm/plugins/adapters/contracts/AbstractAdapter.js

@@ -16,10 +16,12 @@ import { swapFromTransfers } from "../transferHelpers.js";
 class AbstractAdapterContract extends BaseContract {
   #targetContract;
   #creditManager;
+  #sdk;
   protocolAbi;
-  constructor(options, args) {
+  constructor(sdk, args) {
     const { baseParams, protocolAbi, ...rest } = args;
-    super(options, { ...rest, ...baseParams });
+    super(sdk, { ...rest, ...baseParams });
+    this.#sdk = sdk;
     this.protocolAbi = protocolAbi;
     if (baseParams.serializedParams) {
       const [cm, tc] = decodeAbiParameters(
@@ -30,6 +32,12 @@ class AbstractAdapterContract extends BaseContract {
       this.#targetContract = tc;
     }
   }
+  /**
+   * The SDK instance this adapter belongs to.
+   */
+  get sdk() {
+    return this.#sdk;
+  }
   get targetContract() {
     if (this.#targetContract === void 0) {
       throw new MissingSerializedParamsError("targetContract");
@@ -104,15 +112,17 @@ class AbstractAdapterContract extends BaseContract {
     }
   }
   /**
-   * Applies the balance changes implied by a router-generated diff-style
-   * adapter call to a map of credit-account token balances.
+   * Applies the balance changes implied by an adapter call to a map of
+   * credit-account token balances. Handles both router-generated diff-style
+   * calls and exact-input calls (e.g. those emitted by the withdrawal
+   * compressor).
    *
    * @param balances - Token balances before the call. Mutated in place.
    * @param calldata - Raw ABI-encoded adapter calldata.
    * @throws When the calldata cannot be decoded or the adapter (or the
    * specific function) has no balance-changes support.
    */
-  previewBalanceChanges(balances, calldata) {
+  async previewBalanceChanges(balances, calldata) {
     let decoded;
     try {
       decoded = decodeFunctionData({ abi: this.abi, data: calldata });
@@ -122,23 +132,67 @@ class AbstractAdapterContract extends BaseContract {
         { cause: e }
       );
     }
-    const leftovers = this.decodeDiffLeftovers(decoded, balances);
-    for (const { tokenIn, leftoverAmount } of leftovers) {
-      balances.upsert(tokenIn, leftoverAmount);
-    }
+    await this.applyBalanceChanges(balances, decoded);
   }
   /**
-   * Resolves which tokens the diff-style call spends down to which leftovers.
+   * Applies the balance changes of a decoded adapter call to the running
+   * balances, mutating them in place. Overrides should express changes via
+   * {@link setLeftover} (diff-style calls) and {@link spendExact}
+   * (exact-input calls), and should resolve all tokens/amounts before the
+   * first write so a throw never leaves balances partially mutated.
    *
+   * @param _balances - Token balances before the call. Mutated in place.
    * @param decoded - The viem-decoded adapter call
-   * @param _balances - Token balances before the call
    * @throws When the adapter (or the specific function) has no balance-changes support
    */
-  decodeDiffLeftovers(decoded, _balances) {
+  async applyBalanceChanges(_balances, decoded) {
     throw new Error(
       `previewBalanceChanges is not supported for ${decoded.functionName} on ${this.contractType} adapter at ${this.address}`
     );
   }
+  /**
+   * When the given adapter calldata is a delayed-withdrawal request (a call
+   * created by the withdrawal compressor that mints a withdrawal phantom
+   * token), returns its descriptor; `undefined` for any other call.
+   *
+   * The base implementation returns `undefined`; delayed-withdrawal adapters
+   * (Securitize, Midas, Mellow) override it next to their
+   * {@link applyBalanceChanges} cases for the same methods.
+   */
+  parseDelayedWithdrawalRequest(_calldata) {
+    return void 0;
+  }
+  /**
+   * When the given adapter calldata is a delayed-withdrawal claim (a call
+   * created by the withdrawal compressor that burns a withdrawal phantom
+   * token and receives the claim token from a redeemer), returns its
+   * descriptor; `undefined` for any other call.
+   *
+   * The base implementation returns `undefined`; adapters whose gateways log
+   * redemption intents (Securitize, Midas) override it.
+   */
+  parseDelayedWithdrawalClaim(_calldata) {
+    return void 0;
+  }
+  /**
+   * Diff-call semantics of a diff-style adapter call: the call spends the
+   * consumed token down to the exact `leftoverAmount` encoded in its
+   * calldata, so the post-call balance of `tokenIn` is `leftoverAmount`
+   * regardless of the actual amount spent.
+   */
+  setLeftover(balances, tokenIn, leftoverAmount) {
+    balances.upsert(tokenIn, leftoverAmount);
+  }
+  /**
+   * Exact-input call semantics: spends exactly `amount` of `tokenIn` from
+   * the running balance, clamping at zero. Used for non-diff adapter calls
+   * (e.g. those generated by the withdrawal compressor) whose spent amount
+   * is present in calldata.
+   */
+  spendExact(balances, tokenIn, amount) {
+    const balance = balances.getOrZero(tokenIn);
+    balances.upsert(tokenIn, balance > amount ? balance - amount : 0n);
+  }
   /**
    * Classifies an adapter call into a {@link LegacyAdapterOperation} using the
    * parsed call and credit-account transfer deltas.

package/dist/esm/plugins/adapters/contracts/AbstractCurveAdapter.js

@@ -5,23 +5,26 @@ class AbstractCurveAdapterContract extends AbstractAdapterContract {
   classifyLegacyOperation(parsed, transfers) {
     return classifyCurveOperation(parsed.functionName, transfers) ?? super.classifyLegacyOperation(parsed, transfers);
   }
-  decodeDiffLeftovers(decoded, balances) {
+  async applyBalanceChanges(balances, decoded) {
     const { functionName, args } = decoded;
     switch (functionName) {
       case "exchange_diff": {
         const [i, , leftoverAmount] = args;
-        return [{ tokenIn: this.curveCoin(i), leftoverAmount }];
+        this.setLeftover(balances, this.curveCoin(i), leftoverAmount);
+        break;
       }
       case "add_diff_liquidity_one_coin": {
         const [leftoverAmount, i] = args;
-        return [{ tokenIn: this.curveCoin(i), leftoverAmount }];
+        this.setLeftover(balances, this.curveCoin(i), leftoverAmount);
+        break;
       }
       case "remove_diff_liquidity_one_coin": {
         const [leftoverAmount] = args;
-        return [{ tokenIn: this.lpToken, leftoverAmount }];
+        this.setLeftover(balances, this.lpToken, leftoverAmount);
+        break;
       }
       default:
-        return super.decodeDiffLeftovers(decoded, balances);
+        await super.applyBalanceChanges(balances, decoded);
     }
   }
   /**

package/dist/esm/plugins/adapters/contracts/AccountMigratorAdapterContract.js

@@ -3,8 +3,8 @@ import { AbstractAdapterContract } from "./AbstractAdapter.js";
 const abi = accountMigratorAbi;
 const protocolAbi = accountMigratorAbi;
 class AccountMigratorAdapterContract extends AbstractAdapterContract {
-  constructor(options, args) {
-    super(options, { ...args, abi, protocolAbi });
+  constructor(sdk, args) {
+    super(sdk, { ...args, abi, protocolAbi });
   }
   /** Legacy adapter not present in integrations-v3. */
   classifyLegacyOperation(_parsed, _transfers) {

package/dist/esm/plugins/adapters/contracts/BalancerV3RouterAdapterContract.js

@@ -1,8 +1,8 @@
-import { iBalancerV3RouterAdapterAbi } from "@gearbox-protocol/integrations-v3";
 import {
   decodeAbiParameters
 } from "viem";
 import { MissingSerializedParamsError } from "../../../sdk/index.js";
+import { iBalancerV3RouterAdapterAbi } from "../abi/adapters/index.js";
 import { iBalancerV3RouterAbi } from "../abi/targetContractAbi.js";
 import { AbstractAdapterContract } from "./AbstractAdapter.js";
 var BalancerV3PoolStatus = /* @__PURE__ */ ((BalancerV3PoolStatus2) => {
@@ -17,8 +17,8 @@ const abi = iBalancerV3RouterAdapterAbi;
 const protocolAbi = iBalancerV3RouterAbi;
 class BalancerV3RouterAdapterContract extends AbstractAdapterContract {
   #allowedPools;
-  constructor(options, args) {
-    super(options, { ...args, abi, protocolAbi });
+  constructor(sdk, args) {
+    super(sdk, { ...args, abi, protocolAbi });
     if (args.baseParams.serializedParams) {
       const version = Number(args.baseParams.version);
       if (version <= 310) {
@@ -71,23 +71,25 @@ class BalancerV3RouterAdapterContract extends AbstractAdapterContract {
       }))
     };
   }
-  decodeDiffLeftovers(decoded, balances) {
+  async applyBalanceChanges(balances, decoded) {
     switch (decoded.functionName) {
       case "swapSingleTokenDiffIn": {
         const [, tokenIn, , leftoverAmount] = decoded.args;
-        return [{ tokenIn, leftoverAmount }];
+        this.setLeftover(balances, tokenIn, leftoverAmount);
+        break;
       }
       // BPT (pool token) is spent down to the leftover
       case "removeLiquiditySingleTokenDiff": {
         const [pool, leftoverAmount] = decoded.args;
-        return [{ tokenIn: pool, leftoverAmount }];
+        this.setLeftover(balances, pool, leftoverAmount);
+        break;
       }
       case "addLiquidityUnbalancedDiff":
         throw new Error(
           `previewBalanceChanges cannot resolve pool tokens for addLiquidityUnbalancedDiff on ${this.contractType} adapter at ${this.address}`
         );
       default:
-        return super.decodeDiffLeftovers(decoded, balances);
+        await super.applyBalanceChanges(balances, decoded);
     }
   }
 }

package/dist/esm/plugins/adapters/contracts/BalancerV3WrapperAdapterContract.js

@@ -1,16 +1,16 @@
-import { iBalancerV3WrapperAdapterAbi } from "@gearbox-protocol/integrations-v3";
 import {
   decodeAbiParameters
 } from "viem";
 import { MissingSerializedParamsError } from "../../../sdk/index.js";
+import { iBalancerV3WrapperAdapterAbi } from "../abi/adapters/index.js";
 import { iBalancerV3WrapperAbi } from "../abi/targetContractAbi.js";
 import { AbstractAdapterContract } from "./AbstractAdapter.js";
 const abi = iBalancerV3WrapperAdapterAbi;
 const protocolAbi = iBalancerV3WrapperAbi;
 class BalancerV3WrapperAdapterContract extends AbstractAdapterContract {
   #balancerPoolToken;
-  constructor(options, args) {
-    super(options, { ...args, abi, protocolAbi });
+  constructor(sdk, args) {
+    super(sdk, { ...args, abi, protocolAbi });
     if (args.baseParams.serializedParams) {
       const decoded = decodeAbiParameters(
         [
@@ -34,19 +34,21 @@ class BalancerV3WrapperAdapterContract extends AbstractAdapterContract {
       balancerPoolToken: this.#balancerPoolToken ? this.labelAddress(this.#balancerPoolToken) : void 0
     };
   }
-  decodeDiffLeftovers(decoded, balances) {
+  async applyBalanceChanges(balances, decoded) {
     switch (decoded.functionName) {
       // mint spends the pool token, burn spends the wrapper (target contract)
       case "mintDiff": {
         const [leftoverAmount] = decoded.args;
-        return [{ tokenIn: this.balancerPoolToken, leftoverAmount }];
+        this.setLeftover(balances, this.balancerPoolToken, leftoverAmount);
+        break;
       }
       case "burnDiff": {
         const [leftoverAmount] = decoded.args;
-        return [{ tokenIn: this.targetContract, leftoverAmount }];
+        this.setLeftover(balances, this.targetContract, leftoverAmount);
+        break;
       }
       default:
-        return super.decodeDiffLeftovers(decoded, balances);
+        await super.applyBalanceChanges(balances, decoded);
     }
   }
 }

package/dist/esm/plugins/adapters/contracts/CamelotV3AdapterContract.js

@@ -1,16 +1,16 @@
-import { iCamelotV3AdapterAbi } from "@gearbox-protocol/integrations-v3";
 import {
   decodeAbiParameters
 } from "viem";
 import { MissingSerializedParamsError } from "../../../sdk/index.js";
+import { iCamelotV3AdapterAbi } from "../abi/adapters/index.js";
 import { iCamelotV3RouterAbi } from "../abi/targetContractAbi.js";
 import { AbstractAdapterContract } from "./AbstractAdapter.js";
 const abi = iCamelotV3AdapterAbi;
 const protocolAbi = iCamelotV3RouterAbi;
 class CamelotV3AdapterContract extends AbstractAdapterContract {
   #supportedPools;
-  constructor(options, args) {
-    super(options, { ...args, abi, protocolAbi });
+  constructor(sdk, args) {
+    super(sdk, { ...args, abi, protocolAbi });
     if (args.baseParams.serializedParams) {
       const decoded = decodeAbiParameters(
         [
@@ -47,24 +47,21 @@ class CamelotV3AdapterContract extends AbstractAdapterContract {
       }))
     };
   }
-  decodeDiffLeftovers(decoded, balances) {
+  async applyBalanceChanges(balances, decoded) {
     switch (decoded.functionName) {
       case "exactDiffInputSingle": {
         const [params] = decoded.args;
-        return [
-          {
-            tokenIn: params.tokenIn,
-            leftoverAmount: params.leftoverAmount
-          }
-        ];
+        this.setLeftover(balances, params.tokenIn, params.leftoverAmount);
+        break;
       }
       case "exactDiffInput": {
         const [params] = decoded.args;
         const tokenIn = `0x${params.path.replace("0x", "").slice(0, 40)}`;
-        return [{ tokenIn, leftoverAmount: params.leftoverAmount }];
+        this.setLeftover(balances, tokenIn, params.leftoverAmount);
+        break;
       }
       default:
-        return super.decodeDiffLeftovers(decoded, balances);
+        await super.applyBalanceChanges(balances, decoded);
     }
   }
 }