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

wix-style-react

import addMonths from 'date-fns/addMonths';
import startOfMonth from 'date-fns/startOfMonth';
import endOfMonth from 'date-fns/endOfMonth';
import endOfDay from 'date-fns/endOfDay';
import isWithinInterval from 'date-fns/isWithinInterval';
import isAfter from 'date-fns/isAfter';
import isBefore from 'date-fns/isBefore';
import differenceInCalendarDays from 'date-fns/differenceInCalendarDays';

/**
 * 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 isWithinInterval(date, { start: this.startDate, end: 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);
  }
}