preact-arco-design/es/Calendar/month.js

Arco Design React UI Library.

var __assign = this && this.__assign || function () {
  __assign = Object.assign || function (t) {
    for (var s, i = 1, n = arguments.length; i < n; i++) {
      s = arguments[i];

      for (var p in s) {
        if (Object.prototype.hasOwnProperty.call(s, p)) t[p] = s[p];
      }
    }

    return t;
  };

  return __assign.apply(this, arguments);
};

var __read = this && this.__read || function (o, n) {
  var m = typeof Symbol === "function" && o[Symbol.iterator];
  if (!m) return o;
  var i = m.call(o),
      r,
      ar = [],
      e;

  try {
    while ((n === void 0 || n-- > 0) && !(r = i.next()).done) {
      ar.push(r.value);
    }
  } catch (error) {
    e = {
      error: error
    };
  } finally {
    try {
      if (r && !r.done && (m = i["return"])) m.call(i);
    } finally {
      if (e) throw e.error;
    }
  }

  return ar;
};

var __spreadArray = this && this.__spreadArray || function (to, from, pack) {
  if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
    if (ar || !(i in from)) {
      if (!ar) ar = Array.prototype.slice.call(from, 0, i);
      ar[i] = from[i];
    }
  }
  return to.concat(ar || Array.prototype.slice.call(from));
};

import React from "preact/compat";
import { dayjs, methods } from "../_util/dayjs";
import { newArray } from "../_util/constant";
import { padStart } from "../_util/pad";
import cs from "../_util/classNames";
import WeekList from "./week-list";
import useCellClassName from "./hooks/useCellClassName";
var allDaysInOnePage = 6 * 7;

var getReturn = function getReturn(time) {
  return {
    year: time.year(),
    month: time.month() + 1,
    date: time.date(),
    day: time.day(),
    time: time
  };
};

var getTimeObj = function getTimeObj(time) {
  return {
    start: getReturn(methods.startOf(time, 'month')),
    end: getReturn(methods.endOf(time, 'month')),
    days: time.daysInMonth()
  };
};

export function getAllDaysByTime(props, time) {
  var _a = props.dayStartOfWeek,
      dayStartOfWeek = _a === void 0 ? 0 : _a,
      isWeek = props.isWeek;
  var current = getTimeObj(time);
  var flatRows = newArray(allDaysInOnePage).map(function () {
    return {};
  }); // current.start.day is 0 for Sunday

  var startIndex = dayStartOfWeek === 0 ? current.start.day : (current.start.day || 7) - 1;
  flatRows[startIndex] = __assign(__assign({}, current.start), {
    isCurrent: true
  }); // pre

  for (var i = 0; i < startIndex; i++) {
    flatRows[startIndex - i - 1] = __assign(__assign({}, getReturn(methods.subtract(current.start.time, i + 1, 'day'))), {
      isPrev: true
    });
  } // next


  for (var i = 0; i < allDaysInOnePage - startIndex - 1; i++) {
    flatRows[startIndex + i + 1] = __assign(__assign({}, getReturn(methods.add(current.start.time, i + 1, 'day'))), {
      isCurrent: i < current.days,
      isNext: i >= current.days - 1
    });
  }

  var rows = newArray(6).map(function () {
    return [];
  });

  for (var i = 0; i < 6; i++) {
    rows[i] = flatRows.slice(i * 7, 7 * (i + 1));

    if (isWeek) {
      var weekTime = rows[i][0].time;

      var weekRows = __spreadArray([], __read(rows[i]), false);

      rows[i].unshift({
        weekRows: weekRows,
        weekOfYear: weekTime.week()
      });
    }
  }

  return rows;
}

