zent/src/date-picker/hooks/useShowTimeOption.ts

一套前端设计语言和基于React的实现

import { useRef, useMemo } from 'react';
import { startOfToday, endOfToday } from 'date-fns';
import { formatBase } from '../utils/index';
import {
  IShowTime,
  IShowTimeRange,
  IShowTimeOptionWithDefault,
} from '../types';
import { TIME_FORMAT } from '../constants';

const formatStartDate = (format: string) => formatBase(startOfToday(), format);
const formatEndDate = (format: string) => formatBase(endOfToday(), format);
const DefaultStartTime = formatStartDate(TIME_FORMAT);
const DefaultEndTime = formatEndDate(TIME_FORMAT);
/**
 * 获取showTime范围
 * @param showTime
 */
export function useShowTimeRangeOption(
  showTime?: IShowTimeRange<string>
): Array<IShowTimeOptionWithDefault | undefined> {
  const showTimeRef = useRef(showTime);
  showTimeRef.current = showTime;

  const showTimeOption = useMemo(() => {
    if (!showTimeRef.current) {
      return [undefined, undefined];
    }
    if (typeof showTimeRef.current === 'object') {
      const {
        format = TIME_FORMAT,
        defaultTime,
        ...restOption
      } = showTimeRef.current;
      const defaultTimeStart = defaultTime?.[0] || formatStartDate(format);
      const defaultTimeEnd = defaultTime?.[1] || formatEndDate(format);

      return [
        { format, defaultTime: defaultTimeStart, ...restOption },
        { format, defaultTime: defaultTimeEnd, ...restOption },
      ];
    }
    // default
    return [
      {
        format: TIME_FORMAT,
        defaultTime: DefaultStartTime,
      },
      {
        format: TIME_FORMAT,
        defaultTime: DefaultEndTime,
      },
    ];
  }, [showTimeRef]);

  return showTimeOption;
}

/**
 * 将showTime对象转化
 * @param showTime
 */
export function useShowTimeOption(
  showTime?: IShowTime<string>
): IShowTimeOptionWithDefault | undefined {
  const showTimeRef = useRef(showTime);
  showTimeRef.current = showTime;

  const showTimeOption = useMemo(() => {
    if (!showTimeRef.current) {
      return undefined;
    }
    if (typeof showTimeRef.current === 'object') {
      const {
        format = TIME_FORMAT,
        defaultTime,
        ...restOption
      } = showTimeRef.current;
      const defaultTimeTemp = defaultTime || formatStartDate(format);
      return { format, defaultTime: defaultTimeTemp, ...restOption };
    }
    // default
    return {
      format: TIME_FORMAT,
      defaultTime: DefaultStartTime,
    };
  }, [showTimeRef]);

  return showTimeOption;
}