from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession

from app.models.stammdaten import Fach
from app.models.vorlage import Beladungsvorlage, VorlageStatus, Vorlagenposition
from app.schemas.vorlage import VorlagenpositionCreate


async def _stelle_faecher_sicher(
    db: AsyncSession, *, objekttyp_id: int, positionen: list[VorlagenpositionCreate]
) -> None:
    """Vorlagenposition.fach ist Freitext ohne FK zur festen Fächer-Liste (Fach-
    Tabelle) - ohne diesen Abgleich laufen beide auseinander: ein Fach taucht in
    der Kontrolle auf, existiert aber nicht in der Fächer-Verwaltung des
    Objekttyps (Nutzer-Fund: "Güdeltuben" bei Kontrolle, aber kein Fach dazu)."""
    benoetigt = {pos.fach for pos in positionen if pos.fach}
    if not benoetigt:
        return
    result = await db.execute(select(Fach.name).where(Fach.objekttyp_id == objekttyp_id))
    vorhanden_klein = {name.strip().lower() for name in result.scalars().all()}
    for name in benoetigt:
        if name.strip().lower() not in vorhanden_klein:
            db.add(Fach(objekttyp_id=objekttyp_id, name=name))
    await db.flush()


async def erstelle_vorlage(
    db: AsyncSession, *, objekttyp_id: int, name: str, positionen: list[VorlagenpositionCreate]
) -> Beladungsvorlage:
    await _stelle_faecher_sicher(db, objekttyp_id=objekttyp_id, positionen=positionen)
    vorlage = Beladungsvorlage(
        objekttyp_id=objekttyp_id, name=name, version=1, status=VorlageStatus.aktiv
    )
    db.add(vorlage)
    await db.flush()
    for pos in positionen:
        db.add(Vorlagenposition(vorlage_id=vorlage.id, **pos.model_dump()))
    await db.flush()
    return vorlage


async def neue_version(
    db: AsyncSession, *, aktuelle_vorlage: Beladungsvorlage, positionen: list[VorlagenpositionCreate]
) -> Beladungsvorlage:
    """Prompt 08 Punkt 4: Änderung erzeugt neue Version, alte bleibt unverändert/lesbar
    (Historie-Integrität) statt überschrieben zu werden."""
    await _stelle_faecher_sicher(db, objekttyp_id=aktuelle_vorlage.objekttyp_id, positionen=positionen)
    aktuelle_vorlage.status = VorlageStatus.veraltet
    await db.flush()

    neue = Beladungsvorlage(
        objekttyp_id=aktuelle_vorlage.objekttyp_id,
        name=aktuelle_vorlage.name,
        version=aktuelle_vorlage.version + 1,
        status=VorlageStatus.aktiv,
    )
    db.add(neue)
    await db.flush()
    for pos in positionen:
        db.add(Vorlagenposition(vorlage_id=neue.id, **pos.model_dump()))
    await db.flush()
    return neue


async def loesche_alle_positionen(db: AsyncSession, vorlage_id: int) -> None:
    """Gemeinsame Vorstufe für Vorlage-Löschen und Positionen-Ersetzen: Flush
    zwischen Löschen und dem nächsten Schritt ist Pflicht - Beladungsvorlage/
    Vorlagenposition haben keine ORM-relationship() (nur rohe FK-Spalten),
    SQLAlchemy kennt die Abhängigkeit also nicht automatisch und könnte DELETEs
    in falscher Reihenfolge absetzen (FK-Verletzung, beim echten Löschversuch
    gefunden)."""
    for position in await hole_positionen(db, vorlage_id):
        await db.delete(position)
    await db.flush()


async def aktualisiere_positionen(
    db: AsyncSession, *, vorlage: Beladungsvorlage, positionen: list[VorlagenpositionCreate]
) -> Beladungsvorlage:
    """Nutzer-Vorgabe 2026-09-05: Vorlage muss keine Versionen haben - direktes
    Bearbeiten der bestehenden Version statt Zwang zu `neue_version()`. Wirkt
    sich NICHT auf bereits angelegte Objekte aus, da deren Sollmenge fest in
    `objektposition.sollmenge_vorlage` kopiert ist (siehe services/objekte.py),
    nicht mehr live aus der Vorlage gelesen wird."""
    await _stelle_faecher_sicher(db, objekttyp_id=vorlage.objekttyp_id, positionen=positionen)
    await loesche_alle_positionen(db, vorlage.id)
    for pos in positionen:
        db.add(Vorlagenposition(vorlage_id=vorlage.id, **pos.model_dump()))
    await db.flush()
    return vorlage


async def hole_positionen(db: AsyncSession, vorlage_id: int) -> list[Vorlagenposition]:
    result = await db.execute(select(Vorlagenposition).where(Vorlagenposition.vorlage_id == vorlage_id))
    return list(result.scalars().all())