"""
Высокоуровневый менеджер для работы с банвордами.
Упрощает использование repository в handlers и middleware.
"""
from typing import Set, Optional, List, Dict, Any
from datetime import datetime

from middleware.loggers import logger
from .database import Database, get_db
from .repository import BanWordsRepository
from .models import BanWordType, SpamStat, SpamLog

from sqlalchemy import select, delete, func, desc

__all__ = ("BanWordsManager", "get_manager")


class BanWordsManager:
    """
    Менеджер для удобной работы с банвордами.

    Предоставляет упрощённый API для handlers и middleware.

    Attributes:
        db: Экземпляр Database
        repo: Repository для CRUD операций
    """

    def __init__(self, db: Optional[Database] = None):
        """
        Args:
            db: Экземпляр Database (если None, берётся глобальный)
        """
        self.db = db or get_db()
        self.repo = BanWordsRepository(self.db)

        # Кэш для часто используемых данных
        self._cache_banwords: Optional[dict] = None
        self._cache_whitelist: Optional[Set[str]] = None
        self._cache_admins: Optional[Set[int]] = None
        self._cache_updated_at: Optional[datetime] = None

    async def init(self) -> None:
        """Инициализирует базу данных и загружает кэш"""
        await self.db.init()
        await self.refresh_cache()
        logger.info("BanWordsManager инициализирован", log_type="DATABASE")

    async def close(self) -> None:
        """Закрывает соединение с БД"""
        await self.db.close()

    # === CACHE MANAGEMENT ===

    async def refresh_cache(self) -> None:
        """Обновляет кэш из БД"""
        try:
            self._cache_banwords = await self.repo.get_all_banwords()
            temp_banwords = await self.repo.get_all_temp_banwords()

            # Объединяем постоянные и временные банворды
            for word_type, words in temp_banwords.items():
                if word_type in self._cache_banwords:
                    self._cache_banwords[word_type] |= words

            self._cache_whitelist = await self.repo.get_whitelist()
            self._cache_admins = await self.repo.get_admins()
            self._cache_updated_at = datetime.now()

            logger.debug("Кэш банвордов обновлён", log_type="DATABASE")

        except Exception as e:
            logger.error(f"Ошибка обновления кэша: {e}", log_type="DATABASE")

    def invalidate_cache(self) -> None:
        """Сбрасывает кэш (требует refresh_cache)"""
        self._cache_banwords = None
        self._cache_whitelist = None
        self._cache_admins = None
        self._cache_updated_at = None

    # === BANWORDS (с кэшем) ===

    async def add_banword(
            self,
            word: str,
            word_type: BanWordType,
            added_by: Optional[int] = None,
            reason: Optional[str] = None,
            refresh_cache: bool = True
    ) -> bool:
        """
        Добавляет банворд и обновляет кэш.

        Args:
            word: Слово
            word_type: Тип
            added_by: ID админа
            reason: Причина
            refresh_cache: Обновить кэш после добавления

        Returns:
            bool: True если добавлен
        """
        result = await self.repo.add_banword(word, word_type, added_by, reason)

        if result and refresh_cache:
            await self.refresh_cache()

        return result

    async def remove_banword(
            self,
            word: str,
            word_type: BanWordType,
            refresh_cache: bool = True
    ) -> bool:
        """Удаляет банворд и обновляет кэш"""
        result = await self.repo.remove_banword(word, word_type)

        if result and refresh_cache:
            await self.refresh_cache()

        return result

    def get_banwords_cached(self, word_type: BanWordType) -> Set[str]:
        """
        Получает банворды из кэша (быстро).

        Args:
            word_type: Тип банвордов

        Returns:
            Set[str]: Набор слов из кэша
        """
        if self._cache_banwords is None:
            logger.warning("Кэш не инициализирован", log_type="DATABASE")
            return set()

        return self._cache_banwords.get(word_type, set())

    async def get_banwords(self, word_type: BanWordType) -> Set[str]:
        """Получает банворды напрямую из БД (без кэша)"""
        return await self.repo.get_banwords(word_type)

    # === TEMPORARY BANWORDS ===

    async def add_temp_banword(
            self,
            word: str,
            word_type: BanWordType,
            minutes: int,
            added_by: Optional[int] = None,
            refresh_cache: bool = True
    ) -> bool:
        """Добавляет временный банворд"""
        result = await self.repo.add_temp_banword(
            word, word_type, minutes, added_by
        )

        if result and refresh_cache:
            await self.refresh_cache()

        return result

    async def remove_temp_banword(
            self,
            word: str,
            word_type: BanWordType,
            refresh_cache: bool = True
    ) -> bool:
        """Удаляет временный банворд"""
        result = await self.repo.remove_temp_banword(word, word_type)

        if result and refresh_cache:
            await self.refresh_cache()

        return result

    async def cleanup_expired(self) -> int:
        """
        Очищает истёкшие временные банворды.
        Вызывается периодически (например, раз в минуту).

        Returns:
            int: Количество удалённых записей
        """
        deleted = await self.repo.cleanup_expired_temp_banwords()

        if deleted > 0:
            await self.refresh_cache()

        return deleted

    # === WHITELIST ===

    async def add_whitelist(
            self,
            word: str,
            added_by: Optional[int] = None,
            reason: Optional[str] = None,
            refresh_cache: bool = True
    ) -> bool:
        """Добавляет слово в белый список"""
        result = await self.repo.add_whitelist(word, added_by, reason)

        if result and refresh_cache:
            await self.refresh_cache()

        return result

    async def remove_whitelist(
            self,
            word: str,
            refresh_cache: bool = True
    ) -> bool:
        """Удаляет слово из белого списка"""
        result = await self.repo.remove_whitelist(word)

        if result and refresh_cache:
            await self.refresh_cache()

        return result

    def get_whitelist_cached(self) -> Set[str]:
        """Получает белый список из кэша"""
        if self._cache_whitelist is None:
            logger.warning("Кэш whitelist не инициализирован", log_type="DATABASE")
            return set()

        return self._cache_whitelist

    def is_whitelisted(self, text: str) -> bool:
        """
        Проверяет, содержит ли текст слово из белого списка.

        Args:
            text: Текст для проверки (lowercase)

        Returns:
            bool: True если найдено исключение
        """
        whitelist = self.get_whitelist_cached()
        return any(word in text for word in whitelist)

    # === ADMINS ===

    async def add_admin(
            self,
            user_id: int,
            added_by: Optional[int] = None,
            refresh_cache: bool = True
    ) -> bool:
        """Добавляет администратора"""
        result = await self.repo.add_admin(user_id, added_by)

        if result and refresh_cache:
            await self.refresh_cache()

        return result

    async def remove_admin(
            self,
            user_id: int,
            refresh_cache: bool = True
    ) -> bool:
        """Удаляет администратора"""
        result = await self.repo.remove_admin(user_id)

        if result and refresh_cache:
            await self.refresh_cache()

        return result

    def get_admins_cached(self) -> Set[int]:
        """Получает список админов из кэша"""
        if self._cache_admins is None:
            logger.warning("Кэш админов не инициализирован", log_type="DATABASE")
            return set()

        return self._cache_admins

    def is_admin_cached(self, user_id: int) -> bool:
        """
        Проверяет, является ли пользователь админом (из кэша).

        Args:
            user_id: Telegram ID

        Returns:
            bool: True если админ
        """
        return user_id in self.get_admins_cached()

    async def is_admin(self, user_id: int) -> bool:
        """Проверяет админа напрямую из БД"""
        return await self.repo.is_admin(user_id)

    # === SETTINGS (режимы silence/conflict) ===

    async def set_silence_mode(self, minutes: int) -> datetime:
        """
        Включает режим тишины на указанное время.

        Args:
            minutes: Длительность в минутах

        Returns:
            datetime: Время окончания режима
        """
        expires_at = datetime.now().timestamp() + (minutes * 60)
        await self.repo.set_setting("silence_until", str(expires_at))

        logger.info(
            f"Режим тишины активирован на {minutes} мин",
            log_type="SILENCE"
        )

        return datetime.fromtimestamp(expires_at)

    async def disable_silence_mode(self) -> None:
        """Отключает режим тишины"""
        await self.repo.delete_setting("silence_until")
        logger.info("Режим тишины отключён", log_type="SILENCE")

    async def is_silence_active(self) -> bool:
        """Проверяет, активен ли режим тишины"""
        silence_until_str = await self.repo.get_setting("silence_until")

        if not silence_until_str:
            return False

        try:
            silence_until = float(silence_until_str)
            now = datetime.now().timestamp()

            if now >= silence_until:
                # Время истекло - удаляем настройку
                await self.disable_silence_mode()
                return False

            return True

        except (ValueError, TypeError):
            return False

    async def set_conflict_mode(self, minutes: int) -> datetime:
        """
        Включает режим антиконфликта на указанное время.

        Args:
            minutes: Длительность в минутах

        Returns:
            datetime: Время окончания режима
        """
        expires_at = datetime.now().timestamp() + (minutes * 60)
        await self.repo.set_setting("conflict_until", str(expires_at))

        logger.info(
            f"Режим антиконфликта активирован на {minutes} мин",
            log_type="CONFLICT"
        )

        return datetime.fromtimestamp(expires_at)

    async def disable_conflict_mode(self) -> None:
        """Отключает режим антиконфликта"""
        await self.repo.delete_setting("conflict_until")
        logger.info("Режим антиконфликта отключён", log_type="CONFLICT")

    async def is_conflict_active(self) -> bool:
        """Проверяет, активен ли режим антиконфликта"""
        conflict_until_str = await self.repo.get_setting("conflict_until")

        if not conflict_until_str:
            return False

        try:
            conflict_until = float(conflict_until_str)
            now = datetime.now().timestamp()

            if now >= conflict_until:
                # Время истекло
                await self.disable_conflict_mode()
                return False

            return True

        except (ValueError, TypeError):
            return False

    # === STATISTICS ===

    async def log_spam(
            self,
            user_id: int,
            username: str,
            chat_id: int,
            message_text: str,
            matched_word: str,
            match_type: str
    ) -> None:
        """Логирует удаление спам-сообщения"""
        await self.repo.log_spam_deletion(
            user_id=user_id,
            username=username,
            chat_id=chat_id,
            message_text=message_text,
            matched_word=matched_word,
            match_type=match_type
        )

    async def get_spam_stats(
            self,
            limit: int = 100,
            user_id: Optional[int] = None
    ) -> List[SpamStat]:
        """Получает статистику удалений"""
        return await self.repo.get_spam_stats(limit, user_id)

    async def get_user_spam_count(self, user_id: int) -> int:
        """Получает количество удалённых сообщений пользователя"""
        return await self.repo.get_user_spam_count(user_id)

    async def get_top_spammers(self, limit: int = 10) -> List[tuple[int, int]]:
        """Получает топ спамеров"""
        return await self.repo.get_top_spammers(limit)

    # === INFO ===

    async def get_stats(self) -> dict:
        """Получает общую статистику"""
        db_stats = await self.repo.get_stats()

        # Добавляем информацию о кэше
        cache_info = {
            'cache_active': self._cache_banwords is not None,
            'cache_updated_at': self._cache_updated_at.isoformat() if self._cache_updated_at else None
        }

        return {**db_stats, **cache_info}

    async def get_all_words_list(self) -> dict:
        """
        Получает все слова для команды /listwords.

        Returns:
            dict: Словарь со всеми категориями слов
        """
        banwords = await self.repo.get_all_banwords()
        temp_banwords = await self.repo.get_all_temp_banwords()
        whitelist = await self.repo.get_whitelist()
        admins = await self.repo.get_admins()

        return {
            'substring': banwords.get(BanWordType.SUBSTRING, set()),
            'lemma': banwords.get(BanWordType.LEMMA, set()),
            'part': banwords.get(BanWordType.PART, set()),
            'conflict_substring': banwords.get(BanWordType.CONFLICT_SUBSTRING, set()),
            'conflict_lemma': banwords.get(BanWordType.CONFLICT_LEMMA, set()),
            'temp_substring': temp_banwords.get(BanWordType.SUBSTRING, set()),
            'temp_lemma': temp_banwords.get(BanWordType.LEMMA, set()),
            'whitelist': whitelist,
            'admins': admins
        }

    async def get_top_words(self, limit: int = 10) -> List[Dict[str, Any]]:
        """
        Получает топ N самых часто срабатывающих слов.

        Args:
            limit: Количество слов в топе

        Returns:
            List[Dict]: Список словарей с данными:
                - word: слово
                - count: количество срабатываний
                - type: тип проверки
        """
        async with self.session_maker() as session:
            try:
                # Группируем по matched_word и считаем количество
                query = select(
                    SpamLog.matched_word,
                    SpamLog.match_type,
                    func.count(SpamLog.id).label('count')
                ).where(
                    SpamLog.matched_word.isnot(None)
                ).group_by(
                    SpamLog.matched_word,
                    SpamLog.match_type
                ).order_by(
                    desc('count')
                ).limit(limit)

                result = await session.execute(query)
                rows = result.all()

                # Форматируем результат
                top_words = []
                for row in rows:
                    top_words.append({
                        'word': row.matched_word,
                        'type': row.match_type,
                        'count': row.count
                    })

                logger.debug(
                    f"Получен топ-{limit} слов: {len(top_words)} записей",
                    log_type="DATABASE"
                )

                return top_words

            except Exception as e:
                logger.error(
                    f"Ошибка получения топ-слов: {e}",
                    log_type="DATABASE"
                )
                return []

    async def get_total_spam_count(self) -> int:
        """
        Получает общее количество удалённых сообщений.

        Returns:
            int: Количество записей в SpamLog
        """
        async with self.session_maker() as session:
            try:
                query = select(func.count(SpamLog.id))
                result = await session.execute(query)
                count = result.scalar_one()
                return count
            except Exception as e:
                logger.error(
                    f"Ошибка подсчёта спам-логов: {e}",
                    log_type="DATABASE"
                )
                return 0

    async def reset_spam_stats(self) -> bool:
        """
        Очищает всю статистику спама.

        Returns:
            bool: True если успешно
        """
        async with self.session_maker() as session:
            try:
                # Удаляем все записи
                await session.execute(delete(SpamLog))
                await session.commit()

                logger.info("Статистика спама сброшена", log_type="DATABASE")
                return True

            except Exception as e:
                logger.error(
                    f"Ошибка сброса статистики: {e}",
                    log_type="DATABASE"
                )
                await session.rollback()
                return False


# Глобальный экземпляр менеджера
_manager_instance: Optional[BanWordsManager] = None


def get_manager() -> BanWordsManager:
    """
    Возвращает глобальный экземпляр BanWordsManager (Singleton).

    Returns:
        BanWordsManager: Менеджер банвордов
    """
    global _manager_instance
    if _manager_instance is None:
        _manager_instance = BanWordsManager()
    return _manager_instance