MABEA/backend/app/api/v1/endpoints/vorlagen.py

from fastapi import APIRouter, Depends, HTTPException, status
from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession

from app.api.deps import get_current_user, require_roles
from app.db.session import get_db
from app.models.auth import RolleTyp
from app.models.objekt import Objekt
from app.models.vorlage import Beladungsvorlage
from app.schemas.vorlage import BeladungsvorlageCreate, BeladungsvorlageRead, VorlagenAenderung
from app.services.vorlagen import (
    aktualisiere_positionen,
    erstelle_vorlage,
    hole_positionen,
    loesche_alle_positionen,
    neue_version,
)

router = APIRouter()

_ersteller = require_roles(RolleTyp.administration, RolleTyp.materialverantwortlicher)


async def _mit_positionen(db: AsyncSession, vorlage: Beladungsvorlage) -> BeladungsvorlageRead:
    positionen = await hole_positionen(db, vorlage.id)
    return BeladungsvorlageRead(
        id=vorlage.id,
        objekttyp_id=vorlage.objekttyp_id,
        name=vorlage.name,
        version=vorlage.version,
        status=vorlage.status,
        positionen=positionen,
    )


@router.get("/vorlagen", response_model=list[BeladungsvorlageRead])
async def liste_vorlagen(
    objekttyp_id: int | None = None,
    db: AsyncSession = Depends(get_db),
    _=Depends(get_current_user),
) -> list[BeladungsvorlageRead]:
    stmt = select(Beladungsvorlage)
    if objekttyp_id is not None:
        stmt = stmt.where(Beladungsvorlage.objekttyp_id == objekttyp_id)
    result = await db.execute(stmt)
    return [await _mit_positionen(db, v) for v in result.scalars().all()]


@router.get("/vorlagen/{vorlage_id}", response_model=BeladungsvorlageRead)
async def hole_vorlage(
    vorlage_id: int, db: AsyncSession = Depends(get_db), _=Depends(get_current_user)
) -> BeladungsvorlageRead:
    vorlage = await db.get(Beladungsvorlage, vorlage_id)
    if vorlage is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Vorlage nicht gefunden")
    return await _mit_positionen(db, vorlage)


@router.post("/vorlagen", response_model=BeladungsvorlageRead, status_code=status.HTTP_201_CREATED)
async def erstelle_neue_vorlage(
    payload: BeladungsvorlageCreate, db: AsyncSession = Depends(get_db), _=Depends(_ersteller)
) -> BeladungsvorlageRead:
    vorlage = await erstelle_vorlage(
        db, objekttyp_id=payload.objekttyp_id, name=payload.name, positionen=payload.positionen
    )
    return await _mit_positionen(db, vorlage)


@router.post(
    "/vorlagen/{vorlage_id}/neue-version",
    response_model=BeladungsvorlageRead,
    status_code=status.HTTP_201_CREATED,
)
async def erzeuge_neue_version(
    vorlage_id: int,
    payload: VorlagenAenderung,
    db: AsyncSession = Depends(get_db),
    _=Depends(_ersteller),
) -> BeladungsvorlageRead:
    """Prompt 08 Punkt 4: erzeugt neue Version, alte bleibt unverändert/lesbar."""
    aktuelle_vorlage = await db.get(Beladungsvorlage, vorlage_id)
    if aktuelle_vorlage is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Vorlage nicht gefunden")
    neue = await neue_version(db, aktuelle_vorlage=aktuelle_vorlage, positionen=payload.positionen)
    return await _mit_positionen(db, neue)


@router.put("/vorlagen/{vorlage_id}/positionen", response_model=BeladungsvorlageRead)
async def aktualisiere_vorlage_positionen(
    vorlage_id: int,
    payload: VorlagenAenderung,
    db: AsyncSession = Depends(get_db),
    _=Depends(_ersteller),
) -> BeladungsvorlageRead:
    """Nutzer-Vorgabe 2026-09-05: Vorlage muss keine Versionen haben - ersetzt
    die Positionen der bestehenden Vorlage direkt, ohne neue Version anzulegen.
    Bereits angelegte Objekte sind unbetroffen (sollmenge_vorlage ist an der
    Objektposition fest kopiert, siehe app/services/objekte.py)."""
    vorlage = await db.get(Beladungsvorlage, vorlage_id)
    if vorlage is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Vorlage nicht gefunden")
    aktualisiert = await aktualisiere_positionen(db, vorlage=vorlage, positionen=payload.positionen)
    return await _mit_positionen(db, aktualisiert)


@router.delete("/vorlagen/{vorlage_id}", status_code=status.HTTP_204_NO_CONTENT)
async def loesche_vorlage(
    vorlage_id: int, db: AsyncSession = Depends(get_db), _=Depends(_ersteller)
) -> None:
    """Vorlagen sind bewusst versioniert (Prompt 08) - normalerweise wird eine
    Version nie gelöscht, nur durch eine neue ersetzt (status=veraltet). Echtes
    Löschen ist nur erlaubt, wenn KEIN Objekt diese Version referenziert (sonst
    FK-Verletzung bzw. Datenverlust an echten Objekten) - z.B. versehentlich
    angelegte/Test-Versionen entfernen."""
    vorlage = await db.get(Beladungsvorlage, vorlage_id)
    if vorlage is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Vorlage nicht gefunden")

    result = await db.execute(select(Objekt.id).where(Objekt.vorlage_id == vorlage_id).limit(1))
    if result.scalar_one_or_none() is not None:
        raise HTTPException(
            status_code=status.HTTP_409_CONFLICT,
            detail="Vorlage wird von mindestens einem Objekt verwendet und kann nicht gelöscht werden",
        )

    await loesche_alle_positionen(db, vorlage_id)
    await db.delete(vorlage)