shinkan-jinkendo/backend/routers/exercise_progression_graphs.py

"""
Progressionsgraph zwischen Übungen (Übung → Übung), Migration 032–034.
Optional Übungsvarianten als Knoten-Endpunkte; Sequenz-Bulk-Anlage.
AuthZ analog training_plan_templates (_template_access / _has_planning_role).
"""
from typing import Any, List, Optional

from fastapi import APIRouter, Depends, HTTPException, Query
from pydantic import BaseModel, Field, model_validator
from psycopg2 import IntegrityError

from auth import require_auth
from db import get_db, get_cursor, r2d

from routers.training_planning import _has_planning_role

router = APIRouter(prefix="/api", tags=["exercises"])


class ProgressionGraphCreate(BaseModel):
    name: str = Field(..., min_length=1, max_length=200)
    description: Optional[str] = None
    visibility: str = Field(default="private", pattern="^(private|club|official)$")
    club_id: Optional[int] = None


class ProgressionGraphUpdate(BaseModel):
    name: Optional[str] = Field(None, min_length=1, max_length=200)
    description: Optional[str] = None
    visibility: Optional[str] = Field(None, pattern="^(private|club|official)$")
    club_id: Optional[int] = None


class ProgressionEdgeCreate(BaseModel):
    from_exercise_id: int = Field(..., gt=0)
    to_exercise_id: int = Field(..., gt=0)
    from_exercise_variant_id: Optional[int] = Field(default=None)
    to_exercise_variant_id: Optional[int] = Field(default=None)
    edge_type: str = Field(default="next_exercise", min_length=1, max_length=50)
    notes: Optional[str] = Field(None, max_length=4000)


class ProgressionEdgeUpdate(BaseModel):
    notes: Optional[str] = Field(None, max_length=4000)


class SequenceStep(BaseModel):
    exercise_id: int = Field(..., gt=0)
    variant_id: Optional[int] = Field(default=None)


class ProgressionSequenceCreate(BaseModel):
    steps: List[SequenceStep] = Field(..., min_length=2)
    segment_notes: Optional[List[Optional[str]]] = None
    """Länge muss len(steps)-1 sein, wenn gesetzt; Notiz pro Kante Zwischen je zwei Schritten."""

    @model_validator(mode="after")
    def check_segment_notes_len(self):
        if self.segment_notes is not None and len(self.segment_notes) != len(self.steps) - 1:
            raise ValueError(
                f"segment_notes muss genau {len(self.steps) - 1} Einträge haben (len(steps)-1)"
            )
        return self


class EdgeIdsBatch(BaseModel):
    edge_ids: List[int] = Field(..., min_length=1)


_EDGE_SELECT = """
    SELECT e.id, e.graph_id,
           e.from_exercise_id, e.from_exercise_variant_id,
           e.to_exercise_id, e.to_exercise_variant_id,
           e.edge_type, e.notes, e.created_at,
           ef.title AS from_exercise_title, et.title AS to_exercise_title,
           vf.variant_name AS from_variant_name, vt.variant_name AS to_variant_name
    FROM exercise_progression_edges e
    JOIN exercises ef ON ef.id = e.from_exercise_id
    JOIN exercises et ON et.id = e.to_exercise_id
    LEFT JOIN exercise_variants vf ON vf.id = e.from_exercise_variant_id
    LEFT JOIN exercise_variants vt ON vt.id = e.to_exercise_variant_id
"""


def _graph_access(cur, graph_id: int, profile_id: int, role: str) -> dict:
    cur.execute(
        "SELECT * FROM exercise_progression_graphs WHERE id = %s",
        (graph_id,),
    )
    r = cur.fetchone()
    if not r:
        raise HTTPException(status_code=404, detail="Progressionsgraph nicht gefunden")
    row = r2d(r)
    if role in ("admin", "superadmin"):
        return row
    if row.get("created_by") != profile_id:
        raise HTTPException(status_code=403, detail="Keine Berechtigung für diesen Progressionsgraph")
    return row


def _assert_exercises_exist(cur, *exercise_ids: int) -> None:
    unique_ids = list(dict.fromkeys(exercise_ids))
    if not unique_ids:
        return
    cur.execute(
        f"SELECT id FROM exercises WHERE id IN ({','.join(['%s'] * len(unique_ids))})",
        tuple(unique_ids),
    )
    found = {r["id"] if isinstance(r, dict) else r[0] for r in cur.fetchall()}
    missing = set(unique_ids) - found
    if missing:
        raise HTTPException(
            status_code=400,
            detail=f"Unbekannte Übung(en): {sorted(missing)}",
        )


