'use client';

import React from 'react';
import { Tag } from 'antd';

export function getVulnerabilidadeStatusTag(status: string) {
    const statusConfig: Record<string, { color: string; label: string }> = {
        aberta: { color: 'red', label: 'Aberta' },
        em_analise: { color: 'orange', label: 'Em Análise' },
        corrigida: { color: 'green', label: 'Corrigida' },
        falso_positivo: { color: 'default', label: 'Falso Positivo' },
        aceita: { color: 'purple', label: 'Aceita' },
    };
    const config = statusConfig[status] || { color: 'default', label: status };
    return <Tag color={config.color}>{config.label}</Tag>;
}

export function getVulnerabilidadeSeveridadeTag(severidade: string) {
    const severidadeConfig: Record<string, { color: string; label: string }> = {
        baixa: { color: 'default', label: 'Baixa' },
        media: { color: 'blue', label: 'Média' },
        alta: { color: 'orange', label: 'Alta' },
        critica: { color: 'red', label: 'Crítica' },
    };
    const config = severidadeConfig[severidade] || { color: 'default', label: severidade };
    return <Tag color={config.color}>{config.label}</Tag>;
}

export function getVulnerabilidadeTipoLabel(tipo: string): string {
    const tipos: Record<string, string> = {
        sql_injection: 'SQL Injection',
        xss: 'XSS',
        csrf: 'CSRF',
        autenticacao: 'Autenticação',
        autorizacao: 'Autorização',
        configuracao: 'Configuração',
        outro: 'Outro',
    };
    return tipos[tipo] || tipo;
}