@sentio/sdk 2.43.3-rc.3 → 2.44.0-rc.10

package/src/fuel/fuel-processor.ts

@@ -25,7 +25,7 @@ export class FuelProcessor<TContract extends Contract> implements FuelBaseProces
 
   static bind(config: FuelProcessorConfig): FuelProcessor<any> {
     const processor = new FuelProcessor(config)
-    FuelProcessorState.INSTANCE.addValue(processor)
+    addFuelProcessor(config, processor)
     return processor
   }
 
@@ -49,7 +49,7 @@ export class FuelProcessor<TContract extends Contract> implements FuelBaseProces
   }
 
   public onTransaction(
-    handler: (transaction: FuelTransaction, ctx: FuelContractContext<TContract>) => void | Promise<void>,
+    handler: (transaction: FuelTransaction, ctx: FuelContractContext<TContract>) => PromiseOrVoid,
     config: FuelFetchConfig = DEFAULT_FUEL_FETCH_CONFIG
   ) {
     const callHandler = {
@@ -161,7 +161,7 @@ export class FuelProcessor<TContract extends Contract> implements FuelBaseProces
 
   public onLog<T>(
     logIdFilter: string | string[],
-    handler: (logs: FuelLog<T>, ctx: FuelContractContext<TContract>) => void | Promise<void>
+    handler: (logs: FuelLog<T>, ctx: FuelContractContext<TContract>) => PromiseOrVoid
   ) {
     const logIds = new Set(Array.isArray(logIdFilter) ? logIdFilter : [logIdFilter])
 
@@ -308,3 +308,32 @@ export type FuelProcessorConfig = {
   endBlock?: bigint
   abi: JsonAbi
 }
+
+export function getOptionsSignature(opts: Omit<FuelProcessorConfig, 'abi'>): string {
+  const sig = [opts.address]
+  if (opts.chainId) {
+    sig.push(opts.chainId)
+  }
+  if (opts.name) {
+    sig.push(opts.name)
+  }
+  if (opts.startBlock) {
+    sig.push(opts.startBlock.toString())
+  }
+  if (opts.endBlock) {
+    sig.push(opts.endBlock.toString())
+  }
+  return sig.join('_')
+}
+
+// Dedup processor that bind multiple times
+export function getFuelProcessor(opts: Omit<FuelProcessorConfig, 'abi'>) {
+  const sig = getOptionsSignature(opts)
+  return FuelProcessorState.INSTANCE.getValue(sig)
+}
+
+export function addFuelProcessor(opts: Omit<FuelProcessorConfig, 'abi'>, processor: FuelBaseProcessor<any>) {
+  const sig = getOptionsSignature(opts)
+
+  FuelProcessorState.INSTANCE.getOrSetValue(sig, processor)
+}

package/src/fuel/global-processor.ts

@@ -4,7 +4,7 @@ import { Provider } from 'fuels'
 import { getRpcEndpoint } from './network.js'
 import { decodeFuelTransaction, DEFAULT_FUEL_FETCH_CONFIG, FuelFetchConfig } from './transaction.js'
 import { FuelContext } from './context.js'
-import { FuelProcessorConfig } from './fuel-processor.js'
+import { FuelProcessorConfig, getOptionsSignature } from './fuel-processor.js'
 import { mergeProcessResults } from '@sentio/runtime'
 
 type GlobalFuelProcessorConfig = Omit<FuelProcessorConfig, 'address' | 'abi'>
@@ -17,7 +17,13 @@ export class FuelGlobalProcessor implements FuelBaseProcessor<GlobalFuelProcesso
 
   static bind(config: GlobalFuelProcessorConfig): FuelGlobalProcessor {
     const processor = new FuelGlobalProcessor(config)
-    FuelProcessorState.INSTANCE.addValue(processor)
+    const sig =
+      'global_' +
+      getOptionsSignature({
+        ...config,
+        address: '*'
+      })
+    FuelProcessorState.INSTANCE.getOrSetValue(sig, processor)
     return processor
   }
 

package/src/fuel/index.ts

@@ -1,6 +1,7 @@
 export { FuelPlugin } from './fuel-plugin.js'
 export * from './context.js'
 export * from './fuel-processor.js'
+export * from './fuel-processor-template.js'
 export * from './network.js'
 export * from './transaction.js'
 export * from './base-processor.js'

package/src/fuel/types.ts

@@ -1,4 +1,4 @@
-import { ListStateStorage } from '@sentio/runtime'
+import { MapStateStorage } from '@sentio/runtime'
 import {
   Data_FuelBlock,
   Data_FuelCall,
@@ -17,7 +17,7 @@ export interface FuelBaseProcessor<T> {
   blockHandlers: BlockHandler[]
 }
 
-export class FuelProcessorState extends ListStateStorage<FuelBaseProcessor<any>> {
+export class FuelProcessorState extends MapStateStorage<FuelBaseProcessor<any>> {
   static INSTANCE = new FuelProcessorState()
 }
 

package/src/sui/sui-object-processor-template.ts

@@ -5,6 +5,7 @@ import { SuiMoveObject, SuiTransactionBlockResponse } from '@mysten/sui.js/clien
 import { PromiseOrVoid } from '../core/index.js'
 import {
   DEFAULT_ACCOUNT_FETCH_CONFIG,
+  SuiAccountProcessorState,
   SuiAddressProcessor,
   SuiBaseObjectOrAddressProcessor,
   SuiObjectBindOptions,
@@ -13,7 +14,8 @@ import {
 } from './sui-object-processor.js'
 import { TemplateInstanceState } from '../core/template.js'
 import { SuiBindOptions } from './sui-processor.js'
-import { TransactionFilter } from '../move/index.js'
+import { TransactionFilter, accountAddressString } from '../move/index.js'
+import { ServerError, Status } from 'nice-grpc'
 
 class ObjectHandler<HandlerType> {
   type?: string
@@ -48,7 +50,12 @@ export abstract class SuiObjectOrAddressProcessorTemplate<
   bind(options: OptionType, ctx: SuiContext): void {
     options.network = options.network || ctx.network
     options.startCheckpoint = options.startCheckpoint || ctx.checkpoint
-    const id = (options as SuiObjectBindOptions).objectId || (options as SuiBindOptions).address
+    let id = (options as SuiObjectBindOptions).objectId || (options as SuiBindOptions).address
+
+    if (id === '*') {
+      throw new ServerError(Status.INVALID_ARGUMENT, "can't bind template instance with *")
+    }
+    id = accountAddressString(id)
 
     const sig = [options.network, id].join('_')
     if (this.binds.has(sig)) {
@@ -80,6 +87,69 @@ export abstract class SuiObjectOrAddressProcessorTemplate<
       endBlock: 0n,
       baseLabels: config.baseLabels
     })
+    console.log(`successfully bind template ${sig}`)
+  }
+
+  unbind(options: OptionType, ctx: SuiContext): void {
+    options.network = options.network || ctx.network
+    options.startCheckpoint = options.startCheckpoint || ctx.checkpoint
+    let id = (options as SuiObjectBindOptions).objectId || (options as SuiBindOptions).address
+
+    if (id === '*') {
+      throw new ServerError(Status.INVALID_ARGUMENT, "can't delete template instance bind with *")
+    }
+    id = accountAddressString(id)
+
+    const sig = [options.network, id].join('_')
+    if (!this.binds.has(sig)) {
+      console.log(`the template instance ${sig} not existed or already unbind`)
+      return
+    }
+    this.binds.delete(sig)
+
+    let deleted = 0
+    const oldTemplateInstances = TemplateInstanceState.INSTANCE.unregister()
+    for (const templateInstance of oldTemplateInstances) {
+      if (templateInstance.contract?.chainId === options.network && templateInstance.contract.address == id) {
+        deleted++
+        continue
+      }
+      TemplateInstanceState.INSTANCE.addValue(templateInstance)
+    }
+
+    if (deleted !== 1) {
+      throw new ServerError(
+        Status.INVALID_ARGUMENT,
+        `Failed to delete template instance ${sig}, deleted ${deleted} times`
+      )
+    }
+
+    const oldProcessors = SuiAccountProcessorState.INSTANCE.unregister()
+    deleted = 0
+    for (const processor of oldProcessors) {
+      if (processor.templateId === this.id) {
+        if (processor.config.network == options.network && processor.config.address === id) {
+          deleted++
+          continue
+        }
+      }
+      SuiAccountProcessorState.INSTANCE.addValue(processor)
+    }
+
+    if (deleted !== 1) {
+      throw new ServerError(
+        Status.INVALID_ARGUMENT,
+        `Failed to delete processor for template ${this.id}, ${sig}. deleted ${deleted} times`
+      )
+    }
+
+    console.log(`successfully unbind template ${sig}`)
+
+    ctx.update({
+      states: {
+        configUpdated: true
+      }
+    })
   }
 
   protected onInterval(
@@ -153,6 +223,7 @@ export class SuiAddressProcessorTemplate extends SuiObjectOrAddressProcessorTemp
     for (const handler of this.handlers) {
       p.onTransactionBlock(handler.handler, handler.filter, handler.fetchConfig)
     }
+    p.templateId = this.id
     return p
   }
 
@@ -176,7 +247,9 @@ export class SuiObjectProcessorTemplate extends SuiObjectOrAddressProcessorTempl
   SuiObjectProcessor
 > {
   createProcessor(options: SuiObjectBindOptions): SuiObjectProcessor {
-    return SuiObjectProcessor.bind(options)
+    const p = SuiObjectProcessor.bind(options)
+    p.templateId = this.id
+    return p
   }
 }
 
@@ -186,6 +259,8 @@ export class SuiWrappedObjectProcessorTemplate extends SuiObjectOrAddressProcess
   SuiWrappedObjectProcessor
 > {
   createProcessor(options: SuiObjectBindOptions): SuiWrappedObjectProcessor {
-    return SuiWrappedObjectProcessor.bind(options)
+    const p = SuiWrappedObjectProcessor.bind(options)
+    p.templateId = this.id
+    return p
   }
 }

package/src/sui/sui-object-processor.ts

@@ -46,6 +46,7 @@ export interface SuiInternalObjectsBindOptions extends SuiBindOptions {
 export abstract class SuiBaseObjectOrAddressProcessor<HandlerType> {
   config: IndexConfigure
   ownerType: MoveOwnerType
+  templateId: number | undefined
 
   objectHandlers: ObjectHandler[] = []