import logging
from typing import TYPE_CHECKING

from browser.chrome import ChromeDriver
from common import Singleton

if TYPE_CHECKING:
    from typing import Type, List
    from browser.IDriver import IDriver
    from config import ConfigManager
    from common.items import TabInfo

_log = logging.getLogger("BrowserManager")


class BrowserManager(metaclass=Singleton):
    _config_manager: "ConfigManager"
    _mapping: "dict[str, Type[IDriver]]" = {
        "chrome": ChromeDriver,
    }

    def __init__(self, config_manager: "ConfigManager"):
        self._config_manager = config_manager
        _config = self._config_manager.config["webdriver"]["use"]
        if _config not in self._mapping:
            _log.error("不支持的浏览器：%s", _config)
            raise Exception("不支持的浏览器")
        self._driver: IDriver = self._mapping[_config](self._config_manager)
        self._tabs: "List[TabInfo]" = []
        _log.debug("初始化完毕")

    @property
    def driver(self):
        return self._driver

    def open_tab(self, tab_info: "TabInfo"):
        if not tab_info.tab_handler:
            tab_handler = self._driver.new_tab()
            tab_info.tab_handler = tab_handler
        if not tab_info.tab_type:
            tab_info.tab_type = TabInfo.TAB_TYPE_OTHER
        _log.debug("打开URL：【%s】@%s", tab_info.url, tab_info.tab_handler)
        self.driver.open_url(tab_info.url, tab_info.tab_handler)
        _log.info("打开URL完毕：【%s】@%s", tab_info.url, tab_info.tab_handler)
        if tab_info not in self._tabs:
            self._tabs.append(tab_info)

    def close_tab(self, tab_info: "TabInfo"):
        if tab_info not in self._tabs:
            _log.warning("提供的标签不在标签组中，不予执行")
            return
        _log.debug("关闭标签：%s", tab_info.tab_handler)
        self._driver.close_tab(tab_info.tab_handler)
        _log.info("关闭标签完毕：%s", tab_info.tab_handler)
        self._tabs.remove(tab_info)

    def terminate(self):
        if self._driver:
            self._driver.terminate()