dateParse.ts

import { arrayToMap } from './../array/arrayToMap'
import { Nullable } from '../interface/Nullable'
import { isRange } from '../number/isRange'

/**
 * 日期格式化类
 */
class DateFormat {
  /**
   * 构造函数
   * @param name 日期格式的名称
   * @param format 日期的格式值
   * @param value 格式化得到的值
   * @param index 需要替换位置的索引
   */
  constructor(
    public name: string,
    public format: string,
    public value: string,
    public index: number,
  ) {}
}

/**
 * 日期时间的正则表达式
 */
const dateFormats = new Map()
  .set('year', 'Y{4}|Y{2}|y{4}|y{2}')
  .set('month', 'M{1,2}')
  .set('day', 'D{1,2}|d{1,2}')
  .set('hour', 'h{1,2}')
  .set('minute', 'm{1,2}')
  .set('second', 's{1,2}')
  .set('millieSecond', 'S{1,3}')
/**
 * 如果没有格式化某项的话则设置为默认时间
 */
const defaultDateValues = new Map()
  .set('month', '01')
  .set('day', '01')
  .set('hour', '00')
  .set('minute', '00')
  .set('second', '00')
  .set('millieSecond', '000')

/**
 * 月份日期校验
 */
const monthDayValidate = {
  1: 31,
  3: 31,
  5: 31,
  7: 31,
  8: 31,
  10: 31,
  12: 31,
  4: 30,
  6: 30,
  9: 30,
  11: 30,
  2: 28,
}
/**
 * 解析字符串为 Date 对象
 * @param str 日期字符串
 * @param fmt 日期字符串的格式，目前仅支持使用 y(年),M(月),d(日),h(时),m(分),s(秒),S(毫秒)
 * @returns 解析得到的 Date 对象
 */
export function dateParse(str: string, fmt: string): Nullable<Date> {
  const now = new Date()
  defaultDateValues.set('year', now.getFullYear().toString())
  // 保存对传入的日期字符串进行格式化的全部信息数组列表
  const dateUnits: DateFormat[] = []
  for (const [fmtName, regex] of dateFormats) {
    const regExp = new RegExp(regex)
    if (regExp.test(fmt)) {
      const matchStr = regExp.exec(fmt)![0]
      const regexStr = '`'.repeat(matchStr.length)
      const index = fmt.indexOf(matchStr)
      fmt = fmt.replace(matchStr, regexStr)
      dateUnits.push(
        new DateFormat(fmtName, '\\d'.repeat(matchStr.length), null!, index),
      )
    } else {
      dateUnits.push(
        new DateFormat(fmtName, null!, defaultDateValues.get(fmtName), -1),
      )
    }
  }
  // 进行验证是否真的是符合传入格式的字符串
  fmt = fmt.replace(new RegExp('`', 'g'), '\\d')
  if (!new RegExp(`^${fmt}$`).test(str)) {
    return null
  }
  // 进行一次排序, 依次对字符串进行截取
  dateUnits
    // 过滤掉没有得到格式化的对象
    .filter(({ format }) => format)
    // 按照字符串中日期片段的索引进行排序
    .sort(function(a, b) {
      return a.index - b.index
    })
    // 获取到匹配的日期片段的值
    .map(format => {
      const matchDateUnit = new RegExp(format.format).exec(str)
      if (matchDateUnit !== null && matchDateUnit.length > 0) {
        str = str.replace(matchDateUnit[0], '')
        format.value = matchDateUnit[0]
      }
      return format
    })
    // 覆写到 dateStr 上面
    .forEach(({ format }, i) => {
      const matchDateUnit = new RegExp(format).exec(str)
      if (matchDateUnit !== null && matchDateUnit.length > 0) {
        str = str.replace(matchDateUnit[0], '')
        dateUnits[i].value = matchDateUnit[0]
      }
    })
  // 将截取完成的信息封装成对象并格式化标准的日期字符串
  const map = arrayToMap(dateUnits, item => item.name, item => item.value)
  if (map.get('year')!.length === 2) {
    map.set(
      'year',
      defaultDateValues
        .get('year')
        .substr(0, 2)
        .concat(map.get('year')),
    )
  }
  // 注意：此处使用的是本地时间而非 UTC 时间
  const get = (unit: string) => parseInt(map.get(unit)!)
  const year = get('year')
  const month = get('month')
  const day = get('day')
  const hour = get('hour')
  const minute = get('minute')
  const second = get('second')
  const millieSecond = get('millieSecond')
  if (!isRange(month, 1, 12 + 1)) {
    return null
  }
  if (
    !isRange(
      day,
      1,
      Reflect.get(monthDayValidate, month) +
        (month === 2 && year % 4 === 0 ? 1 : 0) +
        1,
    )
  ) {
    return null
  }
  if (
    !isRange(hour, 0, 24 + 1) ||
    !isRange(minute, 0, 60 + 1) ||
    !isRange(second, 0, 60 + 1) ||
    !isRange(millieSecond, 0, 999 + 1)
  ) {
    return null
  }
  return new Date(year, month - 1, day, hour, minute, second, millieSecond)
}