+ update

2018-10-08 19:44:36 +08:00
parent 3b3212dcdc
commit 91840d6a6b
10 changed files with 261 additions and 216 deletions
--- a/src/main/java/me/skymc/taboolib/TabooLibLoader.java
+++ b/src/main/java/me/skymc/taboolib/TabooLibLoader.java
@@ -1,5 +1,7 @@
 package me.skymc.taboolib;

+import com.google.common.collect.Lists;
+import com.google.common.collect.Maps;
 import com.ilummc.tlib.TLib;
 import com.ilummc.tlib.annotations.Dependency;
 import com.ilummc.tlib.inject.TDependencyInjector;
@@ -8,6 +10,7 @@ import me.skymc.taboolib.bstats.Metrics;
 import me.skymc.taboolib.fileutils.FileUtils;
 import me.skymc.taboolib.listener.TListener;
 import me.skymc.taboolib.listener.TListenerHandler;
+import me.skymc.taboolib.methods.ReflectionUtils;
 import me.skymc.taboolib.playerdata.DataUtils;
 import org.bukkit.Bukkit;
 import org.bukkit.event.EventHandler;
@@ -20,10 +23,7 @@ import org.bukkit.plugin.Plugin;
 import java.io.File;
 import java.io.InputStream;
 import java.net.InetAddress;
-import java.util.Arrays;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Optional;
+import java.util.*;

 /**
 * @Author sky
@@ -32,7 +32,8 @@ import java.util.Optional;
@TListener
 public class TabooLibLoader implements Listener {

-    static HashMap<String, List<Class>> pluginClasses = new HashMap<>();
+    static Map<String, List<Class>> pluginClasses = Maps.newHashMap();
+    static List<Loader> loaders = Lists.newArrayList();

    static void setup() {
        testInternet();
@@ -43,84 +44,69 @@ public class TabooLibLoader implements Listener {

    static void register() {
        setupClasses();
+        loadClasses();
        registerListener();
        registerMetrics();
    }

-    /**
-     * 获取插件所有被读取到的类
-     *
-     * @param plugin 插件
-     * @return List
-     */
+    static void unregister() {
+        unloadClasses();
+    }
+
    public static Optional<List<Class>> getPluginClasses(Plugin plugin) {
        return Optional.ofNullable(pluginClasses.get(plugin.getName()));
    }

-    /**
-     * 初始化插件文件夹
-     */
+    public static List<Class> getPluginClassSafely(Plugin plugin) {
+        List<Class> classes = pluginClasses.get(plugin.getName());
+        return classes == null ? new ArrayList<>() : new ArrayList<>(classes);
+    }
+
+    static boolean isLoader(Class pluginClass) {
+        return !Loader.class.equals(pluginClass) && Loader.class.isAssignableFrom(pluginClass);
+    }
+
+    static void loadClasses() {
+        pluginClasses.forEach((key, classes) -> classes.forEach(pluginClass -> loadClass(Bukkit.getPluginManager().getPlugin(key), pluginClass)));
+    }
+
+    static void unloadClasses() {
+        pluginClasses.forEach((key, classes) -> classes.forEach(pluginClass -> unloadClass(Bukkit.getPluginManager().getPlugin(key), pluginClass)));
+    }
+
+    static void registerListener() {
+        TListenerHandler.setupListeners();
+        Bukkit.getScheduler().runTask(TabooLib.instance(), TListenerHandler::registerListeners);
+    }
+
+    static void registerMetrics() {
+        Metrics metrics = new Metrics(TabooLib.instance());
+        metrics.addCustomChart(new Metrics.SingleLineChart("plugins_using_taboolib", () -> Math.toIntExact(Arrays.stream(Bukkit.getPluginManager().getPlugins()).filter(plugin -> plugin.getDescription().getDepend().contains("TabooLib")).count())));
+    }
+
    static void setupDataFolder() {
        Main.setPlayerDataFolder(FileUtils.folder(Main.getInst().getConfig().getString("DATAURL.PLAYER-DATA")));
        Main.setServerDataFolder(FileUtils.folder(Main.getInst().getConfig().getString("DATAURL.SERVER-DATA")));
    }

-    /**
-     * 载入插件数据库
-     */
    static void setupDatabase() {
        DataUtils.addPluginData("TabooLibrary", null);
-        // 检查是否启用数据库
        Main.setStorageType(Main.getInst().getConfig().getBoolean("MYSQL.ENABLE") ? Main.StorageType.SQL : Main.StorageType.LOCAL);
-        // 初始化数据库
        TabooLibDatabase.init();
    }

