creator/backend/routes.py

import asyncio
import shutil
import uuid
from datetime import datetime, timezone

from fastapi import APIRouter, HTTPException
from fastapi.responses import FileResponse

from agents import provider_available
from config import PROJECTS_DIR, PROVIDERS
from database import (
    create_guide, delete_guide, get_guide, list_guides,
    create_topic, list_topics as db_list_topics, delete_topic,
    list_progress, set_progress, delete_progress,
    create_element, list_elements, get_element, update_element, delete_element,
    get_vertiefung, set_vertiefung, list_vertiefungen,
    list_baustein_progress, set_baustein_score, set_baustein_absolviert, set_baustein_verstanden, set_baustein_gemeistert, delete_baustein_daten,
)
from bausteine import generate_bausteine, cancel_bausteine, bausteine_status, active_bausteine, reset_bausteine
from elements import generate_element, chat_with_guide, chat_with_element, check_element, style_element, refine_suggestion
from lernen import NOETIG, MASTERY, MEISTERN, baustein_chat, baustein_diskussion, baustein_element_anlegen, pruefung_bewertung, pruefung_frage, score_berechnen, vertiefung_generieren
from guide import generate_guide, guide_slot_dateien
from pipeline import cancel_guide
from regeln import FORMATE, formate_stats, guide_lock, ist_absolviert, lade_lernstand
from models import (
    GuideCreateRequest, GuideResponse,
    TopicCreateRequest,
    BausteineCreateRequest, BausteineStatusResponse,
    GuideChatRequest, GuideChatResponse,
    ElementCreateRequest, ElementChatRequest, ElementChatResponse, ElementResponse,
    ElementUpdateRequest, ElementCheckRequest, ElementCheckResponse, ElementStyleResponse,
    ElementRefineRequest, ElementRefineResponse,
    ProgressUpdate, ProgressResponse, ProjectResponse, ProviderInfo,
    VertiefungRequest, VertiefungResponse,
    BausteinChatRequest, BausteinChatResponse,
    BausteinPruefungRequest, BausteinPruefungResponse, BausteinLernstandResponse,
)
from paths import bausteine_topics, guide_content_path, project_dir, topic_dir

router = APIRouter(prefix="/api")


@router.get("/providers", response_model=list[ProviderInfo])
async def get_providers():
    return [{"id": pid, "available": provider_available(pid)} for pid in PROVIDERS]


@router.get("/topics")
async def get_topics():
    db_topics = await db_list_topics()
    guides = await list_guides()
    derived = {g["topic"] for g in guides}
    derived.update(bausteine_topics())
    derived.update(job["topic"] for job in active_bausteine())
    # DB ist führend (Reihenfolge: neueste zuerst); Abgeleitetes ohne DB-Eintrag hinten anhängen
    return db_topics + sorted(derived - set(db_topics))


@router.get("/stats")
async def get_stats():
    """Tracker: Themen-Anzahl + pro Format erstellt/absolviert."""
    guides, progress, bausteine_done = await lade_lernstand()
    themen = set(await db_list_topics()) | {g["topic"] for g in guides} | set(bausteine_topics())
    if PROJECTS_DIR.is_dir():
        themen |= {e.name for e in PROJECTS_DIR.iterdir() if e.is_dir()}
    return {"themen": len(themen), "formate": formate_stats(guides, progress, bausteine_done)}


@router.get("/topics/fortschritt")
async def topic_fortschritt(topic: str):
    """Absolviert-Status pro Format — fürs Freischalten der nächsten Ausbaustufe."""
    guides, progress, bausteine_done = await lade_lernstand()
    return {fmt: ist_absolviert(topic, fmt, guides, progress, bausteine_done) for fmt in FORMATE}


@router.post("/topics")
async def add_topic(req: TopicCreateRequest):
    await create_topic(req.name.strip())
    return {"ok": True}


@router.delete("/topics")
async def remove_topic(topic: str):
    await delete_topic(topic)
    await delete_baustein_daten(topic)
    shutil.rmtree(topic_dir(topic), ignore_errors=True)
    return {"ok": True}


