rsuite
Version:
A suite of react components
31 lines (25 loc) • 693 B
JavaScript
import { addDays, getDay } from 'date-fns';
/**
* Get all weeks of this month
* @params monthDate
* @return date[]
*/
export default function getMonthView(monthDate, isoWeek) {
var firstDayOfMonth = getDay(monthDate);
var distance = 0 - firstDayOfMonth;
if (isoWeek) {
distance = 1 - firstDayOfMonth;
if (firstDayOfMonth === 0) {
distance = -6;
}
}
var firstWeekendDate = addDays(monthDate, distance);
var weeks = [firstWeekendDate];
var nextWeekendDate = addDays(firstWeekendDate, 7);
weeks.push(nextWeekendDate);
while (weeks.length < 6) {
nextWeekendDate = addDays(nextWeekendDate, 7);
weeks.push(nextWeekendDate);
}
return weeks;
}