rsuite/esm/internals/utils/date/useDateMode.js

A suite of react components

'use client';
import { useMemo, useCallback } from 'react';
import { shouldRenderDate, shouldRenderTime, shouldOnlyRenderMonth, shouldOnlyRenderTime } from "./formatCheck.js";
export var DateMode = /*#__PURE__*/function (DateMode) {
  DateMode["Date"] = "date";
  DateMode["Month"] = "month";
  DateMode["Time"] = "time";
  DateMode["DateTime"] = "datetime";
  return DateMode;
}({});

/**
 * Custom hook to determine the date mode and check format parts.
 *
 * @param format - The format string.
 * @returns An object containing the resolved DateMode and a `has` method to check format parts.
 */
export var useDateMode = function useDateMode(format) {
  var mode = useMemo(function () {
    if (shouldRenderDate(format) && shouldRenderTime(format)) {
      return DateMode.DateTime;
    }
    if (shouldOnlyRenderMonth(format)) {
      return DateMode.Month;
    }
    if (shouldOnlyRenderTime(format)) {
      return DateMode.Time;
    }
    if (shouldRenderDate(format)) {
      return DateMode.Date;
    }
    return DateMode.Date; // Default fallback
  }, [format]);

  // Use useCallback to memoize the has method
  var has = useCallback(function (part) {
    switch (part) {
      case 'year':
        return /[Yy]/.test(format);
      case 'month':
        return /[ML]/.test(format);
      case 'day':
        return /[Dd]/.test(format);
      case 'time':
        return /([Hhms])/.test(format);
      default:
        return false;
    }
  }, [format]);
  return {
    mode: mode,
    has: has
  };
};