from typing import Optional

from database.repositories.admin_repository import AdminRepository
from database.repositories.audio_repository import AudioRepository
from database.repositories.blacklist_history_repository import (
    BlacklistHistoryRepository,
)
from database.repositories.blacklist_repository import BlacklistRepository
from database.repositories.message_repository import MessageRepository
from database.repositories.post_repository import PostRepository
from database.repositories.user_repository import UserRepository


class RepositoryFactory:
    """Фабрика для создания репозиториев."""

    def __init__(self, db_path: str):
        self.db_path = db_path
        self._user_repo: Optional[UserRepository] = None
        self._blacklist_repo: Optional[BlacklistRepository] = None
        self._blacklist_history_repo: Optional[BlacklistHistoryRepository] = None
        self._message_repo: Optional[MessageRepository] = None
        self._post_repo: Optional[PostRepository] = None
        self._admin_repo: Optional[AdminRepository] = None
        self._audio_repo: Optional[AudioRepository] = None

    @property
    def users(self) -> UserRepository:
        """Возвращает репозиторий пользователей."""
        if self._user_repo is None:
            self._user_repo = UserRepository(self.db_path)
        return self._user_repo

    @property
    def blacklist(self) -> BlacklistRepository:
        """Возвращает репозиторий черного списка."""
        if self._blacklist_repo is None:
            self._blacklist_repo = BlacklistRepository(self.db_path)
        return self._blacklist_repo

    @property
    def blacklist_history(self) -> BlacklistHistoryRepository:
        """Возвращает репозиторий истории банов/разбанов."""
        if self._blacklist_history_repo is None:
            self._blacklist_history_repo = BlacklistHistoryRepository(self.db_path)
        return self._blacklist_history_repo

    @property
    def messages(self) -> MessageRepository:
        """Возвращает репозиторий сообщений."""
        if self._message_repo is None:
            self._message_repo = MessageRepository(self.db_path)
        return self._message_repo

    @property
    def posts(self) -> PostRepository:
        """Возвращает репозиторий постов."""
        if self._post_repo is None:
            self._post_repo = PostRepository(self.db_path)
        return self._post_repo

    @property
    def admins(self) -> AdminRepository:
        """Возвращает репозиторий администраторов."""
        if self._admin_repo is None:
            self._admin_repo = AdminRepository(self.db_path)
        return self._admin_repo

    @property
    def audio(self) -> AudioRepository:
        """Возвращает репозиторий аудио."""
        if self._audio_repo is None:
            self._audio_repo = AudioRepository(self.db_path)
        return self._audio_repo

    async def create_all_tables(self):
        """Создает все таблицы в базе данных."""
        await self.users.create_tables()
        await self.blacklist.create_tables()
        await self.blacklist_history.create_tables()
        await self.messages.create_tables()
        await self.posts.create_tables()
        await self.admins.create_tables()
        await self.audio.create_tables()

    async def check_database_integrity(self):
        """Проверяет целостность базы данных."""
        await self.users.check_database_integrity()

    async def cleanup_wal_files(self):
        """Очищает WAL файлы."""
        await self.users.cleanup_wal_files()