Improved Search

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

1 files changed, 41 insertions, 0 deletions

diff --git a/src/main/java/me/shedaniel/rei/impl/search/Argument.java b/src/main/java/me/shedaniel/rei/impl/search/Argument.java
new file mode 100644
index 000000000..a2b94b624
--- /dev/null
+++ b/src/main/java/me/shedaniel/rei/impl/search/Argument.java
@@ -0,0 +1,41 @@
+package me.shedaniel.rei.impl.search;
+
+import me.shedaniel.rei.api.EntryStack;
+import org.jetbrains.annotations.ApiStatus;
+import org.jetbrains.annotations.Nullable;
+
+@ApiStatus.Internal
+public abstract class Argument {
+    public Argument() {
+    }
+    
+    private int dataOrdinal = -1;
+    
+    public abstract String getName();
+    
+    @Nullable
+    public String getPrefix() {
+        return null;
+    }
+    
+    public MatchStatus matchesArgumentPrefix(String text) {
+        String prefix = getPrefix();
+        if (prefix == null) return MatchStatus.unmatched();
+        if (text.startsWith("-" + prefix)) return MatchStatus.invertMatched(text.substring(1 + prefix.length()));
+        if (text.startsWith(prefix + "-")) return MatchStatus.invertMatched(text.substring(1 + prefix.length()));
+        return text.startsWith(prefix) ? MatchStatus.matched(text.substring(prefix.length())) : MatchStatus.unmatched();
+    }
+    
+    public final int getDataOrdinal() {
+        if (dataOrdinal == -1) {
+            dataOrdinal = ArgumentsRegistry.ARGUMENTS.indexOf(this);
+        }
+        return dataOrdinal;
+    }
+    
+    public abstract boolean matches(Object[] data, EntryStack stack, String searchText, Object searchData);
+    
+    public Object prepareSearchData(String searchText) {
+        return null;
+    }
+}