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

BaseFlow - 核心 Flow 组件库

function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _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(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
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(arr) { if (Array.isArray(arr)) return arr; }
/**
 * @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.
 */
import { ReactFlowProvider } from "@xyflow/react";
import React, { memo, useContext, useState } from "react";
import { create } from "zustand";
import { devtools } from "zustand/middleware";
import { jsx as _jsx } from "react/jsx-runtime";
// Context 外部提供，支持inner hooks
export var FlowContext = /*#__PURE__*/React.createContext(null);
var FlowProviderWrapper = function FlowProviderWrapper(_ref) {
  var children = _ref.children;
  var _useState = useState(function () {
      return create(devtools(function (set) {
        return {
          onEdgesChangeHandlers: [],
          // onChange 列表
          forceRenderTimes: 0,
          initialNodeIdSet: new Set(),
          nodeTypeMap: new Map(),
          smartEdges: {},
          edgeStore: {}
        };
      }, {
        name: "flow",
        enabled: process.env.NODE_ENV === 'development'
      }));
    }),
    _useState2 = _slicedToArray(_useState, 1),
    useStore = _useState2[0];
  return /*#__PURE__*/_jsx(FlowContext.Provider, {
    value: {
      useStore: useStore
    },
    children: children
  });
};
export var FlowProvider = /*#__PURE__*/memo(function (_ref2) {
  var children = _ref2.children;
  var hasWrapped = useContext(FlowContext);
  if (hasWrapped) {
    return children;
  }
  return /*#__PURE__*/_jsx(ReactFlowProvider, {
    children: /*#__PURE__*/_jsx(FlowProviderWrapper, {
      children: children
    })
  });
});

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