fix: add find by price oracle method

Author: doomsower

src/sdk/market/MarketRegister.ts

@@ -17,7 +17,6 @@ import { MarketConfiguratorContract } from "./MarketConfiguratorContract";
 import { MarketFactory } from "./MarketFactory";
 import type { PoolFactory } from "./PoolFactory";
 import { rawTxToMulticallPriceUpdate } from "./pricefeeds";
-import type { PriceOracleContract } from "./PriceOracleContract";
 
 export class MarketRegister extends SDKConstruct {
   #logger?: ILogger;
@@ -151,15 +150,6 @@ export class MarketRegister extends SDKConstruct {
     throw new Error(`cannot find credit manager ${creditManager}`);
   }
 
-  public findPriceOracle(address: Address): PriceOracleContract {
-    for (const market of this.markets) {
-      if (market.priceOracle.address.toLowerCase() === address.toLowerCase()) {
-        return market.priceOracle;
-      }
-    }
-    throw new Error(`cannot find price oracle ${address}`);
-  }
-
   public findByCreditManager(creditManager: Address): MarketFactory {
     const market = this.markets.find(m =>
       m.creditManagers.some(
@@ -174,6 +164,15 @@ export class MarketRegister extends SDKConstruct {
     return market;
   }
 
+  public findByPriceOracle(address: Address): MarketFactory {
+    for (const market of this.markets) {
+      if (market.priceOracle.address.toLowerCase() === address.toLowerCase()) {
+        return market;
+      }
+    }
+    throw new Error(`cannot find price oracle ${address}`);
+  }
+
   public get markets(): MarketFactory[] {
     return this.#markets.values();
   }

src/sdk/market/PriceOracleContract.ts

@@ -325,12 +325,12 @@ export class PriceOracleContract extends BaseContract<abi> {
   ): [token: Address | undefined, reserve: boolean] {
     for (const [token, pf] of this.mainPriceFeeds.entries()) {
       if (pf.address === priceFeed) {
-        return [token as Address, false];
+        return [token, false];
       }
     }
     for (const [token, pf] of this.reservePriceFeeds.entries()) {
       if (pf.address === priceFeed) {
-        return [token as Address, true];
+        return [token, true];
       }
     }
     // TODO: this is v3.0 legacy code, should be gone after full v.3.1 rollout