102 lines
2.8 KiB
JavaScript
102 lines
2.8 KiB
JavaScript
import { action } from 'easy-peasy';
|
|
import isEqual from 'fast-deep-equal';
|
|
|
|
import { getBoundingBox, getNodesInside, getConnectedEdges } from './graph-utils';
|
|
|
|
export default {
|
|
setOnConnect: action((state, onConnect) => {
|
|
state.onConnect = onConnect;
|
|
}),
|
|
|
|
setNodes: action((state, nodes) => {
|
|
state.nodes = nodes;
|
|
}),
|
|
|
|
setEdges: action((state, edges) => {
|
|
state.edges = edges;
|
|
}),
|
|
|
|
updateNodeData: action((state, { id, ...data }) => {
|
|
state.nodes.forEach((n) => {
|
|
if (n.id === id) {
|
|
n.__rg = {
|
|
...n.__rg,
|
|
...data
|
|
};
|
|
}
|
|
});
|
|
}),
|
|
|
|
updateNodePos: action((state, { id, pos }) => {
|
|
state.nodes.forEach((n) => {
|
|
if (n.id === id) {
|
|
n.__rg = {
|
|
...n.__rg,
|
|
position: pos
|
|
};
|
|
}
|
|
});
|
|
}),
|
|
|
|
setSelection: action((state, isActive) => {
|
|
state.selectionActive = isActive;
|
|
}),
|
|
|
|
setNodesSelection: action((state, { isActive, selection }) => {
|
|
if (!isActive) {
|
|
state.nodesSelectionActive = false;
|
|
state.selectedElements = [];
|
|
|
|
return;
|
|
}
|
|
const selectedNodes = getNodesInside(state.nodes, selection, state.transform);
|
|
const selectedNodesBbox = getBoundingBox(selectedNodes);
|
|
|
|
state.selection = selection;
|
|
state.nodesSelectionActive = true;
|
|
state.selectedNodesBbox = selectedNodesBbox;
|
|
state.nodesSelectionActive = true;
|
|
}),
|
|
|
|
setSelectedElements: action((state, elements) => {
|
|
const selectedElementsArr = Array.isArray(elements) ? elements : [elements];
|
|
const selectedElementsUpdated = !isEqual(selectedElementsArr, state.selectedElements);
|
|
const selectedElements = selectedElementsUpdated ? selectedElementsArr : state.selectedElements;
|
|
|
|
state.selectedElements = selectedElements;
|
|
}),
|
|
|
|
updateSelection: action((state, selection) => {
|
|
const selectedNodes = getNodesInside(state.nodes, selection, state.transform);
|
|
const selectedEdges = getConnectedEdges(selectedNodes, state.edges);
|
|
|
|
const nextSelectedElements = [...selectedNodes, ...selectedEdges];
|
|
const selectedElementsUpdated = !isEqual(nextSelectedElements, state.selectedElements);
|
|
|
|
state.selection = selection;
|
|
state.selectedElements = selectedElementsUpdated ? nextSelectedElements: state.selectedElements
|
|
}),
|
|
|
|
updateTransform: action((state, transform) => {
|
|
state.transform = [transform.x, transform.y, transform.k];
|
|
}),
|
|
|
|
updateSize: action((state, size) => {
|
|
state.width = size.width;
|
|
state.height = size.height;
|
|
}),
|
|
|
|
initD3: action((state, { zoom, selection }) => {
|
|
state.d3Zoom = zoom;
|
|
state.d3Selection = selection;
|
|
state.d3Initialised = true;
|
|
}),
|
|
|
|
setConnectionPosition: action((state, position) => {
|
|
state.connectionPosition = position;
|
|
}),
|
|
|
|
setConnectionSourceId: action((state, sourceId) => {
|
|
state.connectionSourceId = sourceId;
|
|
})
|
|
}; |