function Month(props) {
  var prefixCls = props.prefixCls,
      cell = props.cell,
      value = props.value,
      pageData = props.pageData,
      mergedValue = props.mergedValue,
      isWeek = props.isWeek,
      disabledDate = props.disabledDate,
      selectHandler = props.selectHandler,
      panel = props.panel,
      innerMode = props.innerMode,
      dateRender = props.dateRender,
      onMouseEnterCell = props.onMouseEnterCell,
      onMouseLeaveCell = props.onMouseLeaveCell,
      dateInnerContent = props.dateInnerContent,
      mergedPageShowDate = props.mergedPageShowDate,
      dayStartOfWeek = props.dayStartOfWeek,
      CALENDAR_LOCALE = props.CALENDAR_LOCALE;
  var pageShowDateYear = mergedPageShowDate.year();
  var getCellClassName = useCellClassName(__assign(__assign({}, props), {
    isSameTime: function isSameTime(current, target) {
      return current.isSame(target, 'day');
    }
  }));

  function renderDays(row) {
    return row.map(function (col, index) {
      var _a;

      if (col.time) {
        var disabled_1 = typeof disabledDate === 'function' && disabledDate(col.time);

        var onClickHandler = function onClickHandler() {
          return selectHandler(col.time, disabled_1);
        };

        var tdProps = isWeek ? {
          onClick: onClickHandler
        } : {};
        var tdDivProps = !isWeek ? {
          onClick: onClickHandler
        } : {};
        return React.createElement("div", __assign({
          key: index,
          className: getCellClassName(col, disabled_1),
          onMouseEnter: function onMouseEnter() {
            return onMouseEnterCell && onMouseEnterCell(col.time, disabled_1);
          },
          onMouseLeave: function onMouseLeave() {
            return onMouseLeaveCell && onMouseLeaveCell(col.time, disabled_1);
          }
        }, tdProps), dateRender ? React.cloneElement(dateRender(col.time), tdDivProps) : React.createElement("div", __assign({
          className: "".concat(prefixCls, "-date")
        }, tdDivProps), React.createElement("div", {
          className: "".concat(prefixCls, "-date-value")
        }, panel ? col.date : React.createElement("div", {
          className: "".concat(prefixCls, "-date-circle")
        }, col.date)), !panel && innerMode !== 'year' && React.createElement("div", {
          className: "".concat(prefixCls, "-date-content")
        }, dateInnerContent && dateInnerContent(col.time))));
      }

      if ('weekOfYear' in col) {
        var rowYear_1 = mergedValue && mergedValue.year();
        var rowMonth_1 = mergedValue && mergedValue.month() + 1;
        var rowWeek = mergedValue && mergedValue.week();
        var selectedWeek = mergedValue && col.weekRows.find(function (r) {
          return r.year === rowYear_1 && r.month === rowMonth_1;
        }) && rowWeek === col.weekOfYear;
        return React.createElement("div", {
          key: index,
          className: cs("".concat(prefixCls, "-cell"), "".concat(prefixCls, "-cell-week"), (_a = {}, _a["".concat(prefixCls, "-cell-selected-week")] = selectedWeek, _a["".concat(prefixCls, "-cell-in-range")] = selectedWeek, _a))
        }, React.createElement("div", {
          className: "".concat(prefixCls, "-date")
        }, React.createElement("div", {
          className: "".concat(prefixCls, "-date-value")
        }, col.weekOfYear)));
      }
    });
  }

  var pd = pageData;

  if (typeof value === 'number') {
    pd = getAllDaysByTime(props, dayjs("".concat(pageShowDateYear, "-").concat(padStart(value + 1, 2, '0'), "-01")));
  }

  return React.createElement("div", {
    className: cell ? "".concat(prefixCls, "-month-cell") : "".concat(prefixCls, "-month")
  }, React.createElement(WeekList, {
    prefixCls: prefixCls,
    dayStartOfWeek: dayStartOfWeek,
    isWeek: isWeek,
    CALENDAR_LOCALE: CALENDAR_LOCALE,
    panel: panel,
    innerMode: innerMode
  }), React.createElement("div", {
    className: "".concat(prefixCls, "-month-cell-body")
  }, pd.map(function (row, index) {
    var _a;

    return React.createElement("div", {
      key: index,
      className: cs("".concat(prefixCls, "-month-row"), (_a = {}, _a["".concat(prefixCls, "-row-week")] = isWeek, _a))
    }, renderDays(row));
  })));
}

export default Month;