def _safe_project_name(name: str) -> str:
    if not name or "/" in name or "\\" in name or ".." in name or "\x00" in name:
        raise HTTPException(400, "Ungültiger Projektname")
    return name


@router.get("/projects", response_model=list[ProjectResponse])
async def list_projects():
    if not PROJECTS_DIR.is_dir():
        return []
    return [{"name": entry.name} for entry in sorted(PROJECTS_DIR.iterdir()) if entry.is_dir()]


@router.delete("/projects/{name}")
async def remove_project(name: str):
    _safe_project_name(name)
    pdir = project_dir(name)
    if not pdir.is_dir():
        raise HTTPException(404, "Projekt nicht gefunden")
    shutil.rmtree(pdir)
    return {"ok": True}


# --- Bausteine ---

@router.get("/bausteine/status", response_model=BausteineStatusResponse)
async def get_bausteine_status(topic: str):
    return bausteine_status(topic)


@router.get("/bausteine/active")
async def get_active_bausteine():
    return active_bausteine()


@router.post("/bausteine")
async def create_bausteine(req: BausteineCreateRequest):
    topic = req.topic.strip()
    if bausteine_status(topic)["generating"]:
        return {"ok": True, "status": "already_generating"}
    await create_topic(topic)
    asyncio.create_task(generate_bausteine(topic, req.instructions.strip(), req.provider))
    return {"ok": True}


@router.post("/bausteine/cancel")
async def cancel_bausteine_route(topic: str):
    if not cancel_bausteine(topic):
        raise HTTPException(404, "Keine laufende Generierung")
    return {"ok": True}


@router.delete("/bausteine")
async def remove_bausteine(topic: str):
    reset_bausteine(topic)
    return {"ok": True}


# --- Baustein-Lernen: Vertiefung, Chat, Prüfung ---

async def _bester_text(topic: str, baustein: str) -> str | None:
    """Kontext für Chat/Prüfung: Amateur-Fassung (art 'deepdive') bevorzugt, sonst Vertiefung."""
    return await get_vertiefung(topic, baustein, "deepdive") or await get_vertiefung(topic, baustein, "vertiefung")


@router.get("/bausteine/lernstand", response_model=BausteinLernstandResponse)
async def baustein_lernstand(topic: str):
    """Prüfungs-Stand + Vertiefungs-/Deepdive-Existenz pro Baustein (roher Titel als Key)."""
    progress = await list_baustein_progress(topic)
    texte = await list_vertiefungen(topic)
    bausteine = {
        p["baustein"]: {
            "gute_antworten": p["gute_antworten"],
            "absolviert": p["absolviert"] is not None,
            "verstanden": p["verstanden"] is not None,
            "gemeistert": p["gemeistert"] is not None,
            "vertiefung": "vertiefung" in texte.get(p["baustein"], set()),
            "deepdive": "deepdive" in texte.get(p["baustein"], set()),
        }
        for p in progress
    }
    for b, arten in texte.items():
        if b not in bausteine:
            bausteine[b] = {
                "gute_antworten": 0, "absolviert": False, "verstanden": False, "gemeistert": False,
                "vertiefung": "vertiefung" in arten, "deepdive": "deepdive" in arten,
            }
    return {"bausteine": bausteine}


@router.get("/bausteine/vertiefung", response_model=VertiefungResponse)
async def get_baustein_vertiefung(topic: str, baustein: str, art: str = "vertiefung"):
    if art not in ("vertiefung", "deepdive"):
        raise HTTPException(400, "Unbekannte Art")
    md = await get_vertiefung(topic, baustein, art)
    if md is None:
        raise HTTPException(404, "Kein Text vorhanden")
    return {"md": md}


@router.post("/bausteine/vertiefung", response_model=VertiefungResponse)
async def create_baustein_vertiefung(req: VertiefungRequest):
    md = await vertiefung_generieren(req.topic, req.baustein, req.section, art=req.art, provider=req.provider)
    if md is None:
        raise HTTPException(502, "Generierung fehlgeschlagen — bitte erneut versuchen")
    await set_vertiefung(req.topic, req.baustein, req.art, md)
    return {"md": md}


