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

import uuid

from fastapi import APIRouter, Depends, HTTPException, Response, status
from sqlalchemy import select
from sqlalchemy.exc import IntegrityError
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 KnotenTyp, RolleTyp, Systemknoten
from app.models.fahrzeugdetails import Fahrzeugdetails
from app.models.objekt import Objekt
from app.models.objektposition import Objektposition
from app.models.stammdaten import Material, Objekttyp
from app.models.vorlage import Beladungsvorlage
from app.services.label import generiere_code128_label_pdf
from app.schemas.objekt import (
    FahrzeugdetailsRead,
    FahrzeugdetailsUpsert,
    ObjektCreate,
    ObjektDuplizierenRequest,
    ObjektpositionCreate,
    ObjektpositionRead,
    ObjektpositionUpdate,
    ObjektRead,
    ObjektUpdate,
)
from app.services.objekte import (
    dupliziere_objekt,
    letzte_kontrollen_map,
    naechster_freier_code,
    objekt_aus_vorlage,
    sollmenge_effektiv,
    sollmenge_map,
)


def _objekt_read(objekt: Objekt, letzte_kontrolle_am=None) -> ObjektRead:
    return ObjektRead(
        id=objekt.id,
        code=objekt.code,
        name=objekt.name,
        objekttyp_id=objekt.objekttyp_id,
        vorlage_id=objekt.vorlage_id,
        standort_id=objekt.standort_id,
        status=objekt.status,
        fahrzeug_id=objekt.fahrzeug_id,
        letzte_kontrolle_am=letzte_kontrolle_am,
    )

router = APIRouter()

_admin_only = require_roles(RolleTyp.administration)
_materialverantwortliche = require_roles(RolleTyp.administration, RolleTyp.materialverantwortlicher)


async def _pruefe_zugfahrzeug(db: AsyncSession, fahrzeug_id: int) -> Objekt:
    """Karte 11 Nachtrag (Option A+C): Fahrzeug-Zuordnung nur auf Objekte, deren
    Objekttyp ist_zugfahrzeug=True ist - grenzt Rucksäcke/Geräte serverseitig aus,
    unabhängig vom (evtl. umgangenen) Frontend-Dropdown."""
    ziel = await db.get(Objekt, fahrzeug_id)
    if ziel is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Fahrzeug nicht gefunden")
    objekttyp = await db.get(Objekttyp, ziel.objekttyp_id)
    if objekttyp is None or not objekttyp.ist_zugfahrzeug:
        raise HTTPException(
            status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
            detail="Zielobjekt ist kein Zugfahrzeug (Objekttyp.ist_zugfahrzeug=False)",
        )
    return ziel


@router.get("/objekte", response_model=list[ObjektRead])
async def liste_objekte(
    standort_id: int | None = None,
    db: AsyncSession = Depends(get_db),
    _=Depends(get_current_user),
) -> list[ObjektRead]:
    stmt = select(Objekt)
    if standort_id is not None:
        stmt = stmt.where(Objekt.standort_id == standort_id)
    result = await db.execute(stmt)
    objekte = list(result.scalars().all())
    letzte_map = await letzte_kontrollen_map(db, [o.id for o in objekte])
    return [_objekt_read(o, letzte_map.get(o.id)) for o in objekte]


@router.get("/objekte/code/{code}", response_model=ObjektRead)
async def hole_objekt_per_code(
    code: str, db: AsyncSession = Depends(get_db), _=Depends(get_current_user)
) -> ObjektRead:
    """Lookup per QR/Barcode-Code (Karte 10). Muss VOR /objekte/{id} registriert
    sein, sonst greift FastAPI/Starlette-Routing zuerst die generische ID-Route."""
    result = await db.execute(select(Objekt).where(Objekt.code == code))
    objekt = result.scalar_one_or_none()
    if objekt is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Objekt nicht gefunden")
    letzte_map = await letzte_kontrollen_map(db, [objekt.id])
    return _objekt_read(objekt, letzte_map.get(objekt.id))


@router.get("/objekte/naechster-code")
async def naechster_objekt_code(
    praefix: str, db: AsyncSession = Depends(get_db), _=Depends(get_current_user)
) -> dict[str, str]:
    """Code-Vorschlag beim Anlegen/Duplizieren (Konzept-Ergänzung Karte 10/Prompt 09).
    Vorschlag ist überschreibbar, Eindeutigkeit erzwingt weiterhin objekt.code UNIQUE."""
    return {"code": await naechster_freier_code(db, Objekt.code, praefix)}


