38 lines
1.1 KiB
TypeScript
38 lines
1.1 KiB
TypeScript
import React, { useState } from 'react';
|
|
import ReactFlow, {
|
|
ReactFlowProvider,
|
|
removeElements,
|
|
addEdge,
|
|
Elements,
|
|
Connection,
|
|
Edge,
|
|
OnLoadParams,
|
|
} from 'react-flow-renderer';
|
|
|
|
import Controls from './Controls';
|
|
|
|
import './save.css';
|
|
|
|
const initialElements: Elements = [
|
|
{ id: '1', data: { label: 'Node 1' }, position: { x: 100, y: 100 } },
|
|
{ id: '2', data: { label: 'Node 2' }, position: { x: 100, y: 200 } },
|
|
{ id: 'e1-2', source: '1', target: '2' },
|
|
];
|
|
|
|
const SaveRestore = () => {
|
|
const [rfInstance, setRfInstance] = useState<OnLoadParams>();
|
|
const [elements, setElements] = useState<Elements>(initialElements);
|
|
const onElementsRemove = (elementsToRemove: Elements) => setElements((els) => removeElements(elementsToRemove, els));
|
|
const onConnect = (params: Connection | Edge) => setElements((els) => addEdge(params, els));
|
|
|
|
return (
|
|
<ReactFlowProvider>
|
|
<ReactFlow elements={elements} onElementsRemove={onElementsRemove} onConnect={onConnect} onLoad={setRfInstance}>
|
|
<Controls rfInstance={rfInstance} setElements={setElements} />
|
|
</ReactFlow>
|
|
</ReactFlowProvider>
|
|
);
|
|
};
|
|
|
|
export default SaveRestore;
|