ensure arm underlying tokens have an entry in ERC20.

Author: apexearth

src/templates/erc20/erc20-entry.ts

@@ -0,0 +1,40 @@
+import * as abi from '@abi/erc20'
+import { ERC20 } from '@model'
+import { Context, Processor } from '@originprotocol/squid-utils'
+import { TokenAddress } from '@utils/symbols'
+
+/**
+ * Create ERC20 entry only (no event tracking).
+ */
+export const createERC20Entry = ({ from, address }: { from: number; address: TokenAddress | string }): 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
+    erc20 = await ctx.store.findOne(ERC20, { where: { chainId: ctx.chain.id, address } })
+    try {
+      if (!erc20) {
+        const contract = new abi.Contract(ctx, block.header, address)
+        const [name, symbol, decimals] = await Promise.all([contract.name(), contract.symbol(), contract.decimals()])
+        erc20 = new ERC20({
+          id: `${ctx.chain.id}-${address}`,
+          chainId: ctx.chain.id,
+          address,
+          name,
+          symbol,
+          decimals,
+        })
+        await ctx.store.insert(erc20)
+      }
+    } catch (err: any) {
+      ctx.log.info({ height: block.header.height, err: err.message }, `Failed to get contract name for ${address}`)
+    }
+  }
+  return {
+    from,
+    async process(ctx: Context) {
+      await initialize(ctx)
+    },
+  }
+}

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

@@ -1,30 +1,26 @@
-import dayjs from 'dayjs'
-import { findLast } from 'lodash'
-import { LessThan } from 'typeorm'
-import { formatEther, formatUnits } from 'viem'
+import dayjs from 'dayjs';
+import { findLast } from 'lodash';
+import { LessThan } from 'typeorm';
+import { formatEther, formatUnits } from 'viem';
 
-import * as erc20Abi from '@abi/erc20'
-import * as originOsArmAbi from '@abi/origin-arm'
-import * as originEthenaArmAbi from '@abi/origin-ethena-arm'
-import * as originEtherfiArmAbi from '@abi/origin-etherfi-arm'
-import * as originLidoArmAbi from '@abi/origin-lido-arm'
-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 } from '@shared/post-processors/exchange-rates/mainnetCurrencies'
-import { createERC20EventTracker } from '@templates/erc20/erc20-event'
+
+
+import * as erc20Abi from '@abi/erc20';
+import * as originOsArmAbi from '@abi/origin-arm';
+import * as originEthenaArmAbi from '@abi/origin-ethena-arm';
+import * as originEtherfiArmAbi from '@abi/origin-etherfi-arm';
+import * as originLidoArmAbi from '@abi/origin-lido-arm';
+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 } 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'
 
+
 export const createOriginARMProcessors = ({
   name,
   from,
@@ -471,5 +467,13 @@ export const createOriginARMProcessors = ({
       from,
       address: armAddress,
     }),
+    createERC20Entry({
+      from,
+      address: token0,
+    }),
+    createERC20Entry({
+      from,
+      address: token1,
+    }),
   ]
-}
+}