#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Timestamp: 2026-01-27
# File: src/scitex_writer/_mcp/content.py

"""Content compilation for LaTeX snippets and sections.

Provides quick compilation of raw LaTeX content with color mode support.
"""

from __future__ import annotations

import re
import subprocess
import tempfile
from pathlib import Path
from typing import Literal, Optional

from .utils import resolve_project_path


def compile_content(
    latex_content: str,
    project_dir: Optional[str] = None,
    color_mode: Literal["light", "dark", "sepia", "paper"] = "light",
    section_name: str = "content",
    timeout: int = 60,
    keep_aux: bool = False,
) -> dict:
    """Compile raw LaTeX content to PDF.

    Creates a standalone document from the provided LaTeX content and compiles
    it to PDF. Supports various color modes for comfortable viewing and can
    link to project bibliography for citation rendering.

    Args:
        latex_content: Raw LaTeX content to compile. Can be:
            - A complete document (with \\documentclass)
            - Document body only (will be wrapped automatically)
        project_dir: Optional path to scitex-writer project for bibliography.
        color_mode: Color theme for output:
            - 'light': Default white background
            - 'dark': Dark gray background with light text
            - 'sepia': Warm cream background for comfortable reading
            - 'paper': Pure white, optimized for printing
        section_name: Name for the output (used in filename).
        timeout: Compilation timeout in seconds.
        keep_aux: Keep auxiliary files (.aux, .log, etc.) after compilation.

    Returns:
        Dict with success status, output_pdf path, log, and any errors.
    """
    try:
        temp_dir = Path(tempfile.mkdtemp(prefix=f"scitex_content_{section_name}_"))
        tex_file = temp_dir / f"{section_name}.tex"
        pdf_file = temp_dir / f"{section_name}.pdf"

        is_complete_document = "\\documentclass" in latex_content

        if is_complete_document:
            final_content = _inject_color_mode(latex_content, color_mode)
        else:
            final_content = _create_content_document(
                latex_content, color_mode, project_dir
            )

        tex_file.write_text(final_content, encoding="utf-8")

        if project_dir:
            _setup_bibliography(temp_dir, project_dir)

        cmd = [
            "latexmk",
            "-pdf",
            "-interaction=nonstopmode",
            "-halt-on-error",
            f"-jobname={section_name}",
            str(tex_file),
        ]

        result = subprocess.run(
            cmd,
            cwd=str(temp_dir),
            capture_output=True,
            text=True,
            timeout=timeout,
        )

        log_content = _read_log_file(temp_dir, section_name)

        if not keep_aux:
            _cleanup_aux_files(temp_dir, section_name)

        if result.returncode == 0 and pdf_file.exists():
            return {
                "success": True,
                "output_pdf": str(pdf_file),
                "temp_dir": str(temp_dir),
                "color_mode": color_mode,
                "log": log_content,
                "message": f"Content compiled successfully: {section_name}",
            }
        else:
            return {
                "success": False,
                "output_pdf": None,
                "temp_dir": str(temp_dir),
                "color_mode": color_mode,
                "log": log_content,
                "stdout": result.stdout[-2000:]
                if len(result.stdout) > 2000
                else result.stdout,
                "stderr": result.stderr[-2000:]
                if len(result.stderr) > 2000
                else result.stderr,
                "error": f"Compilation failed with exit code {result.returncode}",
            }

    except subprocess.TimeoutExpired:
        return {
            "success": False,
            "error": f"Content compilation timed out after {timeout} seconds",
        }
    except Exception as e:
        return {
            "success": False,
            "error": str(e),
        }


def _setup_bibliography(temp_dir: Path, project_dir: str) -> None:
    """Set up bibliography files in temp directory."""
    project_path = resolve_project_path(project_dir)
    bib_dir = project_path / "00_shared" / "bib_files"
    if bib_dir.exists():
        for bib_file in bib_dir.glob("*.bib"):
            link_path = temp_dir / bib_file.name
            if not link_path.exists():
                link_path.symlink_to(bib_file)


def _read_log_file(temp_dir: Path, section_name: str) -> str:
    """Read and truncate log file content."""
    log_file = temp_dir / f"{section_name}.log"
    if log_file.exists():
        content = log_file.read_text(encoding="utf-8", errors="replace")
        return content[-5000:] if len(content) > 5000 else content
    return ""


def _cleanup_aux_files(temp_dir: Path, section_name: str) -> None:
    """Remove auxiliary files from compilation."""
    aux_extensions = [
        ".aux",
        ".log",
        ".fls",
        ".fdb_latexmk",
        ".synctex.gz",
        ".out",
        ".bbl",
        ".blg",
        ".toc",
        ".lof",
        ".lot",
    ]
    for ext in aux_extensions:
        aux_file = temp_dir / f"{section_name}{ext}"
        if aux_file.exists():
            aux_file.unlink()


def _inject_color_mode(latex_content: str, color_mode: str) -> str:
    """Inject color mode styling into an existing LaTeX document."""
    color_commands = _get_color_commands(color_mode)
    if not color_commands:
        return latex_content

    begin_doc_pattern = r"(\\begin\{document\})"
    match = re.search(begin_doc_pattern, latex_content)

    if match:
        insert_pos = match.end()
        return (
            latex_content[:insert_pos]
            + "\n"
            + color_commands
            + "\n"
            + latex_content[insert_pos:]
        )
    else:
        return color_commands + "\n" + latex_content


def _get_color_commands(color_mode: str) -> str:
    """Get LaTeX color commands for the specified color mode."""
    color_configs = {
        "light": "",
        "dark": """% Dark mode styling
\\pagecolor{black!95!white}
\\color{white}
\\makeatletter
\\@ifpackageloaded{hyperref}{%
  \\hypersetup{
    colorlinks=true,
    linkcolor=cyan!80!white,
    citecolor=green!70!white,
    urlcolor=blue!60!white,
  }%
}{}
\\makeatother""",
        "sepia": """% Sepia mode styling
\\pagecolor{Sepia!15!white}
\\color{black!85!Sepia}
\\makeatletter
\\@ifpackageloaded{hyperref}{%
  \\hypersetup{
    colorlinks=true,
    linkcolor=brown!70!black,
    citecolor=olive!70!black,
    urlcolor=teal!70!black,
  }%
}{}
\\makeatother""",
        "paper": "",
    }
    return color_configs.get(color_mode, "")


def _create_content_document(
    body_content: str, color_mode: str, project_dir: Optional[str]
) -> str:
    """Create a complete LaTeX document wrapping the body content."""
    color_commands = _get_color_commands(color_mode)
    bib_line = "\\bibliography{bibliography}" if project_dir else ""

    return f"""\\documentclass[11pt]{{article}}

% Essential packages
\\usepackage[english]{{babel}}
\\usepackage[T1]{{fontenc}}
\\usepackage[utf8]{{inputenc}}
\\usepackage[table,svgnames]{{xcolor}}
\\usepackage{{amsmath, amssymb}}
\\usepackage{{graphicx}}
\\usepackage{{booktabs}}
\\usepackage{{hyperref}}
\\usepackage[numbers]{{natbib}}
\\usepackage{{geometry}}
\\geometry{{margin=1in}}
\\usepackage{{pagecolor}}

\\begin{{document}}
{color_commands}
{body_content}

{bib_line}
\\end{{document}}
"""


__all__ = ["compile_content"]

# EOF