@router.get("/objektpositionen/naechster-code")
async def naechster_objektposition_code(
    praefix: str, db: AsyncSession = Depends(get_db), _=Depends(get_current_user)
) -> dict[str, str]:
    """Code-Vorschlag für Geräte-Exemplare (objektposition.code, eigener
    Nummernkreis je Präfix, unabhängig von objekt.code)."""
    return {"code": await naechster_freier_code(db, Objektposition.code, praefix)}


@router.get("/objektpositionen/code/{code}", response_model=ObjektpositionRead)
async def hole_objektposition_per_code(
    code: str, db: AsyncSession = Depends(get_db), _=Depends(get_current_user)
) -> ObjektpositionRead:
    """Geräte-Scan (Konzept-Ergänzung Karte 10/Prompt 09): Code128-Etikett am
    Exemplar identifiziert die Objektposition direkt, unabhängig vom Objekt-Code -
    löst SN-vorbelegte Kontroll-Erfassung ohne Umweg über Objekt-Scan aus."""
    result = await db.execute(select(Objektposition).where(Objektposition.code == code))
    position = result.scalar_one_or_none()
    if position is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Objektposition nicht gefunden")

    objekt = await db.get(Objekt, position.objekt_id)
    vorlagen_sollmengen = await sollmenge_map(db, objekt.vorlage_id if objekt else None)
    return ObjektpositionRead(
        id=position.id,
        objekt_id=position.objekt_id,
        material_id=position.material_id,
        sollmenge_override=position.sollmenge_override,
        sollmenge_effektiv=sollmenge_effektiv(position, vorlagen_sollmengen),
        ist_status=position.ist_status,
        istmenge=position.istmenge,
        seriennummer=position.seriennummer,
        ablaufdatum=position.ablaufdatum,
        chargennummer=position.chargennummer,
        code=position.code,
        pruefintervall_monate=position.pruefintervall_monate,
    )


@router.get("/objekte/{objekt_id}", response_model=ObjektRead)
async def hole_objekt(
    objekt_id: int, db: AsyncSession = Depends(get_db), _=Depends(get_current_user)
) -> ObjektRead:
    objekt = await db.get(Objekt, objekt_id)
    if objekt is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Objekt nicht gefunden")
    letzte_map = await letzte_kontrollen_map(db, [objekt.id])
    return _objekt_read(objekt, letzte_map.get(objekt.id))


@router.get("/objekte/{objekt_id}/label.pdf")
async def objekt_label(
    objekt_id: int, db: AsyncSession = Depends(get_db), _=Depends(get_current_user)
) -> Response:
    """Druckfertiges Code128-Label (Karte 10): Code + Objektname als Klartext-
    Fallback. Vektor-PDF, skaliert verlustfrei auf jede Labelgröße beim Druck."""
    objekt = await db.get(Objekt, objekt_id)
    if objekt is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Objekt nicht gefunden")
    pdf = generiere_code128_label_pdf(objekt.code, objekt.name)
    return Response(
        content=pdf,
        media_type="application/pdf",
        headers={"Content-Disposition": f'inline; filename="{objekt.code}.pdf"'},
    )


@router.get("/objektpositionen/{position_id}/label.pdf")
async def objektposition_label(
    position_id: uuid.UUID, db: AsyncSession = Depends(get_db), _=Depends(get_current_user)
) -> Response:
    """Geräte-Etikett (Karte 10): eigener Code je Exemplar, Klartext-Zusatz ist
    der Materialname (Objektname wäre bei mehreren Geräten am selben Objekt
    nicht unterscheidend genug)."""
    position = await db.get(Objektposition, position_id)
    if position is None or position.code is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Kein Geräte-Code vergeben")
    material = await db.get(Material, position.material_id)
    pdf = generiere_code128_label_pdf(position.code, material.name if material else None)
    return Response(
        content=pdf,
        media_type="application/pdf",
        headers={"Content-Disposition": f'inline; filename="{position.code}.pdf"'},
    )


@router.get("/objekte/{objekt_id}/positionen", response_model=list[ObjektpositionRead])
async def liste_objektpositionen(
    objekt_id: int, db: AsyncSession = Depends(get_db), _=Depends(get_current_user)
) -> list[ObjektpositionRead]:
    objekt = await db.get(Objekt, objekt_id)
    if objekt is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Objekt nicht gefunden")

    vorlagen_sollmengen = await sollmenge_map(db, objekt.vorlage_id)
    result = await db.execute(select(Objektposition).where(Objektposition.objekt_id == objekt_id))
    positionen = result.scalars().all()
    return [
        ObjektpositionRead(
            id=p.id,
            objekt_id=p.objekt_id,
            material_id=p.material_id,
            sollmenge_override=p.sollmenge_override,
            sollmenge_effektiv=sollmenge_effektiv(p, vorlagen_sollmengen),
            ist_status=p.ist_status,
            istmenge=p.istmenge,
            seriennummer=p.seriennummer,
            ablaufdatum=p.ablaufdatum,
            chargennummer=p.chargennummer,
            code=p.code,
            pruefintervall_monate=p.pruefintervall_monate,
        )
        for p in positionen
    ]


