import uuid
from datetime import datetime
from sqlalchemy import Column, String, Integer, Boolean, DateTime, ForeignKey, UniqueConstraint, CheckConstraint, Enum as SqlEnum
from sqlalchemy.orm import relationship
from utils.utils import Base


class Aircraft(Base):
    __tablename__ = "aircraft"

    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    registration_number = Column(String(50), nullable=False, unique=True, index=True)
    model = Column(String(100), nullable=False)
    manufacturer = Column(String(100), nullable=False)
    total_seats = Column(Integer, nullable=False)
    year_manufactured = Column(Integer, nullable=True)
    status = Column(String(50), nullable=False, default="Active", index=True)
    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)

    seats = relationship("Seat", back_populates="aircraft", cascade="all, delete-orphan", passive_deletes=True)
    flights = relationship("Flight", back_populates="aircraft")

    __table_args__ = (
        CheckConstraint("total_seats > 0", name="ck_aircraft_total_seats_positive"),
        CheckConstraint("year_manufactured IS NULL OR year_manufactured > 1900", name="ck_aircraft_year_valid"),
    )


class Seat(Base):
    __tablename__ = "seats"

    id = Column(String(36), primary_key=True, default=lambda: str(uuid.uuid4()))
    aircraft_id = Column(String(36), ForeignKey("aircraft.id", ondelete="CASCADE"), nullable=False, index=True)
    seat_number = Column(String(10), nullable=False)
    class_type = Column(String(50), nullable=False)
    is_window = Column(Boolean, nullable=False, default=False)
    is_aisle = Column(Boolean, nullable=False, default=False)
    is_exit_row = Column(Boolean, nullable=False, default=False)
    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)

    aircraft = relationship("Aircraft", back_populates="seats")
    bookings = relationship("Booking", back_populates="seat")

    __table_args__ = (
        UniqueConstraint("aircraft_id", "seat_number", name="uq_aircraft_seat_number"),
    )