feat: add capabilities

.npmrc

@@ -1,3 +1,4 @@
 node-linker=hoisted
+public-hoist-pattern[]=*@nextui-org/*
 store-dir=.pnpm-store
 

packages/algorithm/.turbo/turbo-build.log

@@ -1,4 +1,4 @@
 
-> bob-the-algorithm@ build /Users/alice/work/private/bob/packages/algorithm
+> @bob-the-algorithm/core@0.1.8 build /home/alice/Git/bob/packages/algorithm
 > tsc --build configs/tsconfig.libs.json
 

packages/algorithm/src/algorithm/calulation.ts

@@ -21,6 +21,10 @@ type RunOptions<TPlugins extends Plugins> = {
   heuristic?: (result: any) => boolean;
 };
 
+type PlanableWithPlugins<TPlugins extends Plugins> = Planable<
+  PluginAttributes<TPlugins>
+>;
+
 class Bob<TPlugins extends Plugins> {
   #options: CalulationOptions<TPlugins>;
   #id: number = 0;
@@ -99,5 +103,5 @@ class Bob<TPlugins extends Plugins> {
   };
 }
 
-export type { CalulationOptions, CalulationResult };
+export type { CalulationOptions, CalulationResult, PlanableWithPlugins };
 export { Bob };

packages/algorithm/src/algorithm/expand-node.ts

@@ -70,7 +70,7 @@ const expandNode = async <
       node.time + node.duration,
       planable.start?.min || 0,
     );
-    return {
+    const nextNode = {
       ...node,
       type: 'planable',
       exploreId: 0,
@@ -83,6 +83,14 @@ const expandNode = async <
       completed: remaining.length === 0,
       parent: node.id,
     };
+    return Object.values(plugins).reduce(
+      // TODO: remove any
+      (acc, plugin) =>
+        (plugin.mutateNode
+          ? plugin.mutateNode(acc as any, planable)
+          : acc) as any,
+      nextNode,
+    ) as any;
   });
 
   return [...planableNodes, ...metaNodes.flat()];

packages/algorithm/src/index.ts

@@ -1,5 +1,8 @@
 export type { GraphNode } from './types/node';
 export type { Planable } from './types/planable';
-export { expandNode } from './algorithm/expand-node';
-export { Bob, type CalulationResult } from './algorithm/calulation';
+export {
+  Bob,
+  type CalulationResult,
+  type PlanableWithPlugins,
+} from './algorithm/calulation';
 export { plugins } from './plugins/index';

packages/algorithm/src/plugins/capabilities.ts

@@ -0,0 +1,57 @@
+import { Type } from '@sinclair/typebox';
+import { createPlugin } from './create';
+
+const capabilities = () =>
+  createPlugin(
+    Type.Object({
+      capabilities: Type.Optional(
+        Type.Object({
+          provides: Type.Optional(Type.Array(Type.String())),
+          consumes: Type.Optional(Type.Array(Type.String())),
+          requires: Type.Optional(Type.Array(Type.String())),
+          perhibits: Type.Optional(Type.Array(Type.String())),
+        }),
+      ),
+    }),
+    Type.Object({
+      capabilities: Type.Array(Type.String()),
+    }),
+    {
+      isPlanable: (node, planable) => {
+        const { requires = [], perhibits = [] } =
+          planable.attributes?.capabilities || {};
+        const capabilities = node.context.capabilities;
+
+        if (requires.length === 0 && perhibits.length === 0) {
+          return true;
+        }
+        const satisfiesRequire = requires.every((c) =>
+          capabilities.includes(c),
+        );
+        const satisfiesPerhibit = !perhibits.some((c) =>
+          capabilities.includes(c),
+        );
+        return satisfiesRequire && satisfiesPerhibit;
+      },
+      mutateNode: (node, planable) => {
+        const { provides = [], consumes = [] } =
+          planable.attributes?.capabilities || {};
+        const capabilities = node.context.capabilities || [];
+
+        const newCapabilities = [
+          ...capabilities.filter((c) => !consumes.includes(c)),
+          ...provides,
+        ];
+
+        return {
+          ...node,
+          context: {
+            ...node.context,
+            capabilities: newCapabilities,
+          },
+        };
+      },
+    },
+  );
+
+export { capabilities };

packages/algorithm/src/plugins/index.ts

@@ -1,8 +1,10 @@
 import { Plugin } from '../types/plugin';
 import { transport } from './transport';
+import { capabilities } from './capabilities';
 
 const plugins = {
   transport,
+  capabilities,
 } satisfies Record<string, (...args: any[]) => Plugin>;
 
 export { plugins };

packages/algorithm/src/plugins/transport.ts

@@ -47,7 +47,7 @@ const transport = ({ getTravelTime }: TransportOptions) =>
               planable: undefined,
               location,
               exploreId: 0,
-              score: node.score - 10,
+              score: node.score - 5,
               time: node.time + node.duration,
               duration: travelTime,
               parent: node.id,

packages/algorithm/src/types/plugin.ts

@@ -12,6 +12,10 @@ type Plugin<TAttributes = any, TContext = any> = {
     node: GraphNode<TAttributes, TContext>,
     planable: Planable<TAttributes>,
   ) => boolean;
+  mutateNode?: (
+    node: GraphNode<TAttributes, TContext>,
+    planable: Planable<TAttributes>,
+  ) => GraphNode<TAttributes, TContext>;
 };
 
 type Plugins = Record<string, Plugin>;

packages/playground/package.json

@@ -11,9 +11,12 @@
   },
   "dependencies": {
     "@bob-the-algorithm/core": "workspace:^",
+    "@nextui-org/react": "^2.1.13",
     "date-fns": "^2.30.0",
+    "framer-motion": "^10.16.4",
     "react": "^18.2.0",
-    "react-dom": "^18.2.0"
+    "react-dom": "^18.2.0",
+    "tailwindcss": "^3.3.3"
   },
   "devDependencies": {
     "@mdx-js/rollup": "^2.3.0",
@@ -23,9 +26,11 @@
     "@typescript-eslint/eslint-plugin": "^6.0.0",
     "@typescript-eslint/parser": "^6.0.0",
     "@vitejs/plugin-react-swc": "^3.3.2",
+    "autoprefixer": "^10.4.16",
     "eslint": "^8.45.0",
     "eslint-plugin-react-hooks": "^4.6.0",
     "eslint-plugin-react-refresh": "^0.4.3",
+    "postcss": "^8.4.30",
     "reagraph": "^4.13.0",
     "typescript": "^5.0.2",
     "vite": "^4.4.5"

packages/playground/postcss.config.js

@@ -0,0 +1,6 @@
+export default {
+  plugins: {
+    tailwindcss: {},
+    autoprefixer: {},
+  },
+}

packages/playground/src/app.tsx

@@ -1,7 +1,12 @@
-import { Page } from './containers/page';
+import { NextUIProvider } from '@nextui-org/react';
+import { Page } from './containers/experiment';
 
 const App: React.FC = () => {
-  return <Page slug=".hello" />;
+  return (
+    <NextUIProvider>
+      <Page slug=".capabilities" />
+    </NextUIProvider>
+  );
 };
 
 export { App };

packages/playground/src/containers/experiment/index.tsx

@@ -1,28 +1,34 @@
 import { useEffect, useState } from 'react';
-import { pages } from '../utils/pages';
-import { RunnerProvider } from '../features/runner';
+import { experiments } from '../../utils/experiments';
+import { ExperimentProvider } from '../../features/experiment/context';
+import { ExperimentView } from './view';
+
 type PageProps = {
   slug: string;
 };
 
+type Experiment = {
+  worker: () => Worker;
+  view: React.ReactElement;
+};
+
 const Page: React.FC<PageProps> = ({ slug }) => {
-  const [Component, setComponent] = useState<React.FC>();
   const [loading, setLoading] = useState(true);
   const [error, setError] = useState<unknown>();
+  const [experiment, setExperiment] = useState<Experiment>();
 
   useEffect(() => {
     setLoading(true);
     setError(undefined);
     const load = async () => {
       try {
-        const page = pages.find((page) => page.slug === slug);
+        const page = experiments.find((page) => page.slug === slug);
         if (!page) {
           throw new Error(`Page not found: ${slug}`);
         }
-        const { default: Component } = (await page.loader()) as {
-          default: React.FC;
-        };
-        setComponent(() => Component);
+        const next = (await page.loader()) as Experiment;
+        console.log('n', next);
+        setExperiment(next);
       } catch (err) {
         setError(err);
       } finally {
@@ -37,14 +43,14 @@ const Page: React.FC<PageProps> = ({ slug }) => {
     return <div>Error: {error.toString()}</div>;
   }
 
-  if (loading || !Component) {
+  if (loading || !experiment) {
     return <div>Loading...</div>;
   }
 
   return (
-    <RunnerProvider>
-      <Component />
-    </RunnerProvider>
+    <ExperimentProvider worker={experiment.worker}>
+      <ExperimentView>{experiment.view}</ExperimentView>
+    </ExperimentProvider>
   );
 };
 

packages/playground/src/containers/experiment/node.tsx

@@ -0,0 +1,26 @@
+import { Accordion, AccordionItem } from '@nextui-org/react';
+import { useSelectedNode } from '../../features/experiment/hooks';
+import { Plan } from '../../presenters/plan';
+
+const NodeView = () => {
+  const node = useSelectedNode();
+  if (!node) {
+    return null;
+  }
+
+  return (
+    <div>
+      <h1>{node.id}</h1>
+      <Accordion>
+        <AccordionItem title="Plan" key="plan">
+          <Plan node={node} />
+        </AccordionItem>
+        <AccordionItem title="Context" key="context">
+          <pre>{JSON.stringify(node.context, null, 2)}</pre>
+        </AccordionItem>
+      </Accordion>
+    </div>
+  );
+};
+
+export { NodeView };

packages/playground/src/containers/experiment/nodes.tsx

@@ -0,0 +1,23 @@
+import { useExperimentResult } from '../../features/experiment';
+import { useSelectNode } from '../../features/experiment/hooks';
+
+const NodesView = () => {
+  const data = useExperimentResult();
+  const selectNode = useSelectNode();
+
+  if (!data) {
+    return null;
+  }
+
+  return (
+    <div>
+      {data.completed.map((node) => (
+        <div onClick={() => selectNode(node)} key={node.id}>
+          {node.id} {node.score}
+        </div>
+      ))}
+    </div>
+  );
+};
+
+export { NodesView };

packages/playground/src/containers/experiment/view.tsx

@@ -0,0 +1,19 @@
+import { Graph } from '../../presenters/graph';
+import { NodeView } from './node';
+import { NodesView } from './nodes';
+
+type ExperimentViewProps = {
+  children: React.ReactNode;
+};
+
+const ExperimentView: React.FC<ExperimentViewProps> = ({ children }) => {
+  return (
+    <>
+      <Graph />
+      <NodesView />
+      <NodeView />
+    </>
+  );
+};
+
+export { ExperimentView };

packages/playground/src/experiments/capabilities/index.tsx

@@ -0,0 +1,12 @@
+const Foo = () => {
+  return <div></div>;
+};
+
+const worker = () =>
+  new Worker(new URL('./script.ts', import.meta.url), {
+    type: 'module',
+  });
+
+const view = <Foo />;
+
+export { worker, view };

packages/playground/src/experiments/capabilities/script.ts

@@ -0,0 +1,181 @@
+import { Bob, plugins } from '@bob-the-algorithm/core';
+import { createWorker } from '../../features/experiment/worker';
+
+const MIN = 1000 * 60;
+const HOUR = 1000 * 60 * 60;
+
+const getTravelTime = async () => 30 * MIN;
+const transport = plugins.transport({
+  getTravelTime,
+});
+
+const realistic = async () => {
+  try {
+    const bob = new Bob({
+      plugins: { transport, capabilities: plugins.capabilities() },
+    });
+    const result = await bob.run({
+      context: {
+        location: 'home',
+        capabilities: ['kids'],
+      },
+      start: 0,
+      heuristic: ({ completed }) => completed.length >= 30,
+      planables: [
+        {
+          id: `Brush teeth`,
+          duration: 2 * MIN,
+          start: {
+            min: 7 * HOUR,
+            max: 8 * HOUR,
+          },
+          attributes: {
+            locations: ['home'],
+          },
+          score: 1,
+        },
+        {
+          id: 'Drop off kids',
+          duration: 30 * MIN,
+          attributes: {
+            locations: ['daycare'],
+            capabilities: {
+              consumes: ['kids'],
+              requires: ['kids'],
+            },
+          },
+          score: 1,
+          start: {
+            min: 7 * HOUR,
+            max: 9 * HOUR,
+          },
+        },
+        {
+          id: 'visit zoo',
+          duration: 1 * HOUR,
+          attributes: {
+            locations: ['zoo'],
+            capabilities: {
+              requires: ['kids'],
+            },
+          },
+          score: 1,
+          start: {
+            min: 10 * HOUR,
+            max: 14 * HOUR,
+          },
+        },
+        {
+          id: 'Pickup the kids',
+          duration: 30 * MIN,
+          attributes: {
+            locations: ['daycare'],
+            capabilities: {
+              provides: ['kids'],
+            },
+          },
+          score: 1,
+          start: {
+            min: 10 * HOUR,
+            max: 15.5 * HOUR,
+          },
+        },
+        {
+          id: 'Do work',
+          duration: 1 * HOUR,
+          count: 5,
+          attributes: {
+            locations: ['work'],
+            capabilities: {
+              perhibits: ['kids'],
+            },
+          },
+          score: 10,
+          start: {
+            min: 8 * HOUR,
+            max: 18 * HOUR,
+          },
+        },
+        {
+          id: 'put kids to bed',
+          duration: 30 * MIN,
+          attributes: {
+            locations: ['home'],
+            capabilities: {
+              consumes: ['kids'],
+              requires: ['kids'],
+            },
+          },
+          score: 1,
+        },
+        {
+          id: 'Read book',
+          duration: 0.5 * HOUR,
+          attributes: {
+            locations: ['home', 'work'],
+            capabilities: {
+              perhibits: ['kids'],
+            },
+          },
+          score: 3,
+          count: 2,
+          start: {
+            min: 8 * HOUR,
+            max: 22 * HOUR,
+          },
+        },
+        {
+          id: 'Meditate',
+          duration: 10 * MIN,
+          score: 1,
+          attributes: {
+            locations: ['home', 'work'],
+            capabilities: {
+              perhibits: ['kids'],
+            },
+          },
+          start: {
+            min: 8 * HOUR,
+            max: 22 * HOUR,
+          },
+        },
+        {
+          id: 'Meeting 1',
+          duration: 1 * HOUR,
+          attributes: {
+            locations: ['work', 'work'],
+            capabilities: {
+              perhibits: ['kids'],
+            },
+          },
+          score: 10,
+          start: {
+            min: 10 * HOUR,
+            max: 10 * HOUR,
+          },
+        },
+        {
+          id: 'Play playstation',
+          duration: 1 * HOUR,
+          attributes: {
+            locations: ['home'],
+            capabilities: {
+              perhibits: ['kids'],
+            },
+          },
+          score: 10,
+          start: {
+            min: 16 * HOUR,
+            max: 24 * HOUR,
+          },
+        },
+      ],
+    });
+    return result;
+  } catch (e) {
+    console.error(e);
+    throw e;
+  }
+};
+
+createWorker(realistic);

packages/playground/src/experiments/hello/index.tsx

@@ -0,0 +1,12 @@
+const Foo = () => {
+  return <div></div>;
+};
+
+const worker = () =>
+  new Worker(new URL('./script.ts', import.meta.url), {
+    type: 'module',
+  });
+
+const view = <Foo />;
+
+export { worker, view };

packages/playground/src/experiments/hello/script.ts

@@ -1,6 +1,5 @@
 import { Bob, plugins } from '@bob-the-algorithm/core';
-import { createWorker } from '../../features/runner/worker';
-import { convertResult } from '../../utils/graph';
+import { createWorker } from '../../features/experiment/worker';
 
 const MIN = 1000 * 60;
 const HOUR = 1000 * 60 * 60;
@@ -13,7 +12,7 @@ const transport = plugins.transport({
 const realistic = async () => {
   try {
     const bob = new Bob({
-      plugins: { transport },
+      plugins: { transport, capabilities: plugins.capabilities() },
     });
     const result = await bob.run({
       context: {
@@ -144,13 +143,11 @@ const realistic = async () => {
         },
       ],
     });
-    return convertResult(result);
+    return result;
   } catch (e) {
     console.error(e);
     throw e;
   }
 };
 
-createWorker({
-  realistic,
-});
+createWorker(realistic);

packages/playground/src/features/experiment/context.tsx

@@ -0,0 +1,99 @@
+import { CalulationResult, GraphNode } from '@bob-the-algorithm/core';
+import {
+  ReactNode,
+  createContext,
+  useCallback,
+  useEffect,
+  useMemo,
+  useState,
+} from 'react';
+
+type ExperimentResult = {
+  payload: CalulationResult<any>;
+};
+
+type ExperimentContextValue = {
+  result?: ExperimentResult;
+  error?: any;
+  loading: boolean;
+  selectNode: (node?: GraphNode) => void;
+  selectedNode?: GraphNode;
+};
+
+type ExperimentProviderProps = {
+  children: ReactNode;
+  worker: () => Worker;
+};
+
+const ExperimentContext = createContext<ExperimentContextValue>({
+  loading: false,
+  selectNode: () => { },
+});
+
+const ExperimentProvider: React.FC<ExperimentProviderProps> = ({
+  children,
+  worker,
+}) => {
+  const [result, setResult] = useState<ExperimentResult>();
+  const [error, setError] = useState<any>();
+  const [loading, setLoading] = useState(false);
+  const [selectedNode, setSelectedNode] = useState<GraphNode>();
+
+  const selectNode = useCallback((node?: GraphNode) => {
+    setSelectedNode(node);
+  }, []);
+
+  useEffect(() => {
+    let workerInstance: Worker | undefined;
+    const run = async () => {
+      setLoading(true);
+      setError(undefined);
+      setResult(undefined);
+      const workerInstance = worker();
+      workerInstance.onmessage = (e) => {
+        switch (e.data.type) {
+          case 'error': {
+            setError(e.data);
+            break;
+          }
+          case 'output': {
+            setResult(e.data);
+            break;
+          }
+        }
+        setLoading(false);
+      };
+      workerInstance.onerror = (e) => {
+        setError(e);
+        setLoading(false);
+      };
+      workerInstance.postMessage({
+        type: 'run',
+      });
+    };
+    run();
+    return () => {
+      workerInstance?.terminate();
+    };
+  }, [worker]);
+
+  const value = useMemo(
+    () => ({
+      result,
+      error,
+      loading,
+      selectNode,
+      selectedNode,
+    }),
+    [result, error, loading, selectNode, selectedNode],
+  );
+
+  return (
+    <ExperimentContext.Provider value={value}>
+      {children}
+    </ExperimentContext.Provider>
+  );
+};
+
+export type { ExperimentContextValue };
+export { ExperimentContext, ExperimentProvider };

packages/playground/src/features/experiment/create.tsx

@@ -0,0 +1,15 @@
+import { ReactNode } from 'react';
+import { ExperimentProvider } from './context';
+
+type CreateExperimentInput = {
+  worker: () => Worker;
+  view: ReactNode;
+};
+
+const createExperiment = (input: CreateExperimentInput) => {
+  return (
+    <ExperimentProvider worker={input.worker}>{input.view}</ExperimentProvider>
+  );
+};
+
+export { createExperiment };

packages/playground/src/features/experiment/hooks.ts

@@ -0,0 +1,19 @@
+import { useContext } from 'react';
+import { ExperimentContext } from './context';
+
+const useExperimentResult = () => {
+  const { result } = useContext(ExperimentContext);
+  return result?.payload;
+};
+
+const useSelectNode = () => {
+  const { selectNode } = useContext(ExperimentContext);
+  return selectNode;
+};
+
+const useSelectedNode = () => {
+  const { selectedNode } = useContext(ExperimentContext);
+  return selectedNode;
+};
+
+export { useExperimentResult, useSelectNode, useSelectedNode };

packages/playground/src/features/experiment/index.ts

@@ -0,0 +1,4 @@
+export { createWorker } from './worker';
+export { ExperimentProvider as Experminent } from './context';
+export { createExperiment } from './create';
+export { useExperimentResult } from './hooks';

packages/playground/src/features/experiment/worker.ts

@@ -0,0 +1,23 @@
+type WorkerFn = (...args: any[]) => any;
+
+const createWorker = (fn: WorkerFn) => {
+  const run = async () => {
+    const startTime = performance.now();
+    try {
+      const result = await fn();
+      const endTime = performance.now();
+      const duration = endTime - startTime;
+      self.postMessage({ type: 'output', payload: result, duration });
+    } catch (error) {
+      self.postMessage({ type: 'error', payload: error });
+    }
+  };
+  self.addEventListener('message', (event) => {
+    const { type } = event.data;
+    if (type === 'run') {
+      run();
+    }
+  });
+};
+
+export { createWorker };

packages/playground/src/features/runner/block.tsx

@@ -1,100 +0,0 @@
-import {
-  useCallback,
-  useContext,
-  useEffect,
-  useMemo,
-  useRef,
-  useState,
-} from 'react';
-import { RunnerContext } from './context';
-
-type BlockProps = {
-  worker: Worker;
-  action: string;
-  presenter?: React.FC<any>;
-};
-
-const id = (function*() {
-  let i = 0;
-  while (true) {
-    yield i++;
-  }
-})();
-
-const Block: React.FC<BlockProps> = ({
-  worker,
-  action,
-  presenter: Presenter,
-}) => {
-  const currentId = useRef(id.next().value);
-  const { vars } = useContext(RunnerContext);
-  const [output, setOutput] = useState<unknown>();
-  const [error, setError] = useState<unknown>();
-  const [running, setRunning] = useState<boolean>();
-  const [duration, setDuration] = useState<number>();
-
-  const view = useMemo(() => {
-    if (error) {
-      return error.toString();
-    }
-    if (Presenter) {
-      return <Presenter output={output} />;
-    }
-    return JSON.stringify(output, null, 2);
-  }, [output, error, Presenter]);
-
-  const runBlock = useCallback(async () => {
-    setRunning(true);
-    setError(undefined);
-    setOutput(undefined);
-
-    try {
-      worker.postMessage({
-        type: 'run',
-        action,
-        vars,
-        id: currentId.current,
-      });
-    } catch (error) {
-      setError(error);
-      console.error(error);
-    }
-
-    setRunning(false);
-  }, [worker, vars, action]);
-
-  useEffect(() => {
-    const listener = (event: MessageEvent) => {
-      const { type, payload, id, duration } = event.data;
-      if (id !== currentId.current) {
-        return;
-      }
-      setDuration(duration);
-      setRunning(false);
-      if (type === 'output') {
-        setOutput(payload);
-      }
-      if (type === 'error') {
-        setError(payload);
-        console.error(payload);
-      }
-    };
-    worker.addEventListener('message', listener);
-    return () => {
-      worker.removeEventListener('message', listener);
-    };
-  }, [worker]);
-
-  return (
-    <div>
-      <button onClick={runBlock} disabled={running}>
-        Run
-      </button>
-      {duration && <div>Duration: {duration.toFixed(2)}ms</div>}
-      {running && <div>Running...</div>}
-      {view}
-    </div>
-  );
-};
-
-export { Block };

packages/playground/src/features/runner/context.tsx

@@ -1,42 +0,0 @@
-import { createContext, useCallback, useMemo } from 'react';
-
-type Vars = Record<string, unknown>;
-
-type RunnerContextValue = {
-  vars: Vars;
-  run: (fn: (vars: Vars) => Promise<void>) => Promise<void>;
-};
-
-type RunnerProviderProps = {
-  vars?: Vars;
-  children: React.ReactNode;
-};
-
-const RunnerContext = createContext<RunnerContextValue>({
-  vars: {},
-  run: async () => {},
-});
-
-const RunnerProvider: React.FC<RunnerProviderProps> = ({
-  vars = {},
-  children,
-}) => {
-  const currentVars = useMemo(() => vars, [vars]);
-
-  const run = useCallback(
-    async (fn: (vars: Vars) => Promise<void>) => {
-      const output = await fn(currentVars);
-      return output;
-    },
-    [currentVars],
-  );
-
-  return (
-    <RunnerContext.Provider value={{ vars, run }}>
-      {children}
-    </RunnerContext.Provider>
-  );
-};
-
-export type { Vars };
-export { RunnerContext, RunnerProvider };

packages/playground/src/features/runner/index.ts

@@ -1,3 +0,0 @@
-export { RunnerProvider } from './context';
-export { Block } from './block';
-export { createWorker } from './worker';

packages/playground/src/features/runner/worker.ts

@@ -1,21 +0,0 @@
-type WorkerFn = Record<string, (...args: any[]) => any>;
-
-const createWorker = (fn: WorkerFn) => {
-  self.addEventListener('message', (event) => {
-    const { action, vars = {}, id } = event.data;
-    const run = async () => {
-      const startTime = performance.now();
-      try {
-        const result = await fn[action](vars);
-        const endTime = performance.now();
-        const duration = endTime - startTime;
-        self.postMessage({ type: 'output', payload: result, id, duration });
-      } catch (error) {
-        self.postMessage({ type: 'error', payload: error, id });
-      }
-    };
-    run();
-  });
-};
-
-export { createWorker };

packages/playground/src/main.css

@@ -0,0 +1,3 @@
+@tailwind base;
+@tailwind components;
+@tailwind utilities;

packages/playground/src/main.tsx

@@ -1,3 +1,4 @@
+import './main.css';
 import React from 'react';
 import ReactDOM from 'react-dom/client';
 import { App } from './app.tsx';

packages/playground/src/pages/hello/index.mdx

@@ -1,7 +0,0 @@
-import { Block } from '../../features/runner'
-import { worker } from './worker';
-import { Presenter } from '../../presenters/graph';
-
-# Hello World
-
-<Block worker={worker} action="realistic" presenter={Presenter} />

packages/playground/src/pages/hello/worker.ts

@@ -1,6 +0,0 @@
-import './script.ts';
-const worker = new Worker(new URL('./script.ts', import.meta.url), {
-  type: 'module',
-});
-
-export { worker };

packages/playground/src/presenters/graph/index.tsx

@@ -1,111 +1,70 @@
 import { useMemo, useState } from 'react';
 import { GraphCanvas } from 'reagraph';
-import { ConvertedResult } from '../../utils/graph';
-import { Plan } from './plan';
+import { useExperimentResult } from '../../features/experiment';
+import { convertResult } from '../../utils/graph';
+import {
+  useSelectNode,
+  useSelectedNode,
+} from '../../features/experiment/hooks';
 
-type PresenterProps = {
-  output: ConvertedResult;
-};
-
-const Presenter: React.FC<PresenterProps> = ({ output }) => {
-  const [currentStep, setCurrentStep] = useState(0);
+const Graph: React.FC = () => {
+  const data = useExperimentResult();
+  const selectedNode = useSelectedNode();
+  const selectNode = useSelectNode();
+  const output = useMemo(() => {
+    if (!data) {
+      return undefined;
+    }
+    return convertResult(data);
+  }, [data]);
   const [visualize, setVisualize] = useState(false);
-  const [selectedNode, setSelectedNode] = useState<string | undefined>(
-    undefined,
-  );
   const selectedPath = useMemo(() => {
     if (!selectedNode) {
       return [];
     }
     const result: string[] = [];
-    let current = output.result.nodes.find((n) => n.id === selectedNode);
+    let current = output?.result.nodes.find((n) => n.id === selectedNode.id);
 
     while (current) {
       result.push(current.id);
       if (!current.parent) {
         break;
       }
-      current = output.result.nodes.find((n) => n.id === current?.parent);
+      current = output?.result.nodes.find((n) => n.id === current?.parent);
     }
 
     return result;
   }, [selectedNode, output]);
-  const completed = useMemo(() => {
-    return (
-      output?.result?.completed
-        .map((c) => ({
-          id: c.id,
-          score: c.score,
-        }))
-        .sort((a, b) => b.score - a.score)
-        .slice(0, 10) || []
-    );
-  }, [output?.result?.completed]);
 
-  const maxStep = useMemo(
-    () => Math.max(...(output?.nodes?.map((n) => n.data?.exploreId) || [])),
-    [output],
-  );
-  const collapsedNodeIds = useMemo(
-    () =>
-      output?.nodes
-        ?.filter((n) => n.data?.exploreId > currentStep)
-        .map((n) => n.id),
-    [output, currentStep],
-  );
   if (!output) {
     return null;
   }
+
+  console.log(output);
+
   return (
     <>
       Nodes count: {output.nodes.length}
       <button onClick={() => setVisualize(!visualize)}>
         {visualize ? 'Hide' : 'Show'} Visualize
       </button>
-      {visualize && (
-        <>
-          <button onClick={() => setCurrentStep(currentStep - 1)}>Prev</button>
-          <input
-            type="range"
-            min={0}
-            max={maxStep}
-            value={currentStep}
-            onChange={(e) => setCurrentStep(parseInt(e.target.value))}
-          />
-          <button onClick={() => setCurrentStep(currentStep + 1)}>Next</button>
-        </>
-      )}
-      {completed.map((c) => (
-        <div key={c.id} onClick={() => setSelectedNode(c.id)}>
-          {c.id} - {c.score}
-        </div>
-      ))}
-      {selectedNode && <Plan id={selectedNode} output={output} />}
       {visualize && (
         <div style={{ position: 'relative', height: '70vh' }}>
           <GraphCanvas
             {...output}
-            collapsedNodeIds={collapsedNodeIds}
             labelType="all"
+            layoutType="hierarchicalTd"
             onNodeClick={(node) => {
-              if (node.id === selectedNode) {
-                setSelectedNode(undefined);
+              if (node.id === selectedNode?.id) {
+                selectNode(undefined);
                 return;
               }
-              setSelectedNode(node.id);
+              const nextNode = data?.nodes.find((n) => n.id === node.id);
+              selectNode(nextNode);
             }}
             selections={selectedPath}
             renderNode={({ size, opacity, node }) => {
               let color = 'gray';
-              if (
-                node.data?.exploreId < currentStep &&
-                node.data?.exploreId > 0
-              ) {
-                color = 'yellow';
-              }
-              if (node.data?.exploreId === currentStep) {
-                color = 'blue';
-              }
               if (node.data?.deadEnd) {
                 color = 'red';
               }
@@ -136,4 +95,4 @@ const Presenter: React.FC<PresenterProps> = ({ output }) => {
   );
 };
 
-export { Presenter };
+export { Graph };

packages/playground/src/presenters/graph/plan.tsx

@@ -40,7 +40,7 @@ const Node = ({ node, output }: NodeProps) => {
   if (node.type === 'travel') {
     return (
       <div>
-        {time} Travel: {node.location}
+        {time} Travel: {node.context.location}
       </div>
     );
   }

packages/playground/src/presenters/plan/index.tsx

@@ -0,0 +1,90 @@
+import { GraphNode } from '@bob-the-algorithm/core';
+import { useMemo } from 'react';
+import { convertResult } from '../../utils/graph';
+import { format } from 'date-fns';
+import { useExperimentResult } from '../../features/experiment';
+import { useSelectNode } from '../../features/experiment/hooks';
+
+type PlanProps = {
+  node: GraphNode;
+};
+
+type NodeProps = {
+  node: GraphNode;
+};
+
+const Node = ({ node }: NodeProps) => {
+  const selectNode = useSelectNode();
+  const data = useExperimentResult();
+  const planable = useMemo(() => {
+    return node.planable
+      ? data?.planables.find((n) => n.id === node.planable)
+      : null;
+  }, [node, data]);
+
+  const time = useMemo(() => {
+    const start = new Date(node.time);
+    const end = new Date(start.getTime() + node.duration);
+    return (
+      <span>
+        {format(start, 'HH:mm')} - {format(end, 'HH:mm')}
+      </span>
+    );
+  }, [node.duration, node.time]);
+
+  if (planable) {
+    return (
+      <div onClick={() => selectNode(node)}>
+        {time} Planable: {planable!.id}
+      </div>
+    );
+  }
+  if (node.type === 'travel') {
+    return (
+      <div onClick={() => selectNode(node)}>
+        {time} Travel: {node.context.location}
+      </div>
+    );
+  }
+
+  return null;
+};
+
+const Plan: React.FC<PlanProps> = ({ node }) => {
+  const data = useExperimentResult();
+  const output = useMemo(() => (data ? convertResult(data) : null), [data]);
+  const nodes = useMemo(() => {
+    if (!output) {
+      return [];
+    }
+    const result: GraphNode[] = [];
+    let current = node;
+    if (!current) {
+      return [];
+    }
+
+    while (current) {
+      result.push(current);
+      if (!current.parent) {
+        break;
+      }
+      current = output.result.nodes.find((n) => n.id === current?.parent)!;
+    }
+
+    return result;
+  }, [output, node]);
+
+  if (!output) {
+    return null;
+  }
+
+  return (
+    <>
+      {nodes.map((n) => (
+        <Node key={n.id} node={n} />
+      ))}
+    </>
+  );
+};
+
+export { Plan };

packages/playground/src/utils/experiments.ts

@@ -0,0 +1,8 @@
+const imports = import.meta.glob('../experiments/*/index.tsx');
+
+const experiments = Object.entries(imports).map(([path, loader]) => {
+  const slug = path.replace('./experiments/', '').replace('/index.tsx', '');
+  return { slug, loader };
+});
+
+export { experiments };

