zp-bee/lib/bee-date-picker/generateRangePicker.js

zp-bee，是一款基于 Dumi，由 React + TypeScript 开发的组件库 🎉。

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = generateRangePicker;

var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));

var _classCallCheck2 = _interopRequireDefault(require("@babel/runtime/helpers/classCallCheck"));

var _createClass2 = _interopRequireDefault(require("@babel/runtime/helpers/createClass"));

var _inherits2 = _interopRequireDefault(require("@babel/runtime/helpers/inherits"));

var _createSuper2 = _interopRequireDefault(require("@babel/runtime/helpers/createSuper"));

var _react = _interopRequireDefault(require("react"));

var _classnames = _interopRequireDefault(require("classnames"));

var _CalendarOutlined = _interopRequireDefault(require("@ant-design/icons/CalendarOutlined"));

var _ClockCircleOutlined = _interopRequireDefault(require("@ant-design/icons/ClockCircleOutlined"));

var _CloseCircleFilled = _interopRequireDefault(require("@ant-design/icons/CloseCircleFilled"));

var _SwapRightOutlined = _interopRequireDefault(require("@ant-design/icons/SwapRightOutlined"));

var _rcPicker = require("rc-picker");

var _zh_CN = _interopRequireDefault(require("../bee-locale/zh_CN"));

var _beeConfig = require("../bee-config");

var _LocaleReceiver = _interopRequireDefault(require("../bee-locale/LocaleReceiver"));

var _util = require("./util");

var _generatePicker = require("./generatePicker");

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;
};

function generateRangePicker(generateConfig) {
  var RangePicker = /*#__PURE__*/function (_React$Component) {
    (0, _inherits2.default)(RangePicker, _React$Component);

    var _super = (0, _createSuper2.default)(RangePicker);

    function RangePicker() {
      var _this;

      (0, _classCallCheck2.default)(this, RangePicker);
      _this = _super.apply(this, arguments);
      _this.pickerRef = /*#__PURE__*/_react.default.createRef();

      _this.focus = function () {
        if (_this.pickerRef.current) {
          _this.pickerRef.current.focus();
        }
      };

      _this.blur = function () {
        if (_this.pickerRef.current) {
          _this.pickerRef.current.blur();
        }
      };

      _this.getDefaultLocale = function () {
        var locale = _this.props.locale;
        var result = Object.assign(Object.assign({}, _zh_CN.default), locale);
        result.lang = Object.assign(Object.assign({}, result.DatePicker.lang), (locale || {}).lang);
        return result;
      };

      _this.renderPicker = function (locale) {
        var _classNames;

        var _this$context = _this.context,
            getPrefixCls = _this$context.getPrefixCls,
            direction = _this$context.direction,
            getPopupContainer = _this$context.getPopupContainer;

        var _a = _this.props,
            customizePrefixCls = _a.prefixCls,
            customGetPopupContainer = _a.getPopupContainer,
            className = _a.className,
            customizeSize = _a.size,
            _a$bordered = _a.bordered,
            bordered = _a$bordered === void 0 ? true : _a$bordered,
            placeholder = _a.placeholder,
            restProps = __rest(_a, ["prefixCls", "getPopupContainer", "className", "size", "bordered", "placeholder"]);

        var _this$props = _this.props,
            format = _this$props.format,
            showTime = _this$props.showTime,
            picker = _this$props.picker;
        var prefixCls = getPrefixCls('picker', customizePrefixCls);
        var additionalOverrideProps = {};
        additionalOverrideProps = Object.assign(Object.assign(Object.assign({}, additionalOverrideProps), showTime ? (0, _generatePicker.getTimeProps)(Object.assign({
          format: format,
          picker: picker
        }, showTime)) : {}), picker === 'time' ? (0, _generatePicker.getTimeProps)(Object.assign(Object.assign({
          format: format
        }, _this.props), {
          picker: picker
        })) : {});
        var rootPrefixCls = getPrefixCls();
        var mergedSize = customizeSize;
        return /*#__PURE__*/_react.default.createElement(_rcPicker.RangePicker, Object.assign({
          separator: /*#__PURE__*/_react.default.createElement("span", {
            "aria-label": "to",
            className: "".concat(prefixCls, "-separator")
          }, /*#__PURE__*/_react.default.createElement(_SwapRightOutlined.default, null)),
          ref: _this.pickerRef,
          placeholder: (0, _util.getRangePlaceholder)(picker, locale, placeholder),
          suffixIcon: picker === 'time' ? /*#__PURE__*/_react.default.createElement(_ClockCircleOutlined.default, null) : /*#__PURE__*/_react.default.createElement(_CalendarOutlined.default, null),
          clearIcon: /*#__PURE__*/_react.default.createElement(_CloseCircleFilled.default, null),
          allowClear: true
        }, restProps, additionalOverrideProps, {
          className: (0, _classnames.default)((_classNames = {}, (0, _defineProperty2.default)(_classNames, "".concat(prefixCls, "-").concat(mergedSize), mergedSize), (0, _defineProperty2.default)(_classNames, "".concat(prefixCls, "-borderless"), !bordered), _classNames), className),
          locale: locale.lang,
          prefixCls: prefixCls,
          getPopupContainer: customGetPopupContainer || getPopupContainer,
          generateConfig: generateConfig,
          prevIcon: /*#__PURE__*/_react.default.createElement("span", {
            className: "".concat(prefixCls, "-prev-icon")
          }),
          nextIcon: /*#__PURE__*/_react.default.createElement("span", {
            className: "".concat(prefixCls, "-next-icon")
          }),
          superPrevIcon: /*#__PURE__*/_react.default.createElement("span", {
            className: "".concat(prefixCls, "-super-prev-icon")
          }),
          superNextIcon: /*#__PURE__*/_react.default.createElement("span", {
            className: "".concat(prefixCls, "-super-next-icon")
          }),
          direction: direction
        }));
      };

      return _this;
    }

    (0, _createClass2.default)(RangePicker, [{
      key: "render",
      value: function render() {
        return /*#__PURE__*/_react.default.createElement(_LocaleReceiver.default, {
          componentName: "DatePicker",
          defaultLocale: this.getDefaultLocale
        }, this.renderPicker);
      }
    }]);
    return RangePicker;
  }(_react.default.Component);

  RangePicker.contextType = _beeConfig.ConfigContext;
  return RangePicker;
}