from sqlalchemy import Column, String, DateTime, Boolean, Text
from datetime import datetime
from backend.database import Base


class Emitter(Base):
    __tablename__ = "emitters"

    id = Column(String, primary_key=True, index=True)
    unit_type = Column(String, nullable=False)
    last_seen = Column(DateTime, default=datetime.utcnow)
    last_file = Column(String, nullable=False)
    status = Column(String, nullable=False)
    notes = Column(String, nullable=True)


class RosterUnit(Base):
    """
    Roster table: represents our *intended assignment* of a unit.
    This is editable from the GUI.
    """
    __tablename__ = "roster"

    id = Column(String, primary_key=True, index=True)
    unit_type = Column(String, default="series3")
    deployed = Column(Boolean, default=True)
    retired = Column(Boolean, default=False)
    note = Column(String, nullable=True)
    project_id = Column(String, nullable=True)
    location = Column(String, nullable=True)
    last_updated = Column(DateTime, default=datetime.utcnow)