packages/playground/src/utils/pages.ts

@@ -1,8 +0,0 @@
-const imports = import.meta.glob('../pages/*/index.mdx');
-
-const pages = Object.entries(imports).map(([path, loader]) => {
-  const slug = path.replace('./pages/', '').replace('/index.mdx', '');
-  return { slug, loader };
-});
-
-export { pages };

packages/playground/tailwind.config.js

@@ -0,0 +1,15 @@
+/** @type {import('tailwindcss').Config} */
+const { nextui } = require('@nextui-org/react');
+
+export default {
+  content: [
+    './index.html',
+    './src/**/*.{js,ts,jsx,tsx}',
+    './node_modules/@nextui-org/theme/dist/**/*.{js,ts,jsx,tsx}',
+  ],
+  theme: {
+    extend: {},
+  },
+  darkMode: 'class',
+  plugins: [nextui()],
+};

packages/playground/vite.config.d.ts

@@ -0,0 +1,2 @@
+declare const _default: import("vite").UserConfig;
+export default _default;

packages/playground/vite.config.js

@@ -0,0 +1,9 @@
+import { defineConfig } from 'vite';
+import react from '@vitejs/plugin-react-swc';
+import mdx from '@mdx-js/rollup';
+var ASSET_URL = process.env.ASSET_URL || '';
+// https://vitejs.dev/config/
+export default defineConfig({
+    base: ASSET_URL,
+    plugins: [mdx(), react()],
+});