Add range-start and range-end class in month range (Hacker0x01#1794)

p-rogas · Jun 25, 2019 · 356ced8 · 356ced8
1 parent 51ed59a
commit 356ced8
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/src/month.jsx b/src/month.jsx
@@ -62,6 +62,22 @@ export default class Month extends React.Component {
     }
   };
 
+  isRangeStart = () => {
+    const { day, startDate, endDate } = this.props;
+    if (!startDate || !endDate) {
+      return false;
+    }
+    return utils.isSameMonth(startDate, day);
+  };
+
+  isRangeEnd = () => {
+    const { day, startDate, endDate } = this.props;
+    if (!startDate || !endDate) {
+      return false;
+    }
+    return utils.isSameMonth(endDate, day);
+  };
+
   isWeekInMonth = startOfWeek => {
     const day = this.props.day;
     const endOfWeek = utils.addDays(startOfWeek, 6);
@@ -161,7 +177,9 @@ export default class Month extends React.Component {
           endDate,
           m,
           day
-        )
+        ),
+        "react-datepicker__month--range-start": this.isRangeStart(),
+        "react-datepicker__month--range-end": this.isRangeEnd(),
       }
     );
   };