Custom Filtering Rules

Signed-off-by: shedaniel <daniel@shedaniel.me>

1 files changed, 15 insertions, 7 deletions

diff --git a/src/main/java/me/shedaniel/rei/impl/search/ArgumentsRegistry.java b/src/main/java/me/shedaniel/rei/impl/search/ArgumentsRegistry.java
index bcbc8b6ff..fe69a5ef1 100644
--- a/src/main/java/me/shedaniel/rei/impl/search/ArgumentsRegistry.java
+++ b/src/main/java/me/shedaniel/rei/impl/search/ArgumentsRegistry.java
@@ -24,20 +24,28 @@
 package me.shedaniel.rei.impl.search;
 
 import com.google.common.collect.Lists;
+import com.google.common.collect.Maps;
 import org.jetbrains.annotations.ApiStatus;
 
 import java.util.List;
+import java.util.Map;
 
 @ApiStatus.Internal
 public final class ArgumentsRegistry {
-    public static final List<Argument> ARGUMENTS = Lists.newArrayList();
+    public static final Map<String, Argument> ARGUMENTS = Maps.newHashMap();
+    public static final List<Argument> ARGUMENT_LIST = Lists.newArrayList();
     
     static {
-        ARGUMENTS.add(AlwaysMatchingArgument.INSTANCE);
-        ARGUMENTS.add(ModArgument.INSTANCE);
-        ARGUMENTS.add(TooltipArgument.INSTANCE);
-        ARGUMENTS.add(TagArgument.INSTANCE);
-        ARGUMENTS.add(RegexArgument.INSTANCE);
-        ARGUMENTS.add(TextArgument.INSTANCE);
+        register(AlwaysMatchingArgument.INSTANCE);
+        register(ModArgument.INSTANCE);
+        register(TooltipArgument.INSTANCE);
+        register(TagArgument.INSTANCE);
+        register(RegexArgument.INSTANCE);
+        register(TextArgument.INSTANCE);
+    }
+    
+    private static void register(Argument argument) {
+        ARGUMENTS.put(argument.getName(), argument);
+        ARGUMENT_LIST.add(argument);
     }
 }