@sentio/sdk 3.0.0-rc-next.1 → 3.0.0-rc-next.3

package/src/stark/starknet-processor.ts

@@ -0,0 +1,164 @@
+import { Data_StarknetEvent, ProcessResult } from '@sentio/protos'
+import { StarknetChainId } from '@sentio/chain'
+import { CallData, events, ParsedEvent, RpcProvider } from 'starknet'
+import { StarknetContext } from './context.js'
+import { StarknetEvent } from './event.js'
+import { ListStateStorage, mergeProcessResults } from '@sentio/runtime'
+import { StarknetProcessorConfig } from './types.js'
+import { StarknetContractView } from './contract.js'
+import { getHandlerName, proxyProcessor } from '../utils/metrics.js'
+import { HandlerOptions } from '../core/handler-options.js'
+
+export class StarknetProcessor {
+  callHandlers: CallHandler<Data_StarknetEvent>[] = []
+
+  private provider: RpcProvider
+
+  static bind(config: StarknetProcessorConfig): StarknetProcessor {
+    const processor = new StarknetProcessor(config)
+    StarknetProcessorState.INSTANCE.addValue(processor)
+    return processor
+  }
+
+  classHash: string
+
+  constructor(readonly config: StarknetProcessorConfig) {
+    return proxyProcessor(this)
+  }
+
+  async configure() {
+    this.provider = new RpcProvider({
+      nodeUrl: getRpcEndpoint(this.config.chainId)
+    })
+    const classHash = await this.provider.getClassHashAt(this.config.address)
+    if (!classHash) {
+      throw new Error("can't find the class hash defined at " + this.config.address)
+    }
+    this.classHash = classHash
+    if (!this.config.abi) {
+      const clazz = await this.provider.getClass(classHash, 'latest')
+      this.config.abi = clazz.abi
+    }
+  }
+
+  public onEvent(
+    event: string | string[],
+    handler: (events: StarknetEvent<ParsedEvent>, ctx: StarknetContext<StarknetContractView>) => void | Promise<void>,
+    handlerOptions?: HandlerOptions<object, StarknetEvent<ParsedEvent>>
+  ) {
+    const eventFilter = Array.isArray(event) ? event : [event]
+    if (!this.config.abi) {
+      throw new Error('abi not found')
+    }
+    const abi = this.config.abi
+    const callHandler = {
+      handlerName: getHandlerName(),
+      handler: async (call: Data_StarknetEvent) => {
+        try {
+          const eventData = [call.result] as any[]
+          const abiEvents = events.getAbiEvents(abi)
+
+          const abiStructs = CallData.getAbiStruct(abi)
+          const abiEnums = CallData.getAbiEnum(abi)
+
+          const parsedEvents = events.parseEvents(eventData, abiEvents, abiStructs, abiEnums)
+          const results: ProcessResult[] = []
+          const { block_hash, block_number, transaction_hash, from_address } = call.result!
+          for (let i = 0; i < parsedEvents.length; i++) {
+            const ctx = new StarknetContext<StarknetContractView>(
+              this.config,
+              this.provider,
+              block_number,
+              block_hash,
+              transaction_hash,
+              i,
+              this.classHash
+            )
+            const e = new StarknetEvent(from_address, transaction_hash, parsedEvents[i])
+            try {
+              await handler(e, ctx)
+            } catch (e) {
+              console.error(e)
+            }
+            results.push(ctx.stopAndGetResult())
+          }
+          return mergeProcessResults(results)
+        } catch (e) {
+          console.error(e)
+          return ProcessResult.fromPartial({})
+        }
+      },
+      eventFilter,
+      partitionHandler: async (call: Data_StarknetEvent): Promise<string | undefined> => {
+        const p = handlerOptions?.partitionKey
+        if (!p) return undefined
+        if (typeof p === 'function') {
+          try {
+            const eventData = [call.result] as any[]
+            const abiEvents = events.getAbiEvents(abi)
+            const abiStructs = CallData.getAbiStruct(abi)
+            const abiEnums = CallData.getAbiEnum(abi)
+            const parsedEvents = events.parseEvents(eventData, abiEvents, abiStructs, abiEnums)
+            if (parsedEvents.length > 0) {
+              const { from_address, transaction_hash } = call.result!
+              const e = new StarknetEvent(from_address, transaction_hash, parsedEvents[0])
+              return p(e)
+            }
+            return undefined
+          } catch (e) {
+            console.error(e)
+            return undefined
+          }
+        }
+        return p
+      }
+    }
+    this.callHandlers.push(callHandler)
+    return this
+  }
+}
+
+export type CallHandler<T> = {
+  handlerName: string
+  handler: (call: T) => Promise<ProcessResult>
+  eventFilter?: string[]
+  partitionHandler?: (call: T) => Promise<string | undefined>
+}
+
+function getRpcEndpoint(chainId: StarknetChainId | string) {
+  switch (chainId) {
+    case StarknetChainId.STARKNET_MAINNET:
+      return 'https://starknet-mainnet.g.alchemy.com/starknet/version/rpc/v0_8/8sD5yitBslIYCPFzSq_Q1ObJHqPlZxFw'
+    default:
+      return 'https://starknet-sepolia.g.alchemy.com/starknet/version/rpc/v0_8/8sD5yitBslIYCPFzSq_Q1ObJHqPlZxFw'
+  }
+}
+
+export class StarknetProcessorState extends ListStateStorage<StarknetProcessor> {
+  static INSTANCE = new StarknetProcessorState()
+}
+
+export abstract class AbstractStarknetProcessor {
+  private readonly processor: StarknetProcessor
+  protected constructor(
+    readonly abi: any,
+    readonly config: StarknetProcessorConfig
+  ) {
+    this.processor = new StarknetProcessor(config)
+    StarknetProcessorState.INSTANCE.addValue(this.processor)
+    return proxyProcessor(this)
+  }
+
+  onEvent<T, C>(
+    eventName: string,
+    structName: string,
+    handler: (event: StarknetEvent<T>, ctx: StarknetContext<C>) => Promise<void>
+  ) {
+    this.processor.onEvent(eventName, async (events, ctx) => {
+      const eventData = events.data[structName] as T
+      const e = new StarknetEvent<T>(events.caller, events.transactionHash, eventData)
+      await handler(e, ctx as StarknetContext<C>)
+    })
+    return this
+  }
+}

