docs: fix custom-multiple example

gpbl · Jan 25, 2023 · 6dcaa4b · 6dcaa4b
1 parent 3e4eefc
commit 6dcaa4b
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/website/examples/custom-multiple.tsx b/website/examples/custom-multiple.tsx
@@ -1,20 +1,22 @@
 import React, { useState } from 'react';
 
+import { isSameDay } from 'date-fns';
 import { DayClickEventHandler, DayPicker } from 'react-day-picker';
 
 export default function App() {
   const [selectedDays, setSelectedDays] = useState<Date[]>([]);
 
   const handleDayClick: DayClickEventHandler = (day, modifiers) => {
-    setSelectedDays((currentValue) => {
-      const days = [...currentValue];
-      if (modifiers.selected) {
-        days.splice(currentValue.indexOf(day), 1);
-      } else {
-        days.push(day);
-      }
-      return days;
-    });
+    const newSelectedDays = [...selectedDays];
+    if (modifiers.selected) {
+      const index = selectedDays.findIndex((selectedDay) =>
+        isSameDay(day, selectedDay)
+      );
+      newSelectedDays.splice(index, 1);
+    } else {
+      newSelectedDays.push(day);
+    }
+    setSelectedDays(newSelectedDays);
   };
 
   const handleResetClick = () => setSelectedDays([]);