@rxflow/base/cjs/providers/FlowProvider.js

BaseFlow - 核心 Flow 组件库

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.useSetState = exports.useSelector = exports.useGetState = exports.FlowProvider = exports.FlowContext = void 0;
var _react = require("@xyflow/react");
var _react2 = _interopRequireWildcard(require("react"));
var _zustand = require("zustand");
var _middleware = require("zustand/middleware");
var _jsxRuntime = require("react/jsx-runtime");
function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
/**
 * @author: yanxianliang
 * @date: 2025-10-16 17:28
 * @modified：2025/10/16 17:28 by yanxianliang
 * @desc: 全局状态管理器
 *
 * Copyright (c) 2025 by yanxianliang, All Rights Reserved.
 */

// Context 外部提供，支持inner hooks
const FlowContext = exports.FlowContext = /*#__PURE__*/_react2.default.createContext(null);
const FlowProviderWrapper = ({
  children
}) => {
  const [useStore] = (0, _react2.useState)(() => (0, _zustand.create)((0, _middleware.devtools)(set => ({
    onEdgesChangeHandlers: [],
    // onChange 列表
    forceRenderTimes: 0,
    initialNodeIdSet: new Set(),
    nodeTypeMap: new Map(),
    smartEdges: {},
    edgeStore: {}
  }), {
    name: "flow",
    enabled: process.env.NODE_ENV === 'development'
  })));
  return /*#__PURE__*/(0, _jsxRuntime.jsx)(FlowContext.Provider, {
    value: {
      useStore
    },
    children: children
  });
};
const FlowProvider = exports.FlowProvider = /*#__PURE__*/(0, _react2.memo)(({
  children
}) => {
  const hasWrapped = (0, _react2.useContext)(FlowContext);
  if (hasWrapped) {
    return children;
  }
  return /*#__PURE__*/(0, _jsxRuntime.jsx)(_react.ReactFlowProvider, {
    children: /*#__PURE__*/(0, _jsxRuntime.jsx)(FlowProviderWrapper, {
      children: children
    })
  });
});

/**
 * hooks 创建
 * selector 创建
 */
const useSelector = selector => {
  const context = (0, _react2.useContext)(FlowContext);
  if (!context) {
    console.error("Missing provider for context:", FlowContext);
    throw new Error("Missing provider for context");
  }
  return context.useStore(selector);
};
exports.useSelector = useSelector;
const useSetState = () => {
  const context = (0, _react2.useContext)(FlowContext);
  if (!context) {
    console.error("Missing provider for context:", FlowContext);
    throw new Error("Missing provider for context");
  }
  return context.useStore.setState;
};
exports.useSetState = useSetState;
const useGetState = () => {
  const context = (0, _react2.useContext)(FlowContext);
  if (!context) {
    console.error("Missing provider for context:", FlowContext);
    throw new Error("Missing provider for context");
  }
  return context.useStore.getState;
};
exports.useGetState = useGetState;