dify/web/app/components/header/maintenance-notice.tsx

'use client'

import { useState } from 'react'
import { useTranslation } from 'react-i18next'
import { X } from '@/app/components/base/icons/src/vender/line/general'
import { useLanguage } from '@/app/components/header/account-setting/model-provider-page/hooks'
import { env } from '@/env'
import { NOTICE_I18N } from '@/i18n-config/language'
import { useHideMaintenanceNotice } from './storage'

function MaintenanceNotice() {
  if (!env.NEXT_PUBLIC_MAINTENANCE_NOTICE) return null

  return <MaintenanceNoticeContent />
}

function MaintenanceNoticeContent() {
  const { t } = useTranslation()
  const locale = useLanguage()

  const [hiddenNoticeValue, setHiddenNoticeValue] = useHideMaintenanceNotice()
  const hiddenNotice = hiddenNoticeValue === '1'
  const [closedInSession, setClosedInSession] = useState(false)
  const showNotice = !hiddenNotice && !closedInSession
  const handleJumpNotice = () => {
    window.open(NOTICE_I18N.href, '_blank')
  }

  const handleCloseNotice = () => {
    setHiddenNoticeValue('1')
    setClosedInSession(true)
  }

  const titleByLocale: { [key: string]: string } = NOTICE_I18N.title
  const descByLocale: { [key: string]: string } = NOTICE_I18N.desc

  if (!showNotice) return null

  return (
    <div className="z-20 flex h-[38px] shrink-0 items-center border-[0.5px] border-b border-b-[#FEF0C7] bg-[#FFFAEB] px-4">
      <div className="mr-2 flex h-[22px] shrink-0 items-center rounded-xl bg-[#F79009] px-2 text-[11px] font-medium text-white">
        {titleByLocale[locale]}
      </div>
      {NOTICE_I18N.href && NOTICE_I18N.href !== '#' ? (
        <button
          type="button"
          className="grow cursor-pointer border-none bg-transparent p-0 text-left text-xs font-medium text-gray-700"
          onClick={handleJumpNotice}
        >
          {descByLocale[locale]}
        </button>
      ) : (
        <div className="grow text-xs font-medium text-gray-700">{descByLocale[locale]}</div>
      )}
      <button
        type="button"
        aria-label={t(($) => $['operation.close'], { ns: 'common' })}
        className="size-4 shrink-0 cursor-pointer border-none bg-transparent p-0 text-gray-500"
        onClick={handleCloseNotice}
      >
        <X className="size-4" aria-hidden="true" />
      </button>
    </div>
  )
}

export default MaintenanceNotice