from __future__ import annotations

from typing import Literal, Optional
from pydantic import BaseModel, Field


class BanNotificationRequest(BaseModel):
    """Запрос на отправку уведомления о бане пользователю"""

    notification_type: Literal["punishment", "enabled", "warning", "network_wifi", "network_mobile"] = Field(
        description="Тип уведомления: punishment (бан за устройства), enabled (разбан), warning (предупреждение), network_wifi (бан за WiFi), network_mobile (бан за мобильную сеть)"
    )
    user_identifier: str = Field(
        description="Email или user_id пользователя из Remnawave Panel"
    )
    username: str = Field(
        description="Имя пользователя для отображения"
    )

    # Данные для punishment
    ip_count: Optional[int] = Field(None, description="Количество устройств")
    limit: Optional[int] = Field(None, description="Лимит устройств")
    ban_minutes: Optional[int] = Field(None, description="Длительность бана в минутах")

    # Данные для warning
    warning_message: Optional[str] = Field(None, description="Текст предупреждения")

    # Данные для network_wifi/network_mobile и punishment
    network_type: Optional[str] = Field(None, description="Тип сети (WiFi/Mobile)")
    node_name: Optional[str] = Field(None, description="Название ноды/сервера с которой пришел бан")

    class Config:
        json_schema_extra = {
            "example": {
                "notification_type": "punishment",
                "user_identifier": "user@example.com",
                "username": "john_doe",
                "ip_count": 5,
                "limit": 3,
                "ban_minutes": 30,
                "node_name": "DE-Server-1"
            }
        }


class BanNotificationResponse(BaseModel):
    """Ответ на запрос отправки уведомления"""

    success: bool = Field(description="Успешно ли отправлено уведомление")
    message: str = Field(description="Сообщение о результате")
    telegram_id: Optional[int] = Field(None, description="Telegram ID получателя")
    sent: bool = Field(False, description="Было ли фактически отправлено сообщение")

    class Config:
        json_schema_extra = {
            "example": {
                "success": True,
                "message": "Уведомление отправлено",
                "telegram_id": 123456789,
                "sent": True
            }
        }