"""
harness.py — porte de qualité déterministe du pipeline.

Miroir importable du harnais de référence
(.claude/skills/validation-harness/harness.py) : mêmes contrôles, même
sévérité, mais API Python (validate_text) au lieu d'un CLI. Toute sortie du
pipeline DOIT passer ici avant d'être rendue ; un verdict rouge déclenche
une (1) boucle de réparation LLM puis, à défaut, un warning fort.

Contrôles :
  1. STATIQUE   — {{ }} = identifiant nu camelCase sans _ ; questionId/Index
                  uniques et contigus dès 0 ; bloc {python} présent.
  2. EXEC       — tous les blocs {python} (concaténés) s'exécutent sur N
                  graines, 0 exception. Stubs pyxiscience légers.
  3. INJECTION  — chaque {{ var }} se résout dans le namespace, 0 inconnue.
  4. INTERDITS  — motifs interdits sur le corps rendu (skill §6).
"""

from __future__ import annotations

import io
import logging
import random
import re
import sys
import time as _time
import traceback
import types
import warnings as _warnings
from contextlib import redirect_stdout

logger = logging.getLogger(__name__)

# Budget par graine (exec + rendu + scans). Un bloc légitime tient en < 1 s ;
# au-delà c'est un code généré pathologique → seed en échec, pas de blocage.
SEED_TIMEOUT_S = 10.0

try:
    import matplotlib
    matplotlib.use("Agg")
    import matplotlib.pyplot as _plt
    _HAS_MPL = True
except Exception:
    _plt = None
    _HAS_MPL = False


# ── Stubs pyxiscience (identiques au harnais de référence) ───────────────────

def install_pyxiscience_stubs() -> None:
    if "pyxiscience" in sys.modules:
        return

    class _Universal:
        def __init__(self, *a, **k): pass
        def __call__(self, *a, **k): return _Universal()
        def __getattr__(self, _): return _Universal()
        def __getitem__(self, _): return _Universal()
        def __iter__(self): return iter(())
        def __str__(self): return ""
        def __repr__(self): return ""
        def __mul__(self, o): return self
        __rmul__ = __add__ = __radd__ = __sub__ = __rsub__ = __mul__

    def _passthrough(*a, **k):
        return str(a[0]) if a else ""

    def _pxs_config(*a, **k):
        # dict VIDE : utilisable comme **config_standard sans faire planter
        # sympy.latex (qui rejette toute clé inconnue).
        return {}

    # Liste canonique partagée avec la sandbox (les deux systèmes de stubs
    # partagent sys.modules — le premier installé sert aux deux).
    from app.validation.sandbox import KNOWN_PXS_HELPERS

    def _make_module(name: str) -> types.ModuleType:
        mod = types.ModuleType(name)

        def __getattr__(attr):  # PEP 562
            if attr == "pxs_config":
                return _pxs_config
            if attr == "pxs_variation_number":
                return 1
            # Classe si CamelCase OU motif pxs_Xxx (pxs_Interval, pxs_Plotable…)
            if attr and (attr[0].isupper() or re.match(r"pxs_[A-Z]", attr)):
                return _Universal
            return _passthrough

        mod.__getattr__ = __getattr__
        mod.__all__ = list(KNOWN_PXS_HELPERS)
        return mod

    root = _make_module("pyxiscience")
    sys.modules["pyxiscience"] = root
    for sub in (
        "Mes_fctions_generalistes_bis",
        "Mes_fctions_d_analyse_bis",
        "Mes_fctions_d_alg_lineaire_bis",
        "Mes_fctions_probabilistes_bis",
        "Classes_Extensions",
    ):
        full = f"pyxiscience.{sub}"
        sys.modules[full] = _make_module(full)
        setattr(root, sub, sys.modules[full])


# ── Extraction (fences 3 ou 4 backticks, lecture tolérante) ──────────────────

