Blame view

天文台pc/tianwentai-ui/node_modules/react-day-picker/src/components/Table/utils/daysToMonthWeeks.ts 1.29 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
  import {
    addDays,
    differenceInCalendarDays,
    endOfISOWeek,
    endOfWeek,
    getISOWeek,
    getWeek,
    Locale,
    startOfISOWeek,
    startOfWeek
  } from 'date-fns';
  
  import { MonthWeek } from './getMonthWeeks';
  
  /** Return the weeks between two dates.  */
  export function daysToMonthWeeks(
    fromDate: Date,
    toDate: Date,
    options?: {
      ISOWeek?: boolean;
      locale?: Locale;
      weekStartsOn?: 0 | 1 | 2 | 3 | 4 | 5 | 6;
      firstWeekContainsDate?: 1 | 4;
    }
  ): MonthWeek[] {
    const toWeek = options?.ISOWeek
      ? endOfISOWeek(toDate)
      : endOfWeek(toDate, options);
    const fromWeek = options?.ISOWeek
      ? startOfISOWeek(fromDate)
      : startOfWeek(fromDate, options);
  
    const nOfDays = differenceInCalendarDays(toWeek, fromWeek);
    const days: Date[] = [];
  
    for (let i = 0; i <= nOfDays; i++) {
      days.push(addDays(fromWeek, i));
    }
  
    const weeksInMonth = days.reduce((result: MonthWeek[], date) => {
      const weekNumber = options?.ISOWeek
        ? getISOWeek(date)
        : getWeek(date, options);
  
      const existingWeek = result.find(
        (value) => value.weekNumber === weekNumber
      );
      if (existingWeek) {
        existingWeek.dates.push(date);
        return result;
      }
      result.push({
        weekNumber,
        dates: [date]
      });
      return result;
    }, []);
  
    return weeksInMonth;
  }