@dxos/app-graph 0.8.3 → 0.8.4-main.1068cf700f

package/src/graph-builder.test.ts

@@ -2,45 +2,113 @@
 // Copyright 2023 DXOS.org
 //
 
-import { Registry, Rx } from '@effect-rx/rx-react';
-import { Option, pipe } from 'effect';
+import { Atom, Registry } from '@effect-atom/atom-react';
+import * as Context from 'effect/Context';
+import * as Effect from 'effect/Effect';
+import * as Function from 'effect/Function';
+import * as Option from 'effect/Option';
 import { describe, expect, onTestFinished, test } from 'vitest';
 
-import { sleep, Trigger } from '@dxos/async';
+import { Trigger, sleep } from '@dxos/async';
+import { Obj } from '@dxos/echo';
+import { TestSchema } from '@dxos/echo/testing';
 
-import { ROOT_ID } from './graph';
-import { createExtension, GraphBuilder } from './graph-builder';
-import { type Node } from './node';
+import * as Graph from './graph';
+import * as GraphBuilder from './graph-builder';
+import * as Node from './node';
+import * as NodeMatcher from './node-matcher';
 
 const exampleId = (id: number) => `dx:test:${id}`;
 const EXAMPLE_ID = exampleId(1);
 const EXAMPLE_TYPE = 'dxos.org/type/example';
 
 describe('GraphBuilder', () => {
+  describe('resolver', () => {
+    test('works', async () => {
+      const registry = Registry.make();
+      const builder = GraphBuilder.make({ registry });
+      const graph = builder.graph;
+
+      {
+        const node = Graph.getNode(graph, EXAMPLE_ID).pipe(Option.getOrNull);
+        expect(node).to.be.null;
+      }
+
+      // Test direct API
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
+          id: 'resolver',
+          resolver: () => {
+            console.log('resolver');
+            return Atom.make({ id: EXAMPLE_ID, type: EXAMPLE_TYPE, data: 1 });
+          },
+        }),
+      );
+      await Graph.initialize(graph, EXAMPLE_ID);
+
+      {
+        const node = Graph.getNode(graph, EXAMPLE_ID).pipe(Option.getOrNull);
+        expect(node?.id).to.equal(EXAMPLE_ID);
+        expect(node?.type).to.equal(EXAMPLE_TYPE);
+        expect(node?.data).to.equal(1);
+      }
+    });
+
+    test('updates', async () => {
+      const registry = Registry.make();
+      const builder = GraphBuilder.make({ registry });
+      const name = Atom.make('default');
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
+          id: 'resolver',
+          resolver: () => Atom.make((get) => ({ id: EXAMPLE_ID, type: EXAMPLE_TYPE, data: get(name) })),
+        }),
+      );
+      const graph = builder.graph;
+      await Graph.initialize(graph, EXAMPLE_ID);
+
+      {
+        const node = Graph.getNode(graph, EXAMPLE_ID).pipe(Option.getOrNull);
+        expect(node?.data).to.equal('default');
+      }
+
+      registry.set(name, 'updated');
+
+      {
+        const node = Graph.getNode(graph, EXAMPLE_ID).pipe(Option.getOrNull);
+        expect(node?.data).to.equal('updated');
+      }
+    });
+  });
+
   describe('connector', () => {
     test('works', () => {
       const registry = Registry.make();
-      const builder = new GraphBuilder({ registry });
-      builder.addExtension(
-        createExtension({
+      const builder = GraphBuilder.make({ registry });
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
           id: 'outbound-connector',
-          connector: () => Rx.make([{ id: 'child', type: EXAMPLE_TYPE, data: 2 }]),
+          connector: () => Atom.make([{ id: 'child', type: EXAMPLE_TYPE, data: 2 }]),
         }),
       );
-      builder.addExtension(
-        createExtension({
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
           id: 'inbound-connector',
           relation: 'inbound',
-          connector: () => Rx.make([{ id: 'parent', type: EXAMPLE_TYPE, data: 0 }]),
+          connector: () => Atom.make([{ id: 'parent', type: EXAMPLE_TYPE, data: 0 }]),
         }),
       );
 
       const graph = builder.graph;
-      graph.expand(ROOT_ID);
-      graph.expand(ROOT_ID, 'inbound');
+      Graph.expand(graph, Node.RootId);
+      Graph.expand(graph, Node.RootId, 'inbound');
 
-      const outbound = registry.get(graph.connections(ROOT_ID));
-      const inbound = registry.get(graph.connections(ROOT_ID, 'inbound'));
+      const outbound = registry.get(graph.connections(Node.RootId));
+      const inbound = registry.get(graph.connections(Node.RootId, 'inbound'));
 
       expect(outbound).has.length(1);
       expect(outbound[0].id).to.equal('child');
@@ -52,52 +120,54 @@ describe('GraphBuilder', () => {
 
     test('updates', () => {
       const registry = Registry.make();
-      const builder = new GraphBuilder({ registry });
-      const state = Rx.make(0);
-      builder.addExtension(
-        createExtension({
+      const builder = GraphBuilder.make({ registry });
+      const state = Atom.make(0);
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
           id: 'connector',
-          connector: () => Rx.make((get) => [{ id: EXAMPLE_ID, type: EXAMPLE_TYPE, data: get(state) }]),
+          connector: () => Atom.make((get) => [{ id: EXAMPLE_ID, type: EXAMPLE_TYPE, data: get(state) }]),
         }),
       );
       const graph = builder.graph;
-      graph.expand(ROOT_ID);
+      Graph.expand(graph, Node.RootId);
 
       {
-        const [node] = registry.get(graph.connections(ROOT_ID));
+        const [node] = registry.get(graph.connections(Node.RootId));
         expect(node.data).to.equal(0);
       }
 
       {
         registry.set(state, 1);
-        const [node] = registry.get(graph.connections(ROOT_ID));
+        const [node] = registry.get(graph.connections(Node.RootId));
         expect(node.data).to.equal(1);
       }
     });
 
     test('subscribes to updates', () => {
       const registry = Registry.make();
-      const builder = new GraphBuilder({ registry });
-      const state = Rx.make(0);
-      builder.addExtension(
-        createExtension({
+      const builder = GraphBuilder.make({ registry });
+      const state = Atom.make(0);
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
           id: 'connector',
-          connector: () => Rx.make((get) => [{ id: EXAMPLE_ID, type: EXAMPLE_TYPE, data: get(state) }]),
+          connector: () => Atom.make((get) => [{ id: EXAMPLE_ID, type: EXAMPLE_TYPE, data: get(state) }]),
         }),
       );
       const graph = builder.graph;
 
       let count = 0;
-      const cancel = registry.subscribe(graph.connections(ROOT_ID), (_) => {
+      const cancel = registry.subscribe(graph.connections(Node.RootId), (_) => {
         count++;
       });
       onTestFinished(() => cancel());
 
       expect(count).to.equal(0);
-      expect(registry.get(graph.connections(ROOT_ID))).to.have.length(0);
+      expect(registry.get(graph.connections(Node.RootId))).to.have.length(0);
       expect(count).to.equal(1);
 
-      graph.expand(ROOT_ID);
+      Graph.expand(graph, Node.RootId);
       expect(count).to.equal(2);
       registry.set(state, 1);
       expect(count).to.equal(3);
@@ -105,19 +175,20 @@ describe('GraphBuilder', () => {
 
     test('updates with new extensions', () => {
       const registry = Registry.make();
-      const builder = new GraphBuilder({ registry });
-      builder.addExtension(
-        createExtension({
+      const builder = GraphBuilder.make({ registry });
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
           id: 'connector',
-          connector: () => Rx.make([{ id: EXAMPLE_ID, type: EXAMPLE_TYPE }]),
+          connector: () => Atom.make([{ id: EXAMPLE_ID, type: EXAMPLE_TYPE }]),
         }),
       );
       const graph = builder.graph;
-      graph.expand(ROOT_ID);
+      Graph.expand(graph, Node.RootId);
 
-      let nodes: Node[] = [];
+      let nodes: Node.Node[] = [];
       let count = 0;
-      const cancel = registry.subscribe(graph.connections(ROOT_ID), (_nodes) => {
+      const cancel = registry.subscribe(graph.connections(Node.RootId), (_nodes) => {
         count++;
         nodes = _nodes;
       });
@@ -125,14 +196,15 @@ describe('GraphBuilder', () => {
 
       expect(nodes).has.length(0);
       expect(count).to.equal(0);
-      registry.get(graph.connections(ROOT_ID));
+      registry.get(graph.connections(Node.RootId));
       expect(nodes).has.length(1);
       expect(count).to.equal(1);
 
-      builder.addExtension(
-        createExtension({
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
           id: 'connector-2',
-          connector: () => Rx.make([{ id: exampleId(2), type: EXAMPLE_TYPE }]),
+          connector: () => Atom.make([{ id: exampleId(2), type: EXAMPLE_TYPE }]),
         }),
       );
       expect(nodes).has.length(2);
@@ -141,22 +213,23 @@ describe('GraphBuilder', () => {
 
     test('removes', () => {
       const registry = Registry.make();
-      const builder = new GraphBuilder({ registry });
-      const nodes = Rx.make([
+      const builder = GraphBuilder.make({ registry });
+      const nodes = Atom.make([
         { id: exampleId(1), type: EXAMPLE_TYPE },
         { id: exampleId(2), type: EXAMPLE_TYPE },
       ]);
-      builder.addExtension(
-        createExtension({
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
           id: 'connector',
-          connector: () => Rx.make((get) => get(nodes)),
+          connector: () => Atom.make((get) => get(nodes)),
         }),
       );
       const graph = builder.graph;
-      graph.expand(ROOT_ID);
+      Graph.expand(graph, Node.RootId);
 
       {
-        const nodes = registry.get(graph.connections(ROOT_ID));
+        const nodes = registry.get(graph.connections(Node.RootId));
         expect(nodes).has.length(2);
         expect(nodes[0].id).to.equal(exampleId(1));
         expect(nodes[1].id).to.equal(exampleId(2));
@@ -165,7 +238,7 @@ describe('GraphBuilder', () => {
       registry.set(nodes, [{ id: exampleId(3), type: EXAMPLE_TYPE }]);
 
       {
-        const nodes = registry.get(graph.connections(ROOT_ID));
+        const nodes = registry.get(graph.connections(Node.RootId));
         expect(nodes).has.length(1);
         expect(nodes[0].id).to.equal(exampleId(3));
       }
@@ -173,15 +246,15 @@ describe('GraphBuilder', () => {
 
     test('nodes are updated when removed', () => {
       const registry = Registry.make();
-      const builder = new GraphBuilder({ registry });
-      const name = Rx.make('removed');
+      const builder = GraphBuilder.make({ registry });
+      const name = Atom.make('removed');
 
-      builder.addExtension([
-        createExtension({
+      GraphBuilder.addExtension(builder, [
+        GraphBuilder.createExtensionRaw({
           id: 'root',
           connector: (node) =>
-            Rx.make((get) =>
-              pipe(
+            Atom.make((get) =>
+              Function.pipe(
                 get(node),
                 Option.flatMap((node) => (node.id === 'root' ? Option.some(get(name)) : Option.none())),
                 Option.filter((name) => name !== 'removed'),
@@ -202,7 +275,7 @@ describe('GraphBuilder', () => {
       });
       onTestFinished(() => cancel());
 
-      graph.expand(ROOT_ID);
+      Graph.expand(graph, Node.RootId);
       expect(count).to.equal(0);
       expect(exists).to.be.false;
 
@@ -221,23 +294,24 @@ describe('GraphBuilder', () => {
 
     test('sort edges', async () => {
       const registry = Registry.make();
-      const builder = new GraphBuilder({ registry });
-      const nodes = Rx.make([
+      const builder = GraphBuilder.make({ registry });
+      const nodes = Atom.make([
         { id: exampleId(1), type: EXAMPLE_TYPE, data: 1 },
         { id: exampleId(2), type: EXAMPLE_TYPE, data: 2 },
         { id: exampleId(3), type: EXAMPLE_TYPE, data: 3 },
       ]);
-      builder.addExtension(
-        createExtension({
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
           id: 'connector',
-          connector: () => Rx.make((get) => get(nodes)),
+          connector: () => Atom.make((get) => get(nodes)),
         }),
       );
       const graph = builder.graph;
-      graph.expand(ROOT_ID);
+      Graph.expand(graph, Node.RootId);
 
       {
-        const nodes = registry.get(graph.connections(ROOT_ID));
+        const nodes = registry.get(graph.connections(Node.RootId));
         expect(nodes).has.length(3);
         expect(nodes[0].id).to.equal(exampleId(1));
         expect(nodes[1].id).to.equal(exampleId(2));
@@ -254,7 +328,7 @@ describe('GraphBuilder', () => {
       await sleep(0);
 
       {
-        const nodes = registry.get(graph.connections(ROOT_ID));
+        const nodes = registry.get(graph.connections(Node.RootId));
         expect(nodes).has.length(3);
         expect(nodes[0].id).to.equal(exampleId(3));
         expect(nodes[1].id).to.equal(exampleId(1));
@@ -264,16 +338,16 @@ describe('GraphBuilder', () => {
 
     test('updates are constrained', () => {
       const registry = Registry.make();
-      const builder = new GraphBuilder({ registry });
-      const name = Rx.make('default');
-      const sub = Rx.make('default');
+      const builder = GraphBuilder.make({ registry });
+      const name = Atom.make('default');
+      const sub = Atom.make('default');
 
-      builder.addExtension([
-        createExtension({
+      GraphBuilder.addExtension(builder, [
+        GraphBuilder.createExtensionRaw({
           id: 'root',
           connector: (node) =>
-            Rx.make((get) =>
-              pipe(
+            Atom.make((get) =>
+              Function.pipe(
                 get(node),
                 Option.flatMap((node) => (node.id === 'root' ? Option.some(get(name)) : Option.none())),
                 Option.filter((name) => name !== 'removed'),
@@ -282,11 +356,11 @@ describe('GraphBuilder', () => {
               ),
             ),
         }),
-        createExtension({
+        GraphBuilder.createExtensionRaw({
           id: 'connector1',
           connector: (node) =>
-            Rx.make((get) =>
-              pipe(
+            Atom.make((get) =>
+              Function.pipe(
                 get(node),
                 Option.flatMap((node) => (node.id === EXAMPLE_ID ? Option.some(get(sub)) : Option.none())),
                 Option.map((sub) => [{ id: exampleId(2), type: EXAMPLE_TYPE, data: sub }]),
@@ -294,11 +368,11 @@ describe('GraphBuilder', () => {
               ),
             ),
         }),
-        createExtension({
+        GraphBuilder.createExtensionRaw({
           id: 'connector2',
           connector: (node) =>
-            Rx.make((get) =>
-              pipe(
+            Atom.make((get) =>
+              Function.pipe(
                 get(node),
                 Option.flatMap((node) => (node.id === EXAMPLE_ID ? Option.some(node.data) : Option.none())),
                 Option.map((data) => [{ id: exampleId(3), type: EXAMPLE_TYPE, data }]),
@@ -329,13 +403,13 @@ describe('GraphBuilder', () => {
       onTestFinished(() => dependentCancel());
 
       // Counts should not increment until the node is expanded.
-      graph.expand(ROOT_ID);
+      Graph.expand(graph, Node.RootId);
       expect(parentCount).to.equal(1);
       expect(independentCount).to.equal(0);
       expect(dependentCount).to.equal(0);
 
       // Counts should increment when the node is expanded.
-      graph.expand(EXAMPLE_ID);
+      Graph.expand(graph, EXAMPLE_ID);
       expect(parentCount).to.equal(1);
       expect(independentCount).to.equal(1);
       expect(dependentCount).to.equal(1);
@@ -353,7 +427,7 @@ describe('GraphBuilder', () => {
       expect(dependentCount).to.equal(2);
 
       // Independent count should update if its state changes even if the parent is removed.
-      Rx.batch(() => {
+      Atom.batch(() => {
         registry.set(name, 'removed');
         registry.set(sub, 'batch');
       });
@@ -368,7 +442,7 @@ describe('GraphBuilder', () => {
       expect(dependentCount).to.equal(3);
 
       // Counts should not increment when the node is expanded again.
-      graph.expand(EXAMPLE_ID);
+      Graph.expand(graph, EXAMPLE_ID);
       expect(parentCount).to.equal(3);
       expect(independentCount).to.equal(3);
       expect(dependentCount).to.equal(3);
@@ -376,13 +450,14 @@ describe('GraphBuilder', () => {
 
     test('eager graph expansion', async () => {
       const registry = Registry.make();
-      const builder = new GraphBuilder({ registry });
-      builder.addExtension(
-        createExtension({
+      const builder = GraphBuilder.make({ registry });
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
           id: 'connector',
           connector: (node) => {
-            return Rx.make((get) =>
-              pipe(
+            return Atom.make((get) =>
+              Function.pipe(
                 get(node),
                 Option.map((node) => (node.data ? node.data + 1 : 1)),
                 Option.filter((data) => data <= 5),
@@ -397,14 +472,14 @@ describe('GraphBuilder', () => {
       let count = 0;
       const trigger = new Trigger();
       builder.graph.onNodeChanged.on(({ id }) => {
-        builder.graph.expand(id);
+        Graph.expand(builder.graph, id);
         count++;
         if (count === 5) {
           trigger.wake();
         }
       });
 
-      builder.graph.expand(ROOT_ID);
+      Graph.expand(builder.graph, Node.RootId);
       await trigger.wait();
       expect(count).to.equal(5);
     });
@@ -412,13 +487,14 @@ describe('GraphBuilder', () => {
 
   describe('explore', () => {
     test('works', async () => {
-      const builder = new GraphBuilder();
-      builder.addExtension(
-        createExtension({
+      const builder = GraphBuilder.make();
+      GraphBuilder.addExtension(
+        builder,
+        GraphBuilder.createExtensionRaw({
           id: 'connector',
           connector: (node) =>
-            Rx.make((get) =>
-              pipe(
+            Atom.make((get) =>
+              Function.pipe(
                 get(node),
                 Option.map((node) => (node.data ? node.data + 1 : 1)),
                 Option.filter((data) => data <= 5),
@@ -430,7 +506,7 @@ describe('GraphBuilder', () => {
       );
 
       let count = 0;
-      await builder.explore({
+      await GraphBuilder.explore(builder, {
         visitor: () => {
           count++;
         },
@@ -439,4 +515,402 @@ describe('GraphBuilder', () => {
       expect(count).to.equal(6);
     });
   });
+
+  describe('helpers', () => {
+    describe('createConnector', () => {
+      test('creates connector with type inference', () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        const matcher = (node: Node.Node) => NodeMatcher.whenId('root')(node);
+        const factory = (node: Node.Node) => [{ id: 'child', type: EXAMPLE_TYPE, data: node.id }];
+
+        const connector = GraphBuilder.createConnector(matcher, factory);
+
+        GraphBuilder.addExtension(
+          builder,
+          GraphBuilder.createExtensionRaw({
+            id: 'test-connector',
+            connector,
+          }),
+        );
+
+        Graph.expand(graph, Node.RootId);
+
+        const connections = registry.get(graph.connections(Node.RootId));
+        expect(connections).has.length(1);
+        expect(connections[0].id).to.equal('child');
+      });
+    });
+
+    describe('createExtension', () => {
+      test('works with Effect connector', () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        const extensions = Effect.runSync(
+          GraphBuilder.createExtension({
+            id: 'test-extension',
+            match: NodeMatcher.whenNodeType(EXAMPLE_TYPE),
+            connector: (node, get) => Effect.succeed([{ id: 'child', type: EXAMPLE_TYPE, data: node.data }]),
+          }),
+        );
+
+        GraphBuilder.addExtension(builder, extensions);
+
+        const writableGraph = graph as Graph.WritableGraph;
+        Graph.addNode(writableGraph, { id: 'parent', type: EXAMPLE_TYPE, properties: {}, data: 'test' });
+        Graph.expand(graph, 'parent');
+
+        const connections = registry.get(graph.connections('parent'));
+        expect(connections).has.length(1);
+        expect(connections[0].id).to.equal('child');
+        expect(connections[0].data).to.equal('test');
+      });
+
+      test('works with Effect actions', () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        const extensions = Effect.runSync(
+          GraphBuilder.createExtension({
+            id: 'test-extension',
+            match: NodeMatcher.whenNodeType(EXAMPLE_TYPE),
+            actions: (node, get) =>
+              Effect.succeed([
+                {
+                  id: 'test-action',
+                  data: () =>
+                    Effect.sync(() => {
+                      console.log('TestAction');
+                    }),
+                  properties: { label: 'Test' },
+                },
+              ]),
+          }),
+        );
+
+        GraphBuilder.addExtension(builder, extensions);
+
+        const writableGraph = graph as Graph.WritableGraph;
+        Graph.addNode(writableGraph, { id: 'parent', type: EXAMPLE_TYPE, properties: {}, data: 'test' });
+        Graph.expand(graph, 'parent');
+
+        const actions = registry.get(graph.actions('parent'));
+        expect(actions).has.length(1);
+        expect(actions[0].id).to.equal('test-action');
+      });
+
+      test('_actionContext captures and provides services to action execution', () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        // Define a test service using Context.GenericTag pattern.
+        interface TestServiceInterface {
+          getValue(): number;
+        }
+        const TestService = Context.GenericTag<TestServiceInterface>('TestService');
+
+        // Track whether the action was executed with the correct context.
+        let executionResult: number | null = null;
+
+        // Create extension with service requirement.
+        // Note: The actions callback must USE the service for R to be inferred correctly.
+        const extensions = Effect.runSync(
+          GraphBuilder.createExtension({
+            id: 'test-extension',
+            match: NodeMatcher.whenNodeType(EXAMPLE_TYPE),
+            actions: (node, get) =>
+              // Use TestService in the callback to include it in R.
+              Effect.gen(function* () {
+                const service = yield* TestService;
+                return [
+                  {
+                    id: 'test-action',
+                    data: () =>
+                      Effect.gen(function* () {
+                        // Action can use the same service from captured context.
+                        const svc = yield* TestService;
+                        executionResult = svc.getValue();
+                      }).pipe(Effect.asVoid),
+                    properties: { label: `Test ${service.getValue()}` },
+                  },
+                ];
+              }),
+          }).pipe(Effect.provideService(TestService, { getValue: () => 42 })),
+        );
+
+        GraphBuilder.addExtension(builder, extensions);
+
+        const writableGraph = graph as Graph.WritableGraph;
+        Graph.addNode(writableGraph, { id: 'parent', type: EXAMPLE_TYPE, properties: {}, data: 'test' });
+        Graph.expand(graph, 'parent');
+
+        const actions = registry.get(graph.actions('parent'));
+        expect(actions).has.length(1);
+
+        // Verify _actionContext is captured.
+        const action = actions[0] as Node.Action;
+        expect(action._actionContext).to.not.be.undefined;
+
+        // Execute the action with the captured context.
+        const actionEffect = action.data();
+        const effectWithContext = action._actionContext
+          ? actionEffect.pipe(Effect.provide(action._actionContext))
+          : actionEffect;
+
+        Effect.runSync(effectWithContext);
+
+        // Verify the service was accessible during execution.
+        expect(executionResult).to.equal(42);
+      });
+
+      test('works with resolver', async () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        const extensions = Effect.runSync(
+          GraphBuilder.createExtension({
+            id: 'test-extension',
+            match: NodeMatcher.whenNodeType(EXAMPLE_TYPE),
+            resolver: (id, get) => Effect.succeed({ id, type: EXAMPLE_TYPE, properties: {}, data: 'resolved' }),
+          }),
+        );
+
+        GraphBuilder.addExtension(builder, extensions);
+        await Graph.initialize(graph, EXAMPLE_ID);
+
+        const node = Graph.getNode(graph, EXAMPLE_ID).pipe(Option.getOrNull);
+        expect(node).to.not.be.null;
+        expect(node?.id).to.equal(EXAMPLE_ID);
+        expect(node?.data).to.equal('resolved');
+      });
+
+      test('works with connector and actions together', () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        const extensions = Effect.runSync(
+          GraphBuilder.createExtension({
+            id: 'test-extension',
+            match: NodeMatcher.whenNodeType(EXAMPLE_TYPE),
+            connector: (node, get) => Effect.succeed([{ id: 'child', type: EXAMPLE_TYPE, data: node.data }]),
+            actions: (node, get) =>
+              Effect.succeed([
+                {
+                  id: 'test-action',
+                  data: () =>
+                    Effect.sync(() => {
+                      console.log('TestAction');
+                    }),
+                  properties: { label: 'Test' },
+                },
+              ]),
+          }),
+        );
+
+        GraphBuilder.addExtension(builder, extensions);
+
+        const writableGraph = graph as Graph.WritableGraph;
+        Graph.addNode(writableGraph, { id: 'parent', type: EXAMPLE_TYPE, properties: {}, data: 'test' });
+        Graph.expand(graph, 'parent');
+
+        const connections = registry.get(graph.connections('parent'));
+        // Should have both the child node and the action node.
+        expect(connections.length).to.be.greaterThanOrEqual(1);
+        const childNode = connections.find((n) => n.id === 'child');
+        expect(childNode).to.not.be.undefined;
+        expect(childNode?.data).to.equal('test');
+
+        const actions = registry.get(graph.actions('parent'));
+        expect(actions).has.length(1);
+        expect(actions[0].id).to.equal('test-action');
+      });
+
+      test('works with reactive connector using get context', () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        const state = Atom.make('initial');
+
+        const extensions = Effect.runSync(
+          GraphBuilder.createExtension({
+            id: 'test-extension',
+            match: NodeMatcher.whenNodeType(EXAMPLE_TYPE),
+            connector: (node, get) => Effect.succeed([{ id: 'child', type: EXAMPLE_TYPE, data: get(state) }]),
+          }),
+        );
+
+        GraphBuilder.addExtension(builder, extensions);
+
+        const writableGraph = graph as Graph.WritableGraph;
+        Graph.addNode(writableGraph, { id: 'parent', type: EXAMPLE_TYPE, properties: {}, data: 'test' });
+        Graph.expand(graph, 'parent');
+
+        {
+          const connections = registry.get(graph.connections('parent'));
+          expect(connections).has.length(1);
+          expect(connections[0].data).to.equal('initial');
+        }
+
+        registry.set(state, 'updated');
+
+        {
+          const connections = registry.get(graph.connections('parent'));
+          expect(connections).has.length(1);
+          expect(connections[0].data).to.equal('updated');
+        }
+      });
+    });
+
+    describe('extension error handling', () => {
+      test('connector failure is caught and logged, returns empty array', () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        const extensions = Effect.runSync(
+          GraphBuilder.createExtension({
+            id: 'failing-extension',
+            match: NodeMatcher.whenNodeType(EXAMPLE_TYPE),
+            connector: (node, get) => Effect.fail(new Error('Connector failed intentionally')),
+          }),
+        );
+
+        GraphBuilder.addExtension(builder, extensions);
+
+        const writableGraph = graph as Graph.WritableGraph;
+        Graph.addNode(writableGraph, { id: 'parent', type: EXAMPLE_TYPE, properties: {}, data: 'test' });
+
+        // Should not throw, error is caught internally.
+        Graph.expand(graph, 'parent');
+
+        // Should return empty connections since the connector failed.
+        const connections = registry.get(graph.connections('parent'));
+        expect(connections).has.length(0);
+      });
+
+      test('actions failure is caught and logged, returns empty array', () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        const extensions = Effect.runSync(
+          GraphBuilder.createExtension({
+            id: 'failing-actions-extension',
+            match: NodeMatcher.whenNodeType(EXAMPLE_TYPE),
+            actions: (node, get) => Effect.fail(new Error('Actions failed intentionally')),
+          }),
+        );
+
+        GraphBuilder.addExtension(builder, extensions);
+
+        const writableGraph = graph as Graph.WritableGraph;
+        Graph.addNode(writableGraph, { id: 'parent', type: EXAMPLE_TYPE, properties: {}, data: 'test' });
+
+        // Should not throw, error is caught internally.
+        Graph.expand(graph, 'parent');
+
+        // Should return empty actions since the actions callback failed.
+        const actions = registry.get(graph.actions('parent'));
+        expect(actions).has.length(0);
+      });
+
+      test('resolver failure is caught and logged, returns null', async () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        const extensions = Effect.runSync(
+          GraphBuilder.createExtension({
+            id: 'failing-resolver-extension',
+            match: NodeMatcher.whenNodeType(EXAMPLE_TYPE),
+            resolver: (id, get) => Effect.fail(new Error('Resolver failed intentionally')),
+          }),
+        );
+
+        GraphBuilder.addExtension(builder, extensions);
+
+        // Should not throw, error is caught internally.
+        await Graph.initialize(graph, EXAMPLE_ID);
+
+        // Should return null/none since the resolver failed.
+        const node = Graph.getNode(graph, EXAMPLE_ID).pipe(Option.getOrNull);
+        expect(node).to.be.null;
+      });
+
+      test('failing extension does not affect other extensions', () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        // Add a failing extension.
+        const failingExtensions = Effect.runSync(
+          GraphBuilder.createExtension({
+            id: 'failing-extension',
+            match: NodeMatcher.whenNodeType(EXAMPLE_TYPE),
+            connector: (node, get) => Effect.fail(new Error('This one fails')),
+          }),
+        );
+
+        // Add a working extension.
+        const workingExtensions = Effect.runSync(
+          GraphBuilder.createExtension({
+            id: 'working-extension',
+            match: NodeMatcher.whenNodeType(EXAMPLE_TYPE),
+            connector: (node, get) =>
+              Effect.succeed([{ id: 'child-from-working', type: EXAMPLE_TYPE, data: 'success' }]),
+          }),
+        );
+
+        GraphBuilder.addExtension(builder, failingExtensions);
+        GraphBuilder.addExtension(builder, workingExtensions);
+
+        const writableGraph = graph as Graph.WritableGraph;
+        Graph.addNode(writableGraph, { id: 'parent', type: EXAMPLE_TYPE, properties: {}, data: 'test' });
+        Graph.expand(graph, 'parent');
+
+        // The working extension should still produce its node.
+        const connections = registry.get(graph.connections('parent'));
+        expect(connections).has.length(1);
+        expect(connections[0].id).to.equal('child-from-working');
+        expect(connections[0].data).to.equal('success');
+      });
+    });
+
+    describe('createTypeExtension', () => {
+      test('creates extension matching by schema type with inferred object type', () => {
+        const registry = Registry.make();
+        const builder = GraphBuilder.make({ registry });
+        const graph = builder.graph;
+
+        const extensions = Effect.runSync(
+          GraphBuilder.createTypeExtension({
+            id: 'type-extension',
+            type: TestSchema.Person,
+            connector: (object) => Effect.succeed([{ id: 'child', type: EXAMPLE_TYPE, data: object }]),
+          }),
+        );
+
+        GraphBuilder.addExtension(builder, extensions);
+
+        const writableGraph = graph as Graph.WritableGraph;
+        const testObject = Obj.make(TestSchema.Person, { name: 'Test' });
+        Graph.addNode(writableGraph, { id: 'parent', type: EXAMPLE_TYPE, properties: {}, data: testObject });
+        Graph.expand(graph, 'parent');
+
+        const connections = registry.get(graph.connections('parent'));
+        expect(connections).has.length(1);
+        expect(connections[0].id).to.equal('child');
+        expect(connections[0].data).to.equal(testObject);
+      });
+    });
+  });
 });