import enum

from sqlalchemy import ForeignKey, String
from sqlalchemy.dialects.postgresql import ENUM as PgEnum
from sqlalchemy.orm import Mapped, mapped_column

from app.db.base import Base


class ObjektStatus(str, enum.Enum):
    aktiv = "aktiv"
    ausser_dienst = "ausser_dienst"


objekt_status_pg = PgEnum(ObjektStatus, name="objekt_status", create_type=False)


class Objekt(Base):
    """Sprint 2 (Prompt 08/09/10): Objekt referenziert eine konkrete Beladungsvorlagen-
    Version. Sollmenge-Auflösung/Duplizieren siehe app/services/objekte.py."""

    __tablename__ = "objekt"

    id: Mapped[int] = mapped_column(primary_key=True)
    code: Mapped[str] = mapped_column(String, unique=True, nullable=False)
    name: Mapped[str] = mapped_column(String, nullable=False)
    objekttyp_id: Mapped[int] = mapped_column(ForeignKey("objekttyp.id"), nullable=False)
    vorlage_id: Mapped[int | None] = mapped_column(ForeignKey("beladungsvorlage.id"))
    standort_id: Mapped[int] = mapped_column(ForeignKey("standort.id"), nullable=False)
    status: Mapped[ObjektStatus] = mapped_column(
        objekt_status_pg, nullable=False, default=ObjektStatus.aktiv
    )
    zustaendiger_server_id: Mapped[int] = mapped_column(
        ForeignKey("systemknoten.id"), nullable=False
    )
    # Zuordnung zu einem Fahrzeug (Nutzer-Vorgabe): jedes Objekt (Rucksack,
    # Gerät, Anhänger, ...) gehört zu jedem Zeitpunkt höchstens einem Fahrzeug
    # (1:n), NULL = eigenständig/keinem Fahrzeug zugeordnet. Selbstreferenzierend,
    # kein eigener Fahrzeug-Objekttyp nötig (Objekt bleibt generisch, Karte 11).
    fahrzeug_id: Mapped[int | None] = mapped_column(
        ForeignKey("objekt.id", ondelete="SET NULL")
    )
    # Personal-Modul (Migration 0012): Zuordnung zu einer Organisationseinheit
    # (Zug/Gruppe), additiv/optional wie fahrzeug_id.
    einheit_id: Mapped[int | None] = mapped_column(ForeignKey("einheit.id"))