@dxos/echo-generator 0.8.2-main.fbd8ed0 → 0.8.2-staging.4d6ad0f

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dxos/echo-generator",
-  "version": "0.8.2-main.fbd8ed0",
+  "version": "0.8.2-staging.4d6ad0f",
   "description": "ECHO data generator for testing.",
   "homepage": "https://dxos.org",
   "bugs": "https://github.com/dxos/dxos/issues",
@@ -24,23 +24,24 @@
     "src"
   ],
   "dependencies": {
-    "@automerge/automerge": "3.0.0-beta.0",
+    "@automerge/automerge": "3.0.0-beta.4",
     "effect": "3.14.21",
-    "@dxos/client": "0.8.2-main.fbd8ed0",
-    "@dxos/invariant": "0.8.2-main.fbd8ed0",
-    "@dxos/echo-schema": "0.8.2-main.fbd8ed0",
-    "@dxos/live-object": "0.8.2-main.fbd8ed0",
-    "@dxos/node-std": "0.8.2-main.fbd8ed0",
-    "@dxos/schema": "0.8.2-main.fbd8ed0",
-    "@dxos/util": "0.8.2-main.fbd8ed0",
-    "@dxos/echo-db": "0.8.2-main.fbd8ed0"
+    "@dxos/client": "0.8.2-staging.4d6ad0f",
+    "@dxos/echo-schema": "0.8.2-staging.4d6ad0f",
+    "@dxos/invariant": "0.8.2-staging.4d6ad0f",
+    "@dxos/live-object": "0.8.2-staging.4d6ad0f",
+    "@dxos/log": "0.8.2-staging.4d6ad0f",
+    "@dxos/echo-db": "0.8.2-staging.4d6ad0f",
+    "@dxos/node-std": "0.8.2-staging.4d6ad0f",
+    "@dxos/schema": "0.8.2-staging.4d6ad0f",
+    "@dxos/util": "0.8.2-staging.4d6ad0f"
   },
   "devDependencies": {
-    "@dxos/random": "0.8.2-main.fbd8ed0"
+    "@dxos/random": "0.8.2-staging.4d6ad0f"
   },
   "peerDependencies": {
     "effect": "^3.13.3",
-    "@dxos/random": "0.8.2-main.fbd8ed0"
+    "@dxos/random": "0.8.2-staging.4d6ad0f"
   },
   "publishConfig": {
     "access": "public"

package/src/data.ts

@@ -6,8 +6,7 @@ import { next as A } from '@automerge/automerge';
 import { Schema } from 'effect';
 
 import { createDocAccessor, type Space } from '@dxos/client/echo';
-import { Ref } from '@dxos/echo-schema';
-import { createEchoSchema } from '@dxos/live-object';
+import { EchoObject, Ref } from '@dxos/echo-schema';
 import { faker } from '@dxos/random';
 
 import { SpaceObjectGenerator, TestObjectGenerator } from './generator';
@@ -21,67 +20,49 @@ import { randomText } from './util';
 export const Status = ['pending', 'active', 'done'];
 export const Priority = [1, 2, 3, 4, 5];
 
+/**
+ * @deprecated
+ */
 export enum TestSchemaType {
-  document = 'example.com/type/document',
-  organization = 'example.com/type/organization',
-  contact = 'example.com/type/contact',
-  project = 'example.com/type/project',
+  document = 'example.com/type/Document',
+  organization = 'example.com/type/Organization',
+  contact = 'example.com/type/Contact',
+  project = 'example.com/type/Project',
 }
 
+/**
+ * @deprecated
+ */
 const testSchemas = (): TestSchemaMap<TestSchemaType> => {
-  const document = createEchoSchema(
-    {
-      typename: TestSchemaType.document,
-      version: '0.1.0',
-    },
-    {
-      title: Schema.String.annotations({ description: 'title of the document' }),
-      content: Schema.String,
-    },
-  );
-
-  const organization = createEchoSchema(
-    {
-      typename: TestSchemaType.organization,
-      version: '0.1.0',
-    },
-    {
-      name: Schema.String.annotations({ description: 'name of the company or organization' }),
-      website: Schema.String.annotations({ description: 'public website URL' }),
-      description: Schema.String.annotations({ description: 'short summary of the company' }),
-    },
-  );
-
-  const contact = createEchoSchema(
-    {
-      typename: TestSchemaType.contact,
-      version: '0.1.0',
-    },
-    {
-      name: Schema.String.annotations({ description: 'name of the person' }),
-      email: Schema.String,
-      org: Ref(organization),
-      lat: Schema.Number,
-      lng: Schema.Number,
-    },
-  );
-
-  const project = createEchoSchema(
-    {
-      typename: TestSchemaType.project,
-      version: '0.1.0',
-    },
-    {
-      name: Schema.String.annotations({ description: 'name of the project' }),
-      description: Schema.String,
-      website: Schema.String,
-      repo: Schema.String,
-      status: Schema.String,
-      priority: Schema.Number,
-      active: Schema.Boolean,
-      org: Ref(organization),
-    },
-  );
+  const document = Schema.Struct({
+    title: Schema.String.annotations({ description: 'title of the document' }),
+    content: Schema.String,
+  }).pipe(EchoObject({ typename: TestSchemaType.document, version: '0.1.0' }));
+
+  const organization = Schema.Struct({
+    name: Schema.String.annotations({ description: 'name of the company or organization' }),
+    website: Schema.optional(Schema.String.annotations({ description: 'public website URL' })),
+    description: Schema.String.annotations({ description: 'short summary of the company' }),
+  }).pipe(EchoObject({ typename: TestSchemaType.organization, version: '0.1.0' }));
+
+  const contact = Schema.Struct({
+    name: Schema.String.annotations({ description: 'name of the person' }),
+    email: Schema.optional(Schema.String),
+    org: Schema.optional(Ref(organization)),
+    lat: Schema.optional(Schema.Number),
+    lng: Schema.optional(Schema.Number),
+  }).pipe(EchoObject({ typename: TestSchemaType.contact, version: '0.1.0' }));
+
+  const project = Schema.Struct({
+    name: Schema.String.annotations({ description: 'name of the project' }),
+    description: Schema.String,
+    website: Schema.String,
+    repo: Schema.String,
+    status: Schema.String,
+    priority: Schema.Number,
+    active: Schema.Boolean,
+    org: Schema.optional(Ref(organization)),
+  }).pipe(EchoObject({ typename: TestSchemaType.project, version: '0.1.0' }));
 
   return {
     [TestSchemaType.document]: document,
@@ -105,12 +86,13 @@ const testObjectGenerators: TestGeneratorMap<TestSchemaType> = {
 
   [TestSchemaType.contact]: async (provider) => {
     const organizations = await provider?.(TestSchemaType.organization);
-    const { location } = faker.datatype.boolean() ? faker.geo.airport() : {};
+    const location = faker.datatype.boolean() ? faker.geo.airport() : {};
+
     return {
       name: faker.person.fullName(),
       email: faker.datatype.boolean({ probability: 0.5 }) ? faker.internet.email() : undefined,
       org:
-        organizations?.length && faker.datatype.boolean({ probability: 0.3 })
+        organizations?.length && faker.datatype.boolean({ probability: 0.8 })
           ? Ref.make(faker.helpers.arrayElement(organizations))
           : undefined,
       ...location,
@@ -119,8 +101,10 @@ const testObjectGenerators: TestGeneratorMap<TestSchemaType> = {
 
   [TestSchemaType.project]: async () => ({
     name: faker.commerce.productName(),
-    repo: faker.datatype.boolean({ probability: 0.3 }) ? faker.internet.url() : undefined,
+    repo: faker.internet.url(),
     status: faker.helpers.arrayElement(Status),
+    description: faker.lorem.sentences(),
+    website: faker.internet.url(),
     priority: faker.helpers.arrayElement(Priority),
     active: faker.datatype.boolean(),
   }),
@@ -153,7 +137,13 @@ const testObjectMutators: TestMutationsMap<TestSchemaType> = {
   },
 };
 
+/**
+ * @deprecated Use generators in schema package.
+ */
 export const createTestObjectGenerator = () => new TestObjectGenerator(testSchemas(), testObjectGenerators);
 
+/**
+ * @deprecated Use generators in schema package.
+ */
 export const createSpaceObjectGenerator = (space: Space) =>
   new SpaceObjectGenerator(space, testSchemas(), testObjectGenerators, testObjectMutators);

package/src/generator.test.ts

@@ -60,14 +60,14 @@ describe('TestObjectGenerator', () => {
       const generator = createSpaceObjectGenerator(space);
       await generator.addSchemas();
       const organization = await generator.createObject({ types: [TestSchemaType.organization] });
-      schemaId.push(getType(organization)!.asEchoDXN()!.echoId!);
+      schemaId.push(getType(organization)!.toString());
     }
 
     {
       const generator = createSpaceObjectGenerator(space);
       await generator.addSchemas();
       const organization = await generator.createObject({ types: [TestSchemaType.organization] });
-      schemaId.push(getType(organization)!.asEchoDXN()!.echoId!);
+      schemaId.push(getType(organization)!.toString());
     }
 
     expect(schemaId[0]).not.to.be.undefined;
@@ -124,4 +124,31 @@ describe('TestObjectGenerator', () => {
     const todo = await generator.createObject({ types: [Types.task] });
     expect(getType(todo)).to.exist;
   });
+
+  test('references', async () => {
+    const { space } = await setupTest();
+    const generator = createSpaceObjectGenerator(space);
+    await generator.addSchemas();
+
+    // Create raw object.
+    const objects = await generator.createObjects({
+      [TestSchemaType.organization]: 1,
+      [TestSchemaType.contact]: 1,
+    });
+    expect(objects).to.exist;
+    expect(objects.length).to.be.eq(2);
+  });
+
+  test('create project', async () => {
+    const generator = createTestObjectGenerator();
+    const project = await generator.createObject({ types: [TestSchemaType.project] });
+    expect(getType(project)).to.exist;
+  });
+
+  test('create object with not type', async () => {
+    // TODO(burdon): Create Client/spaces.
+    const generator = createTestObjectGenerator();
+
+    await generator.createObject();
+  });
 });

package/src/generator.ts

@@ -10,7 +10,7 @@ import { getTypeAnnotation, EchoSchema, getSchema } from '@dxos/echo-schema';
 import { invariant } from '@dxos/invariant';
 import { live, isLiveObject, type Live } from '@dxos/live-object';
 import { faker } from '@dxos/random';
-import { range } from '@dxos/util';
+import { entries, range } from '@dxos/util';
 
 import { type TestSchemaType } from './data';
 import {
@@ -23,6 +23,7 @@ import {
 
 /**
  * Typed object generator.
+ * @deprecated
  */
 export class TestObjectGenerator<T extends string = TestSchemaType> {
   // prettier-ignore
@@ -40,7 +41,7 @@ export class TestObjectGenerator<T extends string = TestSchemaType> {
     return this.schemas.find((schema) => getTypeAnnotation(schema)!.typename === type);
   }
 
-  protected setSchema(type: T, schema: EchoSchema | Schema.Schema.AnyNoContext) {
+  protected setSchema(type: T, schema: EchoSchema | Schema.Schema.AnyNoContext): void {
     this._schemas[type] = schema;
   }
 
@@ -57,6 +58,11 @@ export class TestObjectGenerator<T extends string = TestSchemaType> {
 
   // TODO(burdon): Based on dependencies (e.g., organization before contact).
   async createObjects(map: Partial<Record<T, number>>) {
+    const results: Live<any>[] = [];
+    for (const [type, count] of entries(map)) {
+      results.push(...(await Promise.all(range(count ?? 0, () => this.createObject({ types: [type as T] })))));
+    }
+
     const tasks = Object.entries<number>(map as any)
       .map(([type, count]) => {
         return range(count, () => this.createObject({ types: [type as T] }));
@@ -79,7 +85,8 @@ export class SpaceObjectGenerator<T extends string> extends TestObjectGenerator<
   ) {
     super(schemaMap, generators, async (type: T) => {
       const schema = this.getSchema(type);
-      return (schema && (await this._space.db.query(Filter.type(schema)).run()).objects) ?? [];
+      const { objects } = await this._space.db.query(schema ? Filter.type(schema) : Filter.nothing()).run();
+      return objects;
     });
   }
 
@@ -119,7 +126,7 @@ export class SpaceObjectGenerator<T extends string> extends TestObjectGenerator<
     }
   }
 
-  async mutateObject(object: AnyLiveObject<any>, params: MutationsProviderParams) {
+  async mutateObject(object: AnyLiveObject<any>, params: MutationsProviderParams): Promise<void> {
     invariant(this._mutations, 'Mutations not defined.');
     const type = getTypeAnnotation(getSchema(object)!)!.typename as T;
     invariant(type && this._mutations?.[type], 'Invalid object type.');
@@ -127,7 +134,7 @@ export class SpaceObjectGenerator<T extends string> extends TestObjectGenerator<
     await this._mutations![type](object, params);
   }
 
-  async mutateObjects(objects: AnyLiveObject<any>[], params: MutationsProviderParams) {
+  async mutateObjects(objects: AnyLiveObject<any>[], params: MutationsProviderParams): Promise<void> {
     for (const object of objects) {
       await this.mutateObject(object, params);
     }