from pydantic import BaseModel, ConfigDict, field_validator
from datetime import datetime
from typing import Literal, Optional

AccentColor = Literal["cyan", "blue", "green", "purple", "red", "orange", "pink", "yellow"]


def _validate_pin_length(v: str, label: str = "PIN") -> str:
    if len(v) < 4:
        raise ValueError(f'{label} must be at least 4 characters')
    if len(v) > 72:
        raise ValueError(f'{label} must be at most 72 characters')
    return v


class SettingsCreate(BaseModel):
    pin: str

    @field_validator('pin')
    @classmethod
    def pin_length(cls, v: str) -> str:
        return _validate_pin_length(v)


class SettingsUpdate(BaseModel):
    accent_color: Optional[AccentColor] = None
    upcoming_days: int | None = None
    preferred_name: str | None = None
    weather_city: str | None = None
    weather_lat: float | None = None
    weather_lon: float | None = None


class SettingsResponse(BaseModel):
    id: int
    accent_color: str
    upcoming_days: int
    preferred_name: str | None = None
    weather_city: str | None = None
    weather_lat: float | None = None
    weather_lon: float | None = None
    created_at: datetime
    updated_at: datetime

    model_config = ConfigDict(from_attributes=True)


class ChangePinRequest(BaseModel):
    old_pin: str
    new_pin: str

    @field_validator('new_pin')
    @classmethod
    def new_pin_length(cls, v: str) -> str:
        return _validate_pin_length(v, "New PIN")