UNPKG

@difizen/ai-flow

Version:

Scalable, out-of-the-box, agent-oriented flow

233 lines 10.2 kB
function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); } function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; } function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; } function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; } function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : String(i); } function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); } import { useFlowsManagerStore } from "../../stores/flowsManagerStore"; import classNames from "../../utils/classnames"; import { RiArrowGoBackLine, RiArrowGoForwardLine, RiColorFilterLine, RiContractUpDownLine, RiCursorLine, RiExpandUpDownLine, RiFullscreenLine, RiHand, RiZoomInLine, RiZoomOutLine } from '@remixicon/react'; import { useReactFlow, useViewport } from '@xyflow/react'; import { Divider } from 'antd'; import { memo } from 'react'; import { ControlMode, NodePanelTypeEnum } from "../../interfaces/flow"; import { useFlowStore } from "../../stores/flowStore"; import { Popup } from "../AIBasic/Popup/index"; import { TipPopup } from "../AIBasic/TipPopup/index"; import React from 'react'; import { jsx as _jsx } from "react/jsx-runtime"; import { Fragment as _Fragment } from "react/jsx-runtime"; import { jsxs as _jsxs } from "react/jsx-runtime"; export var HoverBlock = function HoverBlock(props) { var children = props.children, className = props.className, tooltip = props.tooltip, _props$disabled = props.disabled, disabled = _props$disabled === void 0 ? false : _props$disabled, _props$selected = props.selected, selected = _props$selected === void 0 ? false : _props$selected; var Inner = /*#__PURE__*/_jsx("div", _objectSpread(_objectSpread({}, props), {}, { className: classNames('flex items-center justify-center w-8 h-8 rounded-lg cursor-pointer hover:bg-black/5', disabled ? 'cursor-not-allowed opacity-50' : '', selected && 'bg-blue-400/20 hover:bg-blue-400/20', className), children: /*#__PURE__*/_jsx("div", { className: selected ? 'text-blue-400 text:bg-blue-400' : '', children: children }) })); if (!tooltip) { return Inner; } return /*#__PURE__*/_jsx(TipPopup, { title: tooltip, children: Inner }); }; var zoomOptions = [25, 50, 100, 125, 'Fit']; var OperatorRaw = function OperatorRaw() { var _useReactFlow = useReactFlow(), zoomIn = _useReactFlow.zoomIn, zoomOut = _useReactFlow.zoomOut, zoomTo = _useReactFlow.zoomTo, fitView = _useReactFlow.fitView; var _useViewport = useViewport(), zoom = _useViewport.zoom; var handleZoom = function handleZoom(type) { if (typeof type === 'number') { zoomTo(type / 100); return; } fitView(); }; var nodePanel = useFlowStore(function (state) { return state.nodePanel; }); var undo = useFlowsManagerStore(function (state) { return state.undo; }); var redo = useFlowsManagerStore(function (state) { return state.redo; }); var past = useFlowsManagerStore(function (state) { return state.past; }); var future = useFlowsManagerStore(function (state) { return state.future; }); var autoLayout = useFlowsManagerStore(function (state) { return state.autoLayout; }); var setNodesFolded = useFlowsManagerStore(function (state) { return state.setNodesFolded; }); var takeSnapshot = useFlowsManagerStore(function (state) { return state.takeSnapshot; }); var setControlMode = useFlowStore(function (state) { return state.setControlMode; }); var controlMode = useFlowStore(function (state) { return state.controlMode; }); return /*#__PURE__*/_jsxs("div", { className: "flex items-center mt-1 gap-2 absolute left-4 bottom-4 z-[9]", children: [/*#__PURE__*/_jsx("div", { className: "p-0.5 h-9 cursor-pointer text-[13px] text-gray-500 font-medium rounded-lg bg-white shadow-lg border-[0.5px] border-gray-100 w-[100%]", children: /*#__PURE__*/_jsxs("div", { className: "flex items-center justify-between h-8 rounded-lg", children: [/*#__PURE__*/_jsx(HoverBlock, { onClick: function onClick(e) { e.stopPropagation(); zoomOut(); }, children: /*#__PURE__*/_jsx(RiZoomOutLine, { className: "w-4 h-4" }) }), /*#__PURE__*/_jsx(Popup, { panel: /*#__PURE__*/_jsx(_Fragment, { children: zoomOptions.map(function (n) { return /*#__PURE__*/_jsx("div", { onClick: function onClick() { return handleZoom(n); }, className: "flex items-center justify-between px-3 h-8 rounded-lg hover:bg-gray-50 cursor-pointer text-[13px] text-gray-600", children: typeof n === 'number' ? n + '%' : '自适应' }, n); }) }), panelProps: { anchor: { to: 'top', gap: 6 } }, children: /*#__PURE__*/_jsxs(HoverBlock, { className: "w-10", children: [Math.round(zoom * 100), "%"] }) }), /*#__PURE__*/_jsx(HoverBlock, { onClick: function onClick(e) { e.stopPropagation(); zoomIn(); }, children: /*#__PURE__*/_jsx(RiZoomInLine, { className: "w-4 h-4" }) }), /*#__PURE__*/_jsx(HoverBlock, { tooltip: "\u81EA\u9002\u5E94", onClick: function onClick(e) { e.stopPropagation(); fitView(); }, children: /*#__PURE__*/_jsx(RiFullscreenLine, { className: "w-4 h-4" }) })] }) }), /*#__PURE__*/_jsx("div", { className: "p-0.5 h-9 cursor-pointer text-[13px] text-gray-500 font-medium rounded-lg bg-white shadow-lg border-[0.5px] border-gray-100 w-[100%]", children: /*#__PURE__*/_jsxs("div", { className: "flex items-center justify-between h-8 rounded-lg", children: [/*#__PURE__*/_jsx(HoverBlock, { tooltip: "\u64A4\u9500", onClick: function onClick(e) { e.stopPropagation(); undo(); }, disabled: !(past && past.length > 0), children: /*#__PURE__*/_jsx(RiArrowGoBackLine, { className: "w-4 h-4" }) }), /*#__PURE__*/_jsx(HoverBlock, { tooltip: "\u91CD\u505A", disabled: !(future && future.length > 0), onClick: function onClick(e) { e.stopPropagation(); redo(); }, children: /*#__PURE__*/_jsx(RiArrowGoForwardLine, { className: "w-4 h-4" }) }), /*#__PURE__*/_jsx(Divider, { type: "vertical", className: "mx-1" }), /*#__PURE__*/_jsx(HoverBlock, { selected: controlMode === ControlMode.Pointer, tooltip: "\u6307\u9488\u6A21\u5F0F", onClick: function onClick(e) { e.stopPropagation(); setControlMode(ControlMode.Pointer); }, children: /*#__PURE__*/_jsx(RiCursorLine, { className: "w-4 h-4" }) }), /*#__PURE__*/_jsx(HoverBlock, { selected: controlMode === ControlMode.Hand, tooltip: "\u624B\u6A21\u5F0F", onClick: function onClick(e) { e.stopPropagation(); setControlMode(ControlMode.Hand); }, children: /*#__PURE__*/_jsx(RiHand, { className: "w-4 h-4" }) }), nodePanel === NodePanelTypeEnum.InNode && /*#__PURE__*/_jsxs(_Fragment, { children: [/*#__PURE__*/_jsx(Divider, { type: "vertical", className: "mx-1" }), /*#__PURE__*/_jsx(HoverBlock, { tooltip: "\u6298\u53E0\u8282\u70B9", onClick: function onClick(e) { e.stopPropagation(); setNodesFolded(true); }, children: /*#__PURE__*/_jsx(RiContractUpDownLine, { className: "w-4 h-4" }) }), /*#__PURE__*/_jsx(HoverBlock, { tooltip: "\u5C55\u5F00\u8282\u70B9", onClick: function onClick(e) { e.stopPropagation(); setNodesFolded(false); }, children: /*#__PURE__*/_jsx(RiExpandUpDownLine, { className: "w-4 h-4" }) })] }), /*#__PURE__*/_jsx(Divider, { type: "vertical", className: "mx-1" }), /*#__PURE__*/_jsx(HoverBlock, { tooltip: "\u81EA\u52A8\u5E03\u5C40", onClick: function onClick(e) { e.stopPropagation(); autoLayout(); takeSnapshot(); }, children: /*#__PURE__*/_jsx(RiColorFilterLine, { className: "w-4 h-4" }) })] }) })] }); }; export var Operator = /*#__PURE__*/memo(OperatorRaw);