100 lines
4.4 KiB
JavaScript
100 lines
4.4 KiB
JavaScript
"use strict";
|
|
|
|
var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
|
|
var _typeof = require("@babel/runtime/helpers/typeof");
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
exports.createContext = createContext;
|
|
exports.useContext = useContext;
|
|
var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
|
|
var _useEvent = _interopRequireDefault(require("rc-util/lib/hooks/useEvent"));
|
|
var _useLayoutEffect = _interopRequireDefault(require("rc-util/lib/hooks/useLayoutEffect"));
|
|
var _isEqual = _interopRequireDefault(require("rc-util/lib/isEqual"));
|
|
var React = _interopRequireWildcard(require("react"));
|
|
var _reactDom = require("react-dom");
|
|
function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
|
|
function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || _typeof(obj) !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
|
|
function createContext(defaultValue) {
|
|
var Context = /*#__PURE__*/React.createContext(undefined);
|
|
var Provider = function Provider(_ref) {
|
|
var value = _ref.value,
|
|
children = _ref.children;
|
|
var valueRef = React.useRef(value);
|
|
valueRef.current = value;
|
|
var _React$useState = React.useState(function () {
|
|
return {
|
|
getValue: function getValue() {
|
|
return valueRef.current;
|
|
},
|
|
listeners: new Set()
|
|
};
|
|
}),
|
|
_React$useState2 = (0, _slicedToArray2.default)(_React$useState, 1),
|
|
context = _React$useState2[0];
|
|
(0, _useLayoutEffect.default)(function () {
|
|
(0, _reactDom.unstable_batchedUpdates)(function () {
|
|
context.listeners.forEach(function (listener) {
|
|
listener(value);
|
|
});
|
|
});
|
|
}, [value]);
|
|
return /*#__PURE__*/React.createElement(Context.Provider, {
|
|
value: context
|
|
}, children);
|
|
};
|
|
return {
|
|
Context: Context,
|
|
Provider: Provider,
|
|
defaultValue: defaultValue
|
|
};
|
|
}
|
|
|
|
/** e.g. useSelect(userContext) => user */
|
|
|
|
/** e.g. useSelect(userContext, user => user.name) => user.name */
|
|
|
|
/** e.g. useSelect(userContext, ['name', 'age']) => user { name, age } */
|
|
|
|
/** e.g. useSelect(userContext, 'name') => user.name */
|
|
|
|
function useContext(holder, selector) {
|
|
var eventSelector = (0, _useEvent.default)(typeof selector === 'function' ? selector : function (ctx) {
|
|
if (selector === undefined) {
|
|
return ctx;
|
|
}
|
|
if (!Array.isArray(selector)) {
|
|
return ctx[selector];
|
|
}
|
|
var obj = {};
|
|
selector.forEach(function (key) {
|
|
obj[key] = ctx[key];
|
|
});
|
|
return obj;
|
|
});
|
|
var context = React.useContext(holder === null || holder === void 0 ? void 0 : holder.Context);
|
|
var _ref2 = context || {},
|
|
listeners = _ref2.listeners,
|
|
getValue = _ref2.getValue;
|
|
var valueRef = React.useRef();
|
|
valueRef.current = eventSelector(context ? getValue() : holder === null || holder === void 0 ? void 0 : holder.defaultValue);
|
|
var _React$useState3 = React.useState({}),
|
|
_React$useState4 = (0, _slicedToArray2.default)(_React$useState3, 2),
|
|
forceUpdate = _React$useState4[1];
|
|
(0, _useLayoutEffect.default)(function () {
|
|
if (!context) {
|
|
return;
|
|
}
|
|
function trigger(nextValue) {
|
|
var nextSelectorValue = eventSelector(nextValue);
|
|
if (!(0, _isEqual.default)(valueRef.current, nextSelectorValue, true)) {
|
|
forceUpdate({});
|
|
}
|
|
}
|
|
listeners.add(trigger);
|
|
return function () {
|
|
listeners.delete(trigger);
|
|
};
|
|
}, [context]);
|
|
return valueRef.current;
|
|
} |