from datetime import datetime, timezone
from decimal import Decimal
from uuid import UUID

from sqlalchemy.ext.asyncio import AsyncSession

from app.models.ausgabe import Ausgabe, AusgabeStatus


class AusgabeBereitsZurueckError(Exception):
    """Eine bereits zurückgegebene Ausgabe kann nicht erneut zurückgegeben werden."""


async def ausgeben(
    db: AsyncSession,
    *,
    material_id: int,
    geraet_instanz_id: UUID | None,
    menge: Decimal,
    empfaenger_id: int,
    ausgegeben_von: int,
    zweck: str | None,
) -> Ausgabe:
    ausgabe = Ausgabe(
        material_id=material_id,
        geraet_instanz_id=geraet_instanz_id,
        menge=menge,
        empfaenger_id=empfaenger_id,
        ausgegeben_von=ausgegeben_von,
        ausgegeben_am=datetime.now(timezone.utc),
        status=AusgabeStatus.offen,
        zweck=zweck,
    )
    db.add(ausgabe)
    await db.flush()
    return ausgabe


async def zurueckgeben(db: AsyncSession, *, ausgabe: Ausgabe) -> Ausgabe:
    if ausgabe.status == AusgabeStatus.zurueck:
        raise AusgabeBereitsZurueckError
    ausgabe.status = AusgabeStatus.zurueck
    ausgabe.rueckgabe_am = datetime.now(timezone.utc)
    await db.flush()
    return ausgabe