@router.post("/bausteine/chat", response_model=BausteinChatResponse)
async def baustein_chat_route(req: BausteinChatRequest):
    vertiefung = await _bester_text(req.topic, req.baustein)
    reply = await baustein_chat(
        req.topic, req.baustein, req.section, vertiefung,
        [m.model_dump() for m in req.messages], provider=req.provider,
    )
    return {"reply": reply}


@router.post("/bausteine/pruefung", response_model=BausteinPruefungResponse)
async def baustein_pruefung_route(req: BausteinPruefungRequest):
    stand = next(
        (p for p in await list_baustein_progress(req.topic) if p["baustein"] == req.baustein),
        {"gute_antworten": 0, "absolviert": None, "verstanden": None, "gemeistert": None},
    )
    gute = stand["gute_antworten"]
    absolviert = stand["absolviert"] is not None
    verstanden = stand["verstanden"] is not None
    gemeistert = stand["gemeistert"] is not None
    vertiefung = await _bester_text(req.topic, req.baustein)
    msgs = [m.model_dump() for m in req.messages]

    if req.aktion == "frage":
        frage = await pruefung_frage(req.topic, req.baustein, req.section, vertiefung, msgs, provider=req.provider)
        if frage is None:
            raise HTTPException(502, "Frage fehlgeschlagen — bitte erneut versuchen")
        return {"frage": frage, "gute_antworten": gute, "absolviert": absolviert, "verstanden": verstanden, "gemeistert": gemeistert}

    if req.aktion == "diskussion":
        if not req.frage.strip():
            raise HTTPException(400, "Diskussion braucht eine laufende Frage")
        reply = await baustein_diskussion(
            req.topic, req.baustein, req.section, vertiefung,
            req.frage, req.letzte_bewertung or None, msgs, provider=req.provider,
        )
        if reply is None:
            raise HTTPException(502, "Diskussion fehlgeschlagen — bitte erneut versuchen")
        return {"reply": reply, "gute_antworten": gute, "absolviert": absolviert, "verstanden": verstanden, "gemeistert": gemeistert}

    # aktion == "antwort" — mindestens eine Nutzer-Antwort muss im Dialog stehen
    # (nach einer Diskussion endet der Dialog mit dem Tutor; Re-Bewertung bleibt erlaubt).
    if not any(m.get("role") == "user" for m in msgs):
        raise HTTPException(400, "Antwort braucht eine Nutzer-Antwort")
    if not req.frage.strip():
        raise HTTPException(400, "Antwort braucht eine laufende Frage")
    data = await pruefung_bewertung(
        req.topic, req.baustein, req.section, vertiefung, req.frage, msgs, gute,
        nachfrage_runde=req.nachfrage_runde, provider=req.provider,
    )
    if data is None:
        raise HTTPException(502, "Bewertung fehlgeschlagen — bitte erneut versuchen")

    # Mündliche Prüfung: noch unklar → Folgefrage stellen, KEINE Wertung, kein Score.
    if data["status"] == "nachfrage":
        return {"frage": data["frage"], "feedback": data["feedback"], "bewertung": None,
                "gute_antworten": gute, "absolviert": absolviert, "verstanden": verstanden, "gemeistert": gemeistert}

    # Score driftfrei aus dem Basis-Score rechnen (Re-Bewertung ersetzt das vorige Ergebnis).
    score = score_berechnen(
        req.score_vor_frage, data["status"] == "gut", req.tier2, req.tier3, absolviert, gemeistert,
    )
    gute = await set_baustein_score(req.topic, req.baustein, score)
    if score >= NOETIG and not absolviert:
        absolviert = True
        if await set_baustein_absolviert(req.topic, req.baustein):
            asyncio.create_task(baustein_element_anlegen(req.topic, req.baustein, req.section, req.provider))
    if score >= MASTERY and not verstanden:
        await set_baustein_verstanden(req.topic, req.baustein)
        verstanden = True
    if score >= MEISTERN and not gemeistert:
        await set_baustein_gemeistert(req.topic, req.baustein)
        gemeistert = True
    return {"feedback": data["feedback"], "bewertung": data["status"], "gute_antworten": gute, "absolviert": absolviert, "verstanden": verstanden, "gemeistert": gemeistert}


