diff --git a/src/main/java/meteordevelopment/meteorclient/systems/modules/Modules.java b/src/main/java/meteordevelopment/meteorclient/systems/modules/Modules.java index 37ec252a92..ffdae0895d 100644 --- a/src/main/java/meteordevelopment/meteorclient/systems/modules/Modules.java +++ b/src/main/java/meteordevelopment/meteorclient/systems/modules/Modules.java @@ -134,7 +134,7 @@ public T get(Class klass) { } public Module get(String name) { - for (Module module : moduleInstances.values()) { + for (Module module : modules) { if (module.name.equalsIgnoreCase(name)) return module; } @@ -151,7 +151,7 @@ public List getGroup(Category category) { } public Collection getAll() { - return moduleInstances.values(); + return modules; } public List getList() { @@ -159,7 +159,7 @@ public List getList() { } public int getCount() { - return moduleInstances.values().size(); + return modules.size(); } public List getActive() { @@ -171,7 +171,7 @@ public List getActive() { public Set searchTitles(String text) { Map modules = new ValueComparableMap<>(Comparator.naturalOrder()); - for (Module module : this.moduleInstances.values()) { + for (Module module : this.modules) { int score = Utils.searchLevenshteinDefault(module.title, text, false); if (Config.get().moduleAliases.get()) { for (String alias : module.aliases) { @@ -188,7 +188,7 @@ public Set searchTitles(String text) { public Set searchSettingTitles(String text) { Map modules = new ValueComparableMap<>(Comparator.naturalOrder()); - for (Module module : this.moduleInstances.values()) { + for (Module module : this.modules) { int lowest = Integer.MAX_VALUE; for (SettingGroup sg : module.settings) { for (Setting setting : sg) { @@ -288,7 +288,7 @@ private void onMouseButton(MouseButtonEvent event) { private void onAction(boolean isKey, int value, int modifiers, boolean isPress) { if (mc.currentScreen != null || Input.isKeyPressed(GLFW.GLFW_KEY_F3)) return; - for (Module module : moduleInstances.values()) { + for (Module module : modules) { if (module.keybind.matches(isKey, value, modifiers) && (isPress || (module.toggleOnBindRelease && module.isActive()))) { module.toggle(); module.sendToggledMsg(); @@ -302,7 +302,7 @@ private void onAction(boolean isKey, int value, int modifiers, boolean isPress) private void onOpenScreen(OpenScreenEvent event) { if (!Utils.canUpdate()) return; - for (Module module : moduleInstances.values()) { + for (Module module : modules) { if (module.toggleOnBindRelease && module.isActive()) { module.toggle(); module.sendToggledMsg(); @@ -380,7 +380,7 @@ public void add(Module module) { // Remove the previous module with the same name AtomicReference removedModule = new AtomicReference<>(); - if (moduleInstances.values().removeIf(module1 -> { + if (modules.removeIf(module1 -> { if (module1.name.equals(module.name)) { removedModule.set(module1); module1.settings.unregisterColorSettings(); @@ -390,6 +390,7 @@ public void add(Module module) { return false; })) { + moduleInstances.remove(removedModule.getClass(), removedModule.get()); getGroup(removedModule.get().category).remove(removedModule.get()); }