pool booster wip

Author: apexearth

db/migrations/1739244714499-Data.js renamed to db/migrations/1739297862213-Data.js

@@ -361,6 +361,18 @@ type PoolBoosterFeeCollected @entity {
   feeAmount: BigInt!
   txHash: String!
 }
+
+type PoolBoosterTokensRescued @entity {
+  id: ID! @index
+  chainId: Int! @index
+  address: String! @index
+  timestamp: DateTime! @index
+  blockNumber: Int! @index
+  token: String!
+  amount: BigInt!
+  receiver: String!
+  txHash: String!
+}
 type OToken @entity {
   id: ID! @index
   chainId: Int! @index

schema.graphql

@@ -32,6 +32,7 @@ export * from "./bridgeTransferState.model"
 export * from "./transactionDetails.model"
 export * from "./poolBoosterCampaign.model"
 export * from "./poolBoosterFeeCollected.model"
+export * from "./poolBoosterTokensRescued.model"
 export * from "./oToken.model"
 export * from "./woToken.model"
 export * from "./oTokenAsset.model"

src/model/generated/index.ts

@@ -0,0 +1,39 @@
+import {Entity as Entity_, Column as Column_, PrimaryColumn as PrimaryColumn_, IntColumn as IntColumn_, Index as Index_, StringColumn as StringColumn_, DateTimeColumn as DateTimeColumn_, BigIntColumn as BigIntColumn_} from "@subsquid/typeorm-store"
+
+@Entity_()
+export class PoolBoosterTokensRescued {
+    constructor(props?: Partial<PoolBoosterTokensRescued>) {
+        Object.assign(this, props)
+    }
+
+    @PrimaryColumn_()
+    id!: string
+
+    @Index_()
+    @IntColumn_({nullable: false})
+    chainId!: number
+
+    @Index_()
+    @StringColumn_({nullable: false})
+    address!: string
+
+    @Index_()
+    @DateTimeColumn_({nullable: false})
+    timestamp!: Date
+
+    @Index_()
+    @IntColumn_({nullable: false})
+    blockNumber!: number
+
+    @StringColumn_({nullable: false})
+    token!: string
+
+    @BigIntColumn_({nullable: false})
+    amount!: bigint
+
+    @StringColumn_({nullable: false})
+    receiver!: string
+
+    @StringColumn_({nullable: false})
+    txHash!: string
+}

src/model/generated/poolBoosterTokensRescued.model.ts

@@ -20,3 +20,15 @@ type PoolBoosterFeeCollected @entity {
   feeAmount: BigInt!
   txHash: String!
 }
+
+type PoolBoosterTokensRescued @entity {
+  id: ID! @index
+  chainId: Int! @index
+  address: String! @index
+  timestamp: DateTime! @index
+  blockNumber: Int! @index
+  token: String!
+  amount: BigInt!
+  receiver: String!
+  txHash: String!
+}

src/templates/otoken/curve-pool-booster.graphql

@@ -1,12 +1,12 @@
 import * as curvePoolBoosterAbi from '@abi/otoken-curve-pool-booster'
-import { PoolBoosterCampaign, PoolBoosterFeeCollected } from '@model'
+import { PoolBoosterCampaign, PoolBoosterFeeCollected, PoolBoosterTokensRescued } from '@model'
 import { Context, blockFrequencyUpdater, defineProcessor, logFilter } from '@originprotocol/squid-utils'
 import { EvmBatchProcessor } from '@subsquid/evm-processor'
 
 export const createCurvePoolBoosterProcessor = (params: { otokenAddress: string; from: number }) => {
   const frequencyUpdate = blockFrequencyUpdater({ from: params.from })
 
-  const bribeCreatedFilter = logFilter({
+  const feeCollectedFilter = logFilter({
     topic0: [curvePoolBoosterAbi.events.FeeCollected.topic],
     range: { from: params.from },
   })
@@ -22,18 +22,28 @@ export const createCurvePoolBoosterProcessor = (params: { otokenAddress: string;
     topic0: [curvePoolBoosterAbi.events.CampaignClosed.topic],
     range: { from: params.from },
   })
+  const tokensRescuedFilter = logFilter({
+    topic0: [curvePoolBoosterAbi.events.TokensRescued.topic],
+    range: { from: params.from },
+  })
 
   return defineProcessor({
     name: `curve-pool-booster`,
     setup: (processor: EvmBatchProcessor) => {
-      processor.addLog(bribeCreatedFilter.value)
+      processor.addLog(campaignCreatedFilter.value)
+      processor.addLog(campaignIdUpdatedFilter.value)
+      processor.addLog(campaignClosedFilter.value)
+      processor.addLog(feeCollectedFilter.value)
+      processor.addLog(tokensRescuedFilter.value)
     },
     process: async (ctx: Context) => {
       const campaigns = new Map<string, PoolBoosterCampaign>()
       const feesCollected = new Map<string, PoolBoosterFeeCollected>()
+      const tokensRescued = new Map<string, PoolBoosterTokensRescued>()
 
       for (const block of ctx.blocksWithContent) {
         for (const log of block.logs) {
+          // TODO: Validate campaign feeCollector is in our whitelist.
           if (campaignCreatedFilter.matches(log)) {
             const data = curvePoolBoosterAbi.events.CampaignCreated.decode(log)
             const id = `${ctx.chain.id}-${log.address}`
@@ -47,23 +57,20 @@ export const createCurvePoolBoosterProcessor = (params: { otokenAddress: string;
               totalRewardAmount: data.totalRewardAmount,
             })
             campaigns.set(id, campaign)
-          }
-          if (campaignIdUpdatedFilter.matches(log)) {
+          } else if (campaignIdUpdatedFilter.matches(log)) {
             const data = curvePoolBoosterAbi.events.CampaignIdUpdated.decode(log)
             const id = `${ctx.chain.id}-${log.address}`
             let campaign = campaigns.get(id) || (await ctx.store.get(PoolBoosterCampaign, id))
             if (campaign) {
               campaign.campaignId = data.newId
             }
-          }
-          if (campaignClosedFilter.matches(log)) {
+          } else if (campaignClosedFilter.matches(log)) {
             const id = `${ctx.chain.id}-${log.address}`
             let campaign = campaigns.get(id) || (await ctx.store.get(PoolBoosterCampaign, id))
             if (campaign) {
               campaign.closed = true
             }
-          }
-          if (bribeCreatedFilter.matches(log)) {
+          } else if (feeCollectedFilter.matches(log)) {
             const data = curvePoolBoosterAbi.events.FeeCollected.decode(log)
             const feeCollected = new PoolBoosterFeeCollected({
               id: `${ctx.chain.id}-${log.id}`,
@@ -76,11 +83,26 @@ export const createCurvePoolBoosterProcessor = (params: { otokenAddress: string;
               txHash: log.transactionHash,
             })
             feesCollected.set(feeCollected.id, feeCollected)
+          } else if (tokensRescuedFilter.matches(log)) {
+            const data = curvePoolBoosterAbi.events.TokensRescued.decode(log)
+            const tokensRescuedEvent = new PoolBoosterTokensRescued({
+              id: `${ctx.chain.id}-${log.id}`,
+              chainId: ctx.chain.id,
+              address: log.address,
+              token: data.token,
+              amount: data.amount,
+              receiver: data.receiver,
+              timestamp: new Date(block.header.timestamp),
+              blockNumber: block.header.height,
+              txHash: log.transactionHash,
+            })
+            tokensRescued.set(tokensRescuedEvent.id, tokensRescuedEvent)
           }
         }
       }
-      await ctx.store.insert([...campaigns.values()])
+      await ctx.store.upsert([...campaigns.values()])
       await ctx.store.insert([...feesCollected.values()])
+      await ctx.store.insert([...tokensRescued.values()])
     },
   })
 }