account for multichain

Author: apexearth

src/shared/post-processors/exchange-rates/mainnetCurrencies.ts

@@ -1,4 +1,8 @@
 import { invertMap } from '@originprotocol/squid-utils'
+import { baseAddresses } from '@utils/addresses-base'
+import { plumeAddresses } from '@utils/addresses-plume'
+import { sonicAddresses } from '@utils/addresses-sonic'
+
 import {
   BaseCurrencyAddress,
   BaseCurrencySymbol,
@@ -51,3 +55,22 @@ export type CurrencySymbol = MainnetCurrencySymbol | BaseCurrencySymbol | SonicC
 export type CurrencyAddress = MainnetCurrencyAddress | BaseCurrencyAddress | SonicCurrencyAddress | PlumeCurrencyAddress
 
 export type Currency = CurrencySymbol | CurrencyAddress
+
+const tokensByChain: Record<number, Record<string, string>> = {
+  1: mainnetCurrencies,
+  8453: baseAddresses.tokens,
+  146: sonicAddresses.tokens,
+  98865: plumeAddresses.tokens,
+}
+
+/**
+ * Convert a Currency (symbol or address) to its address.
+ */
+export const currencyToAddress = (chainId: number, currency: Currency): string => {
+  if (currency.startsWith('0x')) return currency.toLowerCase()
+  const tokens = tokensByChain[chainId]
+  if (!tokens) throw new Error(`Unknown chainId: ${chainId}`)
+  const address = tokens[currency]
+  if (!address) throw new Error(`Unknown currency symbol: ${currency} for chainId: ${chainId}`)
+  return address.toLowerCase()
+}

src/templates/erc20/erc20-entry.ts

@@ -1,17 +1,18 @@
 import * as abi from '@abi/erc20'
 import { ERC20 } from '@model'
 import { Context, Processor } from '@originprotocol/squid-utils'
-import { TokenAddress } from '@utils/symbols'
+import { Currency, currencyToAddress } from '@shared/post-processors/exchange-rates/mainnetCurrencies'
 
 /**
  * Create ERC20 entry only (no event tracking).
  */
-export const createERC20Entry = ({ from, address }: { from: number; address: TokenAddress | string }): Processor => {
+export const createERC20Entry = ({ from, addressOrSymbol }: { from: number; addressOrSymbol: Currency }): Processor => {
   let erc20: ERC20 | undefined
   const initialize = async (ctx: Context) => {
     if (erc20) return
     const block = ctx.blocks.find((b) => b.header.height >= from)
     if (!block) return
+    const address = currencyToAddress(ctx.chain.id, addressOrSymbol)
     erc20 = await ctx.store.findOne(ERC20, { where: { chainId: ctx.chain.id, address } })
     try {
       if (!erc20) {

src/templates/origin-arm/origin-arm.ts

@@ -14,17 +14,12 @@ import * as originLidoArmCapManagerAbi from '@abi/origin-lido-arm-cap-manager';
 import { Arm, ArmDailyStat, ArmState, ArmSwap, ArmWithdrawalRequest, TraderateChanged } from '@model';
 import { Block, Context, EvmBatchProcessor, Processor, blockFrequencyTracker, calculateAPY, logFilter } from '@originprotocol/squid-utils';
 import { ensureExchangeRate } from '@shared/post-processors/exchange-rates';
-import { Currency, mainnetCurrencies, MainnetCurrencySymbol } from '@shared/post-processors/exchange-rates/mainnetCurrencies';
+import { Currency } from '@shared/post-processors/exchange-rates/mainnetCurrencies';
 import { createERC20Entry } from '@templates/erc20/erc20-entry';
 import { createERC20EventTracker } from '@templates/erc20/erc20-event';
 import { createEventProcessor } from '@templates/events/createEventProcessor'
 import { traceFilter } from '@utils/traceFilter'
 
-const currencyToAddress = (currency: Currency): string => {
-  if (currency.startsWith('0x')) return currency.toLowerCase()
-  return mainnetCurrencies[currency as MainnetCurrencySymbol].toLowerCase()
-}
-
 export const createOriginARMProcessors = ({
   name,
   from,
@@ -473,11 +468,11 @@ export const createOriginARMProcessors = ({
     }),
     createERC20Entry({
       from,
-      address: currencyToAddress(token0),
+      addressOrSymbol: token0,
     }),
     createERC20Entry({
       from,
-      address: currencyToAddress(token1),
+      addressOrSymbol: token1,
     }),
   ]
 }