from pydantic import BaseModel, field_validator
from typing import Optional

from ..lib.categories import VALID_CATEGORY_CODES


class ProfileBase(BaseModel):
    phone: Optional[str] = None
    bio: Optional[str] = None
    avatar_url: Optional[str] = None
    topic_preferences: Optional[str] = None

    @field_validator("topic_preferences")
    @classmethod
    def validate_topic_preferences(cls, v: Optional[str]) -> Optional[str]:
        if v is None or v.strip() == "":
            return None
        tags = [t.strip() for t in v.split(",") if t.strip()]
        invalid = [t for t in tags if t not in VALID_CATEGORY_CODES]
        if invalid:
            raise ValueError(
                f"Invalid arXiv categories: {', '.join(invalid)}. "
                f"Must be valid CS/stat/eess categories (e.g. cs.AI, stat.ML)."
            )
        return ",".join(tags)


class ProfileCreate(ProfileBase):
    pass


class ProfileUpdate(ProfileBase):
    pass


class ProfileResponse(BaseModel):
    id: int
    user_id: int
    phone: Optional[str] = None
    bio: Optional[str] = None
    avatar_url: Optional[str] = None
    topic_preferences: Optional[str] = None

    class Config:
        from_attributes = True