@alitajs/antd-plus/es/range-picker-pro/utils.js

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

function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }

function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread 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 _iterableToArray(iter) { if (typeof Symbol !== "undefined" && Symbol.iterator in Object(iter)) return Array.from(iter); }

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

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

import moment from 'moment';
/**
 * 获取禁用的时间
 * @param start
 * @param end
 */

export function disabledRangeTime(start, end) {
  var result = [];

  for (var i = start; i < end; i++) {
    result.push(i);
  }

  return result;
}
/**
 * Ant Design 日期选择组件，禁用今天之后的时间
 * @param current
 */

export var disabledAfterTodayDate = function disabledAfterTodayDate(current) {
  return moment().endOf('day') < moment(current).endOf('day');
};
/**
 * 获取指定天的开始时间的时间戳
 * @param time
 * @example 2020-01-01 >> 2020-01-01 00:00:00 >> 时间戳
 */

export function startOf(time) {
  var type = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 'day';
  return moment(time).startOf(type).valueOf();
}
/**
 * 获取指定天的结束时间的时间戳
 * @param time
 * @example 2020-01-01 >> 2020-01-01 23:59:59 >> 时间戳
 */

export function endOf(time) {
  var type = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 'day';
  return moment(time).endOf(type).valueOf();
}

var getDisabledTimes = function getDisabledTimes(value) {
  var max = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 60;
  var minutes = [];

  for (var i = 0; i < max; i++) {
    if (i % value !== 0) {
      minutes.push(i);
    }
  }

  return minutes;
};

export function getRangePickerProps(_ref) {
  var _ref$periodType = _ref.periodType,
      periodType = _ref$periodType === void 0 ? 'minute' : _ref$periodType,
      _ref$periodValue = _ref.periodValue,
      periodValue = _ref$periodValue === void 0 ? 1 : _ref$periodValue;

  if (periodType === 'second') {
    var defaultDisabled = getDisabledTimes(periodValue);
    return {
      showTime: {
        format: 'HH:mm:ss'
      },
      format: 'YYYY-MM-DD HH:mm:ss',
      picker: undefined,
      disabledTime: function disabledTime(current) {
        var currentDate = moment();
        var hour = currentDate.hour();
        var minute = currentDate.minute();
        var second = currentDate.second(); // 如果上今天之前的则不限制

        if (current && current.isBefore(currentDate, 'days')) {
          return {
            disabledHours: function disabledHours() {
              return [];
            },
            disabledMinutes: function disabledMinutes() {
              return [];
            },
            disabledSeconds: function disabledSeconds() {
              return defaultDisabled;
            }
          };
        }

        return {
          disabledHours: function disabledHours() {
            return disabledRangeTime(hour + 1, 24);
          },
          disabledMinutes: function disabledMinutes(selectedHour) {
            if (selectedHour === hour) {
              return disabledRangeTime(minute + 1, 60);
            } else if (selectedHour > hour) {
              return disabledRangeTime(1, 60);
            } else {
              return [];
            }
          },
          disabledSeconds: function disabledSeconds(selectedHour, selectedMinute) {
            if (selectedHour === hour) {
              if (selectedMinute === minute) {
                return disabledRangeTime(second + 1, 60).concat(defaultDisabled);
              }

              if (selectedMinute > minute) {
                return disabledRangeTime(1, 60).concat(defaultDisabled);
              }
            }

            if (selectedHour > hour) {
              return disabledRangeTime(1, 60).concat(defaultDisabled);
            }

            return defaultDisabled;
          }
        };
      }
    };
  }

  if (periodType === 'minute') {
    var _defaultDisabled = getDisabledTimes(periodValue);

    return {
      showTime: {
        format: 'HH:mm'
      },
      format: 'YYYY-MM-DD HH:mm',
      picker: undefined,
      disabledTime: function disabledTime(current) {
        var now = moment();
        var hour = now.hour();
        var minute = now.minute();

        if (current && current.isBefore(now, 'days')) {
          return {
            disabledHours: function disabledHours() {
              return [];
            },
            disabledMinutes: function disabledMinutes() {
              return _defaultDisabled;
            }
          };
        }

        return {
          disabledHours: function disabledHours() {
            return disabledRangeTime(hour + 1, 24);
          },
          disabledMinutes: function disabledMinutes(selectedHour) {
            if (selectedHour === hour) {
              return [].concat(_toConsumableArray(_defaultDisabled), _toConsumableArray(disabledRangeTime(minute + 1, 60)));
            } else if (selectedHour > hour) {
              return [].concat(_toConsumableArray(_defaultDisabled), _toConsumableArray(disabledRangeTime(1, 60)));
            }

            return _defaultDisabled;
          }
        };
      }
    };
  }

  if (periodType === 'hour') {
    var _defaultDisabled2 = getDisabledTimes(periodValue, 24);

    return {
      showTime: {
        format: 'HH'
      },
      format: 'YYYY-MM-DD HH',
      picker: undefined,
      disabledTime: function disabledTime(current) {
        var now = moment();
        var hour = now.hour();

        if (current && current.isBefore(now, 'days')) {
          return {
            disabledHours: function disabledHours() {
              return _defaultDisabled2;
            }
          };
        }

        return {
          disabledHours: function disabledHours() {
            return disabledRangeTime(hour + 1, 24);
          }
        };
      }
    };
  }

  if (periodType === 'day') {
    return {
      showTime: false,
      format: 'YYYY-MM-DD',
      picker: undefined
    };
  }

  if (periodType === 'month') {
    return {
      showTime: false,
      format: 'YYYY-MM',
      picker: 'month'
    };
  }

  if (periodType === 'year') {
    return {
      showTime: false,
      format: 'YYYY',
      picker: 'year'
    };
  }

  return {};
}