import { Badge } from '@tidbcloud/uikit'

import { Nextgenv1beta2VpcPeeringState } from 'dbaas/services/api-clients/v2025/tidb-management/v1beta2/interface'

export const VpcPeeringStatusBadge = ({
  status = Nextgenv1beta2VpcPeeringState.PENDING
}: {
  status?: Nextgenv1beta2VpcPeeringState
}) => {
  const statusMap = {
    [Nextgenv1beta2VpcPeeringState.ACTIVE]: {
      color: 'green',
      label: 'Active'
    },
    [Nextgenv1beta2VpcPeeringState.FAILED]: {
      color: 'red',
      label: 'Failed'
    },
    [Nextgenv1beta2VpcPeeringState.PENDING]: {
      color: 'carbon.6',
      label: 'System Checking'
    }
  } as const

  const { color, label } = statusMap[status] ?? statusMap[Nextgenv1beta2VpcPeeringState.PENDING]

  return (
    <Badge color={color} variant="dot" size="lg" sx={{ padding: 0 }}>
      {label}
    </Badge>
  )
}