"""User settings Pydantic models."""

from pydantic import BaseModel


class EnvironmentSettings(BaseModel):
    """Settings for a specific environment (IFT/PSI/PROD)."""

    apiMode: str = "bench"
    bearerToken: str = ""
    systemPlatform: str = ""
    systemPlatformUser: str = ""
    platformUserId: str = ""
    platformId: str = ""
    withClassify: bool = False
    resetSessionMode: bool = True


class UserSettings(BaseModel):
    """User settings for all environments."""

    user_id: str
    settings: dict[str, EnvironmentSettings]  # ift, psi, prod
    updated_at: str


class UserSettingsUpdate(BaseModel):
    """Update user settings request."""

    settings: dict[str, EnvironmentSettings]