import { Severity, type Finding, type FindingsReport } from "../findings";
import { calculateScore, type ScoreBreakdown } from "./score";

export interface ReportMetadata {
  projectName: string;
  projectPath: string;
  language: string;
  framework?: string;
  scanType: string;
  timestamp: string;
}

interface MarkdownReportOptions {
  metadata: ReportMetadata;
  findings: Finding[];
  includeRecommendations?: boolean;
  includeCodeSnippets?: boolean;
  locale?: string;
}

const TRANSLATIONS: Record<string, Record<string, string>> = {
  id: {
    "Security Audit Report": "Laporan Audit Keamanan",
    "Bug Investigation Report": "Laporan Investigasi Bug",
    "Performance Audit Report": "Laporan Audit Performa",
    "Architecture Review Report": "Laporan Review Arsitektur",
    "Dependency Review Report": "Laporan Review Dependensi",
    "Database Review Report": "Laporan Review Database",
    "Release Check Report": "Laporan Pemeriksaan Rilis",
    "Score": "Skor",
    "Executive Summary": "Ringkasan Eksekutif",
    "Detailed Findings": "Temuan Detail",
    "Critical Findings": "Temuan Kritis",
    "High Findings": "Temuan Tinggi",
    "Medium Findings": "Temuan Sedang",
    "Low Findings": "Temuan Rendah",
    "Info Findings": "Temuan Info",
  }
};

export function translate(text: string, locale?: string): string {
  if (!locale) return text;
  return TRANSLATIONS[locale]?.[text] || text;
}

const SEVERITY_EMOJI: Record<Severity, string> = {
  [Severity.Critical]: "\u{1F534}",
  [Severity.High]: "\u{1F7E0}",
  [Severity.Medium]: "\u{1F7E1}",
  [Severity.Low]: "\u{1F535}",
  [Severity.Info]: "\u26AA",
};

export function generateMarkdownReport(options: MarkdownReportOptions): string {
  const { metadata, findings, includeRecommendations = true, includeCodeSnippets = true, locale } = options;
  const score = calculateScore(findings);
  const sections: string[] = [];

  // Header
  sections.push(`# ${translate(metadata.scanType + " Report", locale)}\n`);
  sections.push(`**Project:** ${metadata.projectName}`);
  sections.push(`**Path:** ${metadata.projectPath}`);
  sections.push(`**Language:** ${metadata.language}`);
  if (metadata.framework) sections.push(`**Framework:** ${metadata.framework}`);
  sections.push(`**Date:** ${metadata.timestamp}\n`);

  // Score
  sections.push(`## ${translate("Score", locale)}\n`);
  sections.push(`**${score.score}/100** (Grade: ${score.grade})\n`);
  sections.push(`| Severity | Count |`);
  sections.push(`|----------|-------|`);
  for (const [severity, count] of Object.entries(score.breakdown)) {
    if (count > 0) {
      sections.push(`| ${SEVERITY_EMOJI[severity as Severity]} ${severity} | ${count} |`);
    }
  }
  sections.push("");

  // Executive Summary
  sections.push(`## ${translate("Executive Summary", locale)}\n`);
  sections.push(
    `Found **${findings.length}** issues: **${score.breakdown[Severity.Critical]}** critical, **${score.breakdown[Severity.High]}** high, **${score.breakdown[Severity.Medium]}** medium, **${score.breakdown[Severity.Low]}** low, **${score.breakdown[Severity.Info]}** info.\n`
  );

  // Findings by severity
  const grouped = new Map<Severity, Finding[]>();
  for (const finding of findings) {
    const group = grouped.get(finding.severity) ?? [];
    group.push(finding);
    grouped.set(finding.severity, group);
  }

  for (const severity of [Severity.Critical, Severity.High, Severity.Medium, Severity.Low, Severity.Info]) {
    const group = grouped.get(severity);
    if (!group || group.length === 0) continue;

    sections.push(`## ${SEVERITY_EMOJI[severity]} ${translate(severity + " Findings", locale)}\n`);

    // Table Header
    sections.push(`| Title | Location | Description | Impact |`);
    sections.push(`| :--- | :--- | :--- | :--- |`);

    for (const finding of group) {
      const location = finding.evidence.length > 0
        ? finding.evidence.map(e => `\`${e.file}${e.line ? `:${e.line}` : ""}\``).join("<br>")
        : "-";
      
      const cleanDescription = finding.description.replace(/\r?\n/g, " ").trim();
      const cleanImpact = (finding.impact || finding.recommendation || "").replace(/\r?\n/g, " ").trim();

      sections.push(`| **${finding.title}** | ${location} | ${cleanDescription} | ${cleanImpact} |`);
    }
    sections.push("");

    // Code Snippets (Evidence)
    if (includeCodeSnippets) {
      const findingsWithSnippets = group.filter(f => f.evidence.some(e => e.snippet));
      if (findingsWithSnippets.length > 0) {
        sections.push(`### Code Evidence\n`);
        for (const finding of findingsWithSnippets) {
          for (const ev of finding.evidence) {
            if (ev.snippet) {
              sections.push(`#### ${finding.title} (\`${ev.file}${ev.line ? `:${ev.line}` : ""}\`)\n`);
              sections.push("```");
              sections.push(ev.snippet);
              sections.push("```\n");
            }
          }
        }
      }
    }
  }

  return sections.join("\n");
}