yzsd/lib/notification/useNotification.js

a niubi UI component

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
var _typeof = require("@babel/runtime/helpers/typeof");
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports["default"] = void 0;
var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime/helpers/toConsumableArray"));
var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
var _react = _interopRequireWildcard(require("react"));
var _notifications = _interopRequireDefault(require("./notifications"));
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; }
var __rest = void 0 && (void 0).__rest || function (s, e) {
  var t = {};
  for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0) t[p] = s[p];
  if (s != null && typeof Object.getOwnPropertySymbols === "function") for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
    if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i])) t[p[i]] = s[p[i]];
  }
  return t;
};
var uniqueKey = 0;
function mergeConfig() {
  var clone = {};
  for (var _len = arguments.length, objList = new Array(_len), _key = 0; _key < _len; _key++) {
    objList[_key] = arguments[_key];
  }
  objList.forEach(function (obj) {
    if (obj) {
      Object.keys(obj).forEach(function (key) {
        var val = obj[key];
        if (val !== undefined) {
          clone[key] = val;
        }
      });
    }
  });
  return clone;
}
var useNotification = function useNotification(rootConfig) {
  var _a = rootConfig || {},
    _a$getContainer = _a.getContainer,
    getContainer = _a$getContainer === void 0 ? function () {
      return document.body;
    } : _a$getContainer,
    _a$prefixCls = _a.prefixCls,
    prefixCls = _a$prefixCls === void 0 ? 'yzs-notification' : _a$prefixCls,
    className = _a.className,
    style = _a.style,
    onAllRemoved = _a.onAllRemoved,
    maxCount = _a.maxCount,
    shareConfig = __rest(_a, ["getContainer", "prefixCls", "className", "style", "onAllRemoved", "maxCount"]);
  var _useState = (0, _react.useState)(),
    _useState2 = (0, _slicedToArray2["default"])(_useState, 2),
    container = _useState2[0],
    setContainer = _useState2[1];
  var notificationsRef = (0, _react.useRef)(null);
  var _useState3 = (0, _react.useState)([]),
    _useState4 = (0, _slicedToArray2["default"])(_useState3, 2),
    queueTask = _useState4[0],
    setQueueTask = _useState4[1];
  var api = (0, _react.useMemo)(function () {
    return {
      open: function open(config) {
        var mergedConfig = mergeConfig(shareConfig, config);
        if (!mergedConfig.key) {
          mergedConfig.key = "".concat(prefixCls, "-").concat(uniqueKey);
          uniqueKey += 1;
        }
        setQueueTask(function (task) {
          return [].concat((0, _toConsumableArray2["default"])(task), [{
            type: 'open',
            config: mergedConfig
          }]);
        });
      },
      close: function close() {},
      destroy: function destroy() {}
    };
  }, []);
  var contextHolder = /*#__PURE__*/_react["default"].createElement(_notifications["default"], {
    ref: notificationsRef,
    container: container,
    prefixCls: prefixCls,
    className: className,
    style: style,
    maxCount: maxCount,
    onAllRemoved: onAllRemoved
  });
  (0, _react.useEffect)(function () {
    setContainer(getContainer());
  });
  (0, _react.useEffect)(function () {
    if (notificationsRef.current && queueTask.length) {
      queueTask.forEach(function (task) {
        var _a;
        if (task.type === 'open') {
          (_a = notificationsRef.current) === null || _a === void 0 ? void 0 : _a.open(task.config);
        }
      });
      setQueueTask([]);
    }
  }, [queueTask]);
  return [api, contextHolder];
};
var _default = useNotification;
exports["default"] = _default;