"""
Конфигурация для rate limiting
"""

from dataclasses import dataclass
from typing import Optional


@dataclass
class RateLimitSettings:
    """Настройки rate limiting для разных типов сообщений"""

    # Основные настройки
    messages_per_second: float = 0.5  # Максимум 0.5 сообщений в секунду на чат
    burst_limit: int = 2  # Максимум 2 сообщения подряд
    retry_after_multiplier: float = 1.5  # Множитель для увеличения задержки при retry
    max_retry_delay: float = 30.0  # Максимальная задержка между попытками
    max_retries: int = 3  # Максимальное количество повторных попыток

    # Специальные настройки для разных типов сообщений
    voice_message_delay: float = 2.0  # Дополнительная задержка для голосовых сообщений
    media_message_delay: float = 1.5  # Дополнительная задержка для медиа сообщений
    text_message_delay: float = 1.0  # Дополнительная задержка для текстовых сообщений

    # Настройки для разных типов чатов
    private_chat_multiplier: float = 1.0  # Множитель для приватных чатов
    group_chat_multiplier: float = 0.8  # Множитель для групповых чатов
    channel_multiplier: float = 0.6  # Множитель для каналов

    # Глобальные ограничения
    global_messages_per_second: float = (
        10.0  # Максимум 10 сообщений в секунду глобально
    )
    global_burst_limit: int = 20  # Максимум 20 сообщений подряд глобально


# Конфигурации для разных сценариев использования
DEVELOPMENT_CONFIG = RateLimitSettings(
    messages_per_second=1.0,  # Более мягкие ограничения для разработки
    burst_limit=3,
    retry_after_multiplier=1.2,
    max_retry_delay=15.0,
    max_retries=2,
)

PRODUCTION_CONFIG = RateLimitSettings(
    messages_per_second=0.5,  # Строгие ограничения для продакшена
    burst_limit=2,
    retry_after_multiplier=1.5,
    max_retry_delay=30.0,
    max_retries=3,
    voice_message_delay=2.5,
    media_message_delay=2.0,
    text_message_delay=1.5,
)

STRICT_CONFIG = RateLimitSettings(
    messages_per_second=0.3,  # Очень строгие ограничения
    burst_limit=1,
    retry_after_multiplier=2.0,
    max_retry_delay=60.0,
    max_retries=5,
    voice_message_delay=3.0,
    media_message_delay=2.5,
    text_message_delay=2.0,
)


def get_rate_limit_config(environment: str = "production") -> RateLimitSettings:
    """
    Получает конфигурацию rate limiting в зависимости от окружения

    Args:
        environment: Окружение ('development', 'production', 'strict')

    Returns:
        RateLimitSettings: Конфигурация для указанного окружения
    """
    configs = {
        "development": DEVELOPMENT_CONFIG,
        "production": PRODUCTION_CONFIG,
        "strict": STRICT_CONFIG,
    }

    return configs.get(environment, PRODUCTION_CONFIG)


def get_adaptive_config(
    current_error_rate: float, base_config: Optional[RateLimitSettings] = None
) -> RateLimitSettings:
    """
    Получает адаптивную конфигурацию на основе текущего уровня ошибок

    Args:
        current_error_rate: Текущий уровень ошибок (0.0 - 1.0)
        base_config: Базовая конфигурация

    Returns:
        RateLimitSettings: Адаптированная конфигурация
    """
    if base_config is None:
        base_config = PRODUCTION_CONFIG

    # Если уровень ошибок высокий, ужесточаем ограничения
    if current_error_rate > 0.1:  # Более 10% ошибок
        return RateLimitSettings(
            messages_per_second=base_config.messages_per_second * 0.5,
            burst_limit=max(1, base_config.burst_limit - 1),
            retry_after_multiplier=base_config.retry_after_multiplier * 1.5,
            max_retry_delay=base_config.max_retry_delay * 1.5,
            max_retries=base_config.max_retries + 1,
            voice_message_delay=base_config.voice_message_delay * 1.5,
            media_message_delay=base_config.media_message_delay * 1.3,
            text_message_delay=base_config.text_message_delay * 1.2,
        )

    # Если уровень ошибок низкий, можно немного ослабить ограничения
    elif current_error_rate < 0.01:  # Менее 1% ошибок
        return RateLimitSettings(
            messages_per_second=base_config.messages_per_second * 1.2,
            burst_limit=base_config.burst_limit + 1,
            retry_after_multiplier=base_config.retry_after_multiplier * 0.9,
            max_retry_delay=base_config.max_retry_delay * 0.8,
            max_retries=max(1, base_config.max_retries - 1),
            voice_message_delay=base_config.voice_message_delay * 0.8,
            media_message_delay=base_config.media_message_delay * 0.9,
            text_message_delay=base_config.text_message_delay * 0.9,
        )

    # Возвращаем базовую конфигурацию
    return base_config