@router.post("/objekte", response_model=ObjektRead, status_code=status.HTTP_201_CREATED)
async def erstelle_objekt(
    payload: ObjektCreate, db: AsyncSession = Depends(get_db), _=Depends(_admin_only)
) -> ObjektRead:
    """Prompt 09: neues Objekt aus einer Beladungsvorlage, startet "nicht kontrolliert"
    (Ist-Menge 0 in allen übernommenen Positionen)."""
    vorlage = await db.get(Beladungsvorlage, payload.vorlage_id)
    if vorlage is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Vorlage nicht gefunden")
    if payload.fahrzeug_id is not None:
        await _pruefe_zugfahrzeug(db, payload.fahrzeug_id)

    hauptserver = await db.execute(select(Systemknoten.id).where(Systemknoten.typ == KnotenTyp.haupt))
    zustaendiger_server_id = hauptserver.scalar_one()

    try:
        objekt = await objekt_aus_vorlage(
            db,
            code=payload.code,
            name=payload.name,
            standort_id=payload.standort_id,
            vorlage=vorlage,
            zustaendiger_server_id=zustaendiger_server_id,
            fahrzeug_id=payload.fahrzeug_id,
        )
    except IntegrityError as exc:
        raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Code bereits vergeben") from exc
    return _objekt_read(objekt)


@router.patch("/objekte/{objekt_id}", response_model=ObjektRead)
async def aendere_objekt(
    objekt_id: int, payload: ObjektUpdate, db: AsyncSession = Depends(get_db), _=Depends(_admin_only)
) -> ObjektRead:
    """Fahrzeug-Zuordnung nachträglich setzen/lösen: jedes Objekt gehört zu
    jedem Zeitpunkt höchstens einem Fahrzeug (1:n). Zielobjekt-Existenz und
    direkter Zyklus (A↔B) werden geprüft, mehrstufige Ketten sind fachlich
    nicht vorgesehen und bewusst nicht geprüft (Aufwand/Nutzen)."""
    objekt = await db.get(Objekt, objekt_id)
    if objekt is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Objekt nicht gefunden")

    daten = payload.model_dump(exclude_unset=True)
    ziel_id = daten.get("fahrzeug_id")
    if ziel_id is not None:
        ziel = await _pruefe_zugfahrzeug(db, ziel_id)
        if ziel.fahrzeug_id == objekt_id:
            raise HTTPException(
                status_code=status.HTTP_409_CONFLICT,
                detail="Zyklus: Zielobjekt ist bereits diesem Objekt zugeordnet",
            )

    for feld, wert in daten.items():
        setattr(objekt, feld, wert)
    try:
        await db.flush()
    except IntegrityError as exc:
        raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Ungültige Zuordnung") from exc
    letzte_map = await letzte_kontrollen_map(db, [objekt.id])
    return _objekt_read(objekt, letzte_map.get(objekt.id))


@router.post(
    "/objekte/{objekt_id}/duplizieren", response_model=ObjektRead, status_code=status.HTTP_201_CREATED
)
async def dupliziere(
    objekt_id: int,
    payload: ObjektDuplizierenRequest,
    db: AsyncSession = Depends(get_db),
    _=Depends(_admin_only),
) -> ObjektRead:
    """Prompt 09: Struktur/Sollmengen kopiert, Ist-Menge/SN/Ablauf/Charge NICHT,
    neuer eindeutiger Code Pflicht."""
    quelle = await db.get(Objekt, objekt_id)
    if quelle is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Quell-Objekt nicht gefunden")

    try:
        neues_objekt = await dupliziere_objekt(
            db,
            quelle=quelle,
            neuer_code=payload.neuer_code,
            neuer_name=payload.neuer_name,
            neuer_standort_id=payload.neuer_standort_id,
            zustaendiger_server_id=quelle.zustaendiger_server_id,
        )
    except IntegrityError as exc:
        raise HTTPException(status_code=status.HTTP_409_CONFLICT, detail="Code bereits vergeben") from exc
    return _objekt_read(neues_objekt)


