Initial commit

2020-06-02 21:07:31 +02:00 · 2020-06-02 21:07:31 +02:00 · 392479abde
commit 392479abde
30 changed files with 1622 additions and 0 deletions
--- a/src/main/java/net/frankheijden/serverutils/managers/PluginManager.java
+++ b/src/main/java/net/frankheijden/serverutils/managers/PluginManager.java
@ -0,0 +1,148 @@
+package net.frankheijden.serverutils.managers;
+
+import net.frankheijden.serverutils.ServerUtils;
+import net.frankheijden.serverutils.config.Messenger;
+import net.frankheijden.serverutils.reflection.*;
+import org.bukkit.Bukkit;
+import org.bukkit.command.Command;
+import org.bukkit.command.CommandSender;
+import org.bukkit.plugin.InvalidPluginException;
+import org.bukkit.plugin.Plugin;
+
+import java.io.File;
+import java.lang.reflect.InvocationTargetException;
+import java.util.List;
+import java.util.Map;
+
+public class PluginManager {
+
+    public static LoadResult loadPlugin(String jarFile) {
+        return loadPlugin(new File(ServerUtils.getInstance().getDataFolder().getParent(), jarFile));
+    }
+
+    public static LoadResult loadPlugin(File file) {
+        if (!file.exists()) return new LoadResult(Result.NOT_EXISTS);
+        try {
+            return new LoadResult(Bukkit.getPluginManager().loadPlugin(file), Result.SUCCESS);
+        } catch (InvalidPluginException ex) {
+            if (ex.getCause() instanceof IllegalArgumentException) {
+                IllegalArgumentException e = (IllegalArgumentException) ex.getCause();
+                if (e.getMessage().equalsIgnoreCase("Plugin already initialized!")) {
+                    return new LoadResult(Result.ALREADY_ENABLED);
+                }
+            }
+        } catch (Exception ex) {
+            ex.printStackTrace();
+        }
+        return new LoadResult(Result.ERROR);
+    }
+
+    public static Result disablePlugin(String pluginName) {
+        return disablePlugin(Bukkit.getPluginManager().getPlugin(pluginName));
+    }
+
+    public static Result disablePlugin(Plugin plugin) {
+        if (plugin == null) return Result.NOT_EXISTS;
+        try {
+            Bukkit.getPluginManager().disablePlugin(plugin);
+            RSimplePluginManager.getPlugins(Bukkit.getPluginManager()).remove(plugin);
+        } catch (Exception ex) {
+            ex.printStackTrace();
+            return Result.ERROR;
+        }
+        unregisterCommands(plugin);
+        return Result.SUCCESS;
+    }
+
+    public static Result enablePlugin(Plugin plugin) {
+        if (plugin == null) return Result.NOT_EXISTS;
+        if (Bukkit.getPluginManager().isPluginEnabled(plugin.getName())) return Result.ALREADY_ENABLED;
+        Bukkit.getPluginManager().enablePlugin(plugin);
+        if (Bukkit.getPluginManager().isPluginEnabled(plugin.getName())) {
+            return Result.SUCCESS;
+        }
+        return Result.ERROR;
+    }
+
+    public static Result reloadPlugin(String pluginName) {
+        Plugin plugin = Bukkit.getPluginManager().getPlugin(pluginName);
+        if (plugin == null) return Result.NOT_EXISTS;
+        return reloadPlugin(plugin);
+    }
+
+    public static Result reloadPlugin(Plugin plugin) {
+        Result disableResult = disablePlugin(plugin);
+        if (disableResult != Result.SUCCESS) return disableResult;
+
+        File pluginFile;
+        try {
+            pluginFile = RPlugin.getPluginFile(plugin);
+        } catch (InvocationTargetException | IllegalAccessException ex) {
+            ex.printStackTrace();
+            return Result.ERROR;
+        }
+
+        LoadResult loadResult = loadPlugin(pluginFile);
+        if (!loadResult.isSuccess()) return loadResult.getResult();
+        return enablePlugin(loadResult.getPlugin());
+    }
+
+    public static void unregisterCommands(Plugin plugin) {
+        Map<String, Command> map;
+        try {
+            map = RCommandMap.getKnownCommands(RCraftServer.getCommandMap());
+        } catch (Exception ex) {
+            ex.printStackTrace();
+            return;
+        }
+
+        plugin.getDescription().getCommands().forEach((cmd, data) -> {
+            map.remove(cmd);
+
+            @SuppressWarnings("unchecked")
+            List<String> aliases = (List<String>) data.get("aliases");
+            if (aliases != null) {
+                aliases.forEach(map::remove);
+            }
+        });
+    }
+
+    public static class LoadResult {
+        private final Plugin plugin;
+        private final Result result;
+
+        public LoadResult(Plugin plugin, Result result) {
+            this.plugin = plugin;
+            this.result = result;
+        }
+
+        public LoadResult(Result result) {
+            this(null, result);
+        }
+
+        public Result getResult() {
+            return result;
+        }
+
+        public Plugin getPlugin() {
+            return plugin;
+        }
+
+        public boolean isSuccess() {
+            return plugin != null && result == Result.SUCCESS;
+        }
+    }
+
+    public enum Result {
+        NOT_EXISTS,
+        ALREADY_ENABLED,
+        ERROR,
+        SUCCESS;
+
+        public void sendTo(CommandSender sender, String action, String what) {
+            Messenger.sendMessage(sender, "serverutils." + this.name().toLowerCase(),
+                    "%action%", action,
+                    "%what%", what);
+        }
+    }
+}