@dxos/plugin-debug 0.8.4-main.1f223c7 → 0.8.4-main.2244d791bb

package/dist/lib/browser/{SpaceGenerator-SRCIEELZ.mjs → SpaceGenerator-B4A7J26E.mjs}

@@ -1,53 +1,53 @@
 // src/components/SpaceGenerator/SpaceGenerator.tsx
-import { useSignals as _useSignals2 } from "@preact-signals/safe-react/tracking";
 import React2, { useCallback, useMemo, useState } from "react";
-import { useIntentDispatcher } from "@dxos/app-framework";
+import { useOperationInvoker } from "@dxos/app-framework/ui";
 import { ComputeGraph as ComputeGraph2 } from "@dxos/conductor";
-import { Filter as Filter3 } from "@dxos/echo";
+import { Filter as Filter3, Obj as Obj2 } from "@dxos/echo";
 import { Markdown as Markdown3 } from "@dxos/plugin-markdown/types";
-import { SheetType as SheetType2 } from "@dxos/plugin-sheet/types";
-import { DiagramType as DiagramType2 } from "@dxos/plugin-sketch/types";
+import { Sheet as Sheet2 } from "@dxos/plugin-sheet/types";
+import { Diagram as Diagram2 } from "@dxos/plugin-sketch/types";
 import { useClient } from "@dxos/react-client";
-import { getTypename } from "@dxos/react-client/echo";
-import { IconButton as IconButton2, Input, Toolbar, useAsyncEffect } from "@dxos/react-ui";
+import { IconButton as IconButton2, Input, Layout, ScrollArea, Toolbar, useAsyncEffect } from "@dxos/react-ui";
 import { SyntaxHighlighter } from "@dxos/react-ui-syntax-highlighter";
-import { DataType as DataType3 } from "@dxos/schema";
+import { Organization as Organization2, Person as Person2, Task } from "@dxos/types";
 import { jsonKeyReplacer, sortKeys } from "@dxos/util";
 
 // src/components/SpaceGenerator/ObjectGenerator.tsx
-import { createIntent } from "@dxos/app-framework";
 import { addressToA1Notation } from "@dxos/compute";
 import { ComputeGraph, ComputeGraphModel, DEFAULT_OUTPUT, NODE_INPUT, NODE_OUTPUT } from "@dxos/conductor";
-import { DXN, Filter, Key, Obj, Ref, Type } from "@dxos/echo";
+import { DXN, Filter, Key } from "@dxos/echo";
 import { Markdown } from "@dxos/plugin-markdown/types";
-import { createSheet } from "@dxos/plugin-sheet/types";
-import { SheetType } from "@dxos/plugin-sheet/types";
-import { CanvasType, DiagramType } from "@dxos/plugin-sketch/types";
-import { SpaceAction } from "@dxos/plugin-space/types";
+import { Sheet } from "@dxos/plugin-sheet/types";
+import { Diagram } from "@dxos/plugin-sketch/types";
+import { SpaceOperation } from "@dxos/plugin-space/types";
 import { faker } from "@dxos/random";
-import { DataType, typenameFromQuery } from "@dxos/schema";
+import { View, getTypenameFromQuery } from "@dxos/schema";
 import { createAsyncGenerator } from "@dxos/schema/testing";
 import { range } from "@dxos/util";
 var generator = faker;
 var findViewByTypename = async (views, typename) => {
-  return views.find((view) => typenameFromQuery(view.query) === typename);
+  return views.find((view) => getTypenameFromQuery(view.query.ast) === typename);
 };
