@sentio/sdk 2.39.4-rc.1 → 2.39.4-rc.11

package/src/store/decorators.ts

@@ -1,32 +1,201 @@
+import 'reflect-metadata'
+import {
+  array_,
+  BigDecimalConverter,
+  BigIntConverter,
+  BooleanConverter,
+  BytesConverter,
+  FloatConverter,
+  IDConverter,
+  IntConverter,
+  objectId_,
+  required_,
+  StringConverter,
+  TimestampConverter,
+  TypeConverters,
+  ValueConverter
+} from './convert.js'
+import { RichStruct } from '@sentio/protos'
+import { getStore } from './store.js'
+
 type Constructor = { new (...args: any[]): any }
 
-export function entity(name: string) {
-  return function <T extends Constructor>(constructor: T, context: any) {
-    constructor.prototype.entityName = name
-    return constructor
+function handleError(entity: string, key: string, fn: () => any) {
+  try {
+    return fn()
+  } catch (e) {
+    throw new Error(`Get property "${key}" from Entity "${entity}" failed: ${e.message}`)
+  }
+}
+
+export function Entity(name: string) {
+  return function <T extends Constructor>(BaseClass: T) {
+    Object.defineProperty(BaseClass, 'entityName', {
+      value: name
+    })
+
+    const meta: Record<string, ValueConverter<any>> = (BaseClass as any).meta || {}
+    const target = BaseClass.prototype
+    for (const [propertyKey, type] of Object.entries(meta)) {
+      if (type.isRelation && type.relationName) {
+        const relationName = type.relationName
+        const idGetter = function (this: any) {
+          return handleError(this.constructor.entityName, propertyKey, () => type!.to(this._data.fields[propertyKey]))
+        }
+        const idSetter = function (this: any, value: any) {
+          this._data.fields[propertyKey] = handleError(this.constructor.entityName, propertyKey, () =>
+            type!.from(value)
+          )
+        }
+        const idKey = type.isArray ? propertyKey + 'IDs' : propertyKey + 'ID'
+
+        Reflect.defineProperty(target, idKey, {
+          get: idGetter,
+          set: idSetter
+        })
+        Reflect.defineProperty(target, propertyKey, {
+          get: function () {
+            const ids = idGetter.call(this)
+            if (Array.isArray(ids)) {
+              return Promise.all(
+                ids.map((id) => {
+                  return getStore()?.get(relationName, id)
+                })
+              )
+            } else {
+              return getStore()?.get(relationName, ids)
+            }
+          },
+          set: function (value) {
+            if (value instanceof Promise) {
+              value.then((v) => {
+                idSetter.call(this, v)
+              })
+            } else {
+              idSetter.call(this, value)
+            }
+          }
+        })
+      } else {
+        Reflect.defineProperty(target, propertyKey, {
+          configurable: true,
+          get: function () {
+            return handleError(this.constructor.entityName, propertyKey, () => type!.to(this._data.fields[propertyKey]))
+          },
+          set: function (value) {
+            this._data.fields[propertyKey] = handleError(this.constructor.entityName, propertyKey, () =>
+              type!.from(value)
+            )
+          }
+        })
+      }
+    }
+
+    return class extends BaseClass {
+      readonly _data: RichStruct = { fields: {} }
+      constructor(...args: any[]) {
+        super()
+        for (const key of Object.getOwnPropertyNames(this)) {
+          if (BaseClass.prototype.hasOwnProperty(key)) {
+            delete this[key]
+          }
+        }
+        if (args[0]) {
+          Object.assign(this, args[0])
+        }
+      }
+    }
   }
 }
 
-export function derivedFrom(field: string) {
-  return function (target: any, context: any) {}
+export function Column(type: string) {
+  return function (target: any, propertyKey: string) {
+    let required = false
+    if (type.endsWith('!')) {
+      required = true
+      type = type.slice(0, -1)
+    }
+    let typeConverter = TypeConverters[type]
+    if (!typeConverter) {
+      throw new Error(`Unsupported type ${type}`)
+    }
+    if (required) {
+      typeConverter = required_(typeConverter)
+    }
+    column(typeConverter)(target, propertyKey)
+  }
 }
 
-export function unique(field: string) {
-  return function (target: any, context: any) {}
+export function ListColumn(type: string = '') {
+  return function (target: any, propertyKey: string) {
+    let required = false
+    if (type.endsWith('!')) {
+      required = true
+      type = type.slice(0, -1)
+    }
+
+    let typeConverter = TypeConverters[type]
+    if (!typeConverter) {
+      // support list of list
+      const meta = target.constructor.meta || {}
+      const type = meta[propertyKey]
+      if (type) {
+        typeConverter = type
+      } else {
+        throw new Error(`Can't find inner type convert`)
+      }
+    }
+    if (required) {
+      typeConverter = required_(typeConverter)
+    }
+    column(array_(typeConverter))(target, propertyKey)
+  }
 }
 
-export function index(field: string[]) {
-  return function (target: any, context: any) {}
+export function column(type?: ValueConverter<any>) {
+  return function (target: any, propertyKey: string) {
+    const reflectType = Reflect.getMetadata('design:type', target, propertyKey)
+    if (!type) {
+      const typeName = reflectType.name
+      type = TypeConverters[typeName]
+      if (!type) {
+        throw new Error(`Unsupported type ${typeName}`)
+      }
+    }
+
+    const meta = target.constructor.meta || {}
+    meta[propertyKey] = type
+    target.constructor.meta = meta
+  }
 }
 
-export function fulltext(query: string) {
-  return function (target: any, context: any) {}
+export const IDColumn = column(IDConverter)
+export const IntColumn = column(IntConverter)
+export const FloatColumn = column(FloatConverter)
+export const BigDecimalColumn = column(BigDecimalConverter)
+export const BigIntColumn = column(BigIntConverter)
+export const StringColumn = column(StringConverter)
+export const BooleanColumn = column(BooleanConverter)
+export const TimestampColumn = column(TimestampConverter)
+export const BytesColumn = column(BytesConverter)
+
+export function Required(target: any, propertyKey: string) {
+  const meta = target.constructor.meta || {}
+  const type = meta[propertyKey]
+  if (type) {
+    meta[propertyKey] = required_(type)
+  }
+  target.constructor.meta = meta
 }
 
-export function cardinality(value: number) {
-  return function (target: any, context: any) {}
+export function One(entity: string) {
+  return function (target: any, propertyKey: string) {
+    column(objectId_(entity))(target, propertyKey)
+  }
 }
 
-export function byteWeight(value: number) {
-  return function (target: any, context: any) {}
+export function Many(entity: string) {
+  return function (target: any, propertyKey: string) {
+    column(array_(objectId_(entity)))(target, propertyKey)
+  }
 }

package/src/store/index.ts

@@ -1,6 +1,5 @@
 export * from './decorators.js'
 export * from './types.js'
-export * from './entity.js'
 export * from './store.js'
 export * from './context.js'
 export * from './convert.js'

package/src/store/schema.ts

@@ -1,7 +1,8 @@
 import { buildASTSchema, DocumentNode, extendSchema, GraphQLSchema, parse, validateSchema } from 'graphql/index.js'
 import * as fs from 'node:fs'
+import { GraphQLObjectType, GraphQLOutputType, isListType, isNonNullType } from 'graphql'
 
-const customScalars = ['BigInt', 'BigDecimal', 'DateTime', 'JSON', 'Bytes', 'ID']
+const customScalars = ['BigInt', 'BigDecimal', 'Timestamp', 'JSON', 'Bytes', 'ID']
 
 const baseSchema = buildASTSchema(
   parse(`
@@ -19,12 +20,51 @@ const baseSchema = buildASTSchema(
 `)
 )
 
+function getElemType(type: GraphQLOutputType) {
+  if (isNonNullType(type)) {
+    return getElemType(type.ofType)
+  }
+  if (isListType(type)) {
+    return getElemType(type.ofType)
+  }
+  return type
+}
+
+function checkRelations(schema: GraphQLSchema) {
+  for (const t of Object.values(schema.getTypeMap())) {
+    if (t.name.startsWith('__')) {
+      continue
+    }
+    if (t instanceof GraphQLObjectType) {
+      for (const f of Object.values(t.getFields())) {
+        if (f.astNode) {
+          for (const d of f.astNode.directives ?? []) {
+            if (d.name.value === 'derivedFrom') {
+              const arg = (d.arguments ?? []).find((a) => a.name.value === 'field')
+              if (!arg || !arg.value || arg.value.kind !== 'StringValue') {
+                throw new Error(`@derivedFrom directive must have a 'field' argument`)
+              }
+              const fieldName = arg.value.value
+              const targetType = getElemType(f.type) as GraphQLObjectType
+              // Check if the target type has a field with the same name
+              if (!targetType.getFields()[fieldName]) {
+                throw new Error(`Field '${fieldName}' not found in type '${targetType.name}'`)
+              }
+            }
+          }
+        }
+      }
+    }
+  }
+}
+
 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]
   }
+  checkRelations(schema)
   return schema
 }
 

package/src/store/store.ts

@@ -1,48 +1,68 @@
-import { Entity, EntityClass } from './entity.js'
 import { StoreContext } from './context.js'
 import { DatabaseSchema } from '../core/index.js'
 import { BigDecimal } from '@sentio/bigdecimal'
-import { Bytes, DateTime, Float, ID, Int } from './types.js'
+import { Bytes, Float, ID, Int, Timestamp } from './types.js'
+import type { Entity as EntityStruct, RichValue } from '@sentio/protos'
 import { DBRequest_DBOperator, DBResponse } from '@sentio/protos'
-import type { RichStruct, RichValue } from '@sentio/protos'
 import { toBigInteger } from './convert.js'
+import { PluginManager } from '@sentio/runtime'
 
-type Value = ID | string | Int | Float | boolean | DateTime | Bytes | BigDecimal | bigint
+type Value = ID | string | Int | Float | boolean | Timestamp | Bytes | BigDecimal | bigint
+
+interface Entity {
+  id: ID
+}
+
+export interface EntityClass<T> {
+  new (data: Partial<T>): T
+}
 
 export class Store {
   constructor(private readonly context: StoreContext) {}
 
-  async get<T extends Entity>(entity: EntityClass<T> | string, id: string): Promise<T | undefined> {
+  async get<T extends Entity>(entity: EntityClass<T> | string, id: ID): Promise<T | undefined> {
     const promise = this.context.sendRequest({
       get: {
         entity: typeof entity == 'string' ? entity : entity.prototype.entityName,
-        id
+        id: id.toString()
       }
     })
 
     const data = (await promise) as DBResponse
-    if (data.entities?.entities[0]) {
-      const entityData = data.entities.entities[0]
+    if (data.entityList?.entities[0]) {
+      const entityData = data.entityList?.entities[0]
       return this.newEntity(entity, entityData)
     }
 
     return undefined
   }
 
-  async delete(entity: EntityClass<any>, id: string | string[]): Promise<void> {
-    const toBeDeleted = []
-    if (Array.isArray(id)) {
-      for (const i of id) {
-        toBeDeleted.push({ entity: entity.prototype.entityName, id: i })
+  async delete<T extends Entity>(entity: EntityClass<T> | T | T[], id?: string | string[]): Promise<void> {
+    const request = {
+      entity: [] as string[],
+      id: [] as string[]
+    }
+    if (id) {
+      if (Array.isArray(id)) {
+        for (const i of id) {
+          const items = (entity as any).prototype.entityName
+          request.entity.push(items)
+          request.id.push(i.toString())
+        }
+      } else {
+        request.entity.push((entity as any).prototype.entityName)
+        request.id.push(id)
       }
     } else {
-      toBeDeleted.push({ entity: entity.prototype.entityName, id })
+      const entities = Array.isArray(entity) ? entity : [entity]
+      for (const e of entities) {
+        request.entity.push(e.constructor.prototype.entityName)
+        request.id.push((e as Entity).id.toString())
+      }
     }
+
     await this.context.sendRequest({
-      delete: {
-        entity: toBeDeleted.map((e) => e.entity) as string[],
-        id: toBeDeleted.map((e) => e.id) as string[]
-      }
+      delete: request
     })
   }
 
@@ -52,8 +72,8 @@ export class Store {
       upsert: {
         entity: entities.map((e) => e.constructor.prototype.entityName),
         // data: entities.map((e) => serialize(e.data)),
-        id: entities.map((e) => e.id),
-        entityData: entities.map((e) => e.serialize())
+        id: entities.map((e) => e.id.toString()),
+        entityData: entities.map((e: any) => e._data)
       }
     })
 
@@ -77,7 +97,7 @@ export class Store {
         }
       })
       const response = (await promise) as DBResponse
-      for (const data of response.entities?.entities || []) {
+      for (const data of response.entityList?.entities || []) {
         yield this.newEntity(entity, data)
       }
       if (!response.nextCursor) {
@@ -87,18 +107,21 @@ export class Store {
     }
   }
 
-  private newEntity<T extends Entity>(entity: EntityClass<T> | string, data: RichStruct) {
+  private newEntity<T extends Entity>(entity: EntityClass<T> | string, data: EntityStruct) {
     if (typeof entity == 'string') {
-      const en = DatabaseSchema.findEntity(entity)
+      let en = DatabaseSchema.findEntity(entity)
       if (!en) {
         // it is an interface
-        return new Entity() as T
+        en = DatabaseSchema.findEntity(data.entity)
+        if (!en) {
+          throw new Error(`Entity ${entity} not found`)
+        }
       }
       entity = en
     }
 
     const res = new (entity as EntityClass<T>)({}) as T
-    res.setData(data)
+    ;(res as any)._data = data.data
     return res
   }
 }
@@ -175,3 +198,11 @@ function serializeBigDecimal(v: BigDecimal): RichValue {
     bigdecimalValue: undefined
   }
 }
+
+export function getStore() {
+  const dbContext = PluginManager.INSTANCE.dbContextLocalStorage.getStore()
+  if (dbContext) {
+    return new Store(dbContext)
+  }
+  return undefined
+}

package/src/store/types.ts

@@ -1,7 +1,29 @@
-export type ID = string
+export type ID = (string | Uint8Array) & { __id__?: void }
 export type String = string
-export type Int = number
+export type Int = number & { __int__?: void }
 export type Float = number
 export type Boolean = boolean
-export type DateTime = Date
+export type Timestamp = Date
 export type Bytes = Uint8Array
+export type BigInt = bigint
+
+export class Struct<T = unknown, S = unknown> {
+  readonly TYPE!: T
+  constructor(readonly schema: S) {}
+}
+export type Infer<T extends Struct<any>> = T['TYPE']
+export type ObjectSchema = Record<string, Struct<any, any>>
+
+export function object<S extends ObjectSchema>(schema: S): Struct<ObjectType<S>, S> {
+  return new Struct(schema ?? null)
+}
+
+export type Simplify<T> = T extends any[] | Date ? T : { [K in keyof T]: T[K] } & NonNullable<unknown>
+
+export type ObjectType<S extends ObjectSchema> = Simplify<{ [K in keyof S]: Infer<S[K]> }>
+
+const StringStruct = new Struct<string, null>(null)
+
+const UU = object({
+  test: StringStruct
+})

package/lib/store/entity.d.ts

@@ -1,23 +0,0 @@
-import { ID } from './types.js';
-import { RichStruct } from '@sentio/protos';
-import { ValueConverter } from './convert.js';
-export interface EntityClass<T extends Entity> {
-    new (data: any): T;
-}
-export declare class Entity {
-    private _data;
-    get id(): ID;
-    protected fromPojo(data: any, converters: Record<string, ValueConverter<any>>): void;
-    constructor(_data?: RichStruct);
-    setData(data: RichStruct): void;
-    private getIdFromEntity;
-    private getStore;
-    get<T>(field: string, converter: ValueConverter<T>): T;
-    set<T>(field: string, value: T, serializer: ValueConverter<T>): void;
-    protected getObject<T extends Entity>(entity: EntityClass<T> | string, field: string): Promise<T | undefined>;
-    protected setObject(field: string, value: any): void;
-    protected getFieldObjectArray<T extends Entity>(entity: EntityClass<T>, field: string): Promise<T[]>;
-    toString(): string;
-    serialize(): RichStruct;
-}
-//# sourceMappingURL=entity.d.ts.map

package/lib/store/entity.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"entity.d.ts","sourceRoot":"","sources":["../../src/store/entity.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,YAAY,CAAA;AAG/B,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAA;AAC3C,OAAO,EAAkC,cAAc,EAAE,MAAM,cAAc,CAAA;AAE7E,MAAM,WAAW,WAAW,CAAC,CAAC,SAAS,MAAM;IAC3C,KAAK,IAAI,EAAE,GAAG,GAAG,CAAC,CAAA;CACnB;AAED,qBAAa,MAAM;IAaL,OAAO,CAAC,KAAK;IAZzB,IAAI,EAAE,IAAI,EAAE,CAEX;IAED,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC;gBAQzD,KAAK,GAAE,UAA2B;IAEtD,OAAO,CAAC,IAAI,EAAE,UAAU;IAIxB,OAAO,CAAC,eAAe;IAOvB,OAAO,CAAC,QAAQ;IAQhB,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC;IAKtD,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,IAAI;IAUpE,SAAS,CAAC,SAAS,CAAC,CAAC,SAAS,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,SAAS,CAAC;IAK7G,SAAS,CAAC,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,IAAI;IAIpD,SAAS,CAAC,mBAAmB,CAAC,CAAC,SAAS,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IAMpG,QAAQ,IAAI,MAAM;IAMlB,SAAS,IAAI,UAAU;CAGxB"}

package/lib/store/entity.js

@@ -1,70 +0,0 @@
-import { PluginManager } from '@sentio/runtime';
-import { Store } from './store.js';
-import { array_, IDConverter, required_ } from './convert.js';
-export class Entity {
-    _data;
-    get id() {
-        return this.get('id', required_(IDConverter));
-    }
-    fromPojo(data, converters) {
-        for (const [field, value] of Object.entries(data)) {
-            if (converters[field] !== undefined) {
-                this.set(field, value, converters[field]);
-            }
-        }
-    }
-    constructor(_data = { fields: {} }) {
-        this._data = _data;
-    }
-    setData(data) {
-        this._data = data;
-    }
-    getIdFromEntity(entity) {
-        if (typeof entity === 'object' && entity.id) {
-            return entity.id;
-        }
-        return entity;
-    }
-    getStore() {
-        const dbContext = PluginManager.INSTANCE.dbContextLocalStorage.getStore();
-        if (dbContext) {
-            return new Store(dbContext);
-        }
-        return undefined;
-    }
-    get(field, converter) {
-        const value = this._data.fields[field];
-        return converter.to(value);
-    }
-    set(field, value, serializer) {
-        if (Array.isArray(value)) {
-            this._data.fields[field] = {
-                listValue: { values: value.map((v) => serializer.from(v)) }
-            };
-        }
-        else {
-            this._data.fields[field] = serializer.from(value);
-        }
-    }
-    getObject(entity, field) {
-        const id = this.get(field, IDConverter);
-        return id ? this.getStore()?.get(entity, id) : Promise.resolve(undefined);
-    }
-    setObject(field, value) {
-        this.set(field, this.getIdFromEntity(value), IDConverter);
-    }
-    getFieldObjectArray(entity, field) {
-        const ids = this.get(field, array_(IDConverter));
-        const promises = ids.map((id) => this.getStore()?.get(entity, id));
-        return Promise.all(promises);
-    }
-    toString() {
-        const entityName = this.constructor.prototype.entityName;
-        const id = this.id;
-        return `${entityName}#${id} ${JSON.stringify(this._data)}`;
-    }
-    serialize() {
-        return this._data;
-    }
-}
-//# sourceMappingURL=entity.js.map

package/lib/store/entity.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"entity.js","sourceRoot":"","sources":["../../src/store/entity.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAA;AAC/C,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAA;AAElC,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,EAAkB,MAAM,cAAc,CAAA;AAM7E,MAAM,OAAO,MAAM;IAaG;IAZpB,IAAI,EAAE;QACJ,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,WAAW,CAAC,CAAC,CAAA;IAC/C,CAAC;IAES,QAAQ,CAAC,IAAS,EAAE,UAA+C;QAC3E,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YAClD,IAAI,UAAU,CAAC,KAAK,CAAC,KAAK,SAAS,EAAE,CAAC;gBACpC,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,CAAA;YAC3C,CAAC;QACH,CAAC;IACH,CAAC;IAED,YAAoB,QAAoB,EAAE,MAAM,EAAE,EAAE,EAAE;QAAlC,UAAK,GAAL,KAAK,CAA6B;IAAG,CAAC;IAE1D,OAAO,CAAC,IAAgB;QACtB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAA;IACnB,CAAC;IAEO,eAAe,CAAC,MAAW;QACjC,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,EAAE,EAAE,CAAC;YAC5C,OAAO,MAAM,CAAC,EAAE,CAAA;QAClB,CAAC;QACD,OAAO,MAAM,CAAA;IACf,CAAC;IAEO,QAAQ;QACd,MAAM,SAAS,GAAG,aAAa,CAAC,QAAQ,CAAC,qBAAqB,CAAC,QAAQ,EAAE,CAAA;QACzE,IAAI,SAAS,EAAE,CAAC;YACd,OAAO,IAAI,KAAK,CAAC,SAAS,CAAC,CAAA;QAC7B,CAAC;QACD,OAAO,SAAS,CAAA;IAClB,CAAC;IAED,GAAG,CAAI,KAAa,EAAE,SAA4B;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QACtC,OAAO,SAAS,CAAC,EAAE,CAAC,KAAK,CAAC,CAAA;IAC5B,CAAC;IAED,GAAG,CAAI,KAAa,EAAE,KAAQ,EAAE,UAA6B;QAC3D,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG;gBACzB,SAAS,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE;aAC5D,CAAA;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnD,CAAC;IACH,CAAC;IAES,SAAS,CAAmB,MAA+B,EAAE,KAAa;QAClF,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,WAAW,CAAC,CAAA;QACvC,OAAO,EAAE,CAAC,CAAC,CAAE,IAAI,CAAC,QAAQ,EAAE,EAAE,GAAG,CAAC,MAAM,EAAE,EAAE,CAAgB,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;IAC3F,CAAC;IAES,SAAS,CAAC,KAAa,EAAE,KAAU;QAC3C,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,WAAW,CAAC,CAAA;IAC3D,CAAC;IAES,mBAAmB,CAAmB,MAAsB,EAAE,KAAa;QACnF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,CAAA;QAChD,MAAM,QAAQ,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,EAAU,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAA;QAC1E,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAiB,CAAA;IAC9C,CAAC;IAED,QAAQ;QACN,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,UAAU,CAAA;QACxD,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAA;QAClB,OAAO,GAAG,UAAU,IAAI,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAA;IAC5D,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,KAAK,CAAA;IACnB,CAAC;CACF"}

package/src/store/entity.ts

@@ -1,84 +0,0 @@
-import { ID } from './types.js'
-import { PluginManager } from '@sentio/runtime'
-import { Store } from './store.js'
-import { RichStruct } from '@sentio/protos'
-import { array_, IDConverter, required_, ValueConverter } from './convert.js'
-
-export interface EntityClass<T extends Entity> {
-  new (data: any): T
-}
-
-export class Entity {
-  get id(): ID {
-    return this.get('id', required_(IDConverter))
-  }
-
-  protected fromPojo(data: any, converters: Record<string, ValueConverter<any>>) {
-    for (const [field, value] of Object.entries(data)) {
-      if (converters[field] !== undefined) {
-        this.set(field, value, converters[field])
-      }
-    }
-  }
-
-  constructor(private _data: RichStruct = { fields: {} }) {}
-
-  setData(data: RichStruct) {
-    this._data = data
-  }
-
-  private getIdFromEntity(entity: any): any {
-    if (typeof entity === 'object' && entity.id) {
-      return entity.id
-    }
-    return entity
-  }
-
-  private getStore() {
-    const dbContext = PluginManager.INSTANCE.dbContextLocalStorage.getStore()
-    if (dbContext) {
-      return new Store(dbContext)
-    }
-    return undefined
-  }
-
-  get<T>(field: string, converter: ValueConverter<T>): T {
-    const value = this._data.fields[field]
-    return converter.to(value)
-  }
-
-  set<T>(field: string, value: T, serializer: ValueConverter<T>): void {
-    if (Array.isArray(value)) {
-      this._data.fields[field] = {
-        listValue: { values: value.map((v) => serializer.from(v)) }
-      }
-    } else {
-      this._data.fields[field] = serializer.from(value)
-    }
-  }
-
-  protected getObject<T extends Entity>(entity: EntityClass<T> | string, field: string): Promise<T | undefined> {
-    const id = this.get(field, IDConverter)
-    return id ? (this.getStore()?.get(entity, id) as Promise<T>) : Promise.resolve(undefined)
-  }
-
-  protected setObject(field: string, value: any): void {
-    this.set(field, this.getIdFromEntity(value), IDConverter)
-  }
-
-  protected getFieldObjectArray<T extends Entity>(entity: EntityClass<T>, field: string): Promise<T[]> {
-    const ids = this.get(field, array_(IDConverter))
-    const promises = ids.map((id: string) => this.getStore()?.get(entity, id))
-    return Promise.all(promises) as Promise<T[]>
-  }
-
-  toString(): string {
-    const entityName = this.constructor.prototype.entityName
-    const id = this.id
-    return `${entityName}#${id} ${JSON.stringify(this._data)}`
-  }
-
-  serialize(): RichStruct {
-    return this._data
-  }
-}