from iamai import Bot as _Bot
from typing import Optional, Dict, List, Type, Any, Union
from pathlib import Path
import os
from hydroroll.config import GlobalConfig

current_dir = Path.cwd()
script_file = current_dir.resolve() / __file__
script_dir = script_file.parent

__all__ = ["Bot"]

class Bot:
    def __init__(
        self,
        *,
        config_file: Optional[str] = "config.toml",
        config_dict: Optional[Dict] = None,
        hot_reload: bool = False,
    ) -> None:
        self.bot = _Bot(hot_reload=hot_reload,
                       config_file=config_file,
                       config_dict=config_dict
                       )
        self.bot.load_plugins_from_dirs(Path(f"{script_dir}/plugins"))
        self.create_folder_structure(GlobalConfig._folder_dict)
    
    def run(self) -> None:
        self.bot.run()
    
    def restart(self) -> None:
        self.bot.restart()
        
    def create_folders(self):
            folder_path = os.path.dirname(os.path.abspath('__file__')) # 获取main.py所在文件夹路径
            if not os.path.isdir(os.path.join(folder_path, 'user')):
                os.mkdir(os.path.join(folder_path, 'user'))
            if not os.path.isdir(os.path.join(folder_path, 'data')):
                os.mkdir(os.path.join(folder_path, 'data'))
            if not os.path.isdir(os.path.join(folder_path, 'models')):
                os.mkdir(os.path.join(folder_path, 'models'))
            if not os.path.isdir(os.path.join(folder_path, 'web')):
                os.mkdir(os.path.join(folder_path, 'web'))
            if not os.path.isdir(os.path.join(folder_path, 'config')):
                os.mkdir(os.path.join(folder_path, 'config'))
            if not os.path.isdir(os.path.join(folder_path, 'logs')):
                os.mkdir(os.path.join(folder_path, 'logs'))
            if not os.path.isdir(os.path.join(folder_path, 'rules')):
                os.mkdir(os.path.join(folder_path, 'rules'))