def _assert_variant_for_exercise(cur, exercise_id: int, variant_id: Optional[int]) -> None:
    if variant_id is None:
        return
    cur.execute(
        "SELECT exercise_id FROM exercise_variants WHERE id = %s",
        (variant_id,),
    )
    row = cur.fetchone()
    if not row:
        raise HTTPException(status_code=400, detail=f"Unbekannte Variante: {variant_id}")
    ev_ex = row["exercise_id"] if isinstance(row, dict) else row[0]
    if ev_ex != exercise_id:
        raise HTTPException(status_code=400, detail="Variante gehört nicht zur gewählten Übung")


def _insert_edge_row(
    cur,
    graph_id: int,
    from_exercise_id: int,
    from_variant_id: Optional[int],
    to_exercise_id: int,
    to_variant_id: Optional[int],
    edge_type: str,
    notes: Optional[str],
) -> dict:
    cur.execute(
        """
        INSERT INTO exercise_progression_edges (
            graph_id,
            from_exercise_id, from_exercise_variant_id,
            to_exercise_id, to_exercise_variant_id,
            edge_type, notes
        )
        VALUES (%s, %s, %s, %s, %s, %s, %s)
        RETURNING id
        """,
        (
            graph_id,
            from_exercise_id,
            from_variant_id,
            to_exercise_id,
            to_variant_id,
            edge_type,
            notes,
        ),
    )
    new_id = cur.fetchone()["id"]
    cur.execute(_EDGE_SELECT + " WHERE e.id = %s", (new_id,))
    return r2d(cur.fetchone())


@router.get("/exercise-progression-graphs")
def list_progression_graphs(session: dict = Depends(require_auth)):
    profile_id = session["profile_id"]
    role = session.get("role")
    with get_db() as conn:
        cur = get_cursor(conn)
        if role in ("admin", "superadmin"):
            cur.execute(
                """
                SELECT g.*,
                    (SELECT COUNT(*) FROM exercise_progression_edges e WHERE e.graph_id = g.id) AS edges_count
                FROM exercise_progression_graphs g
                ORDER BY g.updated_at DESC NULLS LAST, g.name
                """
            )
        else:
            cur.execute(
                """
                SELECT g.*,
                    (SELECT COUNT(*) FROM exercise_progression_edges e WHERE e.graph_id = g.id) AS edges_count
                FROM exercise_progression_graphs g
                WHERE g.created_by = %s
                ORDER BY g.updated_at DESC NULLS LAST, g.name
                """,
                (profile_id,),
            )
        return [r2d(r) for r in cur.fetchall()]


@router.get("/exercise-progression-graphs/{graph_id}")
def get_progression_graph(
    graph_id: int,
    include_edges: bool = Query(default=False),
    session: dict = Depends(require_auth),
):
    profile_id = session["profile_id"]
    role = session.get("role")
    with get_db() as conn:
        cur = get_cursor(conn)
        row = _graph_access(cur, graph_id, profile_id, role)
        if include_edges:
            cur.execute(
                _EDGE_SELECT + " WHERE e.graph_id = %s ORDER BY e.id",
                (graph_id,),
            )
            row["edges"] = [r2d(r) for r in cur.fetchall()]
        return row


@router.post("/exercise-progression-graphs", status_code=201)
def create_progression_graph(
    body: ProgressionGraphCreate,
    session: dict = Depends(require_auth),
):
    profile_id = session["profile_id"]
    role = session.get("role")
    if not _has_planning_role(role):
        raise HTTPException(status_code=403, detail="Keine Berechtigung zum Anlegen von Progressionsgraphen")

    name = body.name.strip()
    if not name:
        raise HTTPException(status_code=400, detail="name ist Pflicht")

    with get_db() as conn:
        cur = get_cursor(conn)
        cur.execute(
            """
            INSERT INTO exercise_progression_graphs (name, description, visibility, club_id, created_by)
            VALUES (%s, %s, %s, %s, %s)
            RETURNING id
            """,
            (name, body.description, body.visibility, body.club_id, profile_id),
        )
        gid = cur.fetchone()["id"]
        conn.commit()

    return get_progression_graph(gid, include_edges=False, session=session)


