wix-style-react/src/Calendar/utils.js

wix-style-react

import addMonths from 'date-fns/add_months';
import startOfMonth from 'date-fns/start_of_month';
import endOfMonth from 'date-fns/end_of_month';
import endOfDay from 'date-fns/end_of_day';
import isWithinRange from 'date-fns/is_within_range';
import isAfter from 'date-fns/is_after';
import isBefore from 'date-fns/is_before';
import differenceInCalendarDays from 'date-fns/difference_in_calendar_days';

/**
 * Date utils for a multi-month calendar view.
 */
export class CalendarView {
  /**
   * @param {*} firstMonthDate a date that represents the first displayed month in the calendar view. (Does not have to be the 1st of that month)
   * @param {*} numOfMonths number of months that are displayed in the view
   */
  constructor(firstMonthDate, numOfMonths) {
    this.startDate = startOfMonth(firstMonthDate);
    this.endDate = endOfDay(
      endOfMonth(addMonths(this.startDate, numOfMonths - 1)),
    );
  }

  /**
   * Is the given date contained in the calendar view
   *
   * @param {*} date
   * @returns
   * @memberof CalendarView
   */
  isContained(date) {
    return isWithinRange(date, this.startDate, this.endDate);
  }

  isRangeFits(from, to) {
    return (
      differenceInCalendarDays(to, from) <=
      differenceInCalendarDays(this.endDate, this.startDate)
    );
  }

  /**
   * Is the given date after the view's end date
   */
  isAfterView(date) {
    return isAfter(date, this.endDate);
  }

  /**
   * Is the given date before the view's start date
   */
  isBeforeView(date) {
    return isBefore(date, this.startDate);
  }
}