package com.cryohub.core.module; import com.cryohub.core.event.EventBus; import com.cryohub.core.module.event.ModuleLoadEvent; import com.cryohub.core.module.event.ModuleUnloadEvent; import com.cryohub.core.service.ServiceRegistry; import org.bukkit.plugin.java.JavaPlugin; import java.io.File; import java.net.URL; import java.net.URLClassLoader; import java.util.*; import java.util.jar.JarEntry; import java.util.jar.JarFile; import java.util.logging.Logger; public class ModuleManager { private final JavaPlugin plugin; private final ServiceRegistry serviceRegistry; private final EventBus eventBus; private final Logger logger; private final Map modules = new LinkedHashMap<>(); private final Map moduleClassLoaders = new HashMap<>(); public ModuleManager(JavaPlugin plugin, ServiceRegistry serviceRegistry, EventBus eventBus) { this.plugin = plugin; this.serviceRegistry = serviceRegistry; this.eventBus = eventBus; this.logger = plugin.getLogger(); } public void loadAll() { File modulesFolder = new File(this.plugin.getDataFolder(), "modules"); if (!modulesFolder.exists()) { modulesFolder.mkdirs(); this.logger.info("Created modules directory"); return; } File[] jarFiles = modulesFolder.listFiles((dir, name) -> name.endsWith(".jar")); if (jarFiles == null || jarFiles.length == 0) { this.logger.info("No external modules found"); return; } List loadedModules = new ArrayList<>(); for (File jarFile : jarFiles) { try { Module module = this.loadModuleFromJar(jarFile); if (module != null) { loadedModules.add(module); } } catch (Exception e) { this.logger.severe("Failed to load module from " + jarFile.getName() + ": " + e.getMessage()); e.printStackTrace(); } } List sorted = this.sortByDependencies(loadedModules); for (Module module : sorted) { this.registerModule(module); } } private Module loadModuleFromJar(File jarFile) throws Exception { URLClassLoader classLoader = new URLClassLoader( new URL[]{jarFile.toURI().toURL()}, this.getClass().getClassLoader() ); try (JarFile jar = new JarFile(jarFile)) { Enumeration entries = jar.entries(); while (entries.hasMoreElements()) { JarEntry entry = entries.nextElement(); String name = entry.getName(); if (name.endsWith(".class")) { String className = name.replace('/', '.').substring(0, name.length() - 6); try { Class clazz = classLoader.loadClass(className); if (Module.class.isAssignableFrom(clazz) && !clazz.isInterface()) { Module module = (Module) clazz.getDeclaredConstructor().newInstance(); this.moduleClassLoaders.put(module.getName(), classLoader); return module; } } catch (ClassNotFoundException | NoClassDefFoundError ignored) { } } } } return null; } private List sortByDependencies(List modules) { Map moduleMap = new HashMap<>(); for (Module module : modules) { moduleMap.put(module.getName(), module); } List sorted = new ArrayList<>(); Set visited = new HashSet<>(); Set visiting = new HashSet<>(); for (Module module : modules) { this.resolveDependencies(module, moduleMap, sorted, visited, visiting); } return sorted; } private void resolveDependencies(Module module, Map moduleMap, List sorted, Set visited, Set visiting) { String name = module.getName(); if (visited.contains(name)) { return; } if (visiting.contains(name)) { this.logger.warning("Circular dependency detected for module: " + name); return; } visiting.add(name); for (String dependency : module.getDependencies()) { Module depModule = moduleMap.get(dependency); if (depModule != null) { this.resolveDependencies(depModule, moduleMap, sorted, visited, visiting); } else { this.logger.warning("Module " + name + " depends on missing module: " + dependency); } } visiting.remove(name); visited.add(name); sorted.add(module); } public void registerModule(Module module) { String name = module.getName(); if (this.modules.containsKey(name)) { this.logger.warning("Module " + name + " is already registered"); return; } try { module.load(this.serviceRegistry); this.modules.put(name, module); this.eventBus.post(new ModuleLoadEvent(module)); module.enable(); this.logger.info("Loaded module: " + name + " v" + module.getVersion()); } catch (Exception e) { this.logger.severe("Failed to load module " + name + ": " + e.getMessage()); e.printStackTrace(); if (module instanceof BaseModule) { ((BaseModule) module).state = ModuleState.ERROR; } } } public void unloadModule(String name) { Module module = this.modules.get(name); if (module == null) { return; } ClassLoader classLoader = this.moduleClassLoaders.remove(name); try { module.unload(); this.modules.remove(name); this.eventBus.post(new ModuleUnloadEvent(module)); this.logger.info("Unloaded module: " + name); } catch (Exception e) { this.logger.severe("Failed to unload module " + name + ": " + e.getMessage()); e.printStackTrace(); } finally { if (classLoader instanceof URLClassLoader) { try { ((URLClassLoader) classLoader).close(); } catch (Exception e) { this.logger.warning("Failed to close ClassLoader for " + name + ": " + e.getMessage()); } } } } public void unloadAll() { List moduleNames = new ArrayList<>(this.modules.keySet()); Collections.reverse(moduleNames); for (String name : moduleNames) { this.unloadModule(name); } } public void reloadModule(String name) { Module module = this.modules.get(name); if (module != null) { module.reload(); this.logger.info("Reloaded module: " + name); } } public void reloadAll() { for (Module module : this.modules.values()) { module.reload(); } this.logger.info("Reloaded all modules"); } public Module getModule(String name) { return this.modules.get(name); } public Collection getLoadedModules() { return Collections.unmodifiableCollection(this.modules.values()); } public boolean isLoaded(String name) { return this.modules.containsKey(name); } }