PY_FENCE_ANY_RE = re.compile(r"(?ms)^(`{3,4})\{python\}[ \t]*\n(.*?)\n\1[ \t]*$")
INJECTION_RE = re.compile(r"\{\{(.*?)\}\}", re.DOTALL)
VALID_TOKEN_RE = re.compile(r"^[A-Za-z][A-Za-z0-9]*$")


def extract_python_code(text: str) -> str | None:
    """Concatène TOUS les blocs {python} (namespace partagé, ordre document)."""
    blocks = [m.group(2) for m in PY_FENCE_ANY_RE.finditer(text)]
    return "\n\n".join(blocks) if blocks else None


def strip_python_blocks(text: str) -> str:
    return PY_FENCE_ANY_RE.sub("", text)


# ── Contrôles statiques ──────────────────────────────────────────────────────

def check_injection_tokens(text: str) -> list[str]:
    errs: list[str] = []
    if "{{{" in text:
        errs.append("triple-accolade `{{{` détectée (casse la substitution)")
    for raw in INJECTION_RE.findall(strip_python_blocks(text)):
        tok = raw.strip()
        if not tok:
            errs.append("injection vide `{{ }}`")
            continue
        if VALID_TOKEN_RE.match(tok):
            continue
        why = []
        if "_" in tok:
            why.append("underscore")
        if "(" in tok or ")" in tok:
            why.append("appel de fonction")
        if any(c in tok for c in "+*/\\,"):
            why.append("calcul/opérateur")
        if "." in tok:
            why.append("attribut")
        if " " in tok:
            why.append("espace / plusieurs tokens")
        if not why:
            why.append("pas un identifiant nu")
        errs.append(f"injection invalide `{{{{ {tok} }}}}` : {', '.join(why)}")
    return errs


def check_question_ids(text: str) -> list[str]:
    errs: list[str] = []
    for field in ("questionId", "questionIndex"):
        ids = [int(x) for x in re.findall(rf":{field}:\s*(\d+)", text)]
        if not ids:
            continue
        if len(ids) != len(set(ids)):
            errs.append(f":{field}: contient des doublons : {ids}")
        if sorted(ids) != list(range(len(ids))):
            errs.append(f":{field}: non contigu à partir de 0 : {ids}")
    return errs


# ── Motifs interdits (corps rendu — skill §6) ────────────────────────────────

FORBIDDEN = [
    (r"(?<!\\)\$[0-9]",              r"`$` (non échappé) collé à un chiffre — préfixer par ${} ou passer en display"),
    (r"\{\{\{",                      r"triple-accolade `{{{`"),
    (r"\^\{\s*0\s*\}",               r"`^{0}`"),
    (r"\^\{\s*1\s*\}",               r"`^{1}`"),
    (r"\\sqrt\[1\]",                 r"`\sqrt[1]`"),
    (r"\\sqrt\[2\]",                 r"`\sqrt[2]` (utiliser `\sqrt`)"),
    (r"\\frac\{\s*\}\{",             r"`\frac{}{…}` (numérateur vide)"),
    (r"\\frac\{[^{}]*\}\{\s*1\s*\}", r"`\frac{…}{1}`"),
    (r"\blatex\(",                   r"`latex(` ayant fui dans le rendu"),
    (r"\bround\(",                   r"`round(` dans le rendu"),
    (r"\bpxsl_matrix\b",             r"`pxsl_matrix` dans le rendu"),
    (r"-\s*\+",                      r"double signe `- +`"),
    (r"\+\s*-",                      r"double signe `+ -`"),
]
FORBIDDEN = [(re.compile(p), msg) for p, msg in FORBIDDEN]


def scan_forbidden(body: str) -> list[str]:
    errs = []
    for rx, msg in FORBIDDEN:
        m = rx.search(body)
        if m:
            start = max(0, m.start() - 30)
            ctx = body[start:m.end() + 30].replace("\n", " ")
            errs.append(f"{msg}  …{ctx}…")
    return errs


