@sentio/sdk 3.9.0-rc.9 → 4.0.0-rc.10

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

@@ -1,13 +1,18 @@
 import {
-  Data_SuiCall,
-  Data_SuiObject,
-  Data_SuiObjectChange,
-  HandleInterval,
-  MoveAccountFetchConfig,
-  MoveFetchConfig,
+  type Data_SuiCall,
+  type Data_SuiObject,
+  type Data_SuiObjectChange,
+  type HandleInterval,
+  HandleIntervalSchema,
+  type MoveAccountFetchConfig,
+  MoveAccountFetchConfigSchema,
+  type MoveFetchConfig,
+  MoveFetchConfigSchema,
   MoveOwnerType,
-  ProcessResult
+  type ProcessResult,
+  timestampDate
 } from '@sentio/protos'
+import { create } from '@bufbuild/protobuf'
 import { ListStateStorage } from '@sentio/runtime'
 import { SuiNetwork } from './network.js'
 import { SuiAddressContext, SuiContext, SuiObjectChangeContext, SuiObjectContext } from './context.js'
@@ -17,9 +22,10 @@ import type { SuiMoveObjectInput } from '@typemove/sui'
 import { ALL_ADDRESS, PromiseOrVoid } from '../core/index.js'
 import { configure, DEFAULT_FETCH_CONFIG, IndexConfigure, SuiBindOptions } from './sui-processor.js'
 import { CallHandler, TransactionFilter, accountTypeString, ObjectChangeHandler } from '../move/index.js'
-import { ServerError, Status } from 'nice-grpc'
+import { ConnectError, Code } from '@connectrpc/connect'
 import { TypeDescriptor } from '@typemove/move'
 import { TypedSuiMoveObject } from './models.js'
