dify/web/app/components/header/license-env/index.tsx

'use client'

import { RiHourglass2Fill } from '@remixicon/react'
import { useSuspenseQuery } from '@tanstack/react-query'
import dayjs from 'dayjs'
import { useTranslation } from 'react-i18next'
import { systemFeaturesQueryOptions } from '@/features/system-features/client'
import { LicenseStatus } from '@/features/system-features/constants'
import PremiumBadge from '../../base/premium-badge'

const LicenseNav = () => {
  const { t } = useTranslation()
  const { data: systemFeatures } = useSuspenseQuery(systemFeaturesQueryOptions())

  if (systemFeatures.license?.status === LicenseStatus.EXPIRING) {
    const expiredAt = systemFeatures.license?.expired_at
    const count = dayjs(expiredAt).diff(dayjs(), 'days')
    return (
      <PremiumBadge color="orange" className="select-none">
        <RiHourglass2Fill
          aria-hidden="true"
          className="flex size-3 items-center pl-0.5 text-components-premium-badge-indigo-text-stop-0"
        />
        {count <= 1 && (
          <span className="px-0.5 system-xs-medium">
            {t(($) => $['license.expiring'], { ns: 'common', count })}
          </span>
        )}
        {count > 1 && (
          <span className="px-0.5 system-xs-medium">
            {t(($) => $['license.expiring_plural'], { ns: 'common', count })}
          </span>
        )}
      </PremiumBadge>
    )
  }
  if (systemFeatures.license.status === LicenseStatus.ACTIVE) {
    return (
      <PremiumBadge color="indigo" className="select-none">
        <span className="px-1 system-xs-medium">Enterprise</span>
      </PremiumBadge>
    )
  }
  return null
}

export default LicenseNav