Improved Search

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

1 files changed, 43 insertions, 0 deletions

diff --git a/src/main/java/me/shedaniel/rei/impl/search/ModArgument.java b/src/main/java/me/shedaniel/rei/impl/search/ModArgument.java
new file mode 100644
index 000000000..e3a91a8b7
--- /dev/null
+++ b/src/main/java/me/shedaniel/rei/impl/search/ModArgument.java
@@ -0,0 +1,43 @@
+package me.shedaniel.rei.impl.search;
+
+import me.shedaniel.rei.api.ClientHelper;
+import me.shedaniel.rei.api.EntryStack;
+import net.minecraft.util.Identifier;
+import org.jetbrains.annotations.ApiStatus;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.Locale;
+
+@ApiStatus.Internal
+public final class ModArgument extends Argument {
+    public static final ModArgument INSTANCE = new ModArgument();
+    
+    @Override
+    public String getName() {
+        return "mod";
+    }
+    
+    @Override
+    public @Nullable String getPrefix() {
+        return "@";
+    }
+    
+    @Override
+    public boolean matches(Object[] data, EntryStack stack, String searchText, Object searchData) {
+        if (data[getDataOrdinal()] == null) {
+            data[getDataOrdinal()] = new String[]{
+                    stack.getIdentifier().map(Identifier::getNamespace).orElse("").toLowerCase(Locale.ROOT),
+                    null
+            };
+        }
+        String[] strings = (String[]) data[getDataOrdinal()];
+        if (strings[0].isEmpty() || strings[0].contains(searchText)) return true;
+        if (strings[1] == null) {
+            strings[1] = ClientHelper.getInstance().getModFromModId(strings[0]).toLowerCase(Locale.ROOT);
+        }
+        return strings[1].isEmpty() || strings[1].contains(searchText);
+    }
+    
+    private ModArgument() {
+    }
+}