# --- Guides ---

@router.post("/guides", response_model=GuideResponse)
async def create(req: GuideCreateRequest):
    guides, progress, bausteine_done = await lade_lernstand()
    grund = guide_lock(req.topic.strip(), req.format, guides, progress, bausteine_done)
    if grund:
        raise HTTPException(400 if grund == "Erst Bausteine erstellen" else 409, grund)
    await create_topic(req.topic.strip())
    now = datetime.now(timezone.utc).isoformat()
    guide = {
        "id": str(uuid.uuid4()),
        "topic": req.topic.strip(),
        "format": req.format,
        "instructions": req.instructions.strip(),
        "status": "queued",
        "progress": None,
        "created_at": now,
        "updated_at": now,
    }
    await create_guide(guide)
    asyncio.create_task(generate_guide(guide["id"], guide["topic"], guide["format"], guide["instructions"], req.provider))
    return guide


@router.get("/guides", response_model=list[GuideResponse])
async def list_all():
    return await list_guides()


@router.get("/guides/locks")
async def guide_locks(topic: str):
    """Sperr-Gründe pro Format für den ▶-Button — None = erstellbar."""
    guides, progress, bausteine_done = await lade_lernstand()
    return {fmt: guide_lock(topic, fmt, guides, progress, bausteine_done) for fmt in ("OnePager", *FORMATE)}


@router.get("/guides/{guide_id}", response_model=GuideResponse)
async def get_one(guide_id: str):
    guide = await get_guide(guide_id)
    if guide is None:
        raise HTTPException(404, "Guide nicht gefunden")
    return guide


@router.get("/guides/{guide_id}/content")
async def guide_content(guide_id: str):
    guide = await get_guide(guide_id)
    if guide is None:
        raise HTTPException(404, "Guide nicht gefunden")
    if guide["status"] != "done":
        raise HTTPException(404, "Inhalt nicht verfügbar")
    path = guide_content_path(guide["topic"], guide["format"])
    if not path.exists():
        raise HTTPException(404, "Datei nicht gefunden")
    return FileResponse(path, media_type="application/json")


@router.post("/guides/{guide_id}/chat", response_model=GuideChatResponse)
async def guide_chat(guide_id: str, req: GuideChatRequest):
    guide = await get_guide(guide_id)
    if guide is None:
        raise HTTPException(404, "Guide nicht gefunden")
    reply = await chat_with_guide(
        guide["topic"], guide["format"], req.section, req.outline,
        [m.model_dump() for m in req.messages],
        provider=req.provider,
    )
    return {"reply": reply}


# --- Elemente (persönliche Zusammenfassung) ---

@router.get("/elements", response_model=list[ElementResponse])
async def get_elements(topic: str):
    return await list_elements(topic)


@router.post("/elements", response_model=ElementResponse)
async def post_element(req: ElementCreateRequest):
    fields = await generate_element(req.topic, req.hint, provider=req.provider)
    now = datetime.now(timezone.utc).isoformat()
    element = {"id": str(uuid.uuid4()), "topic": req.topic, **fields, "created_at": now, "updated_at": now}
    await create_element(element)
    return element


@router.post("/elements/{element_id}/chat", response_model=ElementChatResponse)
async def element_chat(element_id: str, req: ElementChatRequest):
    element = await get_element(element_id)
    if element is None:
        raise HTTPException(404, "Element nicht gefunden")
    reply, changes = await chat_with_element(element, [m.model_dump() for m in req.messages], provider=req.provider)
    return {"reply": reply, "changes": changes}


