import { Badge } from '@/components/ui/badge'
import type { McpTool } from '@/features/prototype/types'

const riskConfig = {
  low: { label: '低风险', variant: 'success' },
  medium: { label: '中风险', variant: 'warning' },
  high: { label: '高风险', variant: 'danger' }
} as const

export function AppRiskBadge({ risk }: { risk: McpTool['risk_level'] }) {
  const config = riskConfig[risk]

  return <Badge variant={config.variant}>{config.label}</Badge>
}