diff --git a/src/EventGrouping/DayGrouping.ts b/src/EventGrouping/DayGrouping.ts index ee4b8b6..7e0ccf1 100644 --- a/src/EventGrouping/DayGrouping.ts +++ b/src/EventGrouping/DayGrouping.ts @@ -1,8 +1,16 @@ import EventGrouping from "./EventGrouping"; import { Event, Groups } from "../types"; -import { add, differenceInCalendarDays } from "date-fns"; +import { + add, + differenceInCalendarDays, + isToday as dateFnsIsToday, +} from "date-fns"; export default class DayGrouping extends EventGrouping { + static isToday(date: Date): boolean { + return dateFnsIsToday(date); + } + constructor(sortedEvents: Event[]) { super(sortedEvents); this.groups = this.group(); diff --git a/src/EventGrouping/MonthGrouping.ts b/src/EventGrouping/MonthGrouping.ts index eadcc42..322857c 100644 --- a/src/EventGrouping/MonthGrouping.ts +++ b/src/EventGrouping/MonthGrouping.ts @@ -1,8 +1,16 @@ import EventGrouping from "./EventGrouping"; import { Event, Groups } from "../types"; -import { add, differenceInCalendarMonths } from "date-fns"; +import { + add, + differenceInCalendarMonths, + isThisMonth as dateFnsIsThisMonth, +} from "date-fns"; export default class MonthGrouping extends EventGrouping { + static isThisMonth(date: Date): boolean { + return dateFnsIsThisMonth(date); + } + constructor(sortedEvents: Event[]) { super(sortedEvents); this.groups = this.group(); diff --git a/src/EventGrouping/WeekGrouping.ts b/src/EventGrouping/WeekGrouping.ts index 4af0350..6c028de 100644 --- a/src/EventGrouping/WeekGrouping.ts +++ b/src/EventGrouping/WeekGrouping.ts @@ -1,8 +1,21 @@ import EventGrouping from "./EventGrouping"; import { Event, Groups } from "../types"; -import { add, differenceInCalendarWeeks } from "date-fns"; +import { + add, + differenceInCalendarWeeks, + getWeek as dateFnsGetWeek, + isThisWeek as dateFnsIsThisWeek, +} from "date-fns"; export default class WeekGrouping extends EventGrouping { + static isThisWeek(date: Date): boolean { + return dateFnsIsThisWeek(date); + } + + static getWeek(date: Date): number { + return dateFnsGetWeek(date); + } + constructor(sortedEvents: Event[]) { super(sortedEvents); this.groups = this.group();