# ── Lint de RENDU (cause racine des faux VERT — lot 12) ──────────────────────
# Le harnais n'exécutait que le Python : des bugs de RENDU passaient VERT.
# Ces contrôles sont STATIQUES (propriétés du gabarit, pas des tirages) et
# s'appliquent à TOUS les modes. Fondés sur le skill pyxiscience-pythonisation
# §1 et §4/§11.

_ROLE_SPAN_RE = re.compile(r"\{(?:fr|en)\}`([^`]*)`")


def check_render_static(text: str) -> list[str]:
    """Lint de rendu (tous modes). Exclut le bloc Python et les lignes
    `:solution:` (motif de réponse FGQ, pas du texte affiché).

    NB : le contrôle « `{{ }}` dans un rôle bilingue » est volontairement
    SÉPARÉ (check_injection_in_roles) et NON activé par défaut — 3/33 exemples
    validés le violent (décision en attente)."""
    body = strip_python_blocks(text)
    # Retire les lignes de directive/métadonnée (`:title:`, `:originalSource:`,
    # `::::{…}`, `:solution:`, `:isRightAnswer:`…) : ce n'est PAS du LaTeX
    # affiché, un `%`/`$` y est légitime (citation source, id…).
    body = re.sub(r"(?m)^\s*:.*$", "", body)
    errs: list[str] = []

    # `$` inline déséquilibré (skill §1) : ni `{{ }}` ni `${}` ne contiennent de
    # `$`, donc un nombre IMPAIR de `$` non échappés = span math ouvert non
    # refermé (bug `${{ x }},` du lot 12).
    dollars = len(re.findall(r"(?<!\\)\$", body))
    if dollars % 2 == 1:
        errs.append(f"`$` inline déséquilibré ({dollars} `$` non échappés — "
                    "impair) : un span math est ouvert et jamais refermé (skill §1)")

    # `%` nu (non `\\%`) dans le texte affiché → commentaire LaTeX (skill §1).
    mp = re.search(r"(?<!\\)%", body)
    if mp:
        ctx = body[max(0, mp.start() - 25):mp.start() + 6].replace("\n", " ")
        errs.append(f"`%` nu non échappé (utiliser `\\%` — jamais dans le "
                    f"Python) : …{ctx}…")
    return errs


def check_injection_in_roles(text: str) -> list[str]:
    """`{{ }}` DANS un rôle {fr}`…`/{en}`…` → ne s'évalue pas (skill §4/§11 +
    bug n°1 du lot 12). SÉPARÉ car 3/33 exemples validés le violent : à activer
    dans validate_text une fois la politique confirmée + le filet d'extraction
    en place."""
    body = re.sub(r"(?m)^:solution:.*$", "", strip_python_blocks(text))
    errs: list[str] = []
    for m in _ROLE_SPAN_RE.finditer(body):
        if "{{" in m.group(1):
            errs.append("injection `{{…}}` dans un rôle bilingue (ne s'évalue "
                        f"pas — sortir l'injection du rôle) : …{m.group(0)[:70]}…")
    return errs


# ── Contrôles déclinaisons QCM/QAT (mode `declinaisons`) ─────────────────────

_QUESTION_RE = re.compile(r":::::\{question\}.*?:::::", re.DOTALL)
_MCQ_ANSWER_RE = re.compile(
    r"::::\{mcqAnswer\}\s*\n:isRightAnswer:\s*(true|false)\s*\n(.*?)\n::::",
    re.DOTALL,
)
_NONE_OPTION_RE = re.compile(r"Aucune de ces réponses|None of these answers", re.IGNORECASE)
_SOLUTION_FIELD_RE = re.compile(r"(?m)^:solution:[ \t]*(.+)$")
_INPUT_RE = re.compile(r"\{input\}`")
_QTYPE_RE = re.compile(r":questionType:\s*(\w+)")


def has_declinaison_questions(text: str) -> bool:
    return bool(re.search(r":questionType:\s*(MCQ|FGQ)\b", text))


