feat(controls): add next and prev years

hiwllc · Jul 28, 2022 · ae40369 · ae40369
1 parent f265ef3
commit ae40369
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/hooks/use-calendar.tsx b/src/hooks/use-calendar.tsx
@@ -1,6 +1,7 @@
 import * as React from 'react'
 import {
   addMonths,
+  addYears,
   eachDayOfInterval,
   endOfMonth,
   endOfWeek,
@@ -11,6 +12,7 @@ import {
   startOfMonth,
   startOfWeek,
   subMonths,
+  subYears,
 } from 'date-fns'
 import { Range, RangeSelection, Target } from '../types'
 
@@ -58,6 +60,16 @@ export function useCalendar({
     [state]
   )
 
+  const onPrevYear = React.useCallback(
+    () => setState(() => subYears(state, 1)),
+    [state]
+  )
+
+  const onNextYear = React.useCallback(
+    () => setState(() => addYears(state, 1)),
+    [state]
+  )
+
   const dates = React.useMemo(() => {
     return [...Array(months).keys()].map(index => {
       const currentMonth = addMonths(state, index)
@@ -129,14 +141,18 @@ export function useCalendar({
       value: null,
       onSelectDate,
       onNextMonth,
+      onNextYear,
       onPrevMonth,
+      onPrevYear,
       selected: singleDateSelection ? date : range,
     }
   }, [
     state,
     dates,
     onNextMonth,
     onPrevMonth,
+    onNextYear,
+    onPrevYear,
     onSelectDate,
     range,
     singleDateSelection,
@@ -155,6 +171,8 @@ export function useCalendar({
     getMonthProps,
     onPrevMonth,
     onNextMonth,
+    onPrevYear,
+    onNextYear,
     months: dates,
     dates: range,
   }