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
    )
    # Fahrzeug/Anhänger-Kopplung (Nutzer-Vorgabe): optional, beide Fälle möglich -
    # Anhänger als eigenständiges Objekt (NULL) oder fest einem Zugfahrzeug
    # zugeordnet. Selbstreferenzierend, kein eigener Anhänger-Objekttyp nötig.
    gezogen_von_objekt_id: Mapped[int | None] = mapped_column(
        ForeignKey("objekt.id", ondelete="SET NULL")
    )