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

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

import "antd/es/date-picker/style";
import _DatePicker from "antd/es/date-picker";
import "antd/es/radio/style";
import _Radio from "antd/es/radio";

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

import React, { useState, useEffect, useCallback, useContext } from 'react';
import { isObject } from 'lodash';
import LocaleReceiver from 'antd/es/locale-provider/LocaleReceiver';
import moment from 'moment';
import classNames from 'classnames';
import { ConfigContext } from '../config-provider';
import enUS from './locale/en_US';
import { processQueryTimeRange } from './utils';
var Group = _Radio.Group,
    Button = _Radio.Button;
var RangePicker = _DatePicker.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 = useState(),
      _useState2 = _slicedToArray(_useState, 2),
      currentUnit = _useState2[0],
      setCurrentUnit = _useState2[1];

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

  var _useContext = useContext(ConfigContext),
      getPrefixCls = _useContext.getPrefixCls;

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

  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 = moment().startOf(unit);
    return [startTime, moment()];
  };

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

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

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

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

    return undefined;
  };

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

  var setValueCallback = useCallback(function (nextValue) {
    var range = getTimeRange(nextValue);
    var result = 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 = processQueryTimeRange(dates);
      onChange && onChange({
        startTime: result[0],
        endTime: result[1]
      });
      return;
    }

    onChange && onChange({});
  };

  function disabledDate(current) {
    return current && current >= moment().endOf('day');
  }

  var renderDaysRange = function renderDaysRange(locale) {
    var texts = {
      day: locale.day,
      week: locale.week,
      month: locale.month,
      year: locale.year
    };
    return /*#__PURE__*/React.createElement("span", {
      className: classNames(className, _defineProperty({}, "".concat(prefixCls), true)),
      style: style
    }, /*#__PURE__*/React.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.createElement(Item, {
        key: item,
        value: item
      }, text || texts[item]);
    })), showCustomize && /*#__PURE__*/React.createElement(RangePicker, {
      size: size,
      value: timeRange,
      disabledDate: disabledDate,
      onChange: handleRangePickerChange
    }));
  };

  return /*#__PURE__*/React.createElement(LocaleReceiver, {
    componentName: "DaysRange",
    defaultLocale: enUS
  }, function (locale) {
    return renderDaysRange(locale);
  });
};

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