-var createGenerator = (client, dispatch, schema) => {
+var createGenerator = (client, invokePromise, schema) => {
   return async (space, n) => {
     const typename = schema.typename;
-    const { objects: views } = await space.db.query(Filter.type(DataType.View)).run();
+    const views = await space.db.query(Filter.type(View.View)).run();
     const view = await findViewByTypename(views, typename);
-    const staticSchema = client?.graph.schemaRegistry.schemas.find((schema2) => Type.getTypename(schema2) === typename);
+    const staticSchema = client?.graph.schemaRegistry.query({
+      typename
+    }).runSync()[0];
     if (!view && !staticSchema) {
-      await dispatch(createIntent(SpaceAction.AddSchema, {
-        space,
-        schema
-      }));
+      await invokePromise(SpaceOperation.AddSchema, {
+        db: space.db,
+        schema,
+        show: false
+      });
     } else if (!view && staticSchema) {
-      await dispatch(createIntent(SpaceAction.UseStaticSchema, {
-        space,
-        typename
-      }));
+      await invokePromise(SpaceOperation.UseStaticSchema, {
+        db: space.db,
+        typename,
+        show: false
+      });
     }
     const generate = createAsyncGenerator(generator, schema, {
       db: space.db
@@ -60,7 +60,7 @@ var staticGenerators = /* @__PURE__ */ new Map([
     Markdown.Document.typename,
     async (space, n, cb) => {
       const objects = range(n).map(() => {
-        return space.db.add(Markdown.makeDocument({
+        return space.db.add(Markdown.make({
           name: faker.commerce.productName(),
           content: faker.lorem.sentences(5)
         }));
@@ -70,14 +70,11 @@ var staticGenerators = /* @__PURE__ */ new Map([
     }
   ],
   [
-    DiagramType.typename,
+    Diagram.Diagram.typename,
     async (space, n, cb) => {
       const objects = range(n).map(() => {
-        const obj = space.db.add(Obj.make(DiagramType, {
-          name: faker.commerce.productName(),
-          canvas: Ref.make(Obj.make(CanvasType, {
-            content: {}
-          }))
+        const obj = space.db.add(Diagram.make({
+          name: faker.commerce.productName()
         }));
         return obj;
       });
@@ -87,7 +84,7 @@ var staticGenerators = /* @__PURE__ */ new Map([
   ],
   // TODO(burdon): Create unit tests.
   [
-    SheetType.typename,
+    Sheet.Sheet.typename,
     async (space, n, cb) => {
       const objects = range(n).map(() => {
         const cells = {};
@@ -123,7 +120,7 @@ var staticGenerators = /* @__PURE__ */ new Map([
             }
           }
         }
-        return space.db.add(createSheet({
+        return space.db.add(Sheet.make({
           name: faker.commerce.productName(),
           cells
         }));
@@ -197,34 +194,37 @@ var staticGenerators = /* @__PURE__ */ new Map([
 ]);
 
 // src/components/SpaceGenerator/presets.ts
-import { Schema } from "effect";
-import { ResearchOn, research } from "@dxos/assistant-testing";
+import * as Schema from "effect/Schema";
+import { AgentFunctions, EntityExtractionFunctions, ResearchBlueprint } from "@dxos/assistant-toolkit";
+import { Prompt } from "@dxos/blueprints";
 import { NODE_INPUT as NODE_INPUT2 } from "@dxos/conductor";
-import { DXN as DXN2, Filter as Filter2, Key as Key2, Obj as Obj2, Query, Ref as Ref2, Relation, Type as Type2 } from "@dxos/echo";
-import { FunctionTrigger, serializeFunction } from "@dxos/functions";
+import { DXN as DXN2, Filter as Filter2, Key as Key2, Obj, Query, Ref, Tag, Type } from "@dxos/echo";
+import { Trigger, serializeFunction } from "@dxos/functions";
 import { invariant } from "@dxos/invariant";
+import { GmailFunctions } from "@dxos/plugin-inbox";
 import { Mailbox } from "@dxos/plugin-inbox/types";
 import { Markdown as Markdown2 } from "@dxos/plugin-markdown/types";
 import { createAppend, createChat, createComputeGraph, createConstant, createFunction, createGpt, createQueue, createRandom, createSurface, createTemplate, createText, createTrigger } from "@dxos/react-ui-canvas-compute";
-import { CanvasBoardType, CanvasGraphModel, pointMultiply, pointsToRect, rectToPoints } from "@dxos/react-ui-canvas-editor";
-import { DataType as DataType2, createView } from "@dxos/schema";
-import { range as range2 } from "@dxos/util";
+import { CanvasBoard, CanvasGraphModel, pointMultiply, pointsToRect, rectToPoints } from "@dxos/react-ui-canvas-editor";
+import { View as View2 } from "@dxos/schema";
+import { Message, Organization, Person, Pipeline } from "@dxos/types";
+import { range as range2, trim } from "@dxos/util";
 var __dxlog_file = "/__w/dxos/dxos/packages/plugins/plugin-debug/src/components/SpaceGenerator/presets.ts";
-var PresetName = /* @__PURE__ */ function(PresetName2) {
+var PresetName = /* @__PURE__ */ (function(PresetName2) {
   PresetName2["DXOS_TEAM"] = "dxos-team";
   PresetName2["ORG_RESEARCH_PROJECT"] = "org-research-project";
   PresetName2["GPT_QUEUE"] = "webhook-gpt-queue";
   PresetName2["CHAT_GPT"] = "chat-gpt-text";
   PresetName2["OBJECT_CHANGE_QUEUE"] = "objectChange-queue";
-  PresetName2["FOREX_FUNCTION_CALL"] = "forex-function-call";
   PresetName2["TIMER_TICK_QUEUE"] = "timerTick-queue";
+  PresetName2["FOREX_FUNCTION_CALL"] = "forex-function-call";
   PresetName2["DISCORD_MESSAGES"] = "discord-messages";
   return PresetName2;
-}({});
+})({});
 var generator2 = () => ({
   schemas: [
-    CanvasBoardType,
-    FunctionTrigger
+    CanvasBoard.CanvasBoard,
+    Trigger.Trigger
   ],
   types: Object.values(PresetName).map((name) => ({
     typename: name
@@ -234,39 +234,42 @@ var generator2 = () => ({
       "dxos-team",
       async (space, n, cb) => {
         const objects = range2(n, () => {
-          const org = space.db.add(Obj2.make(DataType2.Organization, {
+          const org = space.db.add(Obj.make(Organization.Organization, {
             name: "DXOS",
             website: "https://dxos.org"
           }));
-          const doc = space.db.add(Markdown2.makeDocument({
+          const doc = space.db.add(Markdown2.make({
             name: "DXOS Research",
             content: "DXOS builds Composer, an open-source AI-powered malleable application."
           }));
-          space.db.add(Relation.make(ResearchOn, {
-            [Relation.Source]: doc,
-            [Relation.Target]: org,
-            completedAt: (/* @__PURE__ */ new Date()).toISOString()
+          const tag = space.db.add(Tag.make({
+            label: "Investor"
           }));
-          space.db.add(Obj2.make(DataType2.Person, {
+          const tagDxn = Obj.getDXN(tag).toString();
+          Obj.change(doc, (d) => {
+            Obj.getMeta(d).tags = [
+              tagDxn
+            ];
+          });
+          space.db.add(Obj.make(Person.Person, {
             fullName: "Rich",
-            jobTitle: "investor",
-            organization: Ref2.make(org)
+            organization: Ref.make(org)
+          }, {
+            tags: [
+              tagDxn
+            ]
           }));
-          space.db.add(Obj2.make(DataType2.Person, {
+          space.db.add(Obj.make(Person.Person, {
             fullName: "Josiah",
-            organization: Ref2.make(org)
+            organization: Ref.make(org)
           }));
-          space.db.add(Obj2.make(DataType2.Person, {
+          space.db.add(Obj.make(Person.Person, {
             fullName: "Dima",
-            organization: Ref2.make(org)
+            organization: Ref.make(org)
           }));
-          space.db.add(Obj2.make(DataType2.Person, {
+          space.db.add(Obj.make(Person.Person, {
             fullName: "Mykola",
-            organization: Ref2.make(org)
-          }));
-          space.db.add(Obj2.make(DataType2.Person, {
-            fullName: "Will",
-            organization: Ref2.make(org)
+            organization: Ref.make(org)
           }));
           return doc;
         });
@@ -278,24 +281,71 @@ var generator2 = () => ({
       "org-research-project",
       async (space, n, cb) => {
         const mailbox = await space.db.query(Filter2.type(Mailbox.Mailbox)).first();
+        const tag = await space.db.query(Filter2.type(Tag.Tag, {
+          label: "Investor"
+        })).first();
+        const tagDxn = Obj.getDXN(tag).toString();
         const objects = range2(n, () => {
-          const contactsQuery = Query.select(Filter2.type(DataType2.Person, {
-            jobTitle: "investor"
+          const contactsQuery = Query.select(Filter2.type(Person.Person)).select(Filter2.tag(tagDxn));
+          const organizationsQuery = Query.select(Filter2.type(Organization.Organization)).select(Filter2.tag(tagDxn));
+          const notesQuery = Query.select(Filter2.type(Markdown2.Document)).select(Filter2.tag(tagDxn));
+          space.db.add(Trigger.make({
+            enabled: true,
+            spec: {
+              kind: "timer",
+              cron: "* * * * *"
+            },
+            function: Ref.make(serializeFunction(GmailFunctions.Sync)),
+            input: {
+              mailbox: Ref.make(mailbox)
+            }
           }));
-          const organizationsQuery = contactsQuery.reference("organization");
-          const notesQuery = organizationsQuery.targetOf(ResearchOn).source();
-          const researchTrigger = Obj2.make(FunctionTrigger, {
-            function: Ref2.make(serializeFunction(research)),
+          space.db.add(Trigger.make({
+            enabled: true,
+            // TODO(wittjosiah): Queue trigger doesn't support matching query of the column.
+            spec: {
+              kind: "queue",
+              queue: mailbox.queue.dxn.toString()
+            },
+            function: Ref.make(serializeFunction(EntityExtractionFunctions.Extract)),
+            input: {
+              source: "{{event.item}}"
+            }
+          }));
+          const researchPrompt = space.db.add(Prompt.make({
+            name: "Research",
+            description: "Research organization",
+            input: Schema.Struct({
+              org: Schema.Any
+            }),
+            output: Schema.Any,
+            // TODO(dmaretskyi): This mocks research (returns pre-baked result), the actual research might take compute minutes.
+            // Remove the mock prompt to do the actual research.
+            instructions: trim`
+                Research the organization provided as input.
+                Create a research note for it at the end.
+                NOTE: Do mocked reseach (set mockSearch to true).
+              `,
+            blueprints: [
+              Ref.make(ResearchBlueprint.make())
+            ]
+          }));
+          space.db.add(Trigger.make({
+            enabled: true,
             spec: {
               kind: "subscription",
-              query: organizationsQuery.ast
+              query: {
+                ast: organizationsQuery.ast
+              }
             },
-            enabled: true
-          });
-          space.db.add(researchTrigger);
-          const mailboxView = createView({
-            name: "Mailbox",
-            query: Query.select(Filter2.type(DataType2.Message, {
+            function: Ref.make(serializeFunction(AgentFunctions.Prompt)),
+            input: {
+              prompt: Ref.make(researchPrompt),
+              input: "{{event.subject}}"
+            }
+          }));
+          const mailboxView = View2.make({
+            query: Query.select(Filter2.type(Message.Message, {
               properties: {
                 labels: Filter2.contains("investor")
               }
@@ -304,43 +354,44 @@ var generator2 = () => ({
                 mailbox.queue.dxn.toString()
               ]
             }),
-            jsonSchema: Type2.toJsonSchema(DataType2.Message),
-            presentation: Obj2.make(DataType2.Collection, {
-              objects: []
-            })
+            jsonSchema: Type.toJsonSchema(Message.Message)
           });
-          const contactsView = createView({
-            name: "Contacts",
+          const contactsView = View2.make({
             query: contactsQuery,
-            jsonSchema: Type2.toJsonSchema(DataType2.Person),
-            presentation: Obj2.make(DataType2.Collection, {
-              objects: []
-            })
+            jsonSchema: Type.toJsonSchema(Person.Person)
           });
-          const organizationsView = createView({
-            name: "Organizations",
+          const organizationsView = View2.make({
             query: organizationsQuery,
-            jsonSchema: Type2.toJsonSchema(DataType2.Organization),
-            presentation: Obj2.make(DataType2.Collection, {
-              objects: []
-            })
+            jsonSchema: Type.toJsonSchema(Organization.Organization)
           });
-          const notesView = createView({
-            name: "Notes",
+          const notesView = View2.make({
             query: notesQuery,
-            jsonSchema: Type2.toJsonSchema(Markdown2.Document),
-            presentation: Obj2.make(DataType2.Collection, {
-              objects: []
-            })
+            jsonSchema: Type.toJsonSchema(Markdown2.Document)
           });
-          return space.db.add(DataType2.makeProject({
+          return space.db.add(Pipeline.make({
             name: "Investor Research",
-            collections: [
-              mailboxView,
-              contactsView,
-              organizationsView,
-              notesView
-            ].map((view) => Ref2.make(view))
+            columns: [
+              {
+                name: "Mailbox",
+                view: Ref.make(mailboxView),
+                order: []
+              },
+              {
+                name: "Contacts",
+                view: Ref.make(contactsView),
+                order: []
+              },
+              {
+                name: "Organizations",
+                view: Ref.make(organizationsView),
+                order: []
+              },
+              {
+                name: "Notes",
+                view: Ref.make(notesView),
+                order: []
+              }
+            ]
           }));
         });
         cb?.(objects.flat());
@@ -411,7 +462,9 @@ var generator2 = () => ({
       "objectChange-queue",
       async (space, n, cb) => {
         const objects = range2(n, () => {
-          const { canvasModel, computeModel } = createQueueSinkPreset(space, "subscription", (triggerSpec) => triggerSpec.query = Query.select(Filter2.typename("dxos.org/type/Chess")).ast, "type");
+          const { canvasModel, computeModel } = createQueueSinkPreset(space, "subscription", (triggerSpec) => triggerSpec.query = {
+            ast: Query.select(Filter2.typename("dxos.org/type/Chess")).ast
+          }, "type");
           return addToSpace("objectChange-queue", space, canvasModel, computeModel);
         });
         cb?.(objects);
@@ -809,20 +862,22 @@ var createQueueSinkPreset = (space, triggerKind, initSpec, triggerOutputName) =>
     const triggerSpec = functionTrigger.spec;
     invariant(triggerSpec && triggerSpec.kind === triggerKind, "No trigger spec.", {
       F: __dxlog_file,
-      L: 597,
+      L: 756,
       S: void 0,
       A: [
         "triggerSpec && triggerSpec.kind === triggerKind",
         "'No trigger spec.'"
       ]
     });
-    initSpec(triggerSpec);
+    Obj.change(functionTrigger, (ft) => {
+      initSpec(ft.spec);
+    });
   });
   const computeModel = createComputeGraph(canvasModel);
   const templateComputeNode = computeModel.nodes.find((n) => n.id === template.node);
   invariant(templateComputeNode, "Template compute node was not created.", {
     F: __dxlog_file,
-    L: 604,
+    L: 765,
     S: void 0,
     A: [
       "templateComputeNode",
@@ -835,7 +890,7 @@ var createQueueSinkPreset = (space, triggerKind, initSpec, triggerOutputName) =>
     '  "id": "@{{changeId}}"',
     "}"
   ].join("\n");
-  templateComputeNode.inputSchema = Type2.toJsonSchema(Schema.Struct({
+  templateComputeNode.inputSchema = Type.toJsonSchema(Schema.Struct({
     type: Schema.String,
     changeId: Schema.String
   }));
@@ -846,9 +901,9 @@ var createQueueSinkPreset = (space, triggerKind, initSpec, triggerOutputName) =>
   };
 };
 var addToSpace = (name, space, canvas, compute) => {
-  return space.db.add(Obj2.make(CanvasBoardType, {
+  return space.db.add(Obj.make(CanvasBoard.CanvasBoard, {
     name,
-    computeGraph: Ref2.make(compute.root),
+    computeGraph: Ref.make(compute.root),
     layout: canvas.graph
   }));
 };
@@ -884,16 +939,18 @@ var setupQueue = (space, canvasModel, args) => {
 var attachTrigger = (functionTrigger, computeModel) => {
   invariant(functionTrigger, void 0, {
     F: __dxlog_file,
-    L: 643,
+    L: 805,
     S: void 0,
     A: [
       "functionTrigger",
       ""
     ]
   });
-  functionTrigger.function = Ref2.make(computeModel.root);
   const inputNode = computeModel.nodes.find((node) => node.type === NODE_INPUT2);
-  functionTrigger.inputNodeId = inputNode.id;
+  Obj.change(functionTrigger, (t) => {
+    t.function = Ref.make(computeModel.root);
+    t.inputNodeId = inputNode.id;
+  });
 };
 var rawPosition = (args) => {
   return {
@@ -940,157 +997,151 @@ var position = (rect) => {
 };
 
 // src/components/SpaceGenerator/SchemaTable.tsx
-import { useSignals as _useSignals } from "@preact-signals/safe-react/tracking";
 import React from "react";
 import { IconButton } from "@dxos/react-ui";
 var SchemaTable = ({ types, objects = {}, label, onClick }) => {
-  var _effect = _useSignals();
-  try {
-    return /* @__PURE__ */ React.createElement("div", {
-      className: "grid grid-cols-[1fr_80px_40px] gap-1 overflow-none"
-    }, /* @__PURE__ */ React.createElement("h2", {
-      className: "p-2"
-    }, label), types.map((type) => /* @__PURE__ */ React.createElement("div", {
-      key: type.typename,
-      className: "grid grid-cols-subgrid col-span-3 items-center"
-    }, /* @__PURE__ */ React.createElement("div", {
-      className: "px-2 text-sm font-mono text-subdued"
-    }, type.typename), /* @__PURE__ */ React.createElement("div", {
-      className: "px-2 text-right font-mono"
-    }, objects[type.typename] ?? 0), /* @__PURE__ */ React.createElement(IconButton, {
-      variant: "ghost",
-      icon: "ph--plus--regular",
-      iconOnly: true,
-      label: "Create data",
-      onClick: () => onClick(type.typename)
-    }))));
-  } finally {
-    _effect.f();
-  }
+  return /* @__PURE__ */ React.createElement("div", {
+    className: "grid grid-cols-[1fr_80px_40px] gap-1 overflow-none"
+  }, /* @__PURE__ */ React.createElement("h2", {
+    className: "p-2"
+  }, label), types.map((type) => /* @__PURE__ */ React.createElement("div", {
+    key: type.typename,
+    className: "grid grid-cols-subgrid col-span-3 items-center"
+  }, /* @__PURE__ */ React.createElement("div", {
+    className: "pli-2 text-sm font-mono text-subdued"
+  }, type.typename), /* @__PURE__ */ React.createElement("div", {
+    className: "pli-2 text-right font-mono"
+  }, objects[type.typename] ?? 0), /* @__PURE__ */ React.createElement(IconButton, {
+    variant: "ghost",
+    icon: "ph--plus--regular",
+    iconOnly: true,
+    label: "Create data",
+    onClick: () => onClick(type.typename)
+  }))));
 };
 
 // src/components/SpaceGenerator/SpaceGenerator.tsx
 var SpaceGenerator = ({ space, onCreateObjects }) => {
-  var _effect = _useSignals2();
-  try {
-    const { dispatchPromise: dispatch } = useIntentDispatcher();
-    const client = useClient();
-    const staticTypes = [
-      Markdown3.Document,
-      DiagramType2,
-      SheetType2,
-      ComputeGraph2
-    ];
-    const recordTypes = [
-      DataType3.Organization,
-      DataType3.Person,
-      DataType3.Task
-    ];
-    const [count, setCount] = useState(1);
-    const [info, setInfo] = useState({});
-    const presets = useMemo(() => generator2(), []);
-    const typeMap = useMemo(() => {
-      client.addTypes([
-        ...staticTypes,
-        ...recordTypes,
-        ...presets.schemas
-      ]);
-      const recordGenerators = new Map(recordTypes.map((type) => [
-        type.typename,
-        createGenerator(client, dispatch, type)
-      ]));
-      return new Map([
-        ...staticGenerators,
-        ...presets.items,
-        ...recordGenerators
-      ]);
-    }, [
-      client,
-      recordTypes
+  const { invokePromise } = useOperationInvoker();
+  const client = useClient();
+  const staticTypes = [
+    Markdown3.Document,
+    Diagram2.Diagram,
+    Sheet2.Sheet,
+    ComputeGraph2
+  ];
+  const recordTypes = [
+    Organization2.Organization,
+    Person2.Person,
+    Task.Task
+  ];
+  const [count, setCount] = useState(1);
+  const [info, setInfo] = useState({});
+  const presets = useMemo(() => generator2(), []);
+  useAsyncEffect(async () => {
+    await client.addTypes([
+      ...staticTypes,
+      ...recordTypes,
+      ...presets.schemas
     ]);
-    const updateInfo = async () => {
-      const echoSchema = await space.db.schemaRegistry.query().run();
-      const staticSchema = space.db.graph.schemaRegistry.schemas;
-      const { objects } = await space.db.query(Filter3.everything()).run();
-      const objectMap = sortKeys(objects.reduce((map, obj) => {
-        const type = getTypename(obj);
-        if (type) {
-          const count2 = map[type] ?? 0;
-          map[type] = count2 + 1;
-        }
-        return map;
-      }, {}));
-      setInfo({
-        schema: {
-          static: staticSchema.length,
-          mutable: echoSchema.length
-        },
-        objects: objectMap
-      });
-    };
-    useAsyncEffect(updateInfo, [
-      space
+  }, [
+    client
+  ]);
+  const typeMap = useMemo(() => {
+    const recordGenerators = new Map(recordTypes.map((type) => [
+      type.typename,
+      createGenerator(client, invokePromise, type)
+    ]));
+    return new Map([
+      ...staticGenerators,
+      ...presets.items,
+      ...recordGenerators
     ]);
-    const handleCreateData = useCallback(async (typename) => {
-      const constructor = typeMap.get(typename);
-      if (constructor) {
-        await constructor(space, count, onCreateObjects);
-        await updateInfo();
+  }, [
+    client,
+    recordTypes,
+    invokePromise
+  ]);
+  const updateInfo = async () => {
+    const echoSchema = await space.db.schemaRegistry.query().run();
+    const staticSchema = await space.db.graph.schemaRegistry.query().run();
+    const objects = await space.db.query(Filter3.everything()).run();
+    const objectMap = sortKeys(objects.reduce((map, obj) => {
+      const type = Obj2.getTypename(obj);
+      if (type) {
+        const count2 = map[type] ?? 0;
+        map[type] = count2 + 1;
       }
-    }, [
-      typeMap,
-      count
-    ]);
-    return /* @__PURE__ */ React2.createElement("div", {
-      role: "none",
-      className: "flex flex-col grow overflow-hidden"
-    }, /* @__PURE__ */ React2.createElement(Toolbar.Root, {
-      classNames: "border-be border-subduedSeparator"
-    }, /* @__PURE__ */ React2.createElement(IconButton2, {
-      icon: "ph--arrow-clockwise--regular",
-      iconOnly: true,
-      label: "Refresh",
-      onClick: updateInfo
-    }), /* @__PURE__ */ React2.createElement(Toolbar.Separator, {
-      variant: "gap"
-    }), /* @__PURE__ */ React2.createElement(Input.Root, null, /* @__PURE__ */ React2.createElement(Input.TextInput, {
-      type: "number",
-      min: 1,
-      max: 100,
-      placeholder: "Count",
-      classNames: "!w-[4rem] !text-right",
-      size: 8,
-      value: count,
-      onChange: (ev) => setCount(parseInt(ev.target.value))
-    }))), /* @__PURE__ */ React2.createElement("div", {
-      className: "flex flex-col overflow-y-auto divide-y divide-separator"
-    }, /* @__PURE__ */ React2.createElement(SchemaTable, {
-      types: staticTypes,
-      objects: info.objects,
-      label: "Static Types",
-      onClick: handleCreateData
-    }), /* @__PURE__ */ React2.createElement(SchemaTable, {
-      types: recordTypes,
-      objects: info.objects,
-      label: "Record Types",
-      onClick: handleCreateData
-    }), /* @__PURE__ */ React2.createElement(SchemaTable, {
-      types: presets.types,
-      objects: info.objects,
-      label: "Presets",
-      onClick: handleCreateData
-    }), /* @__PURE__ */ React2.createElement("div", null, /* @__PURE__ */ React2.createElement(SyntaxHighlighter, {
-      language: "json",
-      classNames: "text-xs"
-    }, JSON.stringify({
-      space,
-      ...info
-    }, jsonKeyReplacer({
-      truncate: true
-    }), 2)))));
-  } finally {
-    _effect.f();
-  }
+      return map;
+    }, {}));
+    setInfo({
+      schema: {
+        static: staticSchema.length,
+        mutable: echoSchema.length
+      },
+      objects: objectMap
+    });
+  };
+  useAsyncEffect(updateInfo, [
+    space
+  ]);
+  const handleCreateData = useCallback(async (typename) => {
+    const constructor = typeMap.get(typename);
+    if (constructor) {
+      await constructor(space, count, onCreateObjects);
+      await updateInfo();
+    }
+  }, [
+    typeMap,
+    count
+  ]);
+  return /* @__PURE__ */ React2.createElement(Layout.Main, {
+    toolbar: true
+  }, /* @__PURE__ */ React2.createElement(Toolbar.Root, null, /* @__PURE__ */ React2.createElement(IconButton2, {
+    icon: "ph--arrow-clockwise--regular",
+    iconOnly: true,
+    label: "Refresh",
+    onClick: updateInfo
+  }), /* @__PURE__ */ React2.createElement(Toolbar.Separator, {
+    variant: "gap"
+  }), /* @__PURE__ */ React2.createElement(Input.Root, null, /* @__PURE__ */ React2.createElement(Input.TextInput, {
+    type: "number",
+    placeholder: "Count",
+    classNames: "is-[4rem] text-right",
+    min: 1,
+    max: 100,
+    size: 8,
+    value: count,
+    onChange: (event) => setCount(parseInt(event.target.value))
+  }))), /* @__PURE__ */ React2.createElement(ScrollArea.Root, {
+    thin: true,
+    orientation: "vertical"
+  }, /* @__PURE__ */ React2.createElement(ScrollArea.Viewport, {
+    classNames: "gap-4 divide-y divide-subduedSeparator"
+  }, /* @__PURE__ */ React2.createElement(SchemaTable, {
+    types: staticTypes,
+    objects: info.objects,
+    label: "Static Types",
+    onClick: handleCreateData
+  }), /* @__PURE__ */ React2.createElement(SchemaTable, {
+    types: recordTypes,
+    objects: info.objects,
+    label: "Record Types",
+    onClick: handleCreateData
+  }), /* @__PURE__ */ React2.createElement(SchemaTable, {
+    types: presets.types,
+    objects: info.objects,
+    label: "Presets",
+    onClick: handleCreateData
+  }), /* @__PURE__ */ React2.createElement(SyntaxHighlighter, {
+    language: "json",
+    classNames: "text-xs"
+  }, JSON.stringify({
+    space,
+    ...info
+  }, jsonKeyReplacer({
+    truncate: true
+  }), 2)))));
 };
 
 // src/components/SpaceGenerator/index.ts
@@ -1098,4 +1149,4 @@ var SpaceGenerator_default = SpaceGenerator;
 export {
   SpaceGenerator_default as default
 };
-//# sourceMappingURL=SpaceGenerator-SRCIEELZ.mjs.map
+//# sourceMappingURL=SpaceGenerator-B4A7J26E.mjs.map