@sentio/sdk 2.36.2-rc.1 → 2.37.0-rc.3

package/src/store/schema.ts

@@ -0,0 +1,35 @@
+import { buildASTSchema, DocumentNode, extendSchema, GraphQLSchema, parse, validateSchema } from 'graphql/index.js'
+import * as fs from 'node:fs'
+
+const customScalars = ['BigInt', 'BigDecimal', 'DateTime', 'JSON', 'Bytes', 'ID']
+
+const baseSchema = buildASTSchema(
+  parse(`
+    directive @entity on OBJECT
+    directive @query on INTERFACE
+    directive @derivedFrom(field: String!) on FIELD_DEFINITION
+    directive @unique on FIELD_DEFINITION
+    directive @index(fields: [String!] unique: Boolean) repeatable on OBJECT | FIELD_DEFINITION
+    directive @fulltext(query: String!) on FIELD_DEFINITION
+    directive @cardinality(value: Int!) on OBJECT | FIELD_DEFINITION
+    directive @byteWeight(value: Float!) on FIELD_DEFINITION
+    directive @variant on OBJECT # legacy
+    directive @jsonField on OBJECT # legacy
+    ${customScalars.map((name) => 'scalar ' + name).join('\n')}
+`)
+)
+
+export function buildSchema(doc: DocumentNode): GraphQLSchema {
+  const schema = extendSchema(baseSchema, doc)
+  const errors = validateSchema(schema).filter((err) => !/query root/i.test(err.message))
+  if (errors.length > 0) {
+    throw errors[0]
+  }
+  return schema
+}
+
+export function schemaFromFile(filePath: string) {
+  const source = fs.readFileSync(filePath, 'utf-8')
+  const doc = parse(source)
+  return { schema: buildSchema(doc), source }
+}

package/src/store/store.ts

@@ -0,0 +1,68 @@
+import { Entity, EntityClass } from './entity.js'
+import { StoreContext } from './context.js'
+
+export class Store {
+  constructor(private readonly context: StoreContext) {}
+
+  async get<T extends Entity>(entity: EntityClass<T> | string, id: string): Promise<T | undefined> {
+    const promise = this.context.sendRequest({
+      get: {
+        entity: typeof entity == 'string' ? entity : entity.name,
+        id
+      }
+    })
+
+    const data =  await promise as any
+    if (data?.["id"] != null) {
+      return this.newEntity(entity as EntityClass<T>, data)
+    }
+    return undefined
+  }
+
+  async delete(entity: EntityClass<any>, id: string | string[]): Promise<void> {
+    await this.context.sendRequest({
+      delete: {
+        entity: entity.name,
+        id: Array.isArray(id) ? id : [id]
+      }
+    })
+  }
+
+  async upsert<T extends Entity>(entity: T | T[]): Promise<T> {
+    const promise = this.context.sendRequest({
+      upsert: {
+        entity: entity.constructor.name,
+        data: Array.isArray(entity) ? entity.map(e=>e.data) : [entity.data]
+      }
+    })
+
+    if (Array.isArray(entity)) {
+      entity.forEach((e) => (e.store = this))
+    } else {
+      entity.store = this
+    }
+
+    return promise as Promise<T>
+  }
+
+  async list<T extends Entity>(entity: EntityClass<T>, limit?: number, offset?: number): Promise<T[]> {
+    const promise = this.context.sendRequest({
+      list: {
+        entity: entity.name,
+        limit,
+        offset
+      }
+    })
+
+    const list = await promise as any[]
+    return list.map((data) => {
+      return this.newEntity(entity, data)
+    })
+  }
+
+  private newEntity<T extends Entity>(entity: EntityClass<T>, data: any) {
+    const e = new (entity as EntityClass<T>)(data)
+    e.store = this
+    return e
+  }
+}

package/src/store/types.ts

@@ -0,0 +1,10 @@
+export type { BigDecimal } from '@sentio/bigdecimal'
+
+export type ID = string
+export type String = string
+export type Int = number
+export type Float = number
+export type Boolean = boolean
+export type DateTime = Date
+export type Json = any
+export type Bytes = Uint8Array

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

@@ -2,12 +2,16 @@ import {
   AccountConfig,
   ContractConfig,
   DataBinding,
+  DeepPartial,
   Empty,
   ProcessBindingResponse,
   ProcessBindingsRequest,
   ProcessConfigRequest,
   ProcessConfigResponse,
   ProcessorServiceImplementation,
+  ProcessStreamRequest,
+  ProcessStreamResponse,
+  ServerStreamingMethodResult,
   StartRequest
 } from '@sentio/protos'
 import { CallContext } from 'nice-grpc-common'
@@ -80,7 +84,14 @@ export class TestProcessorServer implements ProcessorServiceImplementation {
     return this.service.processBindings({ bindings: [request] }, context)
   }
 
-  processBindingsStream(request: AsyncIterable<DataBinding>, context: CallContext) {
-    return this.service.processBindingsStream(request, context)
+  processBindingsStream(
+    requests: AsyncIterable<ProcessStreamRequest>,
+    context: CallContext
+  ): ServerStreamingMethodResult<DeepPartial<ProcessStreamResponse>> {
+    throw new Error('Method not implemented.')
   }
+
+  // processBindingsStream(request: AsyncIterable<ProcessStreamRequest>, context: CallContext) {
+  //   return this.service.processBindingsStream(request, context)
+  // }
 }