import logging
from pathlib import Path
from typing import Optional

from pyrogram import Client

from utils.misc import modules_help
from utils.scripts import load_module


class ModuleManager:
    _instance: Optional['ModuleManager'] = None

    def __init__(self):
        self.success_modules = 0
        self.failed_modules = 0
        self.help_navigator = None

    @classmethod
    def get_instance(cls) -> 'ModuleManager':
        if cls._instance is None:
            cls._instance = ModuleManager()
        return cls._instance

    async def load_modules(self, app: Client):
        """Load all modules and initialize help navigator"""
        for path in Path('modules').rglob('*.py'):
            try:
                await load_module(path.stem, app, core='custom_modules' not in path.parent.parts)
            except Exception:
                logging.warning("Can't import module %s", path.stem, exc_info=True)
                self.failed_modules += 1
            else:
                self.success_modules += 1

        logging.info('Imported %d modules', self.success_modules)
        if self.failed_modules:
            logging.warning('Failed to import %d modules', self.failed_modules)

        self.help_navigator = HelpNavigator()
        return self.help_navigator


class HelpNavigator:
    def __init__(self):
        self.current_page = 1
        self.module_list = list(modules_help.keys())
        self.total_pages = (len(modules_help) + 9) // 10
        logging.info('Initialized HelpNavigator with %d modules', len(self.module_list))

    async def send_page(self, message):
        from utils.misc import prefix

        start_index = (self.current_page - 1) * 10
        end_index = start_index + 10
        page_modules = self.module_list[start_index:end_index]
        text = '<b>Help</b>\n'
        text += f'For more help on how to use a command, type <code>{prefix}help [module]</code>\n\n'
        text += f'Help Page No: {self.current_page}/{self.total_pages}\n\n'
        for module_name in page_modules:
            commands = modules_help[module_name]
            text += f'<b>• {module_name.title()}:</b> {", ".join([f"<code>{prefix + cmd_name.split()[0]}</code>" for cmd_name in commands])}\n'
        text += f'\n<b>The number of modules in the userbot: {len(modules_help)}</b>'
        await message.edit(text, disable_web_page_preview=True)

    def next_page(self) -> bool:
        if self.current_page < self.total_pages:
            self.current_page += 1
            return True
        return False

    def prev_page(self) -> bool:
        if self.current_page > 1:
            self.current_page -= 1
            return True
        return False