yueche_app/uni_modules/wot-design-uni/components/wd-datetime-picker-view/util.ts

import { type DateTimeType } from './types'

/**
 * @description 根据传入的值和类型，获取当前的选项数组，便于传入 pickerView
 * @param value
 * @param type picker类型
 * @param useSecond 是否使用秒，仅在 time 和 datetime 类型下生效
 */
export function getPickerValue(value: string | number, type: DateTimeType, useSecond: boolean = false) {
  const values: number[] = []
  const date = new Date(value)
  if (type === 'time') {
    const pair = String(value).split(':')
    values.push(parseInt(pair[0]), parseInt(pair[1]))
    if (useSecond && pair[2]) {
      values.push(parseInt(pair[2]))
    }
  } else {
    values.push(date.getFullYear(), date.getMonth() + 1)
    if (type === 'date') {
      values.push(date.getDate())
    } else if (type === 'datetime') {
      values.push(date.getDate(), date.getHours(), date.getMinutes())
      if (useSecond) {
        values.push(date.getSeconds())
      }
    }
  }
  return values
}