@strkfarm/sdk 2.0.0-dev.26 → 2.0.0-dev.28

package/src/strategies/universal-adapters/avnu-adapter.ts

@@ -3,6 +3,7 @@ import {
   DepositParams,
   WithdrawParams,
   BaseAdapterConfig,
+  SwapPriceInfo,
 } from "./baseAdapter";
 import { toBigInt } from "./adapter-utils";
 import { Protocols } from "@/interfaces";
@@ -19,9 +20,12 @@ import { Quote } from "@avnu/avnu-sdk";
 import { hash, uint256 } from "starknet";
 import { AvnuWrapper } from "@/modules/avnu";
 import axios from "axios";
-import {SIMPLE_SANITIZER } from "./adapter-utils";
+import { SIMPLE_SANITIZER } from "./adapter-utils";
 import { returnFormattedAmount } from "../vesu-extended-strategy/utils/helper";
-import { logger } from "@/utils";
+import { assert, logger } from "@/utils";
+import { Global } from "@/global";
+import { TokenInfo } from "@/interfaces";
+import { ERC20 } from "@/modules";
 export interface AvnuAdapterConfig extends BaseAdapterConfig {
   baseUrl: string;
   avnuContract: ContractAddr; //0x04270219d365d6b017231b52e92b3fb5d7c8378b05e9abc97724537a80e93b0f
@@ -33,23 +37,163 @@ export interface AvnuAdapterConfig extends BaseAdapterConfig {
 export class AvnuAdapter extends BaseAdapter<DepositParams, WithdrawParams> {
   readonly config: AvnuAdapterConfig;
   protected avnuWrapper: AvnuWrapper;
+  lastSwapPriceInfo: SwapPriceInfo | null = null;
+
+  private _depositApproveProofReadableId(): string {
+    return `approve_${this.config.supportedPositions[0].asset.symbol}`;
+  }
+
+  private _depositSwapProofReadableId(): string {
+    return `asutb_${this.config.supportedPositions[0].asset.symbol}_${this.config.supportedPositions[1].asset.symbol}`;
+  }
+
+  private _withdrawApproveProofReadableId(): string {
+    return `approve_${this.config.supportedPositions[1].asset.symbol}`;
+  }
+
+  private _withdrawSwapProofReadableId(): string {
+    const fromToken = this.config.supportedPositions[1].asset;
+    return `asbtu_${fromToken.symbol}_${fromToken.symbol}`;
+  }
+
+  private buildSwapLeafConfigs(
+    fromToken: TokenInfo,
+    toToken: TokenInfo,
+    swapId: string,
+  ): {
+    target: ContractAddr;
+    method: string;
+    packedArguments: bigint[];
+    sanitizer: ContractAddr;
+    id: string;
+  }[] {
+    const vaultAllocator = ContractAddr.from(this.config.vaultAllocator.address);
+    return [
+      {
+        target: fromToken.address,
+        method: "approve",
+        packedArguments: [AVNU_EXCHANGE.toBigInt()],
+        sanitizer: SIMPLE_SANITIZER,
+        id: `approve_${fromToken.symbol}`,
+      },
+      {
+        target: AVNU_EXCHANGE,
+        method: "multi_route_swap",
+        packedArguments: [
+          fromToken.address.toBigInt(),
+          toToken.address.toBigInt(),
+          vaultAllocator.toBigInt(),
+        ],
+        sanitizer: SIMPLE_SANITIZER,
+        id: swapId,
+      },
+    ];
+  }
+
+  private async buildSwapCalls(
+    params: DepositParams | WithdrawParams,
+    fromToken: TokenInfo,
+    toToken: TokenInfo,
+    usdcToBtc: boolean,
+  ): Promise<ManageCall[]> {
+    const vaultAllocator = ContractAddr.from(this.config.vaultAllocator.address);
+    const quote = await this.avnuWrapper.getQuotes(
+      fromToken.address.toString(),
+      toToken.address.toString(),
+      params.amount.toWei(),
+      vaultAllocator.address.toString(),
+    );
+
+    if (!quote) {
+      logger.error("No quotes available for this swap, error in quotes avnu");
+      return [];
+    }
+
+    const fromAmt = Web3Number.fromWei(quote.sellAmount.toString(), fromToken.decimals).toNumber();
+    const toAmt = Web3Number.fromWei(quote.buyAmount.toString(), toToken.decimals).toNumber();
+    this.lastSwapPriceInfo = {
+      source: 'avnu',
+      fromTokenSymbol: fromToken.symbol,
+      toTokenSymbol: toToken.symbol,
+      fromAmount: fromAmt,
+      toAmount: toAmt,
+      effectivePrice: toAmt !== 0 ? fromAmt / toAmt : 0,
+    };
+    logger.verbose(
+      `${AvnuAdapter.name}::buildSwapCalls stored price info: ` +
+        `${fromAmt} ${fromToken.symbol} → ${toAmt} ${toToken.symbol}, ` +
+        `effectivePrice=${this.lastSwapPriceInfo.effectivePrice}`,
+    );
+
+    const getCalldata = await this.avnuWrapper.getSwapCallData(
+      quote,
+      vaultAllocator.address,
+    );
+    const swapCallData = getCalldata[0];
+    const approveAmount = uint256.bnToUint256(params.amount.toWei());
+
+    return [
+      {
+        proofReadableId: usdcToBtc
+          ? this._depositApproveProofReadableId()
+          : this._withdrawApproveProofReadableId(),
+        sanitizer: SIMPLE_SANITIZER,
+        call: {
+          contractAddress: fromToken.address,
+          selector: hash.getSelectorFromName("approve"),
+          calldata: [
+            AVNU_EXCHANGE.toBigInt(),
+            toBigInt(approveAmount.low.toString()),
+            toBigInt(approveAmount.high.toString()),
+          ],
+        },
+      },
+      {
+        proofReadableId: usdcToBtc
+          ? this._depositSwapProofReadableId()
+          : this._withdrawSwapProofReadableId(),
+        sanitizer: SIMPLE_SANITIZER,
+        call: {
+          contractAddress: AVNU_EXCHANGE,
+          selector: hash.getSelectorFromName("multi_route_swap"),
+          calldata: swapCallData,
+        },
+      },
+    ];
+  }
 
   constructor(config: AvnuAdapterConfig) {
     super(config, AvnuAdapter.name, Protocols.AVNU);
     this.config = config as AvnuAdapterConfig;
     this.avnuWrapper = new AvnuWrapper();
+    assert(this.config.supportedPositions.length === 2, "AvnuAdapter must have 2 supported positions");
   }
   //abstract means the method has no implementation in this class; instead, child classes must implement it.
   protected async getAPY(
-    supportedPosition: SupportedPosition
+    supportedPosition: SupportedPosition,
   ): Promise<PositionAPY> {
     return Promise.resolve({ apy: 0, type: APYType.BASE });
   }
 
   protected async getPosition(
-    supportedPosition: SupportedPosition
-  ): Promise<PositionAmount> {
-    return Promise.resolve({ amount: new Web3Number(0, 0), remarks: "Avnu Positions" });
+    supportedPosition: SupportedPosition,
+  ): Promise<PositionAmount | null> {
+    const toToken = this.config.supportedPositions[1].asset;
+    if (supportedPosition.asset.symbol != toToken.symbol) {
+      return null;
+    }
+    try {
+      // only measure balance of toToken, bcz from token usually gets trakced in unused balance or a previous avnu adapter
+      const balance = await new ERC20(this.config.networkConfig).balanceOf(
+        supportedPosition.asset.address,
+        this.config.vaultAllocator.address,
+        supportedPosition.asset.decimals,
+      );
+      return { amount: balance, remarks: `Swapped Unused balance (VA)` };
+    } catch (_e) {
+      logger.error(`${AvnuAdapter.name}::getPosition: failed for ${toToken.symbol}`);
+      throw new Error(`${AvnuAdapter.name}: failed to get balance for ${toToken.symbol}`);
+    }
   }
 
   async maxDeposit(amount?: Web3Number): Promise<PositionInfo> {
@@ -82,31 +226,11 @@ export class AvnuAdapter extends BaseAdapter<DepositParams, WithdrawParams> {
     sanitizer: ContractAddr;
     id: string;
   }[] {
-    const vaultAllocator = ContractAddr.from(
-      this.config.vaultAllocator.address
+    return this.buildSwapLeafConfigs(
+      this.config.supportedPositions[0].asset,
+      this.config.supportedPositions[1].asset,
+      this._depositSwapProofReadableId(),
     );
-    return [
-      {
-        target: this.config.supportedPositions[0].asset.address,
-        method: "approve",
-        packedArguments: [
-          AVNU_EXCHANGE.toBigInt(),
-        ],
-        sanitizer: SIMPLE_SANITIZER,
-        id: `approve_${this.config.supportedPositions[0].asset.symbol}`,
-      },
-      {
-        target: AVNU_EXCHANGE,
-        method: "multi_route_swap",
-        packedArguments: [
-          this.config.supportedPositions[0].asset.address.toBigInt(), //usdc
-          this.config.supportedPositions[1].asset.address.toBigInt(), //wbtc
-          vaultAllocator.toBigInt(),
-        ],
-        sanitizer: SIMPLE_SANITIZER,
-        id: `asutb_${this.config.supportedPositions[0].asset.symbol}_${this.config.supportedPositions[1].asset.symbol}`,
-      },
-    ];
   }
 
   protected _getWithdrawLeaf(): {
@@ -116,33 +240,13 @@ export class AvnuAdapter extends BaseAdapter<DepositParams, WithdrawParams> {
     sanitizer: ContractAddr;
     id: string;
   }[] {
-    const vaultAllocator = ContractAddr.from(
-      this.config.vaultAllocator.address
-    );
     const toToken = this.config.supportedPositions[0].asset;
     const fromToken = this.config.supportedPositions[1].asset;
-    return [
-      {
-        target: fromToken.address,
-        method: "approve",
-        packedArguments: [
-          AVNU_EXCHANGE.toBigInt(),
-        ],
-        sanitizer: SIMPLE_SANITIZER,
-        id: `approve_${fromToken.symbol}`,
-      },
-      {
-        target: AVNU_EXCHANGE,
-        method: "multi_route_swap",
-        packedArguments: [
-          fromToken.address.toBigInt(), //wbtc
-          toToken.address.toBigInt(), //usdc
-          vaultAllocator.toBigInt(),
-        ],
-        sanitizer: SIMPLE_SANITIZER,
-        id: `asbtu_${fromToken.symbol}_${fromToken.symbol}`,
-      },
-    ];
+    return this.buildSwapLeafConfigs(
+      fromToken,
+      toToken,
+      this._withdrawSwapProofReadableId(),
+    );
   }
 
   protected _getLegacySwapLeaf(): {
@@ -159,54 +263,13 @@ export class AvnuAdapter extends BaseAdapter<DepositParams, WithdrawParams> {
     try {
       const fromToken = this.config.supportedPositions[0].asset; //usdc
       const toToken = this.config.supportedPositions[1].asset; //wbtc
-      const vaultAllocator = ContractAddr.from(
-        this.config.vaultAllocator.address
-      );
-      const quote = await this.getQuotesAvnu(
-        fromToken.address.toString(),
-        toToken.address.toString(),
-        params.amount.toNumber(),
-        vaultAllocator.address.toString(),
-        toToken.decimals,
-        true
-      )
-      //console.log(`${AvnuAdapter.name}::getDepositCall quote: ${JSON.stringify(quote)}`);
-      if (!quote) {
-        logger.error("error getting quote from avnu");
-        return [];
-      }
-
-      const getCalldata = await this.avnuWrapper.getSwapCallData(
-        quote,
-        vaultAllocator.address
+      const calls = await this.buildSwapCalls(
+        params,
+        fromToken,
+        toToken,
+        true,
       );
-      const dataObject = quote;
-      logger.info(`${AvnuAdapter.name}::getQuotesAvnu finalAmountOfWbtcOut : ${dataObject.avnuFees} ${parseInt(dataObject.buyAmount.toString(), 16)} ${parseInt(dataObject.sellAmount.toString(), 16)} ${parseInt(dataObject.sellAmount.toString(), 16)}`);
-      const swapCallData = getCalldata[0];
-      // const approveCallData = getCalldata[0];
-      const amount = uint256.bnToUint256(quote.sellAmountInUsd*10**7)
-      return [
-        {
-          sanitizer: SIMPLE_SANITIZER,
-          call: {
-            contractAddress: fromToken.address,
-            selector: hash.getSelectorFromName("approve"),
-            calldata:[
-              AVNU_EXCHANGE.toBigInt(),
-              toBigInt(amount.low.toString()), // amount low
-              toBigInt(amount.high.toString()), // amount high
-            ] ,
-          },
-        },
-        {
-          sanitizer: SIMPLE_SANITIZER,
-          call: {
-            contractAddress: AVNU_EXCHANGE,
-            selector: hash.getSelectorFromName("multi_route_swap"),
-            calldata: swapCallData,
-          },
-        },
-      ];
+      return calls;
     } catch (error) {
       logger.error(`Error getting Avnu quote: ${error}`);
       return [];
@@ -218,49 +281,13 @@ export class AvnuAdapter extends BaseAdapter<DepositParams, WithdrawParams> {
     try {
       const toToken = this.config.supportedPositions[0].asset;
       const fromToken = this.config.supportedPositions[1].asset;
-      const vaultAllocator = ContractAddr.from(
-        this.config.vaultAllocator.address
-      );
-      const quote = await this.getQuotesAvnu(
-        fromToken.address.toString(),
-        toToken.address.toString(),
-        params.amount.toNumber(),
-        vaultAllocator.address.toString(),
-        fromToken.decimals,
-        false
-      );
-      if (!quote) {
-        logger.error("No quotes available for this swap, error in quotes avnu");
-        return [];
-      }
-      const getCalldata = await this.avnuWrapper.getSwapCallData(
-        quote,
-        vaultAllocator.address
+      const calls = await this.buildSwapCalls(
+        params,
+        fromToken,
+        toToken,
+        false,
       );
-      const swapCallData = getCalldata[0];
-      const amount = uint256.bnToUint256(params.amount.toWei())
-      return [
-        {
-          sanitizer: SIMPLE_SANITIZER,
-          call: {
-            contractAddress:fromToken.address,
-            selector: hash.getSelectorFromName("approve"),
-            calldata: [
-              AVNU_EXCHANGE.toBigInt(),
-              toBigInt(amount.low.toString()), // amount low
-              toBigInt(amount.high.toString()), // amount high
-            ],
-          },
-        },
-        {
-          sanitizer: SIMPLE_SANITIZER,
-          call: {
-            contractAddress: AVNU_EXCHANGE,
-            selector: hash.getSelectorFromName("multi_route_swap"),
-            calldata: swapCallData,
-          },
-        },
-      ];
+      return calls;
     } catch (error) {
       logger.error(`Error getting Avnu quote: ${error}`);
       return [];
@@ -268,7 +295,10 @@ export class AvnuAdapter extends BaseAdapter<DepositParams, WithdrawParams> {
   }
 
   async getSwapCallData(quote: Quote): Promise<bigint[][]> {
-    return await this.avnuWrapper.getSwapCallData(quote, this.config.vaultAllocator.address);
+    return await this.avnuWrapper.getSwapCallData(
+      quote,
+      this.config.vaultAllocator.address,
+    );
   }
 
   async getHealthFactor(): Promise<number> {
@@ -277,7 +307,7 @@ export class AvnuAdapter extends BaseAdapter<DepositParams, WithdrawParams> {
 
   async fetchQuoteWithRetry(
     params: Record<string, any>,
-    retries: number = 5
+    retries: number = 5,
   ): Promise<any> {
     for (let attempt = 0; attempt < retries; attempt++) {
       try {
@@ -296,125 +326,4 @@ export class AvnuAdapter extends BaseAdapter<DepositParams, WithdrawParams> {
     }
     throw new Error("Failed to fetch quote after retries");
   }
-
-  async getQuotesAvnu (
-    from_token_address: string,
-    to_token_address: string,
-    amount: number, //amount in btc units
-    takerAddress: string,
-    toTokenDecimals: number,
-    usdcToBtc: boolean,
-    maxIterations: number = 5,
-    tolerance: number = 5000
-  ): Promise<Quote | null>{
-    try {
-      const fromToken = this.config.supportedPositions[0].asset; //usdc
-      const toToken = this.config.supportedPositions[1].asset; //wbtc
-      if(!usdcToBtc) {
-        const sellAmount = returnFormattedAmount(amount, toTokenDecimals);
-        const params: Record<string, any> = {
-          sellTokenAddress: from_token_address,
-          buyTokenAddress: to_token_address,
-          takerAddress,
-          sellAmount:sellAmount,
-        };
-        const finalQuote = await this.fetchQuoteWithRetry(params);
-        if (!finalQuote.data.length) {
-          logger.error("No quotes available for this swap, error in quotes avnu");
-          return null;
-        }
-        const dataObject: Quote = finalQuote.data[0];
-        return dataObject;
-      }
-      const btcPrice = await this.getPriceOfToken(toToken.address.toString());
-      if (!btcPrice) {
-        logger.error(`error getting btc price: ${btcPrice}`);
-        return null;
-      }
-      const estimatedUsdcAmount = Math.floor(amount * btcPrice);
-      logger.info(`${AvnuAdapter.name}::getQuotesAvnu estimatedUsdcAmount: ${estimatedUsdcAmount}`);
-      const targetBtcBig = BigInt(returnFormattedAmount(amount, toTokenDecimals));
-      logger.info(`${AvnuAdapter.name}::getQuotesAvnu targetBtcBig: ${targetBtcBig}`);
-      let low = BigInt(
-        Math.floor(
-          (estimatedUsdcAmount * 10 ** fromToken.decimals) * 0.9
-        )
-      );
-      let high = BigInt(
-        Math.floor(
-          (estimatedUsdcAmount * 10 ** fromToken.decimals) * 1.1
-        )
-      );
-      let mid = 0n;
-      for (let i = 0; i < maxIterations; i++) {
-        mid = (low + high) / 2n;
-        const sellAmount = returnFormattedAmount(Number(mid), 0);
-        const quote = await this.fetchQuoteWithRetry({
-          sellTokenAddress: from_token_address,
-          buyTokenAddress: to_token_address,
-          takerAddress,
-          sellAmount,
-        });
-  
-        const gotBtc = BigInt(quote.data[0].buyAmount);
-        if (gotBtc === targetBtcBig) return quote.data[0];
-  
-        if (gotBtc > targetBtcBig) {
-          high = mid;
-        } else {
-          low = mid;
-        }
-  
-        if (
-          gotBtc >= targetBtcBig &&
-          gotBtc <= targetBtcBig + BigInt(tolerance)
-        ) {
-          return quote.data[0];
-        }
-      }
-      let sellAmount = returnFormattedAmount(
-        Number(mid),
-        0
-      );
-      const params: Record<string, any> = {
-        sellTokenAddress: from_token_address,
-        buyTokenAddress: to_token_address,
-        takerAddress,
-        sellAmount: sellAmount,
-      };
-      const finalQuote = await this.fetchQuoteWithRetry(params);
-      if (!finalQuote.data.length) {
-        logger.error("No quotes available for this swap, error in quotes avnu");
-        return null;
-      }
-      const dataObject: Quote = finalQuote.data[0];
-      const wbtcAmountOut = parseInt(dataObject.buyAmount.toString(), 16);
-      logger.info(`${AvnuAdapter.name}::getQuotesAvnu finalAmountOfWbtcOut : ${wbtcAmountOut} ${dataObject.buyAmount} ${dataObject.sellAmount.toString()} ${dataObject.sellAmount.toString()}`);
-      return dataObject;
-    } catch (err) {
-      logger.error(`No quotes available for this swap: ${err}`);
-      return null;
-    }
-  };
-
-  async getPriceOfToken (
-    tokenAddress: string,
-    retries: number = MAX_RETRIES
-  ): Promise<number | null> {
-    try {
-      const url = `https://starknet.impulse.avnu.fi/v1/tokens/${tokenAddress}/prices/line`;
-      const response = await axios.get(url);
-      const length = response.data.length;
-      return response.data[length - 1].value;
-    } catch (err) {
-      if (retries > 0) {
-        await new Promise((resolve) => setTimeout(resolve, MAX_DELAY));
-        return this.getPriceOfToken(tokenAddress, retries - 1);
-      } else {
-        logger.error(`Failed to fetch price for ${tokenAddress} after ${MAX_RETRIES} attempts`);
-        return null;
-      }
-    }
-  };
-
-}
+}