blob: 0cfdf5ce806987782257f1089cd3f0ce26f766a2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
import isWeekend from '../isWeekend/index'
import toDate from '../toDate/index'
import toInteger from '../_lib/toInteger/index'
import requiredArgs from '../_lib/requiredArgs/index'
import isSunday from '../isSunday/index'
import isSaturday from '../isSaturday/index'
/**
* @name addBusinessDays
* @category Day Helpers
* @summary Add the specified number of business days (mon - fri) to the given date.
*
* @description
* Add the specified number of business days (mon - fri) to the given date, ignoring weekends.
*
* @param {Date|Number} date - the date to be changed
* @param {Number} amount - the amount of business days to be added. Positive decimals will be rounded using `Math.floor`, decimals less than zero will be rounded using `Math.ceil`.
* @returns {Date} the new date with the business days added
* @throws {TypeError} 2 arguments required
*
* @example
* // Add 10 business days to 1 September 2014:
* const result = addBusinessDays(new Date(2014, 8, 1), 10)
* //=> Mon Sep 15 2014 00:00:00 (skipped weekend days)
*/
export default function addBusinessDays(
dirtyDate: Date | number,
dirtyAmount: number
): Date {
requiredArgs(2, arguments)
const date = toDate(dirtyDate)
const startedOnWeekend = isWeekend(date)
const amount = toInteger(dirtyAmount)
if (isNaN(amount)) return new Date(NaN)
const hours = date.getHours()
const sign = amount < 0 ? -1 : 1
const fullWeeks = toInteger(amount / 5)
date.setDate(date.getDate() + fullWeeks * 7)
// Get remaining days not part of a full week
let restDays = Math.abs(amount % 5)
// Loops over remaining days
while (restDays > 0) {
date.setDate(date.getDate() + sign)
if (!isWeekend(date)) restDays -= 1
}
// If the date is a weekend day and we reduce a dividable of
// 5 from it, we land on a weekend date.
// To counter this, we add days accordingly to land on the next business day
if (startedOnWeekend && isWeekend(date) && amount !== 0) {
// If we're reducing days, we want to add days until we land on a weekday
// If we're adding days we want to reduce days until we land on a weekday
if (isSaturday(date)) date.setDate(date.getDate() + (sign < 0 ? 2 : -1))
if (isSunday(date)) date.setDate(date.getDate() + (sign < 0 ? 1 : -2))
}
// Restore hours to avoid DST lag
date.setHours(hours)
return date
}
|
