Function getDateArray

src/shared/src/date.ts:55–99

(
  data: T[],
  startDate: ConfigType,
  endDate: ConfigType,
  unit: DateUnit,
  timezone?: string
)

 * @param timezone - If not provided, timezone will not be processed, otherwise the output date will be converted to the corresponding timezone
 */
export function getDateArray<T extends { date: string }>(
  data: T[],
  startDate: ConfigType,
  endDate: ConfigType,
  unit: DateUnit,
  timezone?: string
): T[] {
  if (data.length === 0) {
    return [];
  }

  const defaultItem: Omit<T, 'date'> = Object.keys(data[0]).reduce(
    (acc: any, key) => {
      if (key === 'date') {
        return acc;
      }

      acc[key] = 0;

      return acc;
    },
    {}
  );

  const arr: T[] = [];
  const { diff, add, normalize } = createDateUnitFn(unit, timezone);
  const n = diff(endDate, startDate) + 1;

  const dataMap = new Map<number, T>();
  data.forEach((item) => {
    const timestamp = timezone
      ? normalize(dayjs.tz(item.date, timezone)).valueOf()
      : normalize(dayjs(item.date)).valueOf();
    dataMap.set(timestamp, item);
  });

  for (let i = 0; i < n; i++) {
    const t = normalize(add(startDate, i));
    const target = dataMap.get(t.valueOf());

    arr.push({ ...defaultItem, ...target, date: formatDate(t, timezone) } as T);
  }

  return arr;
}

/**
 * Infer the time range from the dates