import os
import sys
import tomli
import shutil
from .logger import logger
from typing import Optional


class Config:
    platform: str = "qq"
    nickname: Optional[str] = None
    server_host: str = "localhost"
    server_port: int = 8095
    napcat_heartbeat_interval: int = 30

    def __init__(self):
        self._get_config_path()

    def _get_config_path(self):
        current_file_path = os.path.abspath(__file__)
        src_path = os.path.dirname(current_file_path)
        self.root_path = os.path.join(src_path, "..")
        self.config_path = os.path.join(self.root_path, "config.toml")

    def load_config(self):  # sourcery skip: extract-method, move-assign
        include_configs = ["Napcat_Server", "MaiBot_Server", "Chat", "Voice", "Debug"]
        if not os.path.exists(self.config_path):
            logger.error("配置文件不存在！")
            logger.info("正在创建配置文件...")
            shutil.copy(
                os.path.join(self.root_path, "template", "template_config.toml"),
                os.path.join(self.root_path, "config.toml"),
            )
            logger.info("配置文件创建成功，请修改配置文件后重启程序。")
            sys.exit(1)
        with open(self.config_path, "rb") as f:
            try:
                raw_config = tomli.load(f)
            except tomli.TOMLDecodeError as e:
                logger.critical(f"配置文件bot_config.toml填写有误，请检查第{e.lineno}行第{e.colno}处：{e.msg}")
                sys.exit(1)
        for key in include_configs:
            if key not in raw_config:
                logger.error(f"配置文件中缺少必需的字段: '{key}'")
                sys.exit(1)

        self.server_host = raw_config["Napcat_Server"].get("host", "localhost")
        self.server_port = raw_config["Napcat_Server"].get("port", 8095)
        self.napcat_heartbeat_interval = raw_config["Napcat_Server"].get("heartbeat", 30)

        self.mai_host = raw_config["MaiBot_Server"].get("host", "localhost")
        self.mai_port = raw_config["MaiBot_Server"].get("port", 8000)
        self.platform = raw_config["MaiBot_Server"].get("platform_name")
        if not self.platform:
            logger.critical("请在配置文件中指定平台")
            sys.exit(1)

        self.list_type: str = raw_config["Chat"].get("list_type")
        self.group_list: list = raw_config["Chat"].get("group_list", [])
        self.user_list: list = raw_config["Chat"].get("user_list", [])
        if not self.list_type or self.list_type not in ["whitelist", "blacklist"]:
            logger.critical("请在配置文件中指定list_type或list_type填写错误")
            sys.exit(1)

        self.use_tts = raw_config["Voice"].get("use_tts", False)

        self.debug_level = raw_config["Debug"].get("level", "INFO")


global_config = Config()
global_config.load_config()