import datetime
import enum

from models.assets import Save, Screenshot, State
from models.base import BaseModel
from sqlalchemy import Boolean, Column, DateTime, Enum, Integer, String
from sqlalchemy.orm import Mapped, relationship
from starlette.authentication import SimpleUser


class Role(enum.Enum):
    VIEWER = "viewer"
    EDITOR = "editor"
    ADMIN = "admin"


class User(BaseModel, SimpleUser):
    from models.rom import RomNote

    __tablename__ = "users"
    __table_args__ = {"extend_existing": True}

    id = Column(Integer(), primary_key=True, autoincrement=True)

    username: str = Column(String(length=255), unique=True, index=True)
    hashed_password: str = Column(String(length=255))
    enabled: bool = Column(Boolean(), default=True)
    role: Role = Column(Enum(Role), default=Role.VIEWER)
    avatar_path: str = Column(String(length=255), default="")
    last_login: Mapped[datetime] = Column(DateTime(timezone=True), nullable=True)
    last_active: Mapped[datetime] = Column(DateTime(timezone=True), nullable=True)

    saves: Mapped[list[Save]] = relationship(
        "Save",
        lazy="selectin",
        back_populates="user",
    )
    states: Mapped[list[State]] = relationship(
        "State", lazy="selectin", back_populates="user"
    )
    screenshots: Mapped[list[Screenshot]] = relationship(
        "Screenshot", lazy="selectin", back_populates="user"
    )
    notes: Mapped[list[RomNote]] = relationship(
        "RomNote", lazy="selectin", back_populates="user"
    )

    @property
    def oauth_scopes(self):
        from handler.auth.base_handler import DEFAULT_SCOPES, FULL_SCOPES, WRITE_SCOPES

        if self.role == Role.ADMIN:
            return FULL_SCOPES

        if self.role == Role.EDITOR:
            return WRITE_SCOPES

        return DEFAULT_SCOPES

    @property
    def fs_safe_folder_name(self):
        # Uses the ID to avoid issues with username changes
        return f"User:{self.id}".encode("utf-8").hex()

    def set_last_active(self):
        from handler.database import db_user_handler

        db_user_handler.update_user(self.id, {"last_active": datetime.datetime.now()})