_LEGACY_PATTERNS = [
    (re.compile(r"\\py\{"), r"résidu légacy `\py{`"),
    (re.compile(r"\\qcm\b|\\qat\b|\\qcl\b"), r"résidu légacy `\qcm`/`\qat`/`\qcl`"),
    (re.compile(r"\\input\{null\}"), r"résidu légacy `\input{null}`"),
    (re.compile(r"\\begin\{align\*?\}"), r"`\begin{align*}` interdit (utiliser equation*)"),
    (re.compile(r"(?<!\\)\$\$"), r"`$$` interdit (utiliser equation*)"),
]


def check_declinaison_static(text: str) -> list[str]:
    """Contrôles statiques MCQ/FGQ (indépendants des graines)."""
    import json as _json
    errs: list[str] = []
    body = strip_python_blocks(text)

    # (`{{ }}` dans un rôle bilingue est désormais contrôlé par
    # check_render_static, pour TOUS les modes — pas seulement les déclinaisons.)

    # Résidus de l'ancienne syntaxe plateforme.
    errs += [msg for rx, msg in _LEGACY_PATTERNS if rx.search(body)]

    # UN SEUL bloc {python} en déclinaison (un bloc additionnel qui RE-TIRE de
    # l'aléatoire rendrait les valeurs incohérentes entre questions ; les blocs
    # sans re-tirage sont fusionnés en amont par merge_decl_python_blocks).
    n_blocks = len(PY_FENCE_ANY_RE.findall(text))
    if n_blocks > 1:
        errs.append(f"{n_blocks} blocs {{python}} (un seul attendu en déclinaison — "
                    "bloc additionnel avec re-tirage aléatoire probable)")

    # Approximations dans :solution: (valeurs exactes uniquement).
    for m in _SOLUTION_FIELD_RE.finditer(text):
        if "\\approx" in m.group(1) or "≈" in m.group(1):
            errs.append("approximation (≈) dans un champ `:solution:` — valeurs exactes uniquement")
            break

    for qi, qm in enumerate(_QUESTION_RE.finditer(text)):
        q = qm.group(0)
        tm = _QTYPE_RE.search(q)
        qtype = tm.group(1) if tm else "?"

        if qtype == "MCQ":
            answers = list(_MCQ_ANSWER_RE.finditer(q))
            n_true = sum(1 for a in answers if a.group(1) == "true")
            if n_true != 1:
                errs.append(f"question {qi} (MCQ) : {n_true} option(s) `:isRightAnswer: true` (attendu : exactement 1)")
            if not (3 <= len(answers) <= 6):
                errs.append(f"question {qi} (MCQ) : {len(answers)} options (attendu : 3 à 5)")
            none_idx = [i for i, a in enumerate(answers) if _NONE_OPTION_RE.search(a.group(2))]
            if none_idx and none_idx[-1] != len(answers) - 1:
                errs.append(f"question {qi} (MCQ) : option « None/Aucune » pas en dernière position")
            if _SOLUTION_FIELD_RE.search(q) or _INPUT_RE.search(q):
                errs.append(f"question {qi} (MCQ) : `:solution:`/`{{input}}` interdits en MCQ")

        elif qtype == "FGQ":
            sol = _SOLUTION_FIELD_RE.search(q)
            n_inputs = len(_INPUT_RE.findall(q))
            if not sol:
                errs.append(f"question {qi} (FGQ) : champ `:solution:` manquant")
                continue
            if n_inputs == 0:
                errs.append(f"question {qi} (FGQ) : aucun `{{input}}` dans l'énoncé")
            raw = sol.group(1).strip()
            if "pxsl_matrix" in raw:
                errs.append(f"question {qi} (FGQ) : `pxsl_matrix` dans `:solution:` (interdit)")
            # Parse JSON structurel : on remplace les {{var}} par un littéral neutre.
            probe = INJECTION_RE.sub("V", raw)
            try:
                data = _json.loads(probe)
                assert (isinstance(data, list) and len(data) == 2
                        and isinstance(data[0], list) and isinstance(data[1], list)
                        and data[0] and data[0][0] in ("ord", "CL"))
                n_vals = len(data[0]) - 1
                n_tols = len(data[1])
                if not (n_vals == n_tols == n_inputs):
                    errs.append(f"question {qi} (FGQ) : arité incohérente — "
                                f"{n_inputs} input(s), {n_vals} valeur(s), {n_tols} tolérance(s)")
                if any(str(t).strip() != "0" for t in data[1]):
                    errs.append(f"question {qi} (FGQ) : tolérance ≠ \"0\"")
            except Exception:
                errs.append(f"question {qi} (FGQ) : `:solution:` non parseable en JSON : {raw[:80]}")

    return errs


