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

'use client'

import { useAtomValue } from 'jotai'
import { useTranslation } from 'react-i18next'
import { TerminalSquare } from '@/app/components/base/icons/src/vender/solid/development'
import { Beaker02 } from '@/app/components/base/icons/src/vender/solid/education'
import { langGeniusVersionInfoAtom } from '@/context/version-state'

const headerEnvClassName: { [k: string]: string } = {
  DEVELOPMENT: 'bg-[#FEC84B] border-[#FDB022] text-[#93370D]',
  TESTING: 'bg-[#A5F0FC] border-[#67E3F9] text-[#164C63]',
}

const EnvNav = () => {
  const { t } = useTranslation()
  const langGeniusVersionInfo = useAtomValue(langGeniusVersionInfoAtom)
  const showEnvTag =
    langGeniusVersionInfo.current_env === 'TESTING' ||
    langGeniusVersionInfo.current_env === 'DEVELOPMENT'

  if (!showEnvTag) return null

  return (
    <div
      className={`mr-1 flex h-[22px] items-center rounded-md border px-2 text-xs font-medium ${headerEnvClassName[langGeniusVersionInfo.current_env]} `}
    >
      {langGeniusVersionInfo.current_env === 'TESTING' && (
        <>
          <Beaker02 className="size-3" />
          <div className="ml-1 max-[1280px]:hidden">
            {t(($) => $['environment.testing'], { ns: 'common' })}
          </div>
        </>
      )}
      {langGeniusVersionInfo.current_env === 'DEVELOPMENT' && (
        <>
          <TerminalSquare className="size-3" />
          <div className="ml-1 max-[1280px]:hidden">
            {t(($) => $['environment.development'], { ns: 'common' })}
          </div>
        </>
      )}
    </div>
  )
}

export default EnvNav