zarm/lib/calendar/Calendar.js

基于 React 的移动端UI库

"use strict";

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

var _typeof = require("@babel/runtime/helpers/typeof");

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

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

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

var _bem = require("@zarm-design/bem");

var _dayjs = _interopRequireDefault(require("dayjs"));

var _throttle = _interopRequireDefault(require("lodash/throttle"));

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

var _reactTransitionGroup = require("react-transition-group");

var _carousel = _interopRequireDefault(require("../carousel"));

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

var _useScroll = _interopRequireDefault(require("../use-scroll"));

var _Header = _interopRequireDefault(require("./Header"));

var _Month = _interopRequireDefault(require("./Month"));

var _parseState = _interopRequireDefault(require("./utils/parseState"));

var _Week = _interopRequireDefault(require("./Week"));

function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(e) { return e ? t : r; })(e); }

function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != _typeof(e) && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) { if ("default" !== u && {}.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } } return n.default = e, t && t.set(e, n), n; }

function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }

function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { (0, _defineProperty2.default)(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }

var Calendar = /*#__PURE__*/_react.default.forwardRef(function (props, ref) {
  var className = props.className,
      dateRender = props.dateRender,
      disabledDate = props.disabledDate,
      onChange = props.onChange,
      onSelect = props.onSelect,
      mode = props.mode,
      maxDate = props.max,
      minDate = props.min,
      direction = props.direction,
      header = props.header;

  var container = ref || /*#__PURE__*/_react.default.createRef();

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

  var carouselRef = (0, _react.useRef)(null);
  var bem = (0, _bem.createBEM)('calendar', {
    prefixCls: prefixCls
  });
  var cls = bem([className]);

  var _useState = (0, _react.useState)(function () {
    return _objectSpread(_objectSpread({}, (0, _parseState.default)(props)), {}, {
      step: 0
    });
  }),
      _useState2 = (0, _slicedToArray2.default)(_useState, 2),
      state = _useState2[0],
      setState = _useState2[1];

  var _useMemo = (0, _react.useMemo)(function () {
    var minDay = minDate ? (0, _dayjs.default)(minDate).toDate() : (0, _dayjs.default)().toDate();
    var maxDay = maxDate ? (0, _dayjs.default)(maxDate).toDate() : (0, _dayjs.default)().add(1, 'year').toDate();
    var duration = [minDay, maxDay].sort(function (item1, item2) {
      return +item1 - +item2;
    });
    return duration;
  }, [maxDate, minDate]),
      _useMemo2 = (0, _slicedToArray2.default)(_useMemo, 2),
      min = _useMemo2[0],
      max = _useMemo2[1];

  var steps = (0, _react.useMemo)(function () {
    return mode === 'range' ? 2 : 1;
  }, [mode]);
  var value = state.value;
  var nodes = (0, _react.useRef)({});

  var scrollBodyRef = /*#__PURE__*/_react.default.createRef();

  var weekRef = _react.default.useRef();

  var _useState3 = (0, _react.useState)(),
      _useState4 = (0, _slicedToArray2.default)(_useState3, 2),
      scrollDate = _useState4[0],
      setScrollDate = _useState4[1];

  var _useState5 = (0, _react.useState)(false),
      _useState6 = (0, _slicedToArray2.default)(_useState5, 2),
      scrolling = _useState6[0],
      setScrolling = _useState6[1];

  var isHorizontal = (0, _react.useMemo)(function () {
    return direction === 'horizontal';
  }, [direction]);
  var months = (0, _react.useMemo)(function () {
    var month = [];
    var dateMax = (0, _dayjs.default)(max);
    var dateMin = (0, _dayjs.default)(min);
    var len = (dateMax.year() - dateMin.year()) * 12 + dateMax.month() - dateMin.month();
    var i = 0;

    do {
      month.push((0, _dayjs.default)(min).add(i, 'month').toDate());
      i += 1;
    } while (i <= len);

    return month;
  }, [max, min]);
  var currentMonthIndex = (0, _react.useMemo)(function () {
    var currentTime = (0, _dayjs.default)(value[0] || new Date());
    return months.findIndex(function (current) {
      return (0, _dayjs.default)(current).isSame(currentTime, 'month');
    });
  }, [value]);

  var _useState7 = (0, _react.useState)(currentMonthIndex),
      _useState8 = (0, _slicedToArray2.default)(_useState7, 2),
      currentMonth = _useState8[0],
      setCurrentMonth = _useState8[1]; // 月历定位


  var scrollIntoView = (0, _react.useRef)(false);

  var anchor = function anchor() {
    var _node$el, _node$el$scrollIntoVi;

    var target = value[0] || new Date();
    var key = "".concat(target.getFullYear(), "-").concat(target.getMonth() + 1);
    var node = nodes.current[key];
    node === null || node === void 0 ? void 0 : (_node$el = node.el()) === null || _node$el === void 0 ? void 0 : (_node$el$scrollIntoVi = _node$el.scrollIntoView) === null || _node$el$scrollIntoVi === void 0 ? void 0 : _node$el$scrollIntoVi.call(_node$el);
  };

  var handleDateClick = (0, _react.useCallback)(function (date) {
    var step = state.step;
    var currentStep = step + 1;
    var idx = value.map(Number).indexOf(Number(date));

    if (currentStep === 1 && mode !== 'multiple') {
      value.splice(0, value.length);
    }

    if (mode === 'range') {
      value[currentStep - 1] = date;
    } else if (idx > -1) {
      value.splice(idx, 1);
    } else if (mode === 'multiple') {
      value.push(date);
    } else {
      value[currentStep - 1] = date;
    }

    value.sort(function (item1, item2) {
      return +item1 - +item2;
    });
    setState(function (prevState) {
      return _objectSpread(_objectSpread({}, prevState), {}, {
        value: value,
        step: currentStep === steps ? 0 : currentStep
      });
    });

    if ((currentStep >= steps || mode === 'multiple') && typeof onChange === 'function') {
      onChange(value);
    }

    if (typeof onSelect === 'function') {
      onSelect(value);
    }
  }, [mode, state, value, onChange]);
  var renderMonth = (0, _react.useCallback)(function (dateMonth) {
    var key = "".concat(dateMonth.getFullYear(), "-").concat(dateMonth.getMonth() + 1);
    return /*#__PURE__*/_react.default.createElement(_Month.default, {
      key: key,
      min: min,
      max: max,
      mode: mode,
      value: value,
      dateMonth: dateMonth,
      dateRender: dateRender,
      disabledDate: disabledDate,
      onDateClick: handleDateClick,
      ref: function ref(n) {
        nodes.current[key] = n;
      }
    });
  }, [min, max, mode, value, dateRender, disabledDate, handleDateClick, nodes]);
  var content = (0, _react.useMemo)(function () {
    return months.map(function (item) {
      return renderMonth(item);
    });
  }, [renderMonth]);
  var showHeader = (0, _react.useMemo)(function () {
    return direction === 'horizontal' && header;
  }, [direction, header]);
  var bodyScroll = (0, _throttle.default)(function () {
    var weekNode = weekRef === null || weekRef === void 0 ? void 0 : weekRef.current;
    var body = scrollBodyRef.current;

    if (!body) {
      return false;
    }

    var _ref = body,
        scrollTop = _ref.scrollTop;
    var keys = Object.keys(nodes === null || nodes === void 0 ? void 0 : nodes.current);
    keys = keys.sort(function (a, b) {
      return new Date(a).getTime() - new Date(b).getTime();
    });

    for (var i = 0; i < keys.length; i++) {
      var _nodes$current$keys$i;

      var el = nodes === null || nodes === void 0 ? void 0 : (_nodes$current$keys$i = nodes.current[keys[i]]) === null || _nodes$current$keys$i === void 0 ? void 0 : _nodes$current$keys$i.el();

      if ((el === null || el === void 0 ? void 0 : el.offsetTop) + (el === null || el === void 0 ? void 0 : el.clientHeight) - (weekNode === null || weekNode === void 0 ? void 0 : weekNode.clientHeight) > scrollTop) {
        setScrollDate(el.getAttribute('title'));
        return;
      }
    }
  }, 150);
  var monthsContent = (0, _react.useMemo)(function () {
    if (isHorizontal) {
      return /*#__PURE__*/_react.default.createElement(_carousel.default, {
        className: bem('body'),
        showPagination: false,
        activeIndex: currentMonth,
        onChange: setCurrentMonth,
        ref: carouselRef
      }, content);
    }

    return /*#__PURE__*/_react.default.createElement("div", {
      className: bem('body'),
      ref: scrollBodyRef,
      onScroll: bodyScroll
    }, content, /*#__PURE__*/_react.default.createElement(_reactTransitionGroup.Transition, {
      in: scrolling,
      timeout: 500
    }, function (tState) {
      return /*#__PURE__*/_react.default.createElement("div", {
        className: bem('scroll-month', [(0, _defineProperty2.default)({}, tState, true)])
      }, scrollDate);
    }));
  }, [mode, currentMonth, content, isHorizontal, scrollDate, scrolling]);
  (0, _react.useEffect)(function () {
    !isHorizontal && anchor();
  }, [direction, minDate, maxDate]);
  var timer = (0, _react.useRef)();
  (0, _useScroll.default)({
    container: scrollBodyRef,
    onScroll: function onScroll() {
      !scrollIntoView.current && setScrolling(true);

      if (timer.current) {
        clearTimeout(timer.current);
      }

      timer.current = setTimeout(function () {
        setScrolling(false);
        scrollIntoView.current = false;
      }, 250);
    }
  });
  return /*#__PURE__*/_react.default.createElement("div", {
    className: cls,
    ref: container
  }, showHeader && /*#__PURE__*/_react.default.createElement(_Header.default, {
    changeMonth: function changeMonth(idx) {
      var _carouselRef$current;

      // @ts-ignore
      carouselRef === null || carouselRef === void 0 ? void 0 : (_carouselRef$current = carouselRef.current) === null || _carouselRef$current === void 0 ? void 0 : _carouselRef$current.onSlideTo(idx);
    },
    months: months,
    currentMonth: currentMonth
  }), /*#__PURE__*/_react.default.createElement(_Week.default, {
    ref: weekRef
  }), monthsContent);
});

Calendar.defaultProps = {
  mode: 'single',
  dateRender: function dateRender(date) {
    return date.getDate();
  },
  disabledDate: function disabledDate() {
    return false;
  },
  direction: 'vertical',
  header: false
};
var _default = Calendar;
exports.default = _default;