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

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


async def erstelle_vorlage(
    db: AsyncSession, *, objekttyp_id: int, name: str, positionen: list[VorlagenpositionCreate]
) -> Beladungsvorlage:
    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."""
    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 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."""
    bisherige = await hole_positionen(db, vorlage.id)
    for position in bisherige:
        await db.delete(position)
    # Flush zwischen Löschen und Neuanlegen: gleiches Muster wie beim Löschen
    # einer ganzen Vorlage (DELETE /vorlagen/{id}) - keine relationship(), also
    # keine automatische Abhängigkeits-Reihenfolge durch SQLAlchemy.
    await db.flush()
    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())