@xrenders/xflow
Version:
一款功能强大、易用灵活的流程编辑器框架，帮助你轻松构建复杂的工作流和流程产品
164 lines • 7.07 kB
JavaScript
function _slicedToArray(r, e) { return _arrayWithHoles(r) || _iterableToArrayLimit(r, e) || _unsupportedIterableToArray(r, e) || _nonIterableRest(); }
function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _unsupportedIterableToArray(r, a) { if (r) { if ("string" == typeof r) return _arrayLikeToArray(r, a); var t = {}.toString.call(r).slice(8, -1); return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0; } }
function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length); for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e]; return n; }
function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
function _arrayWithHoles(r) { if (Array.isArray(r)) return r; }
import { CloseOutlined, PlusOutlined } from '@ant-design/icons';
import { BezierEdge, EdgeLabelRenderer, getBezierPath, useReactFlow } from '@xyflow/react';
import React, { memo, useContext, useState } from 'react';
import { shallow } from 'zustand/shallow';
import { useFlow } from '../../hooks/useFlow';
import { useStore } from '../../hooks/useStore';
import { ConfigContext } from '../../models/context';
import { uuid, uuid4 } from '../../utils';
import NodeSelectPopover from '../NodesPopover';
import "./index.css";
export default /*#__PURE__*/memo(function (edge) {
var _a, _b, _c, _d, _e, _f;
var id = edge.id,
selected = edge.selected,
sourceX = edge.sourceX,
sourceY = edge.sourceY,
targetX = edge.targetX,
targetY = edge.targetY,
source = edge.source,
target = edge.target,
sourceHandleId = edge.sourceHandleId;
var reactflow = useReactFlow();
var _useState = useState(false),
_useState2 = _slicedToArray(_useState, 2),
isHovered = _useState2[0],
setIsHovered = _useState2[1];
var _getBezierPath = getBezierPath({
sourceX: sourceX,
sourceY: sourceY,
targetX: targetX,
targetY: targetY
}),
_getBezierPath2 = _slicedToArray(_getBezierPath, 3),
edgePath = _getBezierPath2[0],
labelX = _getBezierPath2[1],
labelY = _getBezierPath2[2];
var _useContext = useContext(ConfigContext),
globalConfig = _useContext.globalConfig,
settingMap = _useContext.settingMap,
readOnly = _useContext.readOnly;
var hideEdgeAddBtn = (_b = (_a = globalConfig === null || globalConfig === void 0 ? void 0 : globalConfig.edge) === null || _a === void 0 ? void 0 : _a.hideEdgeAddBtn) !== null && _b !== void 0 ? _b : false;
var hideEdgeDelBtn = (_d = (_c = globalConfig === null || globalConfig === void 0 ? void 0 : globalConfig.edge) === null || _c === void 0 ? void 0 : _c.hideEdgeDelBtn) !== null && _d !== void 0 ? _d : false;
var deletable = (_f = (_e = globalConfig === null || globalConfig === void 0 ? void 0 : globalConfig.edge) === null || _e === void 0 ? void 0 : _e.deletable) !== null && _f !== void 0 ? _f : true;
var _useStore = useStore(function (state) {
return {
layout: state.layout,
nodes: state.nodes,
edges: state.edges,
mousePosition: state.mousePosition,
addEdges: state.addEdges,
onEdgesChange: state.onEdgesChange
};
}, shallow),
nodes = _useStore.nodes,
edges = _useStore.edges,
addEdges = _useStore.addEdges,
mousePosition = _useStore.mousePosition,
onEdgesChange = _useStore.onEdgesChange,
layout = _useStore.layout;
var _useFlow = useFlow(),
addNodes = _useFlow.addNodes;
var handleAddNode = function handleAddNode(data) {
var _a;
var screenToFlowPosition = reactflow.screenToFlowPosition;
var _screenToFlowPosition = screenToFlowPosition({
x: mousePosition.pageX,
y: mousePosition.pageY
}),
x = _screenToFlowPosition.x,
y = _screenToFlowPosition.y;
var targetId = uuid();
var title = ((_a = settingMap[data === null || data === void 0 ? void 0 : data._nodeType]) === null || _a === void 0 ? void 0 : _a.title) || (data === null || data === void 0 ? void 0 : data._nodeType);
var newNodes = {
id: targetId,
type: 'custom',
data: Object.assign({
title: "".concat(title, "_").concat(uuid4())
}, data),
position: {
x: x,
y: y
}
};
var newEdges = [Object.assign({
id: uuid(),
source: source,
target: targetId,
deletable: deletable
}, sourceHandleId && {
sourceHandle: sourceHandleId
}), {
id: uuid(),
source: targetId,
deletable: deletable,
target: target
}];
addNodes(newNodes);
addEdges(newEdges);
onEdgesChange([{
id: id,
type: 'remove'
}]);
};
var edgeExtra = {
sourceX: edge.sourceX - 15,
targetX: edge.targetX + 10
};
if (layout === 'TB') {
edgeExtra = {
sourceY: edge.sourceY - 15,
targetY: edge.targetY + 13
};
}
return /*#__PURE__*/React.createElement("g", {
onMouseEnter: function onMouseEnter() {
return setIsHovered(true);
},
onMouseLeave: function onMouseLeave() {
return setIsHovered(false);
}
}, /*#__PURE__*/React.createElement(BezierEdge, Object.assign({}, edge, edgeExtra, {
edgePath: edgePath,
label: isHovered && (/*#__PURE__*/React.createElement(EdgeLabelRenderer, null, /*#__PURE__*/React.createElement("div", {
className: "custom-edge-line",
style: {
transform: "translate(-50%, -50%) translate(".concat(labelX, "px,").concat(labelY, "px)")
}
}, /*#__PURE__*/React.createElement("div", {
className: "line-content"
}, !hideEdgeDelBtn && !readOnly && (/*#__PURE__*/React.createElement("div", {
className: "line-icon-box",
onClick: function onClick() {
if (readOnly) {
return;
}
onEdgesChange([{
id: id,
type: 'remove'
}]);
}
}, /*#__PURE__*/React.createElement(CloseOutlined, {
style: {
color: '#fff',
fontSize: 10
}
}))), !hideEdgeAddBtn && !readOnly && (/*#__PURE__*/React.createElement(NodeSelectPopover, {
placement: "right",
addNode: handleAddNode
}, /*#__PURE__*/React.createElement("div", {
className: "line-icon-box"
}, /*#__PURE__*/React.createElement(PlusOutlined, {
style: {
color: '#fff',
fontSize: 10
}
}))))))))
})));
});