@alitajs/antd-plus/es/days-range/days-range.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; }

var __rest = this && this.__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;
};

import React, { useState, useEffect, useCallback, useContext } from 'react';
import { isNumber, isInteger, isObject, isNil } from 'lodash';
import moment from 'moment';
import classNames from 'classnames';
import { ConfigContext } from '../config-provider';
import { processQueryTimeRange, transformTimeRangeToDay, transformDayToTimeRange } from './utils';
import Fast from './fast';
var Group = _Radio.Group,
    Button = _Radio.Button;
var RangePicker = _DatePicker.RangePicker;

var DaysRange = function DaysRange(_a) {
  var className = _a.className,
      style = _a.style,
      value = _a.value,
      onChange = _a.onChange,
      buttonStyle = _a.buttonStyle,
      size = _a.size,
      type = _a.type,
      formatter = _a.formatter,
      showCustomize = _a.showCustomize,
      isMountChange = _a.isMountChange,
      rest = __rest(_a, ["className", "style", "value", "onChange", "buttonStyle", "size", "type", "formatter", "showCustomize", "isMountChange"]);

  var _rest$marks = rest.marks,
      marks = _rest$marks === void 0 ? [] : _rest$marks;
  var isCustomize = false;

  var _useState = useState(true),
      _useState2 = _slicedToArray(_useState, 2),
      isFirst = _useState2[0],
      setIsFirst = _useState2[1];

  var _useState3 = useState(marks[0]),
      _useState4 = _slicedToArray(_useState3, 2),
      dayType = _useState4[0],
      setDayType = _useState4[1];

  var _useState5 = useState([]),
      _useState6 = _slicedToArray(_useState5, 2),
      timeRange = _useState6[0],
      setTimeRange = _useState6[1];

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

  var prefixCls = getPrefixCls('days-range');
  marks = marks.filter(function (item) {
    return item > 0;
  }).filter(function (item) {
    return isInteger(item);
  }).sort(function (a, b) {
    return a - b;
  });
  marks.indexOf(dayType) === -1 ? isCustomize = true : isCustomize = false;
  var Item = type === 'button' ? Button : _Radio;
  useEffect(function () {
    handleInit(value || marks[0]); // 兼容表单设置转换

    if (isNumber(value) && !isFirst) {
      setValueCallback(value);
    }
  }, [value, JSON.stringify(marks)]);

  var handleInit = function handleInit(val) {
    if (isObject(val) && !isNil(val.startTime) && !isNil(val.endTime)) {
      val = transformTimeRangeToDay(val);
    }

    if (isNumber(val) && val >= 1) {
      if (marks.indexOf(val) === -1) {
        isCustomize = true;
      }

      var defaultVal = val || marks[0];

      if (isCustomize) {
        var _timeRange = transformDayToTimeRange(val);

        setTimeRange(_timeRange);
      }

      setDayType(defaultVal);

      if (isFirst) {
        isMountChange && setValueCallback(defaultVal);
        setIsFirst(false);
      }
    }
  };

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

  var setValueCallback = useCallback(function (nextValue) {
    if (isNumber(nextValue)) {
      var range = transformDayToTimeRange(nextValue);
      var result = processQueryTimeRange(range);
      onChange && onChange({
        startTime: result[0],
        endTime: result[1]
      });
    }

    setDayType(nextValue);
  }, [setDayType]);

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

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

    onChange === null || onChange === void 0 ? void 0 : onChange({});
  };

  function disabledDate(current) {
    // Can not select days before today and today
    return current && current >= moment().endOf('day');
  }

  return /*#__PURE__*/React.createElement("span", {
    className: classNames(className, _defineProperty({}, "".concat(prefixCls), true)),
    style: style
  }, /*#__PURE__*/React.createElement(Group, {
    size: size,
    buttonStyle: buttonStyle,
    value: isCustomize ? 'customize' : dayType,
    onChange: handleChange
  }, marks.map(function (item) {
    var text;

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

    text = text || (item === 1 ? '今日' : "".concat(item, "\u5929"));
    return /*#__PURE__*/React.createElement(Item, {
      key: item,
      value: item
    }, text);
  }), showCustomize && /*#__PURE__*/React.createElement(Item, {
    key: "customize",
    value: "customize"
  }, (formatter === null || formatter === void 0 ? void 0 : formatter('customize')) ? formatter === null || formatter === void 0 ? void 0 : formatter('customize') : '自定义')), isCustomize && /*#__PURE__*/React.createElement(RangePicker, {
    size: size,
    value: timeRange,
    disabledDate: disabledDate,
    onChange: handleRangePickerChange
  }));
};

DaysRange.defaultProps = {
  type: 'button',
  marks: [7, 30],
  showCustomize: true,
  isMountChange: true
};
DaysRange.Fast = Fast;
export default DaysRange;