@sentio/sdk 2.59.0-rc.9 → 2.59.1-rc.1

package/src/store/schema.ts

@@ -6,7 +6,7 @@ const customScalars = ['BigInt', 'BigDecimal', 'Timestamp', 'JSON', 'Bytes', 'ID
 
 const baseSchema = buildASTSchema(
   parse(`
-    directive @entity(immutable: Boolean! = false, sparse: Boolean! = false) on OBJECT
+    directive @entity(immutable: Boolean! = false, sparse: Boolean! = false, timeseries: Boolean! = false) on OBJECT
     directive @query on INTERFACE
     directive @derivedFrom(field: String!) on FIELD_DEFINITION
     directive @unique on FIELD_DEFINITION
@@ -16,6 +16,8 @@ const baseSchema = buildASTSchema(
     directive @byteWeight(value: Float!) on FIELD_DEFINITION
     directive @variant on OBJECT # legacy
     directive @jsonField on OBJECT # legacy
+    directive @aggregation(intervals: [String!]! = ["hour"], source: String!) on OBJECT
+    directive @aggregate(fn: String!, arg: String, cumulative: Boolean = false) on FIELD_DEFINITION
     ${customScalars.map((name) => 'scalar ' + name).join('\n')}
 `)
 )

package/src/store/store.ts

@@ -1,7 +1,24 @@
 import { DatabaseSchema } from '../core/index.js'
 import { BigDecimal } from '@sentio/bigdecimal'
-import { AbstractEntity, AbstractEntity as Entity, Bytes, Float, ID, Int, Timestamp } from './types.js'
-import type { DBRequest, Entity as EntityStruct, RichValue } from '@sentio/protos'
+import {
+  AbstractEntity,
+  AbstractEntity as Entity,
+  Bytes,
+  Float,
+  ID,
+  Int,
+  Timestamp,
+  AddOp,
+  MultiplyOp,
+  UpdateValues
+} from './types.js'
+import {
+  DBRequest,
+  DBRequest_DBUpdate,
+  Entity as EntityStruct,
+  EntityUpdateData_Operator,
+  RichValue
+} from '@sentio/protos'
 import { DBRequest_DBOperator, DBResponse } from '@sentio/protos'
 import { IStoreContext, PluginManager } from '@sentio/runtime'
 import { Cursor } from './cursor.js'
@@ -102,6 +119,42 @@ export class Store {
     })
   }
 
