from datetime import datetime

from sqlalchemy import Boolean, DateTime, Integer, String, JSON
from sqlalchemy.orm import Mapped, mapped_column

from app.enums.user import FrontendUserKind
from app.models.base import Base, TimestampMixin


class User(Base, TimestampMixin):
    __tablename__ = "users"
    __allow_unmapped__ = True

    id: Mapped[str] = mapped_column(String(32), primary_key=True)
    username: Mapped[str] = mapped_column(String(64), unique=True, index=True)
    email: Mapped[str | None] = mapped_column(String(255), unique=True, nullable=True)
    phone: Mapped[str | None] = mapped_column(String(20), unique=True, nullable=True)
    hashed_password: Mapped[str | None] = mapped_column(String(128), nullable=True)
    avatar: Mapped[str | None] = mapped_column(String(512), nullable=True)
    is_active: Mapped[bool] = mapped_column(Boolean, default=True)
    is_admin: Mapped[bool] = mapped_column(Boolean, default=False)
    user_type: Mapped[str] = mapped_column(String(16), default="frontend", index=True)
    frontend_user_kind: Mapped[str] = mapped_column(
        String(16),
        default=FrontendUserKind.EXTERNAL.value,
        server_default=FrontendUserKind.EXTERNAL.value,
        index=True,
        nullable=False,
    )
    team_id: Mapped[str | None] = mapped_column(String(32), nullable=True, index=True)
    last_login_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
    password_set_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
    first_membership_paid_at: Mapped[datetime | None] = mapped_column(
        DateTime(timezone=True), nullable=True, index=True
    )
    allowed_menus: Mapped[list | None] = mapped_column(JSON, nullable=True)
    private_portrait_asset_limit: Mapped[int] = mapped_column(
        Integer, default=50, server_default="50", nullable=False
    )

    @property
    def credits(self) -> float:
        return float(getattr(self, "_credits_snapshot", 0.0) or 0.0)

    @credits.setter
    def credits(self, value: float | int) -> None:
        self._credits_snapshot = round(float(value or 0.0), 2)

    @property
    def must_set_password(self) -> bool:
        return self.user_type == "frontend" and not self.hashed_password