feat: method to update prices

Author: doomsower

src/sdk/accounts/CreditAccountsService.ts

@@ -1,10 +1,9 @@
 import type { Address, ContractFunctionArgs } from "viem";
-import { decodeFunctionData, encodeFunctionData } from "viem";
+import { encodeFunctionData } from "viem";
 
 import {
   iCreditAccountCompressorAbi,
   iCreditFacadeV3MulticallAbi,
-  iUpdatablePriceFeedAbi,
 } from "../abi";
 import { type CreditAccountData, SDKConstruct } from "../base";
 import {
@@ -14,12 +13,13 @@ import {
   MIN_INT96,
 } from "../constants";
 import type { GearboxSDK } from "../GearboxSDK";
-import type {
-  CreditFactory,
-  IPriceFeedContract,
-  OnDemandPriceUpdate,
-  PriceOracleContract,
-  UpdatePriceFeedsResult,
+import {
+  type CreditFactory,
+  type IPriceFeedContract,
+  type OnDemandPriceUpdate,
+  type PriceOracleContract,
+  rawTxToMulticallPriceUpdate,
+  type UpdatePriceFeedsResult,
 } from "../market";
 import type { MultiCall, RawTx } from "../types";
 import { simulateMulticall } from "../utils/viem";
@@ -434,23 +434,3 @@ export class CreditAccountsService extends SDKConstruct {
     return this.sdk.marketRegister.poolState.map(p => p.pool.address);
   }
 }
-
-/**
- * Helper method to convert our RawTx into viem's multicall format
- * Involves decoding what was previously encoded, but it's better than adding another method to PriceFeedFactory
- * @param tx
- * @returns
- */
-function rawTxToMulticallPriceUpdate(tx: RawTx) {
-  const { to, callData } = tx;
-  const { args, functionName } = decodeFunctionData({
-    abi: iUpdatablePriceFeedAbi,
-    data: callData,
-  });
-  return {
-    abi: iUpdatablePriceFeedAbi,
-    address: to,
-    functionName,
-    args,
-  };
-}

src/sdk/constants/address-provider.ts

@@ -11,6 +11,7 @@ export const AP_PRICE_ORACLE = "PRICE_ORACLE";
 export const AP_ACCOUNT_FACTORY = "ACCOUNT_FACTORY";
 export const AP_DATA_COMPRESSOR = "DATA_COMPRESSOR";
 export const AP_MARKET_COMPRESSOR = "MARKET_COMPRESSOR";
+export const AP_PRICE_FEED_COMPRESSOR = "PRICE_FEED_COMPRESSOR";
 export const AP_CREDIT_ACCOUNT_COMPRESSOR = "CREDIT_ACCOUNT_COMPRESSOR";
 export const AP_TREASURY = "TREASURY";
 export const AP_GEAR_TOKEN = "GEAR_TOKEN";

src/sdk/market/PriceOracleContract.ts

@@ -1,18 +1,28 @@
 import type { Address, Hex } from "viem";
 import { decodeFunctionData } from "viem";
 
-import { iUpdatablePriceFeedAbi, priceOracleV3Abi } from "../abi";
-import type { PriceFeedTreeNode, PriceOracleData } from "../base";
+import {
+  iPriceFeedCompressorAbi,
+  iUpdatablePriceFeedAbi,
+  priceOracleV3Abi,
+} from "../abi";
+import type {
+  PriceFeedMapEntry,
+  PriceFeedTreeNode,
+  PriceOracleData,
+} from "../base";
 import { BaseContract } from "../base";
+import { AP_PRICE_FEED_COMPRESSOR } from "../constants";
 import type { GearboxSDK } from "../GearboxSDK";
 import type { PriceOracleState } from "../state";
 import { AddressMap } from "../utils";
+import { simulateMulticall } from "../utils/viem";
 import type {
   IPriceFeedContract,
   PriceFeedUsageType,
   UpdatePriceFeedsResult,
 } from "./pricefeeds";
-import { PriceFeedRef } from "./pricefeeds";
+import { PriceFeedRef, rawTxToMulticallPriceUpdate } from "./pricefeeds";
 
 type abi = typeof priceOracleV3Abi;
 
@@ -104,8 +114,8 @@ export class PriceOracleContract extends BaseContract<abi> {
     const reserve = opts?.reserve ?? true;
     return tokens
       .flatMap(t => [
-        main ? this.mainPriceFeeds[t].priceFeed : undefined,
-        reserve ? this.reservePriceFeeds[t].priceFeed : undefined,
+        main ? this.mainPriceFeeds[t]?.priceFeed : undefined,
+        reserve ? this.reservePriceFeeds[t]?.priceFeed : undefined,
       ])
       .filter((f): f is IPriceFeedContract => !!f);
   }
@@ -205,83 +215,43 @@ export class PriceOracleContract extends BaseContract<abi> {
     return (amount * fromPrice * toScale) / (toPrice * fromScale);
   }
 
-  // async loadPrices(
-  //   priceUpdatesTxs: Array<RawTx>,
-  //   block: bigint,
-  // ): Promise<{
-  //   mainPrices: Record<Address, bigint>;
-  //   reservePrices: Record<Address, bigint>;
-  // }> {
-  //   const priceUpdateCalls: Array<MultiCallStruct> = priceUpdatesTxs.map(
-  //     tx => ({
-  //       target: tx.to,
-  //       callData: tx.callData,
-  //       allowFailure: false,
-  //     }),
-  //   );
-
-  //   const getPricesRawCalls = (reserve: boolean): Array<MultiCallStruct> => {
-  //     return Object.keys(
-  //       reserve ? this.reservePriceFeeds : this.mainPriceFeeds,
-  //     ).map(token => ({
-  //       target: this.address,
-  //       callData: encodeFunctionData({
-  //         functionName: "getPriceRaw",
-  //         args: [token as Address, reserve],
-  //         abi: this.abi,
-  //       }),
-  //       allowFailure: true,
-  //     }));
-  //   };
-
-  //   const { result } = await this.v3.publicClient.simulateContract({
-  //     address: MULTICALL_ADDRESS,
-  //     abi: multicall3Abi,
-  //     functionName: "aggregate3",
-  //     args: [
-  //       [
-  //         ...priceUpdateCalls,
-  //         ...getPricesRawCalls(false),
-  //         ...getPricesRawCalls(true),
-  //       ],
-  //     ],
-  //     chain: this.v3.publicClient.chain!,
-  //     account: this.v3.walletClient.account!,
-  //     gas: 550_000_000n,
-  //     // blockNumber: BigInt(block),
-  //   });
-
-  //   const returnRawPrices = (
-  //     result as Array<{ success: boolean; returnData: Hex }>
-  //   ).slice(priceUpdateCalls.length);
-
-  //   const prices = returnRawPrices.map(callReturn =>
-  //     callReturn.success
-  //       ? decodeFunctionResult({
-  //           functionName: "getPrice",
-  //           abi: this.abi,
-  //           data: callReturn.returnData! as Hex,
-  //         })
-  //       : 0n,
-  //   ) as Array<bigint>;
-
-  //   const mainPrices: Record<Address, bigint> = {};
-  //   const reservePrices: Record<Address, bigint> = {};
-
-  //   const mainPFlength = Object.keys(this.mainPriceFeeds).length;
-
-  //   prices.forEach((price, i) => {
-  //     if (i < mainPFlength) {
-  //       mainPrices[Object.keys(this.mainPriceFeeds)[i] as Address] = price;
-  //     } else {
-  //       reservePrices[
-  //         Object.keys(this.reservePriceFeeds)[i - mainPFlength] as Address
-  //       ] = price;
-  //     }
-  //   });
+  /**
+   * Loads new prices for this oracle from PriceFeedCompressor
+   * Does not update price feeds, only updates prices
+   */
+  public async updatePrices(): Promise<void> {
+    const { txs } = await this.updatePriceFeeds();
+    const resp = await simulateMulticall(this.provider.publicClient, {
+      contracts: [
+        ...txs.map(rawTxToMulticallPriceUpdate),
+        {
+          abi: iPriceFeedCompressorAbi,
+          address: this.sdk.addressProvider.getLatestVersion(
+            AP_PRICE_FEED_COMPRESSOR,
+          ),
+          functionName: "getPriceFeeds",
+          args: [this.address],
+        },
+      ],
+      allowFailure: false,
+      gas: 550_000_000n,
+      batchSize: 0, // we cannot have price updates and compressor request in different batches
+    });
+    const [entries, tree] = resp.pop() as [
+      PriceFeedMapEntry[],
+      PriceFeedTreeNode[],
+    ];
 
-  //   return { mainPrices, reservePrices };
-  // }
+    entries.forEach(({ token, priceFeed, reserve }) => {
+      const price = tree.find(n => n.baseParams.addr === priceFeed)?.answer
+        ?.price;
+      if (reserve && price) {
+        this.reservePrices.upsert(token, price);
+      } else if (price) {
+        this.mainPrices.upsert(token, price);
+      }
+    });
+  }
 
   #labelPriceFeed(
     address: Address,

src/sdk/market/pricefeeds/index.ts

@@ -13,6 +13,7 @@ export * from "./PriceFeedRef";
 export * from "./PriceFeedsRegister";
 export * from "./RedstonePriceFeed";
 export * from "./types";
+export * from "./utils";
 export * from "./WstETHPriceFeed";
 export * from "./YearnPriceFeed";
 export * from "./ZeroPriceFeed";

src/sdk/market/pricefeeds/utils.ts

@@ -0,0 +1,24 @@
+import { decodeFunctionData } from "viem";
+
+import { iUpdatablePriceFeedAbi } from "../../abi";
+import type { RawTx } from "../../types";
+
+/**
+ * Helper method to convert our RawTx into viem's multicall format
+ * Involves decoding what was previously encoded, but it's better than adding another method to PriceFeedFactory
+ * @param tx
+ * @returns
+ */
+export function rawTxToMulticallPriceUpdate(tx: RawTx) {
+  const { to, callData } = tx;
+  const { args, functionName } = decodeFunctionData({
+    abi: iUpdatablePriceFeedAbi,
+    data: callData,
+  });
+  return {
+    abi: iUpdatablePriceFeedAbi,
+    address: to,
+    functionName,
+    args,
+  };
+}