+  async update<T extends Entity>(entity: EntityClass<T>, values: UpdateValues<any>): Promise<void> {
+    if (values.id) {
+      const update: DBRequest_DBUpdate = {
+        entity: [getEntityName(entity)],
+        id: [values.id.toString()],
+        entityData: [{ fields: {} }]
+      }
+      for (const [key, value] of Object.entries(values)) {
+        if (key !== 'id') {
+          const field = getEntityField(entity, key)
+          if (value instanceof AddOp) {
+            update.entityData[0].fields[field] = {
+              op: EntityUpdateData_Operator.ADD,
+              value: serializeRichValue(value.value)
+            }
+          } else if (value instanceof MultiplyOp) {
+            update.entityData[0].fields[field] = {
+              op: EntityUpdateData_Operator.MULTIPLY,
+              value: serializeRichValue(value.value)
+            }
+          } else if (value !== undefined) {
+            update.entityData[0].fields[field] = {
+              op: EntityUpdateData_Operator.SET,
+              value: serializeRichValue(value)
+            }
+          }
+        }
+      }
+      await this.context.sendRequest({
+        update
+      })
+    } else {
+      throw new Error('Update must have id field')
+    }
+  }
+
   async upsert<T extends Entity>(entity: T | T[]): Promise<void> {
     const entities = Array.isArray(entity) ? entity : [entity]
     const request = {
@@ -250,7 +303,7 @@ export type Operators<T> =
         : T extends Bytes
           ? '=' | '!=' | 'in' | 'not in'
           : T extends ID
-            ? '=' | '!=' | 'like' | 'not like' | 'in' | 'not in'
+            ? '=' | '!=' | 'like' | 'not like' | 'in' | 'not in' | '<' | '<=' | '>' | '>='
             : T extends string
               ? '=' | '!=' | 'like' | 'not like' | 'in' | 'not in'
               : T extends Timestamp
@@ -321,5 +374,5 @@ export function getStore() {
   if (dbContext) {
     return new Store(dbContext)
   }
-  return undefined
+  throw new Error('Store not found in context, please ensure you are calling this in a handler function')
 }

package/src/store/types.ts

@@ -2,8 +2,9 @@ import { RichStruct, RichValue } from '@sentio/protos'
 import { BigDecimalConverter, BigIntConverter } from './convert.js'
 import { getEntityName, Store } from './store.js'
 import { PluginManager } from '@sentio/runtime'
+import { BigDecimal } from '@sentio/bigdecimal'
 
-export type ID = string | Uint8Array
+export type ID = string | Uint8Array | Int8
 export type String = string
 export type Int = number
 export type Int8 = bigint
@@ -13,6 +14,34 @@ export type Timestamp = Date
 export type Bytes = Uint8Array
 export type BigInt = bigint
 
+export type ValueType = String | Int | Int8 | Float | Boolean | Timestamp | Bytes | BigInt | BigDecimal | null
+
+export abstract class UpdateOp<T> {}
+
+export class AddOp<T> extends UpdateOp<T> {
+  constructor(readonly value: T) {
+    super()
+  }
+}
+
+export class MultiplyOp<T> extends UpdateOp<T> {
+  constructor(readonly value: T) {
+    super()
+  }
+}
+
+export type UpdateValues<T> = {
+  [K in keyof T]?: T[K] | UpdateOp<T[K]>
+} & { id: ID }
+
+export function add<K extends ValueType>(value: K): UpdateOp<K> {
+  return new AddOp<K>(value)
+}
+
+export function multiply<K extends ValueType>(value: K): UpdateOp<K> {
+  return new MultiplyOp<K>(value)
+}
+
 export abstract class AbstractEntity {
   abstract id: ID
   private readonly _data: RichStruct

package/src/sui/builtin/0x1.ts

@@ -36,6 +36,7 @@ import {
 } from "@typemove/sui";
 
 import { CallFilter, MoveFetchConfig, EventFilter } from "@sentio/sdk/move";
+import { HandlerOptions } from "@sentio/sdk";
 import {
   SuiBindOptions,
   SuiBaseProcessor,
@@ -722,13 +723,16 @@ export class bit_vector extends SuiBaseProcessor {
 
   onEventBitVector(
     func: (event: bit_vector.BitVectorInstance, ctx: SuiContext) => void,
-    fetchConfig?: Partial<MoveFetchConfig>,
+    handlerOptions?: HandlerOptions<
+      MoveFetchConfig,
+      bit_vector.BitVectorInstance
+    >,
     eventFilter?: Omit<EventFilter, "type" | "account">,
   ): bit_vector {
     this.onMoveEvent(
       func,
       { ...(eventFilter ?? {}), type: "bit_vector::BitVector" },
-      fetchConfig,
+      handlerOptions,
     );
     return this;
   }
@@ -1066,13 +1070,16 @@ export class fixed_point32 extends SuiBaseProcessor {
 
   onEventFixedPoint32(
     func: (event: fixed_point32.FixedPoint32Instance, ctx: SuiContext) => void,
-    fetchConfig?: Partial<MoveFetchConfig>,
+    handlerOptions?: HandlerOptions<
+      MoveFetchConfig,
+      fixed_point32.FixedPoint32Instance
+    >,
     eventFilter?: Omit<EventFilter, "type" | "account">,
   ): fixed_point32 {
     this.onMoveEvent(
       func,
       { ...(eventFilter ?? {}), type: "fixed_point32::FixedPoint32" },
-      fetchConfig,
+      handlerOptions,
     );
     return this;
   }
@@ -2003,13 +2010,13 @@ export class string$ extends SuiBaseProcessor {
 
   onEventString(
     func: (event: string$.StringInstance, ctx: SuiContext) => void,
-    fetchConfig?: Partial<MoveFetchConfig>,
+    handlerOptions?: HandlerOptions<MoveFetchConfig, string$.StringInstance>,
     eventFilter?: Omit<EventFilter, "type" | "account">,
   ): string$ {
     this.onMoveEvent(
       func,
       { ...(eventFilter ?? {}), type: "string::String" },
-      fetchConfig,
+      handlerOptions,
     );
     return this;
   }
@@ -2509,13 +2516,16 @@ export class type_name extends SuiBaseProcessor {
 
   onEventTypeName(
     func: (event: type_name.TypeNameInstance, ctx: SuiContext) => void,
-    fetchConfig?: Partial<MoveFetchConfig>,
+    handlerOptions?: HandlerOptions<
+      MoveFetchConfig,
+      type_name.TypeNameInstance
+    >,
     eventFilter?: Omit<EventFilter, "type" | "account">,
   ): type_name {
     this.onMoveEvent(
       func,
       { ...(eventFilter ?? {}), type: "type_name::TypeName" },
-      fetchConfig,
+      handlerOptions,
     );
     return this;
   }
@@ -4578,13 +4588,13 @@ export class uq32_32 extends SuiBaseProcessor {
 
   onEventUQ32_32(
     func: (event: uq32_32.UQ32_32Instance, ctx: SuiContext) => void,
-    fetchConfig?: Partial<MoveFetchConfig>,
+    handlerOptions?: HandlerOptions<MoveFetchConfig, uq32_32.UQ32_32Instance>,
     eventFilter?: Omit<EventFilter, "type" | "account">,
   ): uq32_32 {
     this.onMoveEvent(
       func,
       { ...(eventFilter ?? {}), type: "uq32_32::UQ32_32" },
-      fetchConfig,
+      handlerOptions,
     );
     return this;
   }
@@ -5092,13 +5102,13 @@ export class uq64_64 extends SuiBaseProcessor {
 
   onEventUQ64_64(
     func: (event: uq64_64.UQ64_64Instance, ctx: SuiContext) => void,
-    fetchConfig?: Partial<MoveFetchConfig>,
+    handlerOptions?: HandlerOptions<MoveFetchConfig, uq64_64.UQ64_64Instance>,
     eventFilter?: Omit<EventFilter, "type" | "account">,
   ): uq64_64 {
     this.onMoveEvent(
       func,
       { ...(eventFilter ?? {}), type: "uq64_64::UQ64_64" },
-      fetchConfig,
+      handlerOptions,
     );
     return this;
   }