-    /**
-     * 读取插件类
-     */
-    static void setupClasses() {
-        Arrays.stream(Bukkit.getPluginManager().getPlugins()).forEach(TabooLibLoader::setupClasses);
-    }
-
-    /**
-     * 读取插件类
-     */
-    static void setupClasses(Plugin plugin) {
-        if (!(TabooLib.isTabooLib(plugin) || TabooLib.isDependTabooLib(plugin))) {
-            return;
-        }
-        try {
-            long time = System.currentTimeMillis();
-            List<Class> classes = FileUtils.getClasses(plugin);
-            TLocale.Logger.info("DEPENDENCY.LOAD-CLASSES", plugin.getName(), String.valueOf(classes.size()), String.valueOf(System.currentTimeMillis() - time));
-            pluginClasses.put(plugin.getName(), classes);
-        } catch (Exception ignored) {
-        }
-    }
-
-    /**
-     * 初始化插件依赖库
-     */
    static void setupLibraries() {
-        if (!Main.isOfflineVersion()) {
-            return;
-        }
-        for (Dependency dependency : TDependencyInjector.getDependencies(TLib.getTLib())) {
-            if (dependency.type() == Dependency.Type.LIBRARY && dependency.maven().matches(".*:.*:.*")) {
-                String fileName = String.join("-", dependency.maven().split(":")) + ".jar";
+        if (Main.isOfflineVersion()) {
+            Arrays.stream(TDependencyInjector.getDependencies(TLib.getTLib())).filter(dependency -> dependency.type() == Dependency.Type.LIBRARY && dependency.maven().matches(".*:.*:.*")).map(dependency -> String.join("-", dependency.maven().split(":")) + ".jar").forEach(fileName -> {
                File targetFile = FileUtils.file(TLib.getTLib().getLibsFolder(), fileName);
                InputStream inputStream = FileUtils.getResource("libs/" + fileName);
                if (!targetFile.exists() && inputStream != null) {
                    FileUtils.inputStreamToFile(inputStream, FileUtils.file(TLib.getTLib().getLibsFolder(), fileName));
                }
-            }
+            });
        }
    }

-    /**
-     * 检查网络连接状态
-     */
    static void testInternet() {
        try {
            InetAddress inetAddress = InetAddress.getByName(Main.getInst().getConfig().getString("TEST-URL", "aliyun.com"));
@@ -129,40 +115,70 @@ public class TabooLibLoader implements Listener {
        }
        if (!Main.isInternetOnline() && !Main.isOfflineVersion() && !Main.isLibrariesExists()) {
            TLocale.Logger.error("TLIB.LOAD-FAIL-OFFLINE", Main.getInst().getDescription().getVersion());
+            for (; ; ) {
+                // 停止主线程
+            }
+        }
+    }
+
+    static void setupClasses(Plugin plugin) {
+        if (TabooLib.isTabooLib(plugin) || TabooLib.isDependTabooLib(plugin)) {
            try {
-                while (true) {
-                    Thread.sleep(1000);
-                }
+                long time = System.currentTimeMillis();
+                List<Class> classes = FileUtils.getClasses(plugin);
+                TLocale.Logger.info("DEPENDENCY.LOAD-CLASSES", plugin.getName(), String.valueOf(classes.size()), String.valueOf(System.currentTimeMillis() - time));
+                pluginClasses.put(plugin.getName(), classes);
            } catch (Exception ignored) {
            }
        }
    }

-    /**
-     * 载入插件监听
-     */
-    static void registerListener() {
-        // 载入所有 TListener 监听器
-        TListenerHandler.setupListeners();
-        // 注册所有 TListener 监听器
-        Bukkit.getScheduler().runTask(TabooLib.instance(), TListenerHandler::registerListeners);
+    static void setupClasses() {
+        Arrays.stream(Bukkit.getPluginManager().getPlugins()).forEach(TabooLibLoader::setupClasses);
+        pluginClasses.get("TabooLib").stream().filter(TabooLibLoader::isLoader).forEach(pluginClass -> {
+            try {
+                loaders.add((Loader) ReflectionUtils.instantiateObject(pluginClass));
+            } catch (Exception e) {
+                e.printStackTrace();
+            }
+        });
    }

-    /**
-     * 注册插件统计
-     */
-    static void registerMetrics() {
-        Metrics metrics = new Metrics(TabooLib.instance());
-        metrics.addCustomChart(new Metrics.SingleLineChart("plugins_using_taboolib", () -> Math.toIntExact(Arrays.stream(Bukkit.getPluginManager().getPlugins()).filter(plugin -> plugin.getDescription().getDepend().contains("TabooLib")).count())));
+    static void loadClass(Plugin plugin, Class<?> loadClass) {
+        loaders.forEach(loader -> {
+            try {
+                loader.load(plugin, loadClass);
+            } catch (Throwable ignored) {
+            }
+        });
+    }
+
+    static void unloadClass(Plugin plugin, Class<?> loadClass) {
+        loaders.forEach(loader -> {
+            try {
+                loader.unload(plugin, loadClass);
+            } catch (Throwable ignored) {
+            }
+        });
    }

    @EventHandler(priority = EventPriority.LOWEST)
    public void onEnable(PluginEnableEvent e) {
        setupClasses(e.getPlugin());
+        Optional.ofNullable(pluginClasses.get(e.getPlugin().getName())).ifPresent(classes -> classes.forEach(pluginClass -> loadClass(e.getPlugin(), pluginClass)));
    }

    @EventHandler(priority = EventPriority.MONITOR)
    public void onDisable(PluginDisableEvent e) {
-        pluginClasses.remove(e.getPlugin().getName());
+        Optional.ofNullable(pluginClasses.remove(e.getPlugin().getName())).ifPresent(classes -> classes.forEach(pluginClass -> unloadClass(e.getPlugin(), pluginClass)));
+    }
+
+    public interface Loader {
+
+        default void load(Plugin plugin, Class<?> loadClass) {
+        }
+
+        default void unload(Plugin plugin, Class<?> cancelClass) {
+        }
    }
 }