@sentio/sdk 1.8.0 → 1.8.2

package/src/base-processor.ts

@@ -5,21 +5,29 @@ import { BaseContract, EventFilter } from '@ethersproject/contracts'
 import Long from 'long'
 
 import { BoundContractView, Context, ContractView } from './context'
-import { O11yResult } from './gen/processor/protos/processor'
+import { ProcessResult } from './gen/processor/protos/processor'
 import { BindInternalOptions, BindOptions } from './bind-options'
 import { PromiseOrVoid } from './promise-or-void'
+import { Trace } from './trace'
+import { utils } from 'ethers'
 
 export class EventsHandler {
   filters: EventFilter[]
-  handler: (event: Log) => Promise<O11yResult>
+  handler: (event: Log) => Promise<ProcessResult>
+}
+
+export class TraceHandler {
+  signature: string
+  handler: (trace: Trace) => Promise<ProcessResult>
 }
 
 export abstract class BaseProcessor<
   TContract extends BaseContract,
   TBoundContractView extends BoundContractView<TContract, ContractView<TContract>>
 > {
-  blockHandlers: ((block: Block) => Promise<O11yResult>)[] = []
+  blockHandlers: ((block: Block) => Promise<ProcessResult>)[] = []
   eventHandlers: EventsHandler[] = []
+  traceHandlers: TraceHandler[] = []
 
   name: string
   config: BindInternalOptions
@@ -88,11 +96,13 @@ export abstract class BaseProcessor<
           return {
             gauges: ctx.gauges,
             counters: ctx.counters,
+            logs: [],
           }
         }
         return {
           gauges: [],
           counters: [],
+          logs: [],
         }
       },
     })
@@ -109,6 +119,7 @@ export abstract class BaseProcessor<
       return {
         gauges: ctx.gauges,
         counters: ctx.counters,
+        logs: [],
       }
     })
     return this
@@ -125,4 +136,30 @@ export abstract class BaseProcessor<
       return handler(log, ctx)
     }, _filters)
   }
+
+  public onTrace(
+    signature: string,
+    handler: (trace: Trace, ctx: Context<TContract, TBoundContractView>) => PromiseOrVoid
+  ) {
+    const chainId = this.getChainId()
+    const contractView = this.CreateBoundContractView()
+
+    this.traceHandlers.push({
+      signature,
+      handler: async function (trace: Trace) {
+        const contractInterface = contractView.rawContract.interface
+        const fragment = contractInterface.getFunction(signature)
+        trace.args = contractInterface._abiCoder.decode(fragment.inputs, utils.hexDataSlice(trace.action.input, 4))
+
+        const ctx = new Context<TContract, TBoundContractView>(contractView, chainId, undefined, undefined, trace)
+        await handler(trace, ctx)
+        return {
+          gauges: ctx.gauges,
+          counters: ctx.counters,
+          logs: [],
+        }
+      },
+    })
+    return this
+  }
 }

package/src/context.ts

@@ -1,19 +1,30 @@
-import { CounterResult, GaugeResult } from './gen/processor/protos/processor'
+import { CounterResult, GaugeResult, LogResult } from './gen/processor/protos/processor'
 import { BaseContract, EventFilter } from 'ethers'
 import { Block, Log } from '@ethersproject/abstract-provider'
 import { Meter } from './meter'
 import Long from 'long'
+import { Trace } from './trace'
 
-export class EthContext {
+export class BaseContext {
+  gauges: GaugeResult[] = []
+  counters: CounterResult[] = []
+  logs: LogResult[] = []
+  meter: Meter
+
+  constructor() {
+    this.meter = new Meter(this)
+  }
+}
+
+export class EthContext extends BaseContext {
   chainId: number
   log?: Log
   block?: Block
+  trace?: Trace
   blockNumber: Long
-  gauges: GaugeResult[] = []
-  counters: CounterResult[] = []
-  meter: Meter
 
-  constructor(chainId: number, block?: Block, log?: Log) {
+  constructor(chainId: number, block?: Block, log?: Log, trace?: Trace) {
+    super()
     this.chainId = chainId
     this.log = log
     this.block = block
@@ -22,7 +33,6 @@ export class EthContext {
     } else if (block) {
       this.blockNumber = Long.fromNumber(block.number)
     }
-    this.meter = new Meter(this)
   }
 }
 
@@ -31,11 +41,13 @@ export class Context<
   TContractBoundView extends BoundContractView<TContract, ContractView<TContract>>
 > extends EthContext {
   contract: TContractBoundView
+  address: string
 
-  constructor(view: TContractBoundView, chainId: number, block?: Block, log?: Log) {
-    super(chainId, block, log)
+  constructor(view: TContractBoundView, chainId: number, block?: Block, log?: Log, trace?: Trace) {
+    super(chainId, block, log, trace)
     view.context = this
     this.contract = view
+    this.address = view.rawContract.address
   }
 }
 
@@ -79,14 +91,11 @@ export class BoundContractView<TContract extends BaseContract, TContractView ext
   }
 }
 
-export class SolanaContext {
-  gauges: GaugeResult[] = []
-  counters: CounterResult[] = []
-  meter: Meter
-
+export class SolanaContext extends BaseContext {
   address: string
 
   constructor(address: string) {
+    super()
     this.meter = new Meter(this)
     this.address = address
   }