from datetime import datetime

from app.models.model import UTCBaseModel
from app.models.score import GameMode
from app.utils import utcnow

from .beatmap import Beatmap
from .lazer_user import User

from sqlalchemy import Column, DateTime, Index
from sqlalchemy.orm import Mapped
from sqlmodel import BigInteger, Field, ForeignKey, Relationship, SQLModel


class ScoreTokenBase(SQLModel, UTCBaseModel):
    score_id: int | None = Field(sa_column=Column(BigInteger), default=None)
    ruleset_id: GameMode
    playlist_item_id: int | None = Field(default=None)  # playlist
    created_at: datetime = Field(default_factory=utcnow, sa_column=Column(DateTime))
    updated_at: datetime = Field(default_factory=utcnow, sa_column=Column(DateTime))


class ScoreToken(ScoreTokenBase, table=True):
    __tablename__: str = "score_tokens"
    __table_args__ = (Index("idx_user_playlist", "user_id", "playlist_item_id"),)

    id: int | None = Field(
        default=None,
        sa_column=Column(
            BigInteger,
            primary_key=True,
            index=True,
            autoincrement=True,
        ),
    )
    user_id: int = Field(sa_column=Column(BigInteger, ForeignKey("lazer_users.id")))
    beatmap_id: int = Field(foreign_key="beatmaps.id")
    user: Mapped[User] = Relationship()
    beatmap: Mapped[Beatmap] = Relationship()


class ScoreTokenResp(ScoreTokenBase):
    id: int
    user_id: int
    beatmap_id: int

    @classmethod
    def from_db(cls, obj: ScoreToken) -> "ScoreTokenResp":
        return cls.model_validate(obj)