mirror of
https://github.com/morten-olsen/bob.git
synced 2026-02-08 01:46:29 +01:00
feat: add capabilities
This commit is contained in:
99
packages/playground/src/features/experiment/context.tsx
Normal file
99
packages/playground/src/features/experiment/context.tsx
Normal file
@@ -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 };
|
||||
Reference in New Issue
Block a user