@sentio/sdk 4.0.0-rc.1 → 4.0.0-rc.3

package/src/aptos/aptos-plugin.ts

@@ -2,21 +2,27 @@ import { errorString, GLOBAL_CONFIG, mergeProcessResults, Plugin, PluginManager,
 import { PartitionHandlerManager } from '../core/index.js'
 import { HandlerRegister } from '../core/handler-register.js'
 import {
-  AccountConfig,
-  ContractConfig,
-  DataBinding,
+  AccountConfigSchema,
+  ContractConfigSchema,
+  type DataBinding,
   HandlerType,
-  InitResponse,
-  MoveCallHandlerConfig,
-  MoveEventHandlerConfig,
+  type InitResponse,
+  type MoveCallHandlerConfig,
+  MoveCallHandlerConfigSchema,
+  type MoveEventHandlerConfig,
+  MoveEventHandlerConfigSchema,
+  MoveOnIntervalConfigSchema,
   MoveOwnerType,
-  ProcessConfigResponse,
-  ProcessResult,
-  ProcessStreamResponse_Partitions,
-  StartRequest
+  MoveResourceChangeConfigSchema,
+  type ProcessConfigResponse,
+  type ProcessResult,
+  type ProcessStreamResponse_Partitions,
+  ProcessStreamResponse_PartitionsSchema,
+  type StartRequest
 } from '@sentio/protos'
+import { create } from '@bufbuild/protobuf'
 
-import { ServerError, Status } from 'nice-grpc'
+import { ConnectError, Code } from '@connectrpc/connect'
 
 import { AptosProcessorState, AptosResourceProcessorState } from './aptos-processor.js'
 
@@ -61,7 +67,7 @@ export class AptosPlugin extends Plugin {
           network: <AptosNetwork>instance.contract?.chainId || AptosNetwork.MAIN_NET,
           startVersion: instance.startBlock || 0n,
           endVersion: instance.endBlock,
-          baseLabels: instance.baseLabels
+          baseLabels: instance.baseLabels as { [key: string]: string } | undefined
         },
         NoopContext
       )