package/src/stark/types.ts

@@ -0,0 +1,11 @@
+import { StarknetChainId } from '@sentio/chain'
+import { Abi } from 'starknet'
+
+export type StarknetProcessorConfig = {
+  address: string
+  name?: string
+  chainId: StarknetChainId | string
+  startBlock?: bigint
+  endBlock?: bigint
+  abi?: Abi
+}

package/src/testing/starknet-facet.ts

@@ -0,0 +1,62 @@
+import { TestProcessorServer } from './test-processor-server.js'
+import { DataBinding, HandlerType } from '@sentio/protos'
+import { StarknetChainId } from '@sentio/chain'
+
+export class StarknetFacet {
+  server: TestProcessorServer
+
+  constructor(server: TestProcessorServer) {
+    this.server = server
+  }
+
+  testOnEvents(events: any, network: StarknetChainId = StarknetChainId.STARKNET_SEPOLIA) {
+    const bindings = this.buildBinding(events, network)
+    if (!bindings) {
+      throw Error('Invalid test : ' + JSON.stringify(events))
+    }
+
+    return this.server.processBindings({
+      bindings
+    })
+  }
+
+  private buildBinding(events: any, network: StarknetChainId): DataBinding[] {
+    const res: DataBinding[] = []
+    const data = Array.isArray(events) ? events : [events]
+    for (const event of data) {
+      for (const config of this.server.contractConfigs) {
+        if (config.contract?.chainId !== network) {
+          continue
+        }
+
+        for (const logConfig of config.starknetEventConfigs) {
+          const binding = {
+            data: {
+              starknetEvents: {
+                result: event,
+                timestamp: new Date()
+              }
+            },
+            handlerIds: [logConfig.handlerId],
+            handlerType: HandlerType.STARKNET_EVENT,
+            chainId: config.contract.chainId
+          }
+
+          const logFilters = logConfig.filters
+          const keys = event.keys as string[]
+
+          for (const key of keys) {
+            for (const logFilter of logFilters) {
+              if (logFilter.keys.includes(key)) {
+                res.push(binding)
+                return res
+              }
+            }
+          }
+        }
+      }
+    }
+
+    return res
+  }
+}