+import { toSuiClientObject } from './to-client-types.js'
 import { getHandlerName, proxyProcessor } from '../utils/metrics.js'
 
 export interface SuiObjectBindOptions {
@@ -47,9 +53,9 @@ interface ObjectHandler {
   handlerName: string
 }
 
-export const DEFAULT_ACCOUNT_FETCH_CONFIG: MoveAccountFetchConfig = {
+export const DEFAULT_ACCOUNT_FETCH_CONFIG: MoveAccountFetchConfig = create(MoveAccountFetchConfigSchema, {
   owned: false
-}
+})
 
 export class SuiAccountProcessorState extends ListStateStorage<SuiBaseObjectOrAddressProcessor<any>> {
   static INSTANCE = new SuiAccountProcessorState()
@@ -114,7 +120,7 @@ export abstract class SuiBaseObjectOrAddressProcessor<HandlerType> {
           data.objectId,
           data.objectVersion,
           data.slot,
-          data.timestamp || new Date(0),
+          data.timestamp ? timestampDate(data.timestamp) : new Date(0),
           processor.config.baseLabels
         )
         await processor.doHandle(handler, data, ctx)
@@ -141,10 +147,10 @@ abstract class SuiBaseObjectOrAddressProcessorInternal<
   ): this {
     return this.onInterval(
       handler,
-      {
+      create(HandleIntervalSchema, {
         recentInterval: timeIntervalInMinutes,
         backfillInterval: backfillTimeIntervalInMinutes
-      },
+      }),
       undefined,
       type,
       fetchConfig
@@ -161,7 +167,10 @@ abstract class SuiBaseObjectOrAddressProcessorInternal<
     return this.onInterval(
       handler,
       undefined,
-      { recentInterval: checkpointInterval, backfillInterval: backfillCheckpointInterval },
+      create(HandleIntervalSchema, {
+        recentInterval: checkpointInterval,
+        backfillInterval: backfillCheckpointInterval
+      }),
       type,
       fetchConfig
     )
@@ -182,7 +191,10 @@ export class SuiAddressProcessor extends SuiBaseObjectOrAddressProcessorInternal
     data: Data_SuiObject,
     ctx: SuiObjectContext
   ) {
-    return handler(data.rawObjects.map((o) => JSON.parse(o)) as SuiMoveObjectInput[], ctx)
+    return handler(
+      data.rawObjects.map((o) => toSuiClientObject(JSON.parse(o))),
+      ctx
+    )
   }
 
   onTransactionBlock(
@@ -190,7 +202,7 @@ export class SuiAddressProcessor extends SuiBaseObjectOrAddressProcessorInternal
     filter?: TransactionFilter,
     fetchConfig?: Partial<MoveFetchConfig>
   ) {
-    const _fetchConfig = MoveFetchConfig.fromPartial({ ...DEFAULT_FETCH_CONFIG, ...fetchConfig })
+    const _fetchConfig = create(MoveFetchConfigSchema, { ...DEFAULT_FETCH_CONFIG, ...fetchConfig })
     const _filter: TransactionFilter = {
       fromAndToAddress: {
         from: '',
@@ -205,7 +217,7 @@ export class SuiAddressProcessor extends SuiBaseObjectOrAddressProcessorInternal
       handlerName: getHandlerName(),
       handler: async function (data) {
         if (!data.rawTransaction) {
-          throw new ServerError(Status.INVALID_ARGUMENT, 'transaction is null')
+          throw new ConnectError('transaction is null', Code.InvalidArgument)
         }
         const tx = JSON.parse(data.rawTransaction) as GrpcTypes.ExecutedTransaction
 
@@ -213,7 +225,7 @@ export class SuiAddressProcessor extends SuiBaseObjectOrAddressProcessorInternal
           'object',
           processor.config.network,
           processor.config.address,
-          data.timestamp || new Date(0),
+          data.timestamp ? timestampDate(data.timestamp) : new Date(0),
           data.slot,
           tx,
           0,
@@ -259,8 +271,8 @@ export class SuiObjectProcessor extends SuiBaseObjectOrAddressProcessorInternal<
       return
     }
     return handler(
-      JSON.parse(data.rawSelf) as SuiMoveObjectInput,
-      data.rawObjects.map((o) => JSON.parse(o)) as SuiMoveObjectInput[],
+      toSuiClientObject(JSON.parse(data.rawSelf)),
+      data.rawObjects.map((o) => toSuiClientObject(JSON.parse(o))),
       ctx
     )
   }
@@ -298,14 +310,18 @@ export class SuiObjectTypeProcessor<T> extends SuiBaseObjectOrAddressProcessor<
       return
     }
     const object = await ctx.coder.filterAndDecodeObjects(this.objectType, [
-      JSON.parse(data.rawSelf) as SuiMoveObjectInput
+      toSuiClientObject(JSON.parse(data.rawSelf))
     ])
-    return handler(object[0], data.rawObjects.map((o) => JSON.parse(o)) as SuiMoveObjectInput[], ctx)
+    return handler(
+      object[0],
+      data.rawObjects.map((o) => toSuiClientObject(JSON.parse(o))),
+      ctx
+    )
   }
 
   public onObjectChange(handler: (changes: SuiObjectChange[], ctx: SuiObjectChangeContext) => PromiseOrVoid): this {
     if (this.config.network === SuiNetwork.TEST_NET) {
-      throw new ServerError(Status.INVALID_ARGUMENT, 'object change not supported in testnet')
+      throw new ConnectError('object change not supported in testnet', Code.InvalidArgument)
     }
     const processor = this
     this.objectChangeHandlers.push({
@@ -314,7 +330,7 @@ export class SuiObjectTypeProcessor<T> extends SuiBaseObjectOrAddressProcessor<
         const ctx = new SuiObjectChangeContext(
           processor.config.network,
           processor.config.address,
-          data.timestamp || new Date(0),
+          data.timestamp ? timestampDate(data.timestamp) : new Date(0),
           data.slot,
           data.txDigest,
           processor.config.baseLabels
@@ -340,10 +356,10 @@ export class SuiObjectTypeProcessor<T> extends SuiBaseObjectOrAddressProcessor<
   ): this {
     return this.onInterval(
       handler,
-      {
+      create(HandleIntervalSchema, {
         recentInterval: timeIntervalInMinutes,
         backfillInterval: backfillTimeIntervalInMinutes
-      },
+      }),
       undefined,
       this.objectType.getSignature(),
       fetchConfig
@@ -363,7 +379,10 @@ export class SuiObjectTypeProcessor<T> extends SuiBaseObjectOrAddressProcessor<
     return this.onInterval(
       handler,
       undefined,
-      { recentInterval: checkpointInterval, backfillInterval: backfillCheckpointInterval },
+      create(HandleIntervalSchema, {
+        recentInterval: checkpointInterval,
+        backfillInterval: backfillCheckpointInterval
+      }),
       this.objectType.qname,
       fetchConfig
     )
@@ -389,6 +408,9 @@ export class SuiWrappedObjectProcessor extends SuiBaseObjectOrAddressProcessorIn
     data: Data_SuiObject,
     ctx: SuiObjectContext
   ) {
-    return handler(data.rawObjects.map((o) => JSON.parse(o)) as SuiMoveObjectInput[], ctx)
+    return handler(
+      data.rawObjects.map((o) => toSuiClientObject(JSON.parse(o))),
+      ctx
+    )
   }
 }

package/src/sui/sui-plugin-part.ts

@@ -1,15 +1,20 @@
 import { USER_PROCESSOR } from '@sentio/runtime'
 import {
-  AccountConfig,
-  ContractConfig,
+  AccountConfigSchema,
+  ContractConfigSchema,
   HandlerType,
-  InitResponse,
-  MoveCallHandlerConfig,
-  MoveEventHandlerConfig,
-  MoveResourceChangeConfig,
-  ProcessConfigResponse,
-  StartRequest
+  type InitResponse,
+  type MoveCallHandlerConfig,
+  MoveCallHandlerConfigSchema,
+  type MoveEventHandlerConfig,
+  MoveEventHandlerConfigSchema,
+  MoveOnIntervalConfigSchema,
+  type MoveResourceChangeConfig,
+  MoveResourceChangeConfigSchema,
+  type ProcessConfigResponse,
+  type StartRequest
 } from '@sentio/protos'
+import { create } from '@bufbuild/protobuf'
 
 import { PartitionHandlerManager } from '../core/index.js'
 import { HandlerRegister } from '../core/handler-register.js'
@@ -75,7 +80,7 @@ export class SuiPluginPart {
       if (forChainId !== undefined && forChainId !== chainId.toString()) {
         continue
       }
-      const contractConfig = ContractConfig.fromPartial({
+      const contractConfig = create(ContractConfigSchema, {
         transactionConfig: [],
         processorType: USER_PROCESSOR,
         contract: {
@@ -90,7 +95,7 @@ export class SuiPluginPart {
       for (const handler of suiProcessor.eventHandlers) {
         const handlerId = this.handlerRegister.register(handler.handler, chainId)
         this.partitionManager.registerPartitionHandler(HandlerType.SUI_EVENT, handlerId, handler.partitionHandler)
-        const eventHandlerConfig: MoveEventHandlerConfig = {
+        const eventHandlerConfig: MoveEventHandlerConfig = create(MoveEventHandlerConfigSchema, {
           filters: handler.filters.map((f) => {
             return {
               type: f.type,
@@ -101,13 +106,13 @@ export class SuiPluginPart {
           fetchConfig: handler.fetchConfig,
           handlerId,
           handlerName: handler.handlerName
-        }
+        })
         contractConfig.moveEventConfigs.push(eventHandlerConfig)
       }
       for (const handler of suiProcessor.callHandlers) {
         const handlerId = this.handlerRegister.register(handler.handler, chainId)
         this.partitionManager.registerPartitionHandler(HandlerType.SUI_CALL, handlerId, handler.partitionHandler)
-        const functionHandlerConfig: MoveCallHandlerConfig = {
+        const functionHandlerConfig: MoveCallHandlerConfig = create(MoveCallHandlerConfigSchema, {
           filters: handler.filters.map((filter) => {
             return {
               function: filter.function,
@@ -121,18 +126,18 @@ export class SuiPluginPart {
           fetchConfig: handler.fetchConfig,
           handlerId,
           handlerName: handler.handlerName
-        }
+        })
         contractConfig.moveCallConfigs.push(functionHandlerConfig)
       }
       // deprecated, use objectType processor instead
       for (const handler of suiProcessor.objectChangeHandlers) {
         const handlerId = this.handlerRegister.register(handler.handler, chainId)
-        const objectChangeHandler: MoveResourceChangeConfig = {
+        const objectChangeHandler: MoveResourceChangeConfig = create(MoveResourceChangeConfigSchema, {
           types: typeof handler.type === 'string' ? [handler.type] : handler.type,
           handlerId,
           handlerName: handler.handlerName,
           includeDeleted: false
-        }
+        })
         contractConfig.moveResourceChangeConfigs.push(objectChangeHandler)
       }
       config.contractConfigs.push(contractConfig)
@@ -143,7 +148,7 @@ export class SuiPluginPart {
       if (forChainId !== undefined && forChainId !== chainId.toString()) {
         continue
       }
-      const accountConfig = AccountConfig.fromPartial({
+      const accountConfig = create(AccountConfigSchema, {
         address: processor.config.address,
         chainId: processor.getChainId(),
         startBlock: processor.config.startCheckpoint, // TODO maybe use another field
@@ -152,39 +157,41 @@ export class SuiPluginPart {
 
       for (const handler of processor.objectChangeHandlers) {
         const handlerId = this.handlerRegister.register(handler.handler, chainId)
-        const objectChangeHandler: MoveResourceChangeConfig = {
+        const objectChangeHandler: MoveResourceChangeConfig = create(MoveResourceChangeConfigSchema, {
           types: typeof handler.type === 'string' ? [handler.type] : handler.type,
           handlerId,
           handlerName: handler.handlerName,
           includeDeleted: false
-        }
+        })
         accountConfig.moveResourceChangeConfigs.push(objectChangeHandler)
       }
 
       for (const handler of processor.objectHandlers) {
         const handlerId = this.handlerRegister.register(handler.handler, chainId)
 
-        accountConfig.moveIntervalConfigs.push({
-          intervalConfig: {
-            handlerId: handlerId,
-            handlerName: handler.handlerName,
-            minutes: 0,
-            minutesInterval: handler.timeIntervalInMinutes,
-            slot: 0,
-            slotInterval: handler.checkPointInterval,
+        accountConfig.moveIntervalConfigs.push(
+          create(MoveOnIntervalConfigSchema, {
+            intervalConfig: {
+              handlerId: handlerId,
+              handlerName: handler.handlerName,
+              minutes: 0,
+              minutesInterval: handler.timeIntervalInMinutes,
+              slot: 0,
+              slotInterval: handler.checkPointInterval,
+              fetchConfig: undefined
+            },
+            type: handler.type || '',
+            ownerType: processor.ownerType,
+            resourceFetchConfig: handler.fetchConfig,
             fetchConfig: undefined
-          },
-          type: handler.type || '',
-          ownerType: processor.ownerType,
-          resourceFetchConfig: handler.fetchConfig,
-          fetchConfig: undefined
-        })
+          })
+        )
       }
 
       if (processor instanceof SuiAddressProcessor) {
         for (const handler of processor.callHandlers) {
           const handlerId = this.handlerRegister.register(handler.handler, chainId)
-          const functionHandlerConfig: MoveCallHandlerConfig = {
+          const functionHandlerConfig: MoveCallHandlerConfig = create(MoveCallHandlerConfigSchema, {
             filters: handler.filters.map((filter) => {
               return {
                 function: filter.function,
@@ -198,7 +205,7 @@ export class SuiPluginPart {
             fetchConfig: handler.fetchConfig,
             handlerId,
             handlerName: handler.handlerName
-          }
+          })
           accountConfig.moveCallConfigs.push(functionHandlerConfig)
         }
       }

package/src/sui/sui-plugin.ts

@@ -1,15 +1,17 @@
 import { errorString, mergeProcessResults, Plugin, PluginManager } from '@sentio/runtime'
 import {
-  DataBinding,
+  type DataBinding,
   HandlerType,
-  InitResponse,
-  ProcessConfigResponse,
-  ProcessResult,
-  ProcessStreamResponse_Partitions,
-  StartRequest
+  type InitResponse,
+  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 { PartitionHandlerManager } from '../core/index.js'
 import { HandlerRegister } from '../core/handler-register.js'
 
@@ -60,7 +62,7 @@ export class SuiPlugin extends Plugin {
       case HandlerType.SUI_OBJECT_CHANGE:
         return this.processSuiObjectChange(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)
     }
   }
 
@@ -68,48 +70,48 @@ export class SuiPlugin extends Plugin {
     let data: any
     switch (request.handlerType) {
       case HandlerType.SUI_EVENT:
-        if (!request.data?.suiEvent) {
-          throw new ServerError(Status.INVALID_ARGUMENT, "suiEvent can't be empty")
+        if (request.data?.value.case !== 'suiEvent') {
+          throw new ConnectError("suiEvent can't be empty", Code.InvalidArgument)
         }
-        data = request.data.suiEvent
+        data = request.data.value.value
         break
       case HandlerType.SUI_CALL:
-        if (!request.data?.suiCall) {
-          throw new ServerError(Status.INVALID_ARGUMENT, "suiCall can't be empty")
+        if (request.data?.value.case !== 'suiCall') {
+          throw new ConnectError("suiCall can't be empty", Code.InvalidArgument)
         }
-        data = request.data.suiCall
+        data = request.data.value.value
         break
       case HandlerType.SUI_OBJECT:
-        if (!request.data?.suiObject) {
-          throw new ServerError(Status.INVALID_ARGUMENT, "suiObject can't be empty")
+        if (request.data?.value.case !== 'suiObject') {
+          throw new ConnectError("suiObject can't be empty", Code.InvalidArgument)
         }
-        data = request.data.suiObject
+        data = request.data.value.value
         break
       case HandlerType.SUI_OBJECT_CHANGE:
-        if (!request.data?.suiObjectChange) {
-          throw new ServerError(Status.INVALID_ARGUMENT, "suiObjectChange can't be empty")
+        if (request.data?.value.case !== 'suiObjectChange') {
+          throw new ConnectError("suiObjectChange can't be empty", Code.InvalidArgument)
         }
-        data = request.data.suiObjectChange
+        data = 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 processSuiEvent(binding: DataBinding): Promise<ProcessResult> {
-    if (!binding.data?.suiEvent) {
-      throw new ServerError(Status.INVALID_ARGUMENT, "Event can't be empty")
+    if (binding.data?.value.case !== 'suiEvent') {
+      throw new ConnectError("Event can't be empty", Code.InvalidArgument)
     }
     const promises: Promise<ProcessResult>[] = []
-    const event = binding.data.suiEvent
+    const event = binding.data.value.value
 
     for (const handlerId of binding.handlerIds) {
       promises.push(
@@ -119,9 +121,9 @@ export class SuiPlugin 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
             )
           })
       )
@@ -130,10 +132,10 @@ export class SuiPlugin extends Plugin {
   }
 
   async processSuiFunctionCall(binding: DataBinding): Promise<ProcessResult> {
-    if (!binding.data?.suiCall) {
-      throw new ServerError(Status.INVALID_ARGUMENT, "Call can't be empty")
+    if (binding.data?.value.case !== 'suiCall') {
+      throw new ConnectError("Call can't be empty", Code.InvalidArgument)
     }
-    const call = binding.data.suiCall
+    const call = binding.data.value.value
 
     const promises: Promise<ProcessResult>[] = []
     for (const handlerId of binding.handlerIds) {
@@ -143,9 +145,9 @@ export class SuiPlugin 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
           )
         })
       promises.push(promise)
@@ -154,10 +156,10 @@ export class SuiPlugin extends Plugin {
   }
 
   async processSuiObject(binding: DataBinding): Promise<ProcessResult> {
-    if (!binding.data?.suiObject) {
-      throw new ServerError(Status.INVALID_ARGUMENT, "Object can't be empty")
+    if (binding.data?.value.case !== 'suiObject') {
+      throw new ConnectError("Object can't be empty", Code.InvalidArgument)
     }
-    const object = binding.data.suiObject
+    const object = binding.data.value.value
 
     const promises: Promise<ProcessResult>[] = []
     for (const handlerId of binding.handlerIds) {
@@ -168,9 +170,9 @@ export class SuiPlugin extends Plugin {
             handlerId
           )(object)
           .catch((e: any) => {
-            throw new ServerError(
-              Status.INTERNAL,
-              'error processing object: ' + JSON.stringify(object) + '\n' + errorString(e)
+            throw new ConnectError(
+              'error processing object: ' + JSON.stringify(object) + '\n' + errorString(e),
+              Code.Internal
             )
           })
       )
@@ -179,10 +181,10 @@ export class SuiPlugin extends Plugin {
   }
 
   async processSuiObjectChange(binding: DataBinding): Promise<ProcessResult> {
-    if (!binding.data?.suiObjectChange) {
-      throw new ServerError(Status.INVALID_ARGUMENT, "Object change can't be empty")
+    if (binding.data?.value.case !== 'suiObjectChange') {
+      throw new ConnectError("Object change can't be empty", Code.InvalidArgument)
     }
-    const objectChange = binding.data.suiObjectChange
+    const objectChange = binding.data.value.value
 
     const promises: Promise<ProcessResult>[] = []
     for (const handlerId of binding.handlerIds) {
@@ -193,9 +195,9 @@ export class SuiPlugin extends Plugin {
             handlerId
           )(objectChange)
           .catch((e: any) => {
-            throw new ServerError(
-              Status.INTERNAL,
-              'error processing object change: ' + JSON.stringify(objectChange) + '\n' + errorString(e)
+            throw new ConnectError(
+              'error processing object change: ' + JSON.stringify(objectChange) + '\n' + errorString(e),
+              Code.Internal
             )
           })
       )