@alitajs/antd-plus/lib/days-range/fast.js

基于 ant-design 封装的偏业务组件

"use strict";

function _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }

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

require("antd/es/date-picker/style");

var _datePicker = _interopRequireDefault(require("antd/es/date-picker"));

require("antd/es/radio/style");

var _radio = _interopRequireDefault(require("antd/es/radio"));

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

var _lodash = require("lodash");

var _LocaleReceiver = _interopRequireDefault(require("antd/es/locale-provider/LocaleReceiver"));

var _moment = _interopRequireDefault(require("moment"));

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

var _configProvider = require("../config-provider");

var _en_US = _interopRequireDefault(require("./locale/en_US"));

var _utils = require("./utils");

function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function _getRequireWildcardCache() { return cache; }; return cache; }

function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } if (obj === null || _typeof(obj) !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (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 _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }

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(arr, i) { if (typeof Symbol === "undefined" || !(Symbol.iterator in Object(arr))) return; var _arr = []; var _n = true; var _d = false; var _e = undefined; try { for (var _i = arr[Symbol.iterator](), _s; !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"] != null) _i["return"](); } finally { if (_d) throw _e; } } return _arr; }

function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }

var Group = _radio.default.Group,
    Button = _radio.default.Button;
var RangePicker = _datePicker.default.RangePicker;

var Fast = function Fast(_ref) {
  var className = _ref.className,
      style = _ref.style,
      value = _ref.value,
      onChange = _ref.onChange,
      buttonStyle = _ref.buttonStyle,
      size = _ref.size,
      type = _ref.type,
      formatter = _ref.formatter,
      showCustomize = _ref.showCustomize,
      _ref$marks = _ref.marks,
      marks = _ref$marks === void 0 ? [] : _ref$marks;

  var _useState = (0, _react.useState)(),
      _useState2 = _slicedToArray(_useState, 2),
      currentUnit = _useState2[0],
      setCurrentUnit = _useState2[1];

  var _useState3 = (0, _react.useState)(),
      _useState4 = _slicedToArray(_useState3, 2),
      timeRange = _useState4[0],
      setTimeRange = _useState4[1];

  var _useContext = (0, _react.useContext)(_configProvider.ConfigContext),
      getPrefixCls = _useContext.getPrefixCls;

  (0, _react.useEffect)(function () {
    if ((0, _lodash.isObject)(value) && value.startTime && value.endTime) {
      var range = [(0, _moment.default)(value.startTime), (0, _moment.default)(value.endTime)];
      setTimeRange(range);
    }
  }, [value]);
  (0, _react.useEffect)(function () {
    setCurrentUnit(getUnitByTimeRange(timeRange));
  }, [timeRange]);
  var prefixCls = getPrefixCls('days-range');
  var Item = type === 'button' ? Button : _radio.default;

  var handleChange = function handleChange(e) {
    setValueCallback(e.target.value);
  };

  var getTimeRange = function getTimeRange() {
    var unit = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : 'day';
    var startTime = (0, _moment.default)().startOf(unit);
    return [startTime, (0, _moment.default)()];
  };

  var getUnitByTimeRange = function getUnitByTimeRange(range) {
    if (range && range.length === 2 && _moment.default.isMoment(range[0]) && _moment.default.isMoment(range[1]) && formatTime(range[1]) === formatTime((0, _moment.default)())) {
      if (formatTime((0, _moment.default)().startOf('day')) === formatTime(range[0])) {
        return 'day';
      }

      if (formatTime((0, _moment.default)().startOf('week')) === formatTime(range[0])) {
        return 'week';
      }

      if (formatTime((0, _moment.default)().startOf('month')) === formatTime(range[0])) {
        return 'month';
      }

      if (formatTime((0, _moment.default)().startOf('year')) === formatTime(range[0])) {
        return 'year';
      }
    }

    return undefined;
  };

  var formatTime = function formatTime(time) {
    return time.format('YYYY-MM-DD');
  };

  var setValueCallback = (0, _react.useCallback)(function (nextValue) {
    var range = getTimeRange(nextValue);
    var result = (0, _utils.processQueryTimeRange)(range);
    onChange && onChange({
      startTime: result[0],
      endTime: result[1]
    });
    setTimeRange(range);
    setCurrentUnit(nextValue);
  }, [setCurrentUnit]);

  var handleRangePickerChange = function handleRangePickerChange(dates) {
    setTimeRange(dates);

    if (dates && dates.length === 2) {
      var result = (0, _utils.processQueryTimeRange)(dates);
      onChange && onChange({
        startTime: result[0],
        endTime: result[1]
      });
      return;
    }

    onChange && onChange({});
  };

  function disabledDate(current) {
    return current && current >= (0, _moment.default)().endOf('day');
  }

  var renderDaysRange = function renderDaysRange(locale) {
    var texts = {
      day: locale.day,
      week: locale.week,
      month: locale.month,
      year: locale.year
    };
    return /*#__PURE__*/_react.default.createElement("span", {
      className: (0, _classnames.default)(className, _defineProperty({}, "".concat(prefixCls), true)),
      style: style
    }, /*#__PURE__*/_react.default.createElement(Group, {
      size: size,
      buttonStyle: buttonStyle,
      value: currentUnit,
      onChange: handleChange
    }, marks.map(function (item) {
      var text;

      if (formatter && formatter(item)) {
        text = formatter(item);
      }

      return /*#__PURE__*/_react.default.createElement(Item, {
        key: item,
        value: item
      }, text || texts[item]);
    })), showCustomize && /*#__PURE__*/_react.default.createElement(RangePicker, {
      size: size,
      value: timeRange,
      disabledDate: disabledDate,
      onChange: handleRangePickerChange
    }));
  };

  return /*#__PURE__*/_react.default.createElement(_LocaleReceiver.default, {
    componentName: "DaysRange",
    defaultLocale: _en_US.default
  }, function (locale) {
    return renderDaysRange(locale);
  });
};

Fast.defaultProps = {
  type: 'button',
  marks: ['day', 'week', 'month'],
  showCustomize: true
};
var _default = Fast;
exports.default = _default;