@dxos/plugin-explorer 0.8.4-main.406dc2a → 0.8.4-main.52d7546f51

package/package.json

@@ -1,31 +1,35 @@
 {
   "name": "@dxos/plugin-explorer",
-  "version": "0.8.4-main.406dc2a",
+  "version": "0.8.4-main.52d7546f51",
   "description": "Braneframe data visualization plugin",
   "homepage": "https://dxos.org",
   "bugs": "https://github.com/dxos/dxos/issues",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/dxos/dxos"
+  },
   "license": "MIT",
   "author": "DXOS.org",
   "sideEffects": true,
   "type": "module",
   "exports": {
     ".": {
-      "source": "./src/index.ts",
-      "types": "./dist/types/src/index.d.ts",
       "browser": "./dist/lib/browser/index.mjs",
-      "node": "./dist/lib/node-esm/index.mjs"
+      "node": "./dist/lib/node-esm/index.mjs",
+      "source": "./src/index.ts",
+      "types": "./dist/types/src/index.d.ts"
     },
     "./meta": {
-      "source": "./src/meta.ts",
-      "types": "./dist/types/src/meta.d.ts",
       "browser": "./dist/lib/browser/meta.mjs",
-      "node": "./dist/lib/node-esm/meta.mjs"
+      "node": "./dist/lib/node-esm/meta.mjs",
+      "source": "./src/meta.ts",
+      "types": "./dist/types/src/meta.d.ts"
     },
     "./types": {
-      "source": "./src/types/index.ts",
-      "types": "./dist/types/src/types/index.d.ts",
       "browser": "./dist/lib/browser/types/index.mjs",
-      "node": "./dist/lib/node-esm/types/index.mjs"
+      "node": "./dist/lib/node-esm/types/index.mjs",
+      "source": "./src/types/index.ts",
+      "types": "./dist/types/src/types/index.d.ts"
     }
   },
   "types": "dist/types/src/index.d.ts",
