dify/web/hooks/use-timestamp.ts

'use client'
import { useQuery } from '@tanstack/react-query'
import dayjs from 'dayjs'
import timezone from 'dayjs/plugin/timezone'
import utc from 'dayjs/plugin/utc'
import { useCallback } from 'react'
import { userProfileQueryOptions } from '@/features/account-profile/client'

dayjs.extend(utc)
dayjs.extend(timezone)

type UseTimestampOptions = {
  timezone?: string
}

const getBrowserTimezone = () => {
  return Intl.DateTimeFormat().resolvedOptions().timeZone
}

const useTimestamp = ({ timezone: timezoneOverride }: UseTimestampOptions = {}) => {
  const { data: accountTimezone } = useQuery({
    ...userProfileQueryOptions(),
    select: data => data.profile.timezone ?? undefined,
    enabled: timezoneOverride === undefined,
  })
  const resolvedTimezone = timezoneOverride ?? accountTimezone ?? getBrowserTimezone()

  const formatTime = useCallback((value: number, format: string) => {
    return dayjs.unix(value).tz(resolvedTimezone).format(format)
  }, [resolvedTimezone])

  const formatDate = useCallback((value: string, format: string) => {
    return dayjs(value).tz(resolvedTimezone).format(format)
  }, [resolvedTimezone])

  return { formatTime, formatDate }
}

export default useTimestamp