def check_mcq_collisions(text: str, env: dict) -> list[str]:
    """Pour UNE graine : deux options d'un même MCQ rendues identiques = échec.
    (Le piège n°1 des QCM randomisés — un distracteur « erreur type » peut
    devenir égal à la bonne réponse sur certaines graines.)"""
    errs: list[str] = []
    for qi, qm in enumerate(_QUESTION_RE.finditer(text)):
        q = qm.group(0)
        if not re.search(r":questionType:\s*MCQ\b", q):
            continue
        rendered: list[str] = []
        for a in _MCQ_ANSWER_RE.finditer(q):
            r, _unres = render_body(a.group(2), env)
            rendered.append(re.sub(r"\s+", " ", r).strip())
        seen: dict[str, int] = {}
        for i, r in enumerate(rendered):
            if r in seen:
                errs.append(f"question {qi} (MCQ) : options {seen[r]} et {i} identiques après rendu : « {r[:70]} »")
            else:
                seen[r] = i
    return errs


# ── Rendu (substitution des injections) ──────────────────────────────────────

def render_body(body_tmpl: str, env: dict) -> tuple[str, list[str]]:
    unresolved: list[str] = []

    def repl(m):
        name = m.group(1).strip()
        if name in env:
            try:
                return str(env[name])
            except Exception:
                return f"<<ERR:{name}>>"
        unresolved.append(name)
        return f"<<{name}>>"

    return INJECTION_RE.sub(repl, body_tmpl), unresolved


# ── Validation principale ────────────────────────────────────────────────────

