Blame view

天文台pc/tianwentai-ui/node_modules/date-fns/addBusinessDays.js 2.32 KB
bc518174   王天杨   提交两个项目文件
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
  "use strict";
  exports.addBusinessDays = addBusinessDays;
  var _index = require("./constructFrom.js");
  var _index2 = require("./isSaturday.js");
  var _index3 = require("./isSunday.js");
  var _index4 = require("./isWeekend.js");
  var _index5 = require("./toDate.js");
  
  /**
   * @name addBusinessDays
   * @category Date Extension 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.
   *
   * @typeParam DateType - The `Date` type, the function operates on. Gets inferred from passed arguments. Allows to use extensions like [`UTCDate`](https://github.com/date-fns/utc).
   *
   * @param date - The date to be changed
   * @param amount - The amount of business days to be added.
   *
   * @returns The new date with the business days added
   *
   * @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)
   */
  function addBusinessDays(date, amount) {
    const _date = (0, _index5.toDate)(date);
    const startedOnWeekend = (0, _index4.isWeekend)(_date);
  
    if (isNaN(amount)) return (0, _index.constructFrom)(date, NaN);
  
    const hours = _date.getHours();
    const sign = amount < 0 ? -1 : 1;
    const fullWeeks = Math.trunc(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 (!(0, _index4.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 && (0, _index4.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 ((0, _index2.isSaturday)(_date))
        _date.setDate(_date.getDate() + (sign < 0 ? 2 : -1));
      if ((0, _index3.isSunday)(_date))
        _date.setDate(_date.getDate() + (sign < 0 ? 1 : -2));
    }
  
    // Restore hours to avoid DST lag
    _date.setHours(hours);
  
    return _date;
  }