@@ -75,7 +81,7 @@ export class AptosPlugin extends Plugin {
       if (forChainId !== undefined && forChainId !== chainId.toString()) {
         continue
       }
-      const contractConfig = ContractConfig.fromPartial({
+      const contractConfig = create(ContractConfigSchema, {
         processorType: USER_PROCESSOR,
         contract: {
           name: aptosProcessor.moduleName,
@@ -90,7 +96,7 @@ export class AptosPlugin extends Plugin {
       for (const handler of aptosProcessor.eventHandlers) {
         const handlerId = this.handlerRegister.register(handler.handler, chainId)
         this.partitionManager.registerPartitionHandler(HandlerType.APT_EVENT, handlerId, handler.partitionHandler)
-        const eventHandlerConfig: MoveEventHandlerConfig = {
+        const eventHandlerConfig: MoveEventHandlerConfig = create(MoveEventHandlerConfigSchema, {
           filters: handler.filters.map((f) => {
             return {
               type: f.type,
@@ -101,7 +107,7 @@ export class AptosPlugin extends Plugin {
           fetchConfig: handler.fetchConfig,
           handlerId,
           handlerName: handler.handlerName
-        }
+        })
         contractConfig.moveEventConfigs.push(eventHandlerConfig)
       }
 
@@ -109,7 +115,7 @@ export class AptosPlugin extends Plugin {
       for (const handler of aptosProcessor.callHandlers) {
         const handlerId = this.handlerRegister.register(handler.handler, chainId)
         this.partitionManager.registerPartitionHandler(HandlerType.APT_CALL, handlerId, handler.partitionHandler)
-        const functionHandlerConfig: MoveCallHandlerConfig = {
+        const functionHandlerConfig: MoveCallHandlerConfig = create(MoveCallHandlerConfigSchema, {
           filters: handler.filters.map((filter) => {
             return {
               function: filter.function,
@@ -123,7 +129,7 @@ export class AptosPlugin extends Plugin {
           fetchConfig: handler.fetchConfig,
           handlerId,
           handlerName: handler.handlerName
-        }
+        })
         contractConfig.moveCallConfigs.push(functionHandlerConfig)
       }
 
@@ -131,21 +137,23 @@ export class AptosPlugin extends Plugin {
       for (const handler of aptosProcessor.transactionIntervalHandlers) {
         const handlerId = this.handlerRegister.register(handler.handler, chainId)
         this.partitionManager.registerPartitionHandler(HandlerType.APT_CALL, handlerId, handler.partitionHandler)
-        contractConfig.moveIntervalConfigs.push({
-          intervalConfig: {
-            handlerId: handlerId,
-            handlerName: handler.handlerName,
-            minutes: 0,
-            minutesInterval: handler.timeIntervalInMinutes,
-            slot: 0,
-            slotInterval: handler.versionInterval,
-            fetchConfig: undefined
-          },
-          ownerType: MoveOwnerType.ADDRESS,
-          fetchConfig: handler.fetchConfig,
-          resourceFetchConfig: undefined,
-          type: ''
-        })
+        contractConfig.moveIntervalConfigs.push(
+          create(MoveOnIntervalConfigSchema, {
+            intervalConfig: {
+              handlerId: handlerId,
+              handlerName: handler.handlerName,
+              minutes: 0,
+              minutesInterval: handler.timeIntervalInMinutes,
+              slot: 0,
+              slotInterval: handler.versionInterval,
+              fetchConfig: undefined
+            },
+            ownerType: MoveOwnerType.ADDRESS,
+            fetchConfig: handler.fetchConfig,
+            resourceFetchConfig: undefined,
+            type: ''
+          })
+        )
       }
 
       config.contractConfigs.push(contractConfig)
@@ -157,7 +165,7 @@ export class AptosPlugin extends Plugin {
       if (forChainId !== undefined && forChainId !== chainId.toString()) {
         continue
       }
-      const accountConfig = AccountConfig.fromPartial({
+      const accountConfig = create(AccountConfigSchema, {
         address: aptosProcessor.config.address,
         chainId: aptosProcessor.getChainId(),
         startBlock: aptosProcessor.config.startVersion,
@@ -166,12 +174,14 @@ export class AptosPlugin extends Plugin {
       for (const handler of aptosProcessor.resourceChangeHandlers) {
         const handlerId = this.handlerRegister.register(handler.handler, chainId)
         this.partitionManager.registerPartitionHandler(HandlerType.APT_RESOURCE, handlerId, handler.partitionHandler)
-        accountConfig.moveResourceChangeConfigs.push({
-          handlerId: handlerId,
-          handlerName: handler.handlerName,
-          types: typeof handler.type === 'string' ? [handler.type] : handler.type,
-          includeDeleted: false
-        })
+        accountConfig.moveResourceChangeConfigs.push(
+          create(MoveResourceChangeConfigSchema, {
+            handlerId: handlerId,
+            handlerName: handler.handlerName,
+            types: typeof handler.type === 'string' ? [handler.type] : handler.type,
+            includeDeleted: false
+          })
+        )
       }
 
       config.accountConfigs.push(accountConfig)
@@ -182,7 +192,7 @@ export class AptosPlugin extends Plugin {
       if (forChainId !== undefined && forChainId !== chainId.toString()) {
         continue
       }
-      const accountConfig = AccountConfig.fromPartial({
+      const accountConfig = create(AccountConfigSchema, {
         address: aptosProcessor.config.address,
         chainId: aptosProcessor.getChainId(),
         startBlock: aptosProcessor.config.startVersion,
@@ -192,29 +202,33 @@ export class AptosPlugin extends Plugin {
         const handlerId = this.handlerRegister.register(handler.handler, chainId)
         this.partitionManager.registerPartitionHandler(HandlerType.APT_RESOURCE, handlerId, handler.partitionHandler)
         if (handler.timeIntervalInMinutes || handler.versionInterval) {
-          accountConfig.moveIntervalConfigs.push({
-            intervalConfig: {
-              handlerId: handlerId,
-              handlerName: handler.handlerName,
-              minutes: 0,
-              minutesInterval: handler.timeIntervalInMinutes,
-              slot: 0,
-              slotInterval: handler.versionInterval,
+          accountConfig.moveIntervalConfigs.push(
+            create(MoveOnIntervalConfigSchema, {
+              intervalConfig: {
+                handlerId: handlerId,
+                handlerName: handler.handlerName,
+                minutes: 0,
+                minutesInterval: handler.timeIntervalInMinutes,
+                slot: 0,
+                slotInterval: handler.versionInterval,
+                fetchConfig: undefined
+              },
+              type: (Array.isArray(handler.type) ? handler.type[0] : handler.type) || '',
+              ownerType: MoveOwnerType.ADDRESS,
+              resourceFetchConfig: handler.fetchConfig,
               fetchConfig: undefined
-            },
-            type: (Array.isArray(handler.type) ? handler.type[0] : handler.type) || '',
-            ownerType: MoveOwnerType.ADDRESS,
-            resourceFetchConfig: handler.fetchConfig,
-            fetchConfig: undefined
-          })
+            })
+          )
         } else if (handler.type) {
           // on resource change
-          accountConfig.moveResourceChangeConfigs.push({
-            handlerId,
-            handlerName: handler.handlerName,
-            types: typeof handler.type == 'string' ? [handler.type] : handler.type,
-            includeDeleted: false
-          })
+          accountConfig.moveResourceChangeConfigs.push(
+            create(MoveResourceChangeConfigSchema, {
+              handlerId,
+              handlerName: handler.handlerName,
+              types: typeof handler.type == 'string' ? [handler.type] : handler.type,
+              includeDeleted: false
+            })
+          )
         }
       }
       config.accountConfigs.push(accountConfig)
@@ -232,7 +246,7 @@ export class AptosPlugin extends Plugin {
       case HandlerType.APT_RESOURCE:
         return this.processAptosResource(request)
       default:
-        throw new ServerError(Status.INVALID_ARGUMENT, 'No handle type registered ' + request.handlerType)
+        throw new ConnectError('No handle type registered ' + request.handlerType, Code.InvalidArgument)
     }
   }
 
@@ -240,42 +254,42 @@ export class AptosPlugin extends Plugin {
     let data: any
     switch (request.handlerType) {
       case HandlerType.APT_EVENT:
-        if (!request.data?.aptEvent) {
-          throw new ServerError(Status.INVALID_ARGUMENT, "aptEvent can't be empty")
+        if (request.data?.value.case !== 'aptEvent') {
+          throw new ConnectError("aptEvent can't be empty", Code.InvalidArgument)
         }
-        data = new AptEvent(request.data.aptEvent)
+        data = new AptEvent(request.data.value.value)
         break
       case HandlerType.APT_CALL:
-        if (!request.data?.aptCall) {
-          throw new ServerError(Status.INVALID_ARGUMENT, "aptCall can't be empty")
+        if (request.data?.value.case !== 'aptCall') {
+          throw new ConnectError("aptCall can't be empty", Code.InvalidArgument)
         }
-        data = new AptCall(request.data.aptCall)
+        data = new AptCall(request.data.value.value)
         break
       case HandlerType.APT_RESOURCE:
-        if (!request.data?.aptResource) {
-          throw new ServerError(Status.INVALID_ARGUMENT, "aptResource can't be empty")
+        if (request.data?.value.case !== 'aptResource') {
+          throw new ConnectError("aptResource can't be empty", Code.InvalidArgument)
         }
-        data = new AptResource(request.data.aptResource)
+        data = new AptResource(request.data.value.value)
         break
       default:
-        throw new ServerError(Status.INVALID_ARGUMENT, 'No handle type registered ' + request.handlerType)
+        throw new ConnectError('No handle type registered ' + request.handlerType, Code.InvalidArgument)
     }
     const partitions = await this.partitionManager.processPartitionForHandlerType(
       request.handlerType,
       request.handlerIds,
       data
     )
-    return {
+    return create(ProcessStreamResponse_PartitionsSchema, {
       partitions
-    }
+    })
   }
 
   async processAptosEvent(binding: DataBinding): Promise<ProcessResult> {
-    if (!binding.data?.aptEvent) {
-      throw new ServerError(Status.INVALID_ARGUMENT, "Event can't be empty")
+    if (binding.data?.value.case !== 'aptEvent') {
+      throw new ConnectError("Event can't be empty", Code.InvalidArgument)
     }
     const promises: Promise<ProcessResult>[] = []
-    const event = new AptEvent(binding.data.aptEvent)
+    const event = new AptEvent(binding.data.value.value)
 
     for (const handlerId of binding.handlerIds) {
       const promise = this.handlerRegister
@@ -284,9 +298,9 @@ export class AptosPlugin extends Plugin {
           handlerId
         )(event)
         .catch((e: any) => {
-          throw new ServerError(
-            Status.INTERNAL,
-            'error processing event: ' + JSON.stringify(event) + '\n' + errorString(e)
+          throw new ConnectError(
+            'error processing event: ' + JSON.stringify(event) + '\n' + errorString(e),
+            Code.Internal
           )
         })
       if (GLOBAL_CONFIG.execution.sequential) {
@@ -298,10 +312,10 @@ export class AptosPlugin extends Plugin {
   }
 
   async processAptosResource(binding: DataBinding): Promise<ProcessResult> {
-    if (!binding.data?.aptResource) {
-      throw new ServerError(Status.INVALID_ARGUMENT, "Resource can't be empty")
+    if (binding.data?.value.case !== 'aptResource') {
+      throw new ConnectError("Resource can't be empty", Code.InvalidArgument)
     }
-    const resource = new AptResource(binding.data.aptResource)
+    const resource = new AptResource(binding.data.value.value)
 
     const promises: Promise<ProcessResult>[] = []
     for (const handlerId of binding.handlerIds) {
@@ -311,9 +325,9 @@ export class AptosPlugin extends Plugin {
           handlerId
         )(resource)
         .catch((e: any) => {
-          throw new ServerError(
-            Status.INTERNAL,
-            'error processing resource: ' + JSON.stringify(resource) + '\n' + errorString(e)
+          throw new ConnectError(
+            'error processing resource: ' + JSON.stringify(resource) + '\n' + errorString(e),
+            Code.Internal
           )
         })
       if (GLOBAL_CONFIG.execution.sequential) {
@@ -325,10 +339,10 @@ export class AptosPlugin extends Plugin {
   }
 
   async processAptosFunctionCall(binding: DataBinding): Promise<ProcessResult> {
-    if (!binding.data?.aptCall) {
-      throw new ServerError(Status.INVALID_ARGUMENT, "Call can't be empty")
+    if (binding.data?.value.case !== 'aptCall') {
+      throw new ConnectError("Call can't be empty", Code.InvalidArgument)
     }
-    const call = new AptCall(binding.data.aptCall)
+    const call = new AptCall(binding.data.value.value)
 
     const promises: Promise<ProcessResult>[] = []
     for (const handlerId of binding.handlerIds) {
@@ -339,9 +353,9 @@ export class AptosPlugin extends Plugin {
           handlerId
         )(call)
         .catch((e: any) => {
-          throw new ServerError(
-            Status.INTERNAL,
-            'error processing call: ' + JSON.stringify(call) + '\n' + errorString(e)
+          throw new ConnectError(
+            'error processing call: ' + JSON.stringify(call) + '\n' + errorString(e),
+            Code.Internal
           )
         })
       if (GLOBAL_CONFIG.execution.sequential) {

package/src/aptos/aptos-processor.ts

@@ -11,8 +11,17 @@ import {
 import { AptosBindOptions, AptosNetwork } from './network.js'
 import { AptosContext, AptosResourcesContext, AptosTransactionContext } from './context.js'
 import { ListStateStorage } from '@sentio/runtime'
-import { Data_AptResource, HandleInterval, MoveAccountFetchConfig, MoveFetchConfig } from '@sentio/protos'
-import { ServerError, Status } from 'nice-grpc'
+import {
+  type Data_AptResource,
+  type HandleInterval,
+  HandleIntervalSchema,
+  type MoveAccountFetchConfig,
+  MoveAccountFetchConfigSchema,
+  type MoveFetchConfig,
+  MoveFetchConfigSchema
+} from '@sentio/protos'
+import { create } from '@bufbuild/protobuf'
+import { ConnectError, Code } from '@connectrpc/connect'
 import {
   accountTypeString,
   CallHandler,
@@ -34,17 +43,17 @@ import { GeneralTransactionResponse, HandlerOptions } from './models.js'
 import { getHandlerName, proxyProcessor } from '../utils/metrics.js'
 import { AptCall, AptEvent, AptResource } from './data.js'
 
-const DEFAULT_FETCH_CONFIG: MoveFetchConfig = {
+const DEFAULT_FETCH_CONFIG: MoveFetchConfig = create(MoveFetchConfigSchema, {
   resourceChanges: false,
   allEvents: true,
   inputs: true,
   // for backward compatibility
   supportMultisigFunc: true
-}
+})
 
-export const DEFAULT_RESOURCE_FETCH_CONFIG: MoveAccountFetchConfig = {
+export const DEFAULT_RESOURCE_FETCH_CONFIG: MoveAccountFetchConfig = create(MoveAccountFetchConfigSchema, {
   owned: true
-}
+})
 
 type IndexConfigure = {
   address: string
@@ -84,7 +93,7 @@ export class AptosTransactionProcessor<T extends GeneralTransactionResponse, CT
     handlerOptions?: HandlerOptions<MoveFetchConfig, T>
   ): this {
     let _filters: EventFilter[] = []
-    const _fetchConfig = MoveFetchConfig.fromPartial({ ...DEFAULT_FETCH_CONFIG, ...handlerOptions })
+    const _fetchConfig = create(MoveFetchConfigSchema, { ...DEFAULT_FETCH_CONFIG, ...handlerOptions })
 
     if (Array.isArray(filter)) {
       _filters = filter
@@ -100,7 +109,7 @@ export class AptosTransactionProcessor<T extends GeneralTransactionResponse, CT
       handlerName: getHandlerName(),
       handler: async function (data) {
         if (!data.rawTransaction) {
-          throw new ServerError(Status.INVALID_ARGUMENT, 'event is null')
+          throw new ConnectError('event is null', Code.InvalidArgument)
         }
         const txn = data.transaction
 
@@ -143,7 +152,7 @@ export class AptosTransactionProcessor<T extends GeneralTransactionResponse, CT
     handlerOptions?: HandlerOptions<MoveFetchConfig, T>
   ): this {
     let _filters: FunctionNameAndCallFilter[] = []
-    const _fetchConfig = MoveFetchConfig.fromPartial({ ...DEFAULT_FETCH_CONFIG, ...handlerOptions })
+    const _fetchConfig = create(MoveFetchConfigSchema, { ...DEFAULT_FETCH_CONFIG, ...handlerOptions })
 
     if (Array.isArray(filter)) {
       _filters = filter
@@ -159,7 +168,7 @@ export class AptosTransactionProcessor<T extends GeneralTransactionResponse, CT
       handlerName: getHandlerName(),
       handler: async function (data) {
         if (!data.rawTransaction) {
-          throw new ServerError(Status.INVALID_ARGUMENT, 'call is null')
+          throw new ConnectError('call is null', Code.InvalidArgument)
         }
         const tx = data.transaction
 
@@ -201,7 +210,7 @@ export class AptosTransactionProcessor<T extends GeneralTransactionResponse, CT
     },
     handleOptions?: HandlerOptions<MoveFetchConfig, UserTransactionResponse>
   ): this {
-    const _fetchConfig = MoveFetchConfig.fromPartial({ ...DEFAULT_FETCH_CONFIG, ...handleOptions })
+    const _fetchConfig = create(MoveFetchConfigSchema, { ...DEFAULT_FETCH_CONFIG, ...handleOptions })
 
     const processor = this
     const filter: FunctionNameAndCallFilter = { function: '', includeFailed: transactionFilter?.includeFailed }
@@ -216,7 +225,7 @@ export class AptosTransactionProcessor<T extends GeneralTransactionResponse, CT
       handlerName: getHandlerName(),
       handler: async function (data) {
         if (!data.rawTransaction) {
-          throw new ServerError(Status.INVALID_ARGUMENT, 'call is null')
+          throw new ConnectError('call is null', Code.InvalidArgument)
         }
         const call = data.transaction
         const ctx = new AptosContext(
@@ -287,9 +296,8 @@ export class AptosTransactionProcessor<T extends GeneralTransactionResponse, CT
       handlerName: getHandlerName(),
       handler: async function (data) {
         if (!data.rawResources || !data.version) {
-          throw new ServerError(Status.INVALID_ARGUMENT, 'resource is null')
+          throw new ConnectError('resource is null', Code.InvalidArgument)
         }
-        const aptResource = new AptResource(data)
         const timestamp = Number(data.timestampMicros)
         const ctx = new AptosResourcesContext(
           processor.config.network,
@@ -298,7 +306,7 @@ export class AptosTransactionProcessor<T extends GeneralTransactionResponse, CT
           timestamp,
           processor.config.baseLabels
         )
-        let resources = await aptResource.decodeResources<T>(processor.coder)
+        let resources = await data.decodeResources<T>(processor.coder)
 
         if (hasAny) {
           resources = resources.filter((r) => {
@@ -337,12 +345,12 @@ export class AptosTransactionProcessor<T extends GeneralTransactionResponse, CT
       handlerName: getHandlerName(),
       handler: async function (data) {
         if (!data.rawTransaction) {
-          throw new ServerError(Status.INVALID_ARGUMENT, 'call is null')
+          throw new ConnectError('call is null', Code.InvalidArgument)
         }
         const transaction = JSON.parse(data.rawTransaction) as T
         const timestampMicros = BigInt(transaction.timestamp)
         if (timestampMicros > Number.MAX_SAFE_INTEGER) {
-          throw new ServerError(Status.INVALID_ARGUMENT, 'timestamp is too large')
+          throw new ConnectError('timestamp is too large', Code.InvalidArgument)
         }
 
         const ctx = new AptosTransactionContext(
@@ -381,10 +389,10 @@ export class AptosTransactionProcessor<T extends GeneralTransactionResponse, CT
   ): this {
     return this.onInterval(
       handler,
-      {
+      create(HandleIntervalSchema, {
         recentInterval: timeIntervalInMinutes,
         backfillInterval: backfillTimeIntervalInMinutes
-      },
+      }),
       undefined,
       handlerOptions
     )
@@ -399,7 +407,7 @@ export class AptosTransactionProcessor<T extends GeneralTransactionResponse, CT
     return this.onInterval(
       handler,
       undefined,
-      { recentInterval: versionInterval, backfillInterval: backfillVersionInterval },
+      create(HandleIntervalSchema, { recentInterval: versionInterval, backfillInterval: backfillVersionInterval }),
       handlerOptions
     )
   }
@@ -521,7 +529,7 @@ export class AptosResourcesProcessor {
       handlerName,
       handler: async function (data) {
         if (data.timestampMicros > Number.MAX_SAFE_INTEGER) {
-          throw new ServerError(Status.INVALID_ARGUMENT, 'timestamp is too large')
+          throw new ConnectError('timestamp is too large', Code.InvalidArgument)
         }
         const aptResource = new AptResource(data)
         const timestamp = Number(data.timestampMicros)
@@ -562,10 +570,10 @@ export class AptosResourcesProcessor {
   ): this {
     return this.onInterval(
       handler,
-      {
+      create(HandleIntervalSchema, {
         recentInterval: timeIntervalInMinutes,
         backfillInterval: backfillTimeIntervalInMinutes
-      },
+      }),
       undefined,
       type,
       handlerOptions
@@ -582,7 +590,7 @@ export class AptosResourcesProcessor {
     return this.onInterval(
       handler,
       undefined,
-      { recentInterval: versionInterval, backfillInterval: backfillVersionInterval },
+      create(HandleIntervalSchema, { recentInterval: versionInterval, backfillInterval: backfillVersionInterval }),
       typePrefix,
       handlerOptions
     )
@@ -622,7 +630,7 @@ export class AptosResourcesProcessor {
         const timestamp = Number(data.timestampMicros)
 
         if (!data.rawResources || !data.version) {
-          throw new ServerError(Status.INVALID_ARGUMENT, 'resource is null')
+          throw new ConnectError('resource is null', Code.InvalidArgument)
         }
         const aptResource = new AptResource(data)
         const ctx = new AptosResourcesContext(

package/src/aptos/aptos-resource-processor-template.ts

@@ -2,7 +2,13 @@ import { ListStateStorage, processMetrics } from '@sentio/runtime'
 import { AptosResourcesContext } from './context.js'
 import { AptosBindOptions } from './network.js'
 import { AptosResourcesProcessor, DEFAULT_RESOURCE_FETCH_CONFIG } from './aptos-processor.js'
-import { HandleInterval, MoveAccountFetchConfig, TemplateInstance } from '@sentio/protos'
+import {
+  type HandleInterval,
+  HandleIntervalSchema,
+  type MoveAccountFetchConfig,
+  TemplateInstanceSchema
+} from '@sentio/protos'
+import { create } from '@bufbuild/protobuf'
 import { MoveResource } from '@aptos-labs/ts-sdk'
 import { PromiseOrVoid } from '../core/index.js'
 import { getHandlerName, proxyProcessor } from '../utils/metrics.js'
@@ -40,7 +46,7 @@ export class AptosResourceProcessorTemplate {
     options.network = options.network || ctx.network
     options.startVersion = options.startVersion || ctx.version
 
-    const instance: TemplateInstance = {
+    const instance = create(TemplateInstanceSchema, {
       templateId: this.id,
       contract: {
         name: '',
@@ -51,7 +57,7 @@ export class AptosResourceProcessorTemplate {
       startBlock: options.startVersion ? BigInt(options.startVersion) : 0n,
       endBlock: options.endVersion ? BigInt(options.endVersion) : 0n,
       baseLabels: options.baseLabels
-    }
+    })
 
     ctx.sendTemplateInstance(instance)
 
@@ -119,10 +125,10 @@ export class AptosResourceProcessorTemplate {
   ): this {
     return this.onInterval(
       handler,
-      {
+      create(HandleIntervalSchema, {
         recentInterval: timeIntervalInMinutes,
         backfillInterval: backfillTimeIntervalInMinutes
-      },
+      }),
       undefined,
       type,
       handlerOptions
@@ -139,7 +145,10 @@ export class AptosResourceProcessorTemplate {
     return this.onInterval(
       handler,
       undefined,
-      { recentInterval: checkpointInterval, backfillInterval: backfillCheckpointInterval },
+      create(HandleIntervalSchema, {
+        recentInterval: checkpointInterval,
+        backfillInterval: backfillCheckpointInterval
+      }),
       type,
       handlerOptions
     )

package/src/aptos/context.ts

@@ -1,4 +1,5 @@
-import { RecordMetaData } from '@sentio/protos'
+import { type RecordMetaData, RecordMetaDataSchema } from '@sentio/protos'
+import { create } from '@bufbuild/protobuf'
 import { type Labels, normalizeLabels } from '../index.js'
 import { MoveCoder, RichAptosClientWithContext } from './index.js'
 import {
@@ -12,7 +13,7 @@ import {
 import { defaultMoveCoder } from './move-coder.js'
 import { AptosNetwork } from './network.js'
 import { Endpoints } from '@sentio/runtime'
-import { ServerError, Status } from 'nice-grpc'
+import { ConnectError, Code } from '@connectrpc/connect'
 import { MoveContext } from '../move/index.js'
 import { GeneralTransactionResponse } from './models.js'
 
@@ -30,7 +31,7 @@ export abstract class AptosBaseContext extends MoveContext<AptosNetwork, MoveMod
   getClient(): RichAptosClientWithContext {
     const chainServer = Endpoints.INSTANCE.chainServer.get(this.network)
     if (!chainServer) {
-      throw new ServerError(Status.INTERNAL, 'RPC endpoint not provided')
+      throw new ConnectError('RPC endpoint not provided', Code.Internal)
     }
     const fullnode = chainServer + '/v1'
 
@@ -86,7 +87,7 @@ export class AptosTransactionContext<T extends GeneralTransactionResponse> exten
   }
 
   getMetaDataInternal(name: string, labels: Labels): RecordMetaData {
-    return {
+    return create(RecordMetaDataSchema, {
       address: this.address,
       contractName: this.moduleName,
       blockNumber: this.version,
@@ -96,7 +97,7 @@ export class AptosTransactionContext<T extends GeneralTransactionResponse> exten
       chainId: this.getChainId(),
       name: name,
       labels: normalizeLabels(labels)
-    }
+    })
   }
 }
 
@@ -120,7 +121,7 @@ export class AptosResourcesContext extends AptosBaseContext {
   }
 
   getMetaDataInternal(name: string, labels: Labels): RecordMetaData {
-    return {
+    return create(RecordMetaDataSchema, {
       address: this.address,
       contractName: 'resources',
       blockNumber: this.version,
@@ -130,6 +131,6 @@ export class AptosResourcesContext extends AptosBaseContext {
       chainId: this.getChainId(),
       name: name,
       labels: normalizeLabels(labels)
-    }
+    })
   }
 }