@router.post("/elements/{element_id}/refine", response_model=ElementRefineResponse)
async def element_refine(element_id: str, req: ElementRefineRequest):
    element = await get_element(element_id)
    if element is None:
        raise HTTPException(404, "Element nicht gefunden")
    change = await refine_suggestion(element, req.suggestion.model_dump(), req.instruction, provider=req.provider)
    if change is None:
        raise HTTPException(502, "Überarbeitung fehlgeschlagen — bitte erneut versuchen")
    return {"change": change}


@router.put("/elements/{element_id}", response_model=ElementResponse)
async def put_element(element_id: str, req: ElementUpdateRequest):
    if await get_element(element_id) is None:
        raise HTTPException(404, "Element nicht gefunden")
    fields = req.model_dump(exclude_unset=True, exclude_none=True)
    if fields:
        now = datetime.now(timezone.utc).isoformat()
        await update_element(element_id, **fields, updated_at=now)
    return await get_element(element_id)


@router.post("/elements/{element_id}/style", response_model=ElementStyleResponse)
async def element_style(element_id: str, req: ElementCheckRequest):
    element = await get_element(element_id)
    if element is None:
        raise HTTPException(404, "Element nicht gefunden")
    changes = await style_element(element, provider=req.provider)
    if changes is None:
        raise HTTPException(502, "Stil-Prüfung fehlgeschlagen — bitte erneut versuchen")
    return {"changes": changes}


@router.post("/elements/{element_id}/check", response_model=ElementCheckResponse)
async def element_check(element_id: str, req: ElementCheckRequest):
    element = await get_element(element_id)
    if element is None:
        raise HTTPException(404, "Element nicht gefunden")
    suggestions = await check_element(element, provider=req.provider)
    if suggestions is None:
        raise HTTPException(502, "Prüfung fehlgeschlagen — bitte erneut versuchen")
    return {"suggestions": suggestions}


@router.delete("/elements/{element_id}")
async def remove_element(element_id: str):
    if not await delete_element(element_id):
        raise HTTPException(404, "Element nicht gefunden")
    return {"ok": True}


@router.post("/guides/{guide_id}/cancel")
async def cancel(guide_id: str):
    cancelled = await cancel_guide(guide_id)
    if not cancelled:
        raise HTTPException(404, "Kein aktiver Prozess gefunden")
    return {"ok": True}


@router.delete("/guides/{guide_id}")
async def remove(guide_id: str, slots: bool = False):
    guide = await get_guide(guide_id)
    if guide is None:
        raise HTTPException(404, "Guide nicht gefunden")
    await delete_progress(guide_id)
    await delete_guide(guide_id)
    # Content-/Schritt-Dateien teilen sich alle Läufe eines Thema+Formats — erst löschen,
    # wenn kein Eintrag sie mehr braucht. Teilfortschritt (Schritt-Dateien ohne fertigen
    # Content) bleibt fürs Resume erhalten, außer es wird explizit verlangt (slots=1).
    rest = [g for g in await list_guides() if g["topic"] == guide["topic"] and g["format"] == guide["format"]]
    if not rest:
        content = guide_content_path(guide["topic"], guide["format"])
        if slots or content.exists():
            for p in guide_slot_dateien(content):
                p.unlink(missing_ok=True)
            content.unlink(missing_ok=True)
    return {"ok": True}


@router.get("/guides/{guide_id}/progress", response_model=ProgressResponse)
async def get_progress(guide_id: str):
    guide = await get_guide(guide_id)
    if guide is None:
        raise HTTPException(404, "Guide nicht gefunden")
    return {"chapters": await list_progress(guide_id)}


@router.post("/guides/{guide_id}/progress", response_model=ProgressResponse)
async def update_progress(guide_id: str, req: ProgressUpdate):
    guide = await get_guide(guide_id)
    if guide is None:
        raise HTTPException(404, "Guide nicht gefunden")
    await set_progress(guide_id, req.chapter, req.done)
    return {"chapters": await list_progress(guide_id)}