from pydantic import BaseModel, field_validator, Field
from kawai_focus_v2.core.messages.errors import ErrorMessage


class TimerModel(BaseModel):
    """Модель схемы данных таймера"""

    id: int
    title: str
    pomodoro_time: int
    break_time: int
    break_long_time: int
    count_pomodoro: int


class NewTimerModel(BaseModel):
    """Модель схемы нового таймера"""

    title: str
    pomodoro_time: int
    break_time: int
    break_long_time: int
    count_pomodoro: int


class UpdateTimerModel(BaseModel):
    """Модель схемы обновления данных таймера"""

    title: str | None = None
    pomodoro_time: int | None = None
    break_time: int | None = None
    break_long_time: int | None = None
    count_pomodoro: int | None = None


class TimerListModel(BaseModel):
    """Модель схемы данных таймера для списка"""

    id: int
    title: str
    pomodoro_time: int
    count_pomodoro: int


class TimerTimeModel(BaseModel):
    """Модель схемы данных времени таймера"""

    hh: int = Field(0, ge=0, le=23)
    mm: int = Field(0, ge=0, le=59)

    @field_validator('hh', 'mm')
    @classmethod
    def check_all_time_fields(cls, value: int) -> int:
        """Метод валидирует все поля времени"""

        # гарантирует, что время не равно 00:00:00
        if value == 0:
            raise ValueError(ErrorMessage.NO_TIME.value)

        return value