Same as v2.3 in Fabricv2.3.0.17

1 files changed, 43 insertions, 0 deletions

diff --git a/src/main/java/me/shedaniel/rei/plugin/PluginManager.java b/src/main/java/me/shedaniel/rei/plugin/PluginManager.java
new file mode 100644
index 000000000..c3fa7b57e
--- /dev/null
+++ b/src/main/java/me/shedaniel/rei/plugin/PluginManager.java
@@ -0,0 +1,43 @@
+package me.shedaniel.rei.plugin;
+
+import com.google.common.collect.Lists;
+import com.google.common.collect.Maps;
+import me.shedaniel.rei.api.IPluginDisabler;
+import me.shedaniel.rei.api.PluginFunction;
+import net.minecraft.util.ResourceLocation;
+
+import java.util.List;
+import java.util.Map;
+
+public class PluginManager implements IPluginDisabler {
+    
+    private static Map<ResourceLocation, List<PluginFunction>> pluginDisabledFunctions = Maps.newHashMap();
+    
+    @Override
+    public void disablePluginFunction(ResourceLocation plugin, PluginFunction function) {
+        List<PluginFunction> list = Lists.newArrayList();
+        if (pluginDisabledFunctions.containsKey(plugin))
+            list = pluginDisabledFunctions.get(plugin);
+        if (!list.contains(function))
+            list.add(function);
+        pluginDisabledFunctions.put(plugin, list);
+    }
+    
+    @Override
+    public void enablePluginFunction(ResourceLocation plugin, PluginFunction function) {
+        List<PluginFunction> list = Lists.newArrayList();
+        if (pluginDisabledFunctions.containsKey(plugin))
+            list = pluginDisabledFunctions.get(plugin);
+        if (list.contains(function))
+            list.remove(function);
+        pluginDisabledFunctions.put(plugin, list);
+        if (list.size() == 0)
+            pluginDisabledFunctions.remove(plugin);
+    }
+    
+    @Override
+    public boolean isFunctionEnabled(ResourceLocation plugin, PluginFunction function) {
+        return !pluginDisabledFunctions.containsKey(plugin) || !pluginDisabledFunctions.get(plugin).contains(function);
+    }
+    
+}
\ No newline at end of file