package/src/testing/test-processor-server.ts

@@ -25,6 +25,7 @@ import { EthFacet } from './eth-facet.js'
 import { SuiFacet } from './sui-facet.js'
 import { FuelFacet } from './fuel-facet.js'
 import { CosmosFacet } from './cosmos-facet.js'
+import { StarknetFacet } from './starknet-facet.js'
 import { Subject } from 'rxjs'
 import { MemoryDatabase } from './memory-database.js'
 import { DatabaseSchemaState } from '../core/database-schema.js'
@@ -53,6 +54,7 @@ export class TestProcessorServer implements ProcessorServiceImplementation {
   iota: IotaFacet
   fuel: FuelFacet
   cosmos: CosmosFacet
+  starknet: StarknetFacet
   _db: MemoryDatabase
 
   constructor(loader: () => Promise<any>, httpEndpoints: Record<string, string> = {}) {
@@ -66,6 +68,7 @@ export class TestProcessorServer implements ProcessorServiceImplementation {
     this.iota = new IotaFacet(this)
     this.fuel = new FuelFacet(this)
     this.cosmos = new CosmosFacet(this)
+    this.starknet = new StarknetFacet(this)
 
     for (const k of Object.keys(ChainInfo)) {
       const http = httpEndpoints[k] || ''

package/src/utils/price.ts

@@ -23,7 +23,7 @@ export function getPriceClient(basePath = Endpoints.INSTANCE.priceFeedAPI) {
     basePath = basePath.slice(0, -1)
   }
   client.setConfig({
-    baseUrl: basePath || 'https://app.sentio.xyz',
+    baseUrl: basePath || 'https://api.sentio.xyz',
     auth: getApiKey()
   })
   return PriceService

package/lib/core/base-context.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=base-context.test.d.ts.map

package/lib/core/base-context.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"base-context.test.d.ts","sourceRoot":"","sources":["../../src/core/base-context.test.ts"],"names":[],"mappings":""}

package/lib/core/base-context.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"base-context.test.js","sourceRoot":"","sources":["../../src/core/base-context.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAA;AACxC,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAA;AACxC,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAA;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAA;AAIvC,iDAAiD;AACjD,MAAM,WAAY,SAAQ,WAAW;IACnC,YAAY,UAAmB;QAC7B,KAAK,CAAC,UAAU,CAAC,CAAA;IACnB,CAAC;IAES,mBAAmB,CAAC,IAAY,EAAE,MAAc;QACxD,OAAO;YACL,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,EAAE;YAChB,MAAM,EAAE,EAAE;YACV,QAAQ,EAAE,CAAC;YACX,eAAe,EAAE,EAAE;YACnB,gBAAgB,EAAE,CAAC;YACnB,OAAO,EAAE,cAAc;YACvB,OAAO,EAAE,UAAU;YACnB,IAAI,EAAE,IAAI;SACX,CAAA;IACH,CAAC;IAED,UAAU;QACR,OAAO,OAAO,CAAC,QAAQ,CAAA;IACzB,CAAC;CACF;AAED,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC5C,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;QACnF,MAAM,GAAG,GAAG,IAAI,WAAW,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;QAChE,MAAM,UAAU,GAAG,OAAO,CAAA;QAC1B,MAAM,UAAU,GAAG,KAAK,CAAA,CAAC,4BAA4B;QACrD,MAAM,KAAK,GAAa,EAAE,CAAA;QAE1B,OAAO,CAAC,GAAG,CAAC,gCAAgC,EAAE,UAAU,EAAE,eAAe,CAAC,CAAA;QAE1E,MAAM,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE,CAAA;QACxC,IAAI,cAAsB,CAAA;QAE1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,8CAA8C;YAC9C,IAAI,CAAC,GAAG,UAAU,KAAK,CAAC,EAAE,CAAC;gBACzB,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE,CAAA;YACpC,CAAC;YAED,sDAAsD;YACtD,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,EAAE;gBAChC,SAAS,EAAE,CAAC;gBACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,IAAI,EAAE,eAAe,CAAC,EAAE;gBACxB,MAAM,EAAE;oBACN,KAAK,EAAE,CAAC,GAAG,CAAC;oBACZ,IAAI,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC;iBAClB;aACF,CAAC,CAAA;YAEF,sCAAsC;YACtC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,UAAU,KAAK,CAAC,EAAE,CAAC;gBAC/B,MAAM,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAA;gBACtC,MAAM,QAAQ,GAAG,YAAY,GAAG,cAAe,CAAA;gBAC/C,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;gBAEpB,sCAAsC;gBACtC,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,EAAE,CAAA;gBACtC,OAAO,CAAC,GAAG,CACT,aAAa,CAAC,GAAG,CAAC,KAAK,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,UAAU,mBAAmB,CAAC,QAAQ,CAAC,QAAQ,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAClI,CAAA;YACH,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAA;QACtC,MAAM,aAAa,GAAG,YAAY,GAAG,cAAc,CAAA;QAEnD,kCAAkC;QAClC,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,CAAA;QACrE,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,EAAE,CAAA;QAClE,MAAM,iBAAiB,GAAG,SAAS,GAAG,UAAU,CAAA;QAEhD,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAA;QAC7C,OAAO,CAAC,GAAG,CAAC,eAAe,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAA;QACxD,OAAO,CAAC,GAAG,CAAC,oBAAoB,UAAU,SAAS,CAAC,CAAA;QACpD,OAAO,CAAC,GAAG,CAAC,uBAAuB,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAA;QAC7D,OAAO,CAAC,GAAG,CAAC,sBAAsB,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAA;QAC3D,OAAO,CAAC,GAAG,CAAC,qCAAqC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAA;QAEjF,yBAAyB;QACzB,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,EAAE,CAAA;QACzC,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,WAAW,CAAC,QAAQ,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAA;QAEvF,mCAAmC;QACnC,MAAM,MAAM,GAAG,GAAG,CAAC,gBAAgB,EAAE,CAAA;QACrC,OAAO,CAAC,GAAG,CAAC,6BAA6B,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAA;QAEhE,uEAAuE;QACvE,IAAI,iBAAiB,GAAG,GAAG,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAA;YACpD,OAAO,CAAC,GAAG,CAAC,kEAAkE,CAAC,CAAA;QACjF,CAAC;QAED,+DAA+D;QAC/D,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,GAAG,GAAG,IAAI,WAAW,EAAE,CAAA;QAC7B,MAAM,WAAW,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,CAAA;QAErD,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAA;QAE9C,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,qCAAqC;YACrC,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;gBACd,MAAM,CAAC,EAAE,EAAE,CAAA;YACb,CAAC;YAED,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAA;YAElD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,YAAY,EAAE;oBACjC,EAAE,EAAE,CAAC;oBACL,OAAO,EAAE,QAAQ,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,+BAA+B;iBAChE,CAAC,CAAA;YACJ,CAAC;YAED,MAAM,SAAS,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAA;YAChD,MAAM,YAAY,GAAG,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,IAAI,GAAG,IAAI,CAAA;YAE5D,OAAO,CAAC,GAAG,CAAC,SAAS,UAAU,YAAY,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAA;QACzE,CAAC;QAED,MAAM,MAAM,GAAG,GAAG,CAAC,gBAAgB,EAAE,CAAA;QACrC,OAAO,CAAC,GAAG,CAAC,8BAA8B,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAA;IACnE,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}