From 3802055673e7877d66614791eba6ea0599488c7a Mon Sep 17 00:00:00 2001
From: shedaniel <daniel@shedaniel.me>
Date: Sat, 6 Jun 2020 16:29:32 +0800
Subject: Improved Search

Signed-off-by: shedaniel <daniel@shedaniel.me>
---
 .../me/shedaniel/rei/impl/search/ModArgument.java  | 43 ++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 src/main/java/me/shedaniel/rei/impl/search/ModArgument.java

(limited to 'src/main/java/me/shedaniel/rei/impl/search/ModArgument.java')

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() {
+    }
+}
-- 
cgit