import uuid
from datetime import datetime, timezone

from sqlalchemy import Column, DateTime, ForeignKey, Integer, String, event
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)
    created_at = Column(DateTime, default=_utcnow)

    api_keys = relationship("ApiKey", back_populates="user", cascade="all, delete-orphan")
    profiles = relationship("Profile", back_populates="user", cascade="all, delete-orphan")
    sessions = relationship("UserSession", back_populates="user", cascade="all, delete-orphan")


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=True)

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


class ApiKey(Base):
    __tablename__ = "api_keys"

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

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

    @staticmethod
    def generate_key() -> str:
        return uuid.uuid4().hex


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)
    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 / received
    # 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
    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