Improved Search

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

1 files changed, 33 insertions, 0 deletions

diff --git a/src/main/java/me/shedaniel/rei/impl/search/TextArgument.java b/src/main/java/me/shedaniel/rei/impl/search/TextArgument.java
new file mode 100644
index 000000000..78c37269d
--- /dev/null
+++ b/src/main/java/me/shedaniel/rei/impl/search/TextArgument.java
@@ -0,0 +1,33 @@
+package me.shedaniel.rei.impl.search;
+
+import me.shedaniel.rei.api.EntryStack;
+import org.jetbrains.annotations.ApiStatus;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.Locale;
+
+@ApiStatus.Internal
+public final class TextArgument extends Argument {
+    public static final TextArgument INSTANCE = new TextArgument();
+    
+    @Override
+    public String getName() {
+        return "text";
+    }
+    
+    @Override
+    public @Nullable String getPrefix() {
+        return "";
+    }
+    
+    @Override
+    public boolean matches(Object[] data, EntryStack stack, String searchText, Object searchData) {
+        if (data[getDataOrdinal()] == null) {
+            data[getDataOrdinal()] = stack.asFormatStrippedText().getString().toLowerCase(Locale.ROOT);
+        }
+        return ((String) data[getDataOrdinal()]).contains(searchText);
+    }
+    
+    private TextArgument() {
+    }
+}