from datetime import date, datetime
from sqlalchemy import Boolean, Integer, Date, DateTime, Text, Numeric, String, ForeignKey, UniqueConstraint, Index, func
from sqlalchemy.orm import Mapped, mapped_column
from database import Base


class User(Base):
    __tablename__ = "users"

    id:              Mapped[int]      = mapped_column(Integer, primary_key=True, autoincrement=True)
    username:        Mapped[str]      = mapped_column(String(64), unique=True, nullable=False, index=True)
    hashed_password: Mapped[str]      = mapped_column(String(255), nullable=False)
    is_admin:        Mapped[bool]     = mapped_column(Boolean, default=False, nullable=False)
    is_disabled:     Mapped[bool]     = mapped_column(Boolean, default=False, nullable=False)
    timezone:        Mapped[str]      = mapped_column(String(64), nullable=False, default='UTC')
    created_at:      Mapped[datetime] = mapped_column(DateTime, server_default=func.now())


class EggCollection(Base):
    __tablename__ = "egg_collections"
    __table_args__ = (UniqueConstraint("user_id", "date", name="uq_user_date"),)

    id:         Mapped[int]      = mapped_column(Integer, primary_key=True, autoincrement=True)
    user_id:    Mapped[int]      = mapped_column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True)
    date:       Mapped[date]     = mapped_column(Date, nullable=False, index=True)
    eggs:       Mapped[int]      = mapped_column(Integer, nullable=False)
    notes:      Mapped[str]      = mapped_column(Text, nullable=True)
    created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())


class FlockHistory(Base):
    __tablename__ = "flock_history"
    __table_args__ = (Index("ix_flock_history_user_date", "user_id", "date"),)

    id:            Mapped[int]      = mapped_column(Integer, primary_key=True, autoincrement=True)
    user_id:       Mapped[int]      = mapped_column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True)
    date:          Mapped[date]     = mapped_column(Date, nullable=False, index=True)
    chicken_count: Mapped[int]      = mapped_column(Integer, nullable=False)
    notes:         Mapped[str]      = mapped_column(Text, nullable=True)
    created_at:    Mapped[datetime] = mapped_column(DateTime, server_default=func.now())


class FeedPurchase(Base):
    __tablename__ = "feed_purchases"
    __table_args__ = (Index("ix_feed_purchases_user_date", "user_id", "date"),)

    id:            Mapped[int]      = mapped_column(Integer, primary_key=True, autoincrement=True)
    user_id:       Mapped[int]      = mapped_column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True)
    date:          Mapped[date]     = mapped_column(Date, nullable=False, index=True)
    bags:          Mapped[float]    = mapped_column(Numeric(5, 2), nullable=False)
    price_per_bag: Mapped[float]    = mapped_column(Numeric(10, 2), nullable=False)
    notes:         Mapped[str]      = mapped_column(Text, nullable=True)
    created_at:    Mapped[datetime] = mapped_column(DateTime, server_default=func.now())


class OtherPurchase(Base):
    __tablename__ = "other_purchases"
    __table_args__ = (Index("ix_other_purchases_user_date", "user_id", "date"),)

    id:         Mapped[int]      = mapped_column(Integer, primary_key=True, autoincrement=True)
    user_id:    Mapped[int]      = mapped_column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True)
    date:       Mapped[date]     = mapped_column(Date, nullable=False, index=True)
    total:      Mapped[float]    = mapped_column(Numeric(10, 2), nullable=False)
    notes:      Mapped[str]      = mapped_column(Text, nullable=True)
    created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())