import { Project, SyntaxKind, JsxOpeningElement, JsxSelfClosingElement } from 'ts-morph';
import * as path from 'path';

// Mapeamento de regras de substituição do HTML Nativo -> Xertica UI
const RULES: Record<string, { replacement: string; reason: string; props?: string }> = {
  button: {
    replacement: 'Button',
    reason: 'Design System exige `<Button>`. Garante acessibilidade e variantes de tema.',
    props: "variant='default' | 'destructive' | 'outline' | 'secondary' | 'ghost' | 'link'",
  },
  input: {
    replacement: 'Input',
    reason:
      'O `<Input>` da Xertica abstrai focus rings, tamanhos (`size`) e cores de fundo do Tema.',
    props: "type='text' | 'email' | 'password', size='sm' | 'md' | 'lg'",
  },
  select: {
    replacement: 'Select (Cascata)',
    reason:
      'O select HTML nativo não aceita estilização profunda. Use a montagem `<Select> > <SelectTrigger> > <SelectContent>`.',
    props: 'onValueChange={...}',
  },
  table: {
    replacement: 'Table (Cascata)',
    reason:
      'Tag nativa quebra layout em resoluções menores. Use `<Table>` envolto nos componentes da Xertica.',
    props: "className='w-full'",
  },
};

function validateFile(filePath: string) {
  console.log(`\n🔍 Analisando: ${filePath}`);

  const project = new Project();
  project.addSourceFileAtPath(filePath);
  const sourceFile = project.getSourceFileOrThrow(filePath);

  let errors = 0;

  // Analisa Tags React: <tag />
  sourceFile.getDescendantsOfKind(SyntaxKind.JsxSelfClosingElement).forEach(element => {
    errors += checkElement(element, sourceFile.getFilePath());
  });

  // Analisa Tags React: <tag>...</tag>
  sourceFile.getDescendantsOfKind(SyntaxKind.JsxOpeningElement).forEach(element => {
    errors += checkElement(element, sourceFile.getFilePath());
  });

  if (errors === 0) {
    console.log('✅ Validação aprovada. Nenhum elemento HTML irregular encontrado.\n');
  } else {
    console.log(
      `\n❌ ${errors} infrações detectadas.\nPor favor, modifique seu código para utilizar Xertica UI.\n`
    );
    process.exit(1);
  }
}

function checkElement(
  element: JsxOpeningElement | JsxSelfClosingElement,
  filePath: string
): number {
  const tagName = element.getTagNameNode().getText();
  const rule = RULES[tagName.toLowerCase()];

  if (rule && tagName === tagName.toLowerCase()) {
    // É uma tag HTML bloqueada pela rule
    const startLine = element.getStartLineNumber();
    console.log(`\n[ERRO] Uso ilegal do elemento nativo: <${tagName}>`);
    console.log(`       📍 Arquivo: ${filePath}:${startLine}`);
    console.log(`       💡 Substituição sugerida: <${rule.replacement}>`);
    console.log(`       📎 Props sugeridas válidas: ${rule.props ?? 'N/A'}`);
    console.log(`       📝 Motivo: ${rule.reason}\n`);
    return 1;
  }
  return 0;
}

// Execução Simples: npx ts-node scripts/ai-validator.ts path/to/file.tsx
const args = process.argv.slice(2);
if (args.length === 0) {
  console.log('Uso: ts-node ai-validator.ts <caminho_do_arquivo>');
  process.exit(1);
}

// Wildcard loop se necessário, mas focado na chamada 1:1 por MCP.
args.forEach(file => {
  const absolutePath = path.resolve(process.cwd(), file);
  validateFile(absolutePath);
});