@@ -45,60 +49,62 @@
   ],
   "dependencies": {
     "3d-force-graph": "^1.76.1",
+    "@effect-atom/atom-react": "^0.5.0",
     "@observablehq/plot": "^0.6.11",
-    "@preact-signals/safe-react": "^0.9.0",
-    "@preact/signals-core": "^1.12.1",
     "d3": "^7.9.0",
-    "effect": "3.18.3",
+    "effect": "3.19.16",
     "force-graph": "^1.49.4",
     "lodash.defaultsdeep": "^4.6.1",
     "react-resize-detector": "^11.0.1",
     "three": "^0.178.0",
     "topojson-client": "^3.1.0",
-    "@dxos/app-framework": "0.8.4-main.406dc2a",
-    "@dxos/client": "0.8.4-main.406dc2a",
-    "@dxos/async": "0.8.4-main.406dc2a",
-    "@dxos/echo": "0.8.4-main.406dc2a",
-    "@dxos/echo-query": "0.8.4-main.406dc2a",
-    "@dxos/graph": "0.8.4-main.406dc2a",
-    "@dxos/invariant": "0.8.4-main.406dc2a",
-    "@dxos/live-object": "0.8.4-main.406dc2a",
-    "@dxos/log": "0.8.4-main.406dc2a",
-    "@dxos/plugin-client": "0.8.4-main.406dc2a",
-    "@dxos/plugin-graph": "0.8.4-main.406dc2a",
-    "@dxos/plugin-search": "0.8.4-main.406dc2a",
-    "@dxos/react-client": "0.8.4-main.406dc2a",
-    "@dxos/plugin-space": "0.8.4-main.406dc2a",
-    "@dxos/react-ui-attention": "0.8.4-main.406dc2a",
-    "@dxos/react-ui-components": "0.8.4-main.406dc2a",
-    "@dxos/react-ui-graph": "0.8.4-main.406dc2a",
-    "@dxos/react-ui-stack": "0.8.4-main.406dc2a",
-    "@dxos/schema": "0.8.4-main.406dc2a",
-    "@dxos/util": "0.8.4-main.406dc2a"
+    "@dxos/app-framework": "0.8.4-main.52d7546f51",
+    "@dxos/async": "0.8.4-main.52d7546f51",
+    "@dxos/client": "0.8.4-main.52d7546f51",
+    "@dxos/echo-query": "0.8.4-main.52d7546f51",
+    "@dxos/app-toolkit": "0.8.4-main.52d7546f51",
+    "@dxos/echo": "0.8.4-main.52d7546f51",
+    "@dxos/invariant": "0.8.4-main.52d7546f51",
+    "@dxos/graph": "0.8.4-main.52d7546f51",
+    "@dxos/log": "0.8.4-main.52d7546f51",
+    "@dxos/operation": "0.8.4-main.52d7546f51",
+    "@dxos/plugin-client": "0.8.4-main.52d7546f51",
+    "@dxos/plugin-search": "0.8.4-main.52d7546f51",
+    "@dxos/plugin-graph": "0.8.4-main.52d7546f51",
+    "@dxos/plugin-space": "0.8.4-main.52d7546f51",
+    "@dxos/react-ui-attention": "0.8.4-main.52d7546f51",
+    "@dxos/react-ui-components": "0.8.4-main.52d7546f51",
+    "@dxos/react-client": "0.8.4-main.52d7546f51",
+    "@dxos/react-ui-graph": "0.8.4-main.52d7546f51",
+    "@dxos/react-ui-mosaic": "0.8.4-main.52d7546f51",
+    "@dxos/schema": "0.8.4-main.52d7546f51",
+    "@dxos/react-ui-stack": "0.8.4-main.52d7546f51",
+    "@dxos/types": "0.8.4-main.52d7546f51",
+    "@dxos/util": "0.8.4-main.52d7546f51"
   },
   "devDependencies": {
     "@types/d3": "^7.4.3",
     "@types/d3-hierarchy": "^3.1.6",
     "@types/lodash.defaultsdeep": "^4.6.6",
-    "@types/react": "~19.2.2",
-    "@types/react-dom": "~19.2.1",
+    "@types/react": "~19.2.7",
+    "@types/react-dom": "~19.2.3",
     "@types/topojson-client": "^3.1.4",
     "@types/topojson-specification": "^1.0.5",
-    "react": "~19.2.0",
-    "react-dom": "~19.2.0",
+    "react": "~19.2.3",
+    "react-dom": "~19.2.3",
     "vite": "7.1.9",
-    "@dxos/react-ui": "0.8.4-main.406dc2a",
-    "@dxos/random": "0.8.4-main.406dc2a",
-    "@dxos/echo-generator": "0.8.4-main.406dc2a",
-    "@dxos/react-ui-theme": "0.8.4-main.406dc2a",
-    "@dxos/storybook-utils": "0.8.4-main.406dc2a"
+    "@dxos/echo-generator": "0.8.4-main.52d7546f51",
+    "@dxos/random": "0.8.4-main.52d7546f51",
+    "@dxos/react-ui": "0.8.4-main.52d7546f51",
+    "@dxos/ui-theme": "0.8.4-main.52d7546f51",
+    "@dxos/storybook-utils": "0.8.4-main.52d7546f51"
   },
   "peerDependencies": {
-    "effect": "^3.13.3",
-    "react": "^19.0.0",
-    "react-dom": "^19.0.0",
-    "@dxos/react-ui": "0.8.4-main.406dc2a",
-    "@dxos/react-ui-theme": "0.8.4-main.406dc2a"
+    "effect": "3.19.16",
+    "react": "~19.2.3",
+    "react-dom": "~19.2.3",
+    "@dxos/react-ui": "0.8.4-main.52d7546f51",
+    "@dxos/ui-theme": "0.8.4-main.52d7546f51"
   },
   "publishConfig": {
     "access": "public"

package/src/ExplorerPlugin.tsx

@@ -2,60 +2,37 @@
 // Copyright 2023 DXOS.org
 //
 
-import { Capabilities, Events, contributes, createIntent, defineModule, definePlugin } from '@dxos/app-framework';
-import { ClientCapabilities, ClientEvents } from '@dxos/plugin-client';
-import { SpaceCapabilities } from '@dxos/plugin-space';
-import { defineObjectForm } from '@dxos/plugin-space/types';
+import * as Effect from 'effect/Effect';
 
-import { IntentResolver, ReactSurface } from './capabilities';
+import { Plugin } from '@dxos/app-framework';
+import { AppPlugin } from '@dxos/app-toolkit';
+import { Type } from '@dxos/echo';
+import { type CreateObject } from '@dxos/plugin-space/types';
+import { View } from '@dxos/schema';
+
+import { ReactSurface } from './capabilities';
 import { meta } from './meta';
 import { translations } from './translations';
 import { ExplorerAction, Graph } from './types';
 
-export const ExplorerPlugin = definePlugin(meta, () => [
-  defineModule({
-    id: `${meta.id}/module/translations`,
-    activatesOn: Events.SetupTranslations,
-    activate: () => contributes(Capabilities.Translations, translations),
-  }),
-  defineModule({
-    id: `${meta.id}/module/metadata`,
-    activatesOn: Events.SetupMetadata,
-    activate: () =>
-      contributes(Capabilities.Metadata, {
-        id: Graph.Graph.typename,
-        metadata: {
-          icon: 'ph--graph--regular',
-        },
-      }),
-  }),
-  defineModule({
-    id: `${meta.id}/module/object-form`,
-    activatesOn: ClientEvents.SetupSchema,
-    activate: () =>
-      contributes(
-        SpaceCapabilities.ObjectForm,
-        defineObjectForm({
-          objectSchema: Graph.Graph,
-          formSchema: ExplorerAction.GraphProps,
-          hidden: true,
-          getIntent: (props, options) => createIntent(ExplorerAction.CreateGraph, { ...props, space: options.space }),
-        }),
-      ),
-  }),
-  defineModule({
-    id: `${meta.id}/module/schema`,
-    activatesOn: ClientEvents.SetupSchema,
-    activate: () => contributes(ClientCapabilities.Schema, [Graph.Graph]),
-  }),
-  defineModule({
-    id: `${meta.id}/module/react-surface`,
-    activatesOn: Events.SetupReactSurface,
-    activate: ReactSurface,
-  }),
-  defineModule({
-    id: `${meta.id}/module/intent-resolver`,
-    activatesOn: Events.SetupIntentResolver,
-    activate: IntentResolver,
+export const ExplorerPlugin = Plugin.define(meta).pipe(
+  AppPlugin.addMetadataModule({
+    metadata: {
+      id: Type.getTypename(Graph.Graph),
+      metadata: {
+        icon: 'ph--graph--regular',
+        iconHue: 'green',
+        inputSchema: ExplorerAction.GraphProps,
+        createObject: ((props, { db }) =>
+          Effect.promise(async () => {
+            const { view } = await View.makeFromDatabase({ db, typename: props.typename });
+            return Graph.make({ name: props.name, view });
+          })) satisfies CreateObject,
+      },
+    },
   }),
-]);
+  AppPlugin.addSchemaModule({ schema: [Graph.Graph] }),
+  AppPlugin.addSurfaceModule({ activate: ReactSurface }),
+  AppPlugin.addTranslationsModule({ translations }),
+  Plugin.make,
+);

package/src/capabilities/index.ts

@@ -2,7 +2,4 @@
 // Copyright 2025 DXOS.org
 //
 
-import { lazy } from '@dxos/app-framework';
-
-export const IntentResolver = lazy(() => import('./intent-resolver'));
-export const ReactSurface = lazy(() => import('./react-surface'));
+export * from './react-surface';

package/src/capabilities/react-surface/index.ts

@@ -0,0 +1,7 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { Capability } from '@dxos/app-framework';
+
+export const ReactSurface = Capability.lazy('ReactSurface', () => import('./react-surface'));

package/src/capabilities/react-surface/react-surface.tsx

@@ -0,0 +1,31 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import * as Effect from 'effect/Effect';
+import React from 'react';
+
+import { Capabilities, Capability } from '@dxos/app-framework';
+import { Surface } from '@dxos/app-framework/ui';
+import { Obj } from '@dxos/echo';
+import { type View } from '@dxos/schema';
+
+import { ExplorerContainer } from '../../components';
+import { meta } from '../../meta';
+import { Graph } from '../../types';
+
+export default Capability.makeModule(() =>
+  Effect.succeed(
+    Capability.contributes(
+      Capabilities.ReactSurface,
+      Surface.create({
+        id: `${meta.id}/article`,
+        role: ['article', 'section'],
+        filter: (data): data is { subject: View.View } => Obj.instanceOf(Graph.Graph, data.subject),
+        component: ({ data, role }) => {
+          return <ExplorerContainer role={role} subject={data.subject} />;
+        },
+      }),
+    ),
+  ),
+);

package/src/components/Chart/Chart.stories.tsx

@@ -6,7 +6,7 @@ import { type Meta } from '@storybook/react-vite';
 import React from 'react';
 
 import { ClientRepeater } from '@dxos/react-client/testing';
-import { withTheme } from '@dxos/react-ui/testing';
+import { withLayout, withTheme } from '@dxos/react-ui/testing';
 
 import CitiesData from '../../../data/cities.js';
 
@@ -34,7 +34,7 @@ export const Default = () => <ClientRepeater component={Story} />;
 const meta = {
   title: 'plugins/plugin-explorer/Chart',
   component: Chart,
-  decorators: [withTheme],
+  decorators: [withTheme(), withLayout()],
   parameters: {
     layout: 'fullscreen',
   },

package/src/components/ExplorerContainer.tsx

@@ -4,33 +4,31 @@
 
 import React, { useCallback, useMemo, useState } from 'react';
 
+import { type SurfaceComponentProps } from '@dxos/app-toolkit/ui';
 import { type Filter } from '@dxos/echo';
 import { QueryBuilder } from '@dxos/echo-query';
 import { useGlobalSearch } from '@dxos/plugin-search';
-import { getSpace } from '@dxos/react-client/echo';
-import { Toolbar } from '@dxos/react-ui';
+import { getSpace, useObject } from '@dxos/react-client/echo';
+import { Layout, Toolbar } from '@dxos/react-ui';
 import { QueryEditor, type QueryEditorProps } from '@dxos/react-ui-components';
-import { StackItem } from '@dxos/react-ui-stack';
-import { type DataType } from '@dxos/schema';
+import { type View } from '@dxos/schema';
 
 import { useGraphModel } from '../hooks';
 
 import { D3ForceGraph } from './Graph';
 
-type ExplorerContainerProps = {
-  role: string;
-  view: DataType.View;
-};
+export type ExplorerContainerProps = SurfaceComponentProps<View.View>;
 
-const ExplorerContainer = ({ role, view }: ExplorerContainerProps) => {
-  const space = getSpace(view);
+const ExplorerContainer = ({ role, subject: view }: ExplorerContainerProps) => {
+  useObject(view);
+  const space = view && getSpace(view);
   const [filter, setFilter] = useState<Filter.Any>();
   const model = useGraphModel(space, filter);
   const { match } = useGlobalSearch();
 
   const builder = useMemo(() => new QueryBuilder(), []);
   const handleChange = useCallback<NonNullable<QueryEditorProps['onChange']>>((value) => {
-    setFilter(builder.build(value));
+    setFilter(builder.build(value).filter);
   }, []);
 
   const showToolbar = role === 'article';
@@ -40,14 +38,14 @@ const ExplorerContainer = ({ role, view }: ExplorerContainerProps) => {
   }
 
   return (
-    <StackItem.Content toolbar={showToolbar}>
+    <Layout.Main role={role} toolbar={showToolbar}>
       {showToolbar && (
         <Toolbar.Root>
           <QueryEditor db={space.db} onChange={handleChange} />
         </Toolbar.Root>
       )}
       <D3ForceGraph model={model} match={match} />
-    </StackItem.Content>
+    </Layout.Main>
   );
 };
 

package/src/components/Globe/Globe.stories.tsx

@@ -10,7 +10,7 @@ import { useResizeDetector } from 'react-resize-detector';
 import { feature } from 'topojson-client';
 
 import { ClientRepeater } from '@dxos/react-client/testing';
-import { withTheme } from '@dxos/react-ui/testing';
+import { withLayout, withTheme } from '@dxos/react-ui/testing';
 
 import CitiesData from '../../../data/cities.js';
 import CountriesData from '../../../data/countries-110m.js';
@@ -81,7 +81,7 @@ const ExtendedStory = () => {
 
 const meta = {
   title: 'plugins/plugin-explorer/Globe',
-  decorators: [withTheme],
+  decorators: [withTheme(), withLayout()],
   parameters: {
     layout: 'fullscreen',
   },

package/src/components/Graph/D3ForceGraph.stories.tsx

@@ -5,15 +5,18 @@
 import { type Meta, type StoryObj } from '@storybook/react-vite';
 import React, { useState } from 'react';
 
+import { Type } from '@dxos/echo';
 import { faker } from '@dxos/random';
 import { useClient } from '@dxos/react-client';
 import { type Space } from '@dxos/react-client/echo';
 import { withClientProvider } from '@dxos/react-client/testing';
 import { useAsyncEffect } from '@dxos/react-ui';
-import { withTheme } from '@dxos/react-ui/testing';
-import { DataType } from '@dxos/schema';
+import { withLayout, withTheme } from '@dxos/react-ui/testing';
+import { View } from '@dxos/schema';
 import { type ValueGenerator } from '@dxos/schema/testing';
+import { withRegistry } from '@dxos/storybook-utils';
 import { render } from '@dxos/storybook-utils';
+import { HasRelationship, Organization, Person, Pipeline } from '@dxos/types';
 
 import { useGraphModel } from '../../hooks';
 import { Graph } from '../../types';
@@ -28,19 +31,20 @@ faker.seed(1);
 const DefaultStory = () => {
   const client = useClient();
   const [space, setSpace] = useState<Space>();
-  const [view, setView] = useState<DataType.View>();
+  const [graph, setGraph] = useState<Graph.Graph>();
 
   useAsyncEffect(async () => {
     const space = client.spaces.default;
     void generate(space, generator);
-    const { view } = await Graph.makeView({ client, space, name: 'Test', typename: Graph.Graph.typename });
-    space.db.add(view);
+    const { view } = await View.makeFromDatabase({ db: space.db, typename: Type.getTypename(Graph.Graph) });
+    const graph = Graph.make({ name: 'Test', view });
+    space.db.add(graph);
     setSpace(space);
-    setView(view);
-  }, []);
+    setGraph(graph);
+  }, [client]);
 
   const model = useGraphModel(space);
-  if (!model || !space || !view) {
+  if (!model || !space || !graph) {
     return null;
   }
 
@@ -52,16 +56,18 @@ const meta = {
   component: D3ForceGraph,
   render: render(DefaultStory),
   decorators: [
-    withTheme,
+    withRegistry,
+    withTheme(),
+    withLayout(),
     withClientProvider({
       createSpace: true,
       types: [
         Graph.Graph,
-        DataType.View,
-        DataType.Organization,
-        DataType.Project,
-        DataType.Person,
-        DataType.HasRelationship,
+        View.View,
+        Organization.Organization,
+        Pipeline.Pipeline,
+        Person.Person,
+        HasRelationship.HasRelationship,
       ],
     }),
   ],

package/src/components/Graph/D3ForceGraph.tsx

@@ -15,8 +15,8 @@ import {
   SVG,
   type SVGContext,
 } from '@dxos/react-ui-graph';
-import { getHashStyles } from '@dxos/react-ui-theme';
 import { type SpaceGraphEdge, type SpaceGraphModel, type SpaceGraphNode } from '@dxos/schema';
+import { getHashStyles } from '@dxos/ui-theme';
 
 import '@dxos/react-ui-graph/styles/graph.css';
 
@@ -52,7 +52,7 @@ export const D3ForceGraph = ({ classNames, model, selection: _selection, grid, .
 
   const graph = useRef<GraphController>(null);
   const selection = useMemo(() => _selection ?? new SelectionModel(), [_selection]);
-  useEffect(() => graph.current?.repaint(), [selection.selected.value]);
+  useEffect(() => selection.subscribe(() => graph.current?.repaint()), [selection]);
 
   const handleSelect = useCallback<NonNullable<GraphProps['onSelect']>>(
     (node) => {

package/src/components/Graph/ForceGraph.stories.tsx

@@ -5,15 +5,18 @@
 import { type Meta, type StoryObj } from '@storybook/react-vite';
 import React, { useState } from 'react';
 
+import { Type } from '@dxos/echo';
 import { faker } from '@dxos/random';
 import { useClient } from '@dxos/react-client';
 import { type Space } from '@dxos/react-client/echo';
 import { withClientProvider } from '@dxos/react-client/testing';
 import { useAsyncEffect } from '@dxos/react-ui';
-import { withTheme } from '@dxos/react-ui/testing';
-import { DataType } from '@dxos/schema';
+import { withLayout, withTheme } from '@dxos/react-ui/testing';
+import { View } from '@dxos/schema';
 import { type ValueGenerator } from '@dxos/schema/testing';
+import { withRegistry } from '@dxos/storybook-utils';
 import { render } from '@dxos/storybook-utils';
+import { HasRelationship, Organization, Person, Pipeline } from '@dxos/types';
 
 import { useGraphModel } from '../../hooks';
 import { Graph } from '../../types';
@@ -28,19 +31,20 @@ faker.seed(1);
 const DefaultStory = () => {
   const client = useClient();
   const [space, setSpace] = useState<Space>();
-  const [view, setView] = useState<DataType.View>();
+  const [graph, setGraph] = useState<Graph.Graph>();
 
   useAsyncEffect(async () => {
     const space = client.spaces.default;
     void generate(space, generator);
-    const { view } = await Graph.makeView({ client, space, name: 'Test', typename: Graph.Graph.typename });
-    space.db.add(view);
+    const { view } = await View.makeFromDatabase({ db: space.db, typename: Type.getTypename(Graph.Graph) });
+    const graph = Graph.make({ name: 'Test', view });
+    space.db.add(graph);
     setSpace(space);
-    setView(view);
-  }, []);
+    setGraph(graph);
+  }, [client]);
 
   const model = useGraphModel(space);
-  if (!model || !space || !view) {
+  if (!model || !space || !graph) {
     return null;
   }
 
@@ -52,16 +56,18 @@ const meta = {
   component: ForceGraph,
   render: render(DefaultStory),
   decorators: [
-    withTheme,
+    withRegistry,
+    withTheme(),
+    withLayout(),
     withClientProvider({
       createSpace: true,
       types: [
         Graph.Graph,
-        DataType.View,
-        DataType.HasRelationship,
-        DataType.Organization,
-        DataType.Project,
-        DataType.Person,
+        View.View,
+        HasRelationship.HasRelationship,
+        Organization.Organization,
+        Pipeline.Pipeline,
+        Person.Person,
       ],
     }),
   ],

package/src/components/Graph/adapter.ts

@@ -25,19 +25,25 @@ export class GraphAdapter implements GraphData {
   private readonly _nodes: GraphNode[] = [];
   private readonly _links: GraphLink[] = [];
 
-  constructor(private readonly graph: Graph) {
-    this._nodes = graph.nodes.map((node) => ({
+  constructor(private readonly graph: Graph.Any) {
+    this._nodes = graph.nodes.map((node: Graph.Node.Any) => ({
       id: node.id,
       type: node.type,
       data: node.data,
     }));
 
-    this._links = graph.edges.map((edge) => ({
-      type: edge.type,
-      source: edge.source,
-      target: edge.target,
-      data: edge.data,
-    }));
+    // Build a set of node IDs for efficient lookup.
+    const nodeIds = new Set(this._nodes.map((node) => node.id));
+
+    // Filter out edges where source or target node doesn't exist.
+    this._links = graph.edges
+      .filter((edge: Graph.Edge.Any) => nodeIds.has(edge.source) && nodeIds.has(edge.target))
+      .map((edge: Graph.Edge.Any) => ({
+        type: edge.type,
+        source: edge.source,
+        target: edge.target,
+        data: edge.data,
+      }));
   }
 
   get nodes() {

package/src/components/Graph/testing.ts

@@ -4,16 +4,16 @@
 
 import { type Space } from '@dxos/client/echo';
 import { type Obj, Query, Relation } from '@dxos/echo';
-import { DataType } from '@dxos/schema';
 import { type TypeSpec, type ValueGenerator, createObjectFactory } from '@dxos/schema/testing';
+import { HasRelationship, Organization, Person, Pipeline } from '@dxos/types';
 import { range } from '@dxos/util';
 
-const getObject = (objects: Obj.Any[]) => objects[Math.floor(Math.random() * objects.length)];
+const getObject = (objects: Obj.Unknown[]) => objects[Math.floor(Math.random() * objects.length)];
 
 const defaultTypes: TypeSpec[] = [
-  { type: DataType.Organization, count: 5 },
-  { type: DataType.Project, count: 5 },
-  { type: DataType.Person, count: 10 },
+  { type: Organization.Organization, count: 5 },
+  { type: Pipeline.Pipeline, count: 5 },
+  { type: Person.Person, count: 10 },
 ];
 
 export type GenerateOptions = {
@@ -24,7 +24,10 @@ export type GenerateOptions = {
   };
 };
 
-const defaultRelations: GenerateOptions['relations'] = { kind: 'friend', count: 10 };
+const defaultRelations: GenerateOptions['relations'] = {
+  kind: 'friend',
+  count: 10,
+};
 
 /**
  * @deprecated Use @dxos/schema.
@@ -38,13 +41,13 @@ export const generate = async (
   await createObjects(spec);
 
   // Add relations between objects.
-  const { objects: contacts } = await space.db.query(Query.type(DataType.Person)).run();
+  const contacts = await space.db.query(Query.type(Person.Person)).run();
   for (const _ of range(relations.count)) {
     const source = getObject(contacts);
     const target = getObject(contacts);
     if (source.id !== target.id) {
       space.db.add(
-        Relation.make(DataType.HasRelationship, {
+        Relation.make(HasRelationship.HasRelationship, {
           [Relation.Source]: source,
           [Relation.Target]: target,
           kind: relations.kind,

package/src/components/Tree/Tree.stories.tsx

@@ -8,7 +8,8 @@ import React, { useEffect, useState } from 'react';
 import { faker } from '@dxos/random';
 import { useClient } from '@dxos/react-client';
 import { type ClientRepeatedComponentProps, ClientRepeater } from '@dxos/react-client/testing';
-import { withTheme } from '@dxos/react-ui/testing';
+import { withLayout, withTheme } from '@dxos/react-ui/testing';
+import { withRegistry } from '@dxos/storybook-utils';
 
 import { Tree, type TreeComponentProps } from './Tree';
 import { Tree as TreeModel, TreeType } from './types';
@@ -46,7 +47,7 @@ const meta = {
   title: 'plugins/plugin-explorer/Tree',
   component: Tree as any,
   render: DefaultStory,
-  decorators: [withTheme],
+  decorators: [withRegistry, withTheme(), withLayout()],
   parameters: {
     layout: 'fullscreen',
   },