from collections.abc import Sequence
from typing import Any

from pydantic import ConfigDict

from models.collection import Collection, SmartCollection

from .base import BaseModel, UTCDatetime


class BaseCollectionSchema(BaseModel):
    name: str
    description: str
    rom_ids: set[int]
    rom_count: int
    path_cover_small: str | None
    path_cover_large: str | None
    path_covers_small: list[str]
    path_covers_large: list[str]
    is_public: bool = False
    is_favorite: bool = False
    is_virtual: bool = False
    is_smart: bool = False
    created_at: UTCDatetime
    updated_at: UTCDatetime


class CollectionSchema(BaseCollectionSchema):
    model_config = ConfigDict(from_attributes=True)

    id: int
    url_cover: str | None
    user_id: int
    owner_username: str

    @classmethod
    def for_user(
        cls, user_id: int, collections: Sequence["Collection"]
    ) -> list["CollectionSchema"]:
        return [
            cls.model_validate(c)
            for c in collections
            if c.user_id == user_id or c.is_public
        ]


class VirtualCollectionSchema(BaseCollectionSchema):
    model_config = ConfigDict(from_attributes=True)

    id: str
    type: str
    is_virtual: bool = True


class SmartCollectionSchema(BaseCollectionSchema):
    model_config = ConfigDict(from_attributes=True)

    id: int
    name: str
    description: str = ""
    filter_criteria: dict[str, Any]
    filter_summary: str
    user_id: int
    owner_username: str
    is_smart: bool = True

    @classmethod
    def for_user(
        cls, user_id: int, smart_collections: Sequence["SmartCollection"]
    ) -> list["SmartCollectionSchema"]:
        """Filter smart collections visible to user and create schemas."""
        return [
            cls.model_validate(c)
            for c in smart_collections
            if c.user_id == user_id or c.is_public
        ]