@router.put("/exercise-progression-graphs/{graph_id}")
def update_progression_graph(
    graph_id: int,
    body: ProgressionGraphUpdate,
    session: dict = Depends(require_auth),
):
    profile_id = session["profile_id"]
    role = session.get("role")
    data = body.model_dump(exclude_unset=True)
    if not data:
        raise HTTPException(status_code=400, detail="Keine Felder zum Aktualisieren")

    with get_db() as conn:
        cur = get_cursor(conn)
        _graph_access(cur, graph_id, profile_id, role)

        fields: List[str] = []
        params: List[Any] = []
        if "name" in data:
            n = (data["name"] or "").strip()
            if not n:
                raise HTTPException(status_code=400, detail="name ist Pflicht")
            fields.append("name = %s")
            params.append(n)
        if "description" in data:
            fields.append("description = %s")
            params.append(data["description"])
        if "visibility" in data:
            fields.append("visibility = %s")
            params.append(data["visibility"])
        if "club_id" in data:
            fields.append("club_id = %s")
            params.append(data["club_id"])

        fields.append("updated_at = NOW()")
        params.append(graph_id)
        cur.execute(
            f"UPDATE exercise_progression_graphs SET {', '.join(fields)} WHERE id = %s",
            tuple(params),
        )
        conn.commit()

    return get_progression_graph(graph_id, include_edges=False, session=session)


@router.delete("/exercise-progression-graphs/{graph_id}")
def delete_progression_graph(graph_id: int, session: dict = Depends(require_auth)):
    profile_id = session["profile_id"]
    role = session.get("role")
    with get_db() as conn:
        cur = get_cursor(conn)
        _graph_access(cur, graph_id, profile_id, role)
        cur.execute("DELETE FROM exercise_progression_graphs WHERE id = %s", (graph_id,))
        conn.commit()
    return {"ok": True}


@router.get("/exercise-progression-graphs/{graph_id}/edges")
def list_progression_edges(
    graph_id: int,
    from_exercise_id: Optional[int] = Query(default=None),
    to_exercise_id: Optional[int] = Query(default=None),
    session: dict = Depends(require_auth),
):
    profile_id = session["profile_id"]
    role = session.get("role")
    with get_db() as conn:
        cur = get_cursor(conn)
        _graph_access(cur, graph_id, profile_id, role)
        q = _EDGE_SELECT + " WHERE e.graph_id = %s"
        params: List[Any] = [graph_id]
        if from_exercise_id is not None:
            q += " AND e.from_exercise_id = %s"
            params.append(from_exercise_id)
        if to_exercise_id is not None:
            q += " AND e.to_exercise_id = %s"
            params.append(to_exercise_id)
        q += " ORDER BY e.id"
        cur.execute(q, tuple(params))
        return [r2d(r) for r in cur.fetchall()]


@router.post("/exercise-progression-graphs/{graph_id}/edges", status_code=201)
def create_progression_edge(
    graph_id: int,
    body: ProgressionEdgeCreate,
    session: dict = Depends(require_auth),
):
    profile_id = session["profile_id"]
    role = session.get("role")

    with get_db() as conn:
        cur = get_cursor(conn)
        _graph_access(cur, graph_id, profile_id, role)
        _assert_exercises_exist(cur, body.from_exercise_id, body.to_exercise_id)
        fv = body.from_exercise_variant_id
        tv = body.to_exercise_variant_id
        _assert_variant_for_exercise(cur, body.from_exercise_id, fv)
        _assert_variant_for_exercise(cur, body.to_exercise_id, tv)
        et = (body.edge_type or "next_exercise").strip() or "next_exercise"
        notes = (body.notes or "").strip() or None
        try:
            row = _insert_edge_row(
                cur,
                graph_id,
                body.from_exercise_id,
                fv,
                body.to_exercise_id,
                tv,
                et,
                notes,
            )
            conn.commit()
        except IntegrityError as e:
            conn.rollback()
            raise HTTPException(
                status_code=409,
                detail="Kante existiert bereits oder Endpunkte unzulässig (gleiche Übung ohne zwei Varianten)",
            ) from e

    return row


