package top.r3944realms.ltdmanager.module

import top.r3944realms.ltdmanager.utils.LoggerUtil

class ModuleManager {

    private val modules = mutableMapOf<String, BaseModule>()

    /**
     * 注册模块到管理器
     */
    fun registerModule(module: BaseModule) {
        if (modules.containsKey(module.name)) {
            LoggerUtil.logger.warn("模块已注册: ${module.name}")
            return
        }
        modules[module.name] = module
        LoggerUtil.logger.info("模块注册: ${module.name}")
    }

    /**
     * 加载指定模块
     */
    fun loadModule(name: String) {
        val module = modules[name]
        if (module == null) {
            LoggerUtil.logger.warn("尝试加载不存在的模块: $name")
            return
        }
        if (module.loaded) {
            LoggerUtil.logger.info("模块已加载: $name")
            return
        }
        try {
            module.load()
            LoggerUtil.logger.info("模块加载: $name")
        } catch (e: Exception) {
            LoggerUtil.logger.error("加载模块 $name 失败", e)
        }
    }

    /**
     * 卸载指定模块
     */
    fun unloadModule(name: String) {
        val module = modules[name]
        if (module == null) {
            LoggerUtil.logger.warn("尝试卸载不存在的模块: $name")
            return
        }
        if (!module.loaded) {
            LoggerUtil.logger.info("模块未加载: $name")
            return
        }
        try {
            module.unload()
            LoggerUtil.logger.info("模块卸载: $name")
        } catch (e: Exception) {
            LoggerUtil.logger.warn("卸载模块 $name 失败", e)
        }
    }

    /**
     * 卸载所有模块
     */
    fun unloadAll() {
        modules.values.forEach { module ->
            try {
                if (module.loaded) {
                    module.unload()
                    LoggerUtil.logger.info("模块卸载: ${module.name}")
                }
            } catch (e: Exception) {
                LoggerUtil.logger.warn("卸载模块 ${module.name} 失败", e)
            }
        }
    }

    /**
     * 获取所有模块名称
     */
    fun getModuleNames(): List<String> = modules.keys.toList()

    /**
     * 检查模块是否已加载
     */
    fun isModuleLoaded(name: String): Boolean {
        return modules[name]?.loaded ?: false
    }
    /**
     * 扩展方法：批量加载模块
     */
    fun ModuleManager.loadModules(vararg names: String) {
        names.forEach { loadModule(it) }
    }

    /**
     * 扩展方法：批量卸载模块
     */
    fun ModuleManager.unloadModules(vararg names: String) {
        names.forEach { unloadModule(it) }
    }
    /**
     * 关闭所有模块
     */
    suspend fun stopAllModules() {
        modules.values.forEach { module ->
            if (module.loaded) {
                module.stop()
            }
        }
    }
}