def validate_text(text: str, seeds: int = 100) -> dict:
    """
    Valide un exercice pythonisé (texte MyST complet).

    Retourne :
      {
        "ok": bool,
        "seeds": int,
        "static_errors": [str],
        "n_exec_errors": int, "n_unresolved": int, "n_forbidden": int,
        "first_failures": [str],     # ≤ 8, lisibles
      }
    """
    static_errors = (check_injection_tokens(text) + check_question_ids(text)
                     + check_render_static(text) + check_injection_in_roles(text))
    declinaison = has_declinaison_questions(text)
    if declinaison:
        static_errors += check_declinaison_static(text)
    code = extract_python_code(text)
    if code is None:
        static_errors.append("aucun bloc `{python}` trouvé")

    report = {
        "ok": False,
        "seeds": seeds,
        "static_errors": static_errors,
        "n_exec_errors": 0,
        "n_unresolved": 0,
        "n_forbidden": 0,
        "n_mcq_collisions": 0,
        "first_failures": [],
    }
    if code is None:
        return report

    body_tmpl = strip_python_blocks(text)
    install_pyxiscience_stubs()
    failures: list[str] = []
    n_timeouts = 0

    from app.validation.sandbox import ExecTimeout, run_with_timeout

    for s in range(seeds):
        random.seed(s)
        # Céder la main entre les graines : en contexte serveur (mono-process),
        # 100 graines de sympy monopolisent le GIL et rendent le suivi de job
        # injoignable (proxy HF → 502) — 1 ms/graine rétablit la réactivité.
        _time.sleep(0.001)
        env: dict = {"rd": random, "random": random}
        out: dict = {}

        def _seed_pass(env=env, out=out):
            with redirect_stdout(io.StringIO()), _warnings.catch_warnings():
                _warnings.simplefilter("ignore")   # plt.show() sous Agg, etc.
                exec(code, env)  # noqa: S102 — sandbox stubs + contenu maison
            # Le rendu et le scan MCQ font des str() sur les valeurs du
            # namespace : time-boxés AVEC l'exec (un str() sympy sur une
            # expression géante peut mouliner des heures — banc 2026-07-02).
            out["rendered"], out["unresolved"] = render_body(body_tmpl, env)
            if declinaison:
                out["mcq"] = check_mcq_collisions(text, env)

        try:
            run_with_timeout(_seed_pass, SEED_TIMEOUT_S)
        except ExecTimeout:
            report["n_exec_errors"] += 1
            n_timeouts += 1
            if len(failures) < 5:
                failures.append(
                    f"seed {s} : timeout ({SEED_TIMEOUT_S:g}s) — exécution ou "
                    "rendu pathologiquement lents (expression sympy géante ?)")
            if _HAS_MPL:
                _plt.close("all")
            if n_timeouts >= 3:
                failures.append("≥ 3 timeouts — graines restantes abandonnées "
                                "(verdict déjà ROUGE)")
                break
            continue
        except Exception:
            report["n_exec_errors"] += 1
            if len(failures) < 5:
                tb = traceback.format_exc().strip().splitlines()
                failures.append(f"seed {s} : exception — {tb[-1]}")
            if _HAS_MPL:
                _plt.close("all")
            continue
        if _HAS_MPL:
            _plt.close("all")

        rendered, unresolved = out["rendered"], out["unresolved"]
        if unresolved:
            report["n_unresolved"] += 1
            if len(failures) < 8:
                failures.append(f"seed {s} : variable(s) non résolue(s) : {sorted(set(unresolved))}")
        # Les lignes `:solution:` (FGQ) sont des motifs de correspondance de
        # réponse, pas du texte affiché : exclues du scan des motifs interdits
        # (`$5$` y est légitime — confirmé par les exemples validés). Leur
        # format est contrôlé séparément (JSON, arité, exactitude).
        scan_target = re.sub(r"(?m)^:solution:.*$", "", rendered)
        fb = scan_forbidden(scan_target)
        if fb:
            report["n_forbidden"] += 1
            if len(failures) < 8:
                failures.append(f"seed {s} : motif interdit : {fb[0]}")
        if declinaison:
            mc = check_mcq_collisions(text, env)
            if mc:
                report["n_mcq_collisions"] += 1
                if len(failures) < 8:
                    failures.append(f"seed {s} : collision d'options MCQ : {mc[0]}")

    report["first_failures"] = failures
    report["ok"] = (
        not static_errors
        and report["n_exec_errors"] == 0
        and report["n_unresolved"] == 0
        and report["n_forbidden"] == 0
        and report["n_mcq_collisions"] == 0
    )
    return report


def format_report(report: dict) -> str:
    """Résumé texte court (pour warnings / prompt de réparation)."""
    lines = []
    for e in report["static_errors"]:
        lines.append(f"[STATIQUE] {e}")
    n = report["seeds"]
    if report["n_exec_errors"]:
        lines.append(f"[EXEC] {report['n_exec_errors']}/{n} graines en exception")
    if report["n_unresolved"]:
        lines.append(f"[INJECTION] {report['n_unresolved']}/{n} graines avec variables non résolues")
    if report["n_forbidden"]:
        lines.append(f"[INTERDITS] {report['n_forbidden']}/{n} graines avec motif interdit")
    if report.get("n_mcq_collisions"):
        lines.append(f"[MCQ] {report['n_mcq_collisions']}/{n} graines avec collision d'options")
    lines.extend(f"  • {f}" for f in report["first_failures"])
    return "\n".join(lines) if lines else "VERT"