@router.post("/exercise-progression-graphs/{graph_id}/edges/sequence", status_code=201)
def create_progression_sequence(
    graph_id: int,
    body: ProgressionSequenceCreate,
    session: dict = Depends(require_auth),
):
    """Legt n−1 Nachfolger-Kanten (next_exercise) für eine geordnete Schrittliste an."""
    profile_id = session["profile_id"]
    role = session.get("role")
    steps = body.steps
    n_seg = len(steps) - 1
    seg_notes = body.segment_notes

    created: List[dict] = []
    with get_db() as conn:
        cur = get_cursor(conn)
        _graph_access(cur, graph_id, profile_id, role)

        ex_ids = [s.exercise_id for s in steps]
        _assert_exercises_exist(cur, *ex_ids)

        try:
            for i in range(n_seg):
                a, b = steps[i], steps[i + 1]
                _assert_variant_for_exercise(cur, a.exercise_id, a.variant_id)
                _assert_variant_for_exercise(cur, b.exercise_id, b.variant_id)
                note = None
                if seg_notes is not None:
                    raw = seg_notes[i]
                    note = (raw or "").strip() or None if raw is not None else None
                row = _insert_edge_row(
                    cur,
                    graph_id,
                    a.exercise_id,
                    a.variant_id,
                    b.exercise_id,
                    b.variant_id,
                    "next_exercise",
                    note,
                )
                created.append(row)
            conn.commit()
        except IntegrityError as e:
            conn.rollback()
            raise HTTPException(
                status_code=409,
                detail="Sequenz konnte nicht vollständig angelegt werden (Duplikat oder ungültige Endpunkte). Keine Teilmenge gespeichert.",
            ) from e

    return {"created": created, "count": len(created)}


@router.put("/exercise-progression-graphs/{graph_id}/edges/{edge_id}")
def update_progression_edge(
    graph_id: int,
    edge_id: int,
    body: ProgressionEdgeUpdate,
    session: dict = Depends(require_auth),
):
    profile_id = session["profile_id"]
    role = session.get("role")
    data = body.model_dump(exclude_unset=True)
    if not data:
        raise HTTPException(status_code=400, detail="Keine Felder zum Aktualisieren")

    with get_db() as conn:
        cur = get_cursor(conn)
        _graph_access(cur, graph_id, profile_id, role)
        cur.execute(
            "SELECT id FROM exercise_progression_edges WHERE id = %s AND graph_id = %s",
            (edge_id, graph_id),
        )
        if not cur.fetchone():
            raise HTTPException(status_code=404, detail="Kante nicht gefunden")

        if "notes" in data:
            n = data["notes"]
            notes_val = (n or "").strip() or None if n is not None else None
            cur.execute(
                "UPDATE exercise_progression_edges SET notes = %s WHERE id = %s AND graph_id = %s",
                (notes_val, edge_id, graph_id),
            )
        conn.commit()
        cur.execute(_EDGE_SELECT + " WHERE e.id = %s AND e.graph_id = %s", (edge_id, graph_id))
        return r2d(cur.fetchone())


@router.delete("/exercise-progression-graphs/{graph_id}/edges/{edge_id}")
def delete_progression_edge(
    graph_id: int,
    edge_id: int,
    session: dict = Depends(require_auth),
):
    profile_id = session["profile_id"]
    role = session.get("role")
    with get_db() as conn:
        cur = get_cursor(conn)
        _graph_access(cur, graph_id, profile_id, role)
        cur.execute(
            """
            DELETE FROM exercise_progression_edges
            WHERE id = %s AND graph_id = %s
            RETURNING id
            """,
            (edge_id, graph_id),
        )
        if not cur.fetchone():
            raise HTTPException(status_code=404, detail="Kante nicht gefunden")
        conn.commit()
    return {"ok": True}


@router.post("/exercise-progression-graphs/{graph_id}/edges/delete-batch")
def delete_progression_edges_batch(
    graph_id: int,
    body: EdgeIdsBatch,
    session: dict = Depends(require_auth),
):
    """Löscht mehrere Kanten (z. B. eine zusammenhängende Kette in einem Schritt)."""
    profile_id = session["profile_id"]
    role = session.get("role")
    ids = body.edge_ids
    clean_ids = list(dict.fromkeys(ids))

    with get_db() as conn:
        cur = get_cursor(conn)
        _graph_access(cur, graph_id, profile_id, role)
        cur.execute(
            f"""
            DELETE FROM exercise_progression_edges
            WHERE graph_id = %s AND id IN ({",".join(["%s"] * len(clean_ids))})
            """,
            (graph_id, *clean_ids),
        )
        deleted = cur.rowcount
        conn.commit()
    return {"ok": True, "deleted": deleted}