@router.patch("/objekte/{objekt_id}/positionen/{position_id}", response_model=ObjektpositionRead)
async def aendere_objektposition(
    objekt_id: int,
    position_id: uuid.UUID,
    payload: ObjektpositionUpdate,
    db: AsyncSession = Depends(get_db),
    _=Depends(_materialverantwortliche),
) -> ObjektpositionRead:
    """Prompt 10: Sollmengen-Override setzen/aufheben, Material als entfernt markieren,
    Ist-Werte (Menge/SN/Ablauf/Charge) pflegen."""
    position = await db.get(Objektposition, position_id)
    if position is None or position.objekt_id != objekt_id:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Objektposition nicht gefunden")

    for feld, wert in payload.model_dump(exclude_unset=True).items():
        setattr(position, feld, wert)
    await db.flush()

    objekt = await db.get(Objekt, objekt_id)
    vorlagen_sollmengen = await sollmenge_map(db, objekt.vorlage_id if objekt else None)
    return ObjektpositionRead(
        id=position.id,
        objekt_id=position.objekt_id,
        material_id=position.material_id,
        sollmenge_override=position.sollmenge_override,
        sollmenge_effektiv=sollmenge_effektiv(position, vorlagen_sollmengen),
        ist_status=position.ist_status,
        istmenge=position.istmenge,
        seriennummer=position.seriennummer,
        ablaufdatum=position.ablaufdatum,
        chargennummer=position.chargennummer,
        code=position.code,
        pruefintervall_monate=position.pruefintervall_monate,
    )


@router.post(
    "/objekte/{objekt_id}/positionen",
    response_model=ObjektpositionRead,
    status_code=status.HTTP_201_CREATED,
)
async def fuege_zusatzposition_hinzu(
    objekt_id: int,
    payload: ObjektpositionCreate,
    db: AsyncSession = Depends(get_db),
    _=Depends(_materialverantwortliche),
) -> ObjektpositionRead:
    """Prompt 10 Punkt 2: zusätzliches, nicht in der Vorlage enthaltenes Material."""
    objekt = await db.get(Objekt, objekt_id)
    if objekt is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Objekt nicht gefunden")

    position = Objektposition(
        objekt_id=objekt_id,
        material_id=payload.material_id,
        sollmenge_override=payload.sollmenge_override,
    )
    db.add(position)
    try:
        await db.flush()
    except IntegrityError as exc:
        raise HTTPException(
            status_code=status.HTTP_409_CONFLICT, detail="Material existiert bereits an diesem Objekt"
        ) from exc

    return ObjektpositionRead(
        id=position.id,
        objekt_id=position.objekt_id,
        material_id=position.material_id,
        sollmenge_override=position.sollmenge_override,
        sollmenge_effektiv=payload.sollmenge_override,
        ist_status=position.ist_status,
        istmenge=position.istmenge,
        seriennummer=position.seriennummer,
        ablaufdatum=position.ablaufdatum,
        chargennummer=position.chargennummer,
        code=position.code,
        pruefintervall_monate=position.pruefintervall_monate,
    )


@router.get("/objekte/{objekt_id}/fahrzeugdetails", response_model=FahrzeugdetailsRead)
async def hole_fahrzeugdetails(
    objekt_id: int, db: AsyncSession = Depends(get_db), _=Depends(get_current_user)
) -> Fahrzeugdetails:
    """Roadmap Phase 3 (Modul Fahrzeuge): 404 statt leerem Default, wenn noch
    keine Fahrzeugdetails erfasst sind - Frontend zeigt dann ein leeres
    Formular statt scheinbar vorhandener Nullwerte."""
    details = await db.get(Fahrzeugdetails, objekt_id)
    if details is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Keine Fahrzeugdetails erfasst")
    return details


@router.put("/objekte/{objekt_id}/fahrzeugdetails", response_model=FahrzeugdetailsRead)
async def setze_fahrzeugdetails(
    objekt_id: int,
    payload: FahrzeugdetailsUpsert,
    db: AsyncSession = Depends(get_db),
    _=Depends(_materialverantwortliche),
) -> Fahrzeugdetails:
    """Upsert (PUT statt PATCH): Fahrzeugdetails existieren typischerweise
    genau einmal je Objekt, ein Formular ersetzt komplett statt Teil-Update -
    einfacher als Anlegen/Ändern zu unterscheiden."""
    objekt = await db.get(Objekt, objekt_id)
    if objekt is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Objekt nicht gefunden")

    details = await db.get(Fahrzeugdetails, objekt_id)
    if details is None:
        details = Fahrzeugdetails(objekt_id=objekt_id)
        db.add(details)
    for feld, wert in payload.model_dump().items():
        setattr(details, feld, wert)
    await db.flush()
    return details