CDNManager/apps/web/src/components/country-flag.tsx

import { cn } from '@cdnmanager/ui/lib/utils'
import { COUNTRY_BY_NAME_RU } from '@cdnmanager/shared'
import { countryCodeFromName, getCountryFlagUrl } from '@/lib/country-labels'

interface CountryFlagProps {
  code?: string
  country?: string
  className?: string
}

export function CountryFlag({ code, country, className }: CountryFlagProps) {
  const resolvedCode =
    code ??
    (country
      ? COUNTRY_BY_NAME_RU[country.trim().toLowerCase()]?.code ??
        countryCodeFromName(country)
      : undefined)
  const url = getCountryFlagUrl(resolvedCode)
  if (!url) return null

  return (
    <img
      src={url}
      alt=""
      className={cn('size-4 shrink-0 rounded-full object-cover', className)}
    />
  )
}