from datetime import datetime, timezone

from sqlalchemy import Boolean, Column, DateTime, ForeignKey, Integer, String
from sqlalchemy.orm import relationship, validates

from app.database import Base


def _utcnow() -> datetime:
    return datetime.now(timezone.utc)


class User(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, index=True)
    username = Column(String(64), unique=True, nullable=False, index=True)
    password_hash = Column(String(128), nullable=False)
    is_admin = Column(Boolean, nullable=False, default=False)
    invited_by_code_id = Column(Integer, ForeignKey("invite_codes.id", ondelete="SET NULL"), nullable=True)
    showcase_mode = Column(String(16), nullable=False, default="profile")
    created_at = Column(DateTime, default=_utcnow)

    profiles = relationship("Profile", back_populates="user", cascade="all, delete-orphan")
    sessions = relationship("UserSession", back_populates="user", cascade="all, delete-orphan")
    invite_codes = relationship("InviteCode", back_populates="creator", cascade="all, delete-orphan")
    invited_by_code = relationship("InviteCode", foreign_keys=[invited_by_code_id])


class InviteCode(Base):
    __tablename__ = "invite_codes"

    id = Column(Integer, primary_key=True, index=True)
    code = Column(String(8), unique=True, nullable=False, index=True)
    max_uses = Column(Integer, nullable=True)  # None = unlimited
    expires_at = Column(DateTime, nullable=True)  # None = no expiration
    use_count = Column(Integer, nullable=False, default=0)
    created_by = Column(Integer, ForeignKey("users.id", ondelete="SET NULL"), nullable=True)
    created_at = Column(DateTime, default=_utcnow)

    creator = relationship("User", back_populates="invite_codes")
    registered_users = relationship("User", back_populates="invited_by_code", foreign_keys="User.invited_by_code_id")


class UserSession(Base):
    __tablename__ = "user_sessions"

    token = Column(String(64), primary_key=True)
    user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
    created_at = Column(DateTime, default=_utcnow)
    expires_at = Column(DateTime, nullable=False)

    user = relationship("User", back_populates="sessions")


class Profile(Base):
    __tablename__ = "profiles"

    id = Column(Integer, primary_key=True, index=True)
    user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
    nickname = Column(String(64), nullable=False)
    showcase_enabled = Column(default=False, nullable=False)
    showcase_visible = Column(String(16), nullable=False, default="both")
    created_at = Column(DateTime, default=_utcnow)

    user = relationship("User", back_populates="profiles")
    postcards = relationship("Postcard", back_populates="profile", cascade="all, delete-orphan")


class Postcard(Base):
    __tablename__ = "postcards"

    id = Column(Integer, primary_key=True, index=True)
    profile_id = Column(Integer, ForeignKey("profiles.id", ondelete="CASCADE"), nullable=False)
    card_number = Column(String(64), nullable=False)
    status = Column(String(16), nullable=False)
    # sent & delivered
    recipient_name = Column(String(64))
    country = Column(String(64))
    send_time = Column(DateTime)
    arrival_time = Column(DateTime)
    # received
    sender_name = Column(String(64))
    receive_time = Column(DateTime)
    # images
    showcase_hidden = Column(default=False, nullable=False)
    image_front = Column(String(256))
    image_back = Column(String(256))
    created_at = Column(DateTime, default=_utcnow)
    updated_at = Column(DateTime, default=_utcnow, onupdate=_utcnow)

    profile = relationship("Profile", back_populates="postcards")

    @validates("country")
    def _upper_country(self, key, value):
        return value.strip().upper() if value else None