add getFilteredList to Util

GamesInTheBox-MC · Sep 5, 2023 · b9aef23 · b9aef23
1 parent 664d78d
commit b9aef23
Showing 1 changed file with 48 additions and 4 deletions.
diff --git a/src/main/java/me/hsgamer/gamesinthebox/util/Util.java b/src/main/java/me/hsgamer/gamesinthebox/util/Util.java
@@ -20,12 +20,12 @@
 import org.bukkit.configuration.ConfigurationSection;
 import org.bukkit.configuration.MemoryConfiguration;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 
-import java.util.Collection;
-import java.util.HashMap;
-import java.util.Map;
-import java.util.Optional;
+import java.util.*;
+import java.util.function.BiPredicate;
 import java.util.function.Consumer;
+import java.util.stream.Collectors;
 
 /**
  * The common utilities
@@ -81,4 +81,48 @@ public static String getRandomColorizedString(@NotNull Collection<@NotNull Strin
         String s = Optional.ofNullable(CollectionUtils.pickRandom(collection)).orElse(defaultValue);
         return ColorUtils.colorize(s);
     }
+
+    /**
+     * Get a filtered list from a list
+     *
+     * @param list      the list
+     * @param filter    the filter
+     * @param predicate the predicate with 2 parameters: the element and the filter
+     * @param <T>       the type of the element in the list
+     * @return the filtered list
+     */
+    @NotNull
+    public static <T> List<T> getFilteredList(@NotNull List<T> list, @Nullable T filter, @NotNull BiPredicate<T, T> predicate) {
+        return filter == null
+                ? Collections.unmodifiableList(list)
+                : list.stream().filter(element -> predicate.test(element, filter)).collect(Collectors.toList());
+    }
+
+    /**
+     * Get a filtered string list from a list
+     *
+     * @param list      the list
+     * @param filter    the filter
+     * @param predicate the predicate with 2 parameters: the element and the filter
+     * @return the filtered list
+     */
+    @NotNull
+    public static List<String> getFilteredList(@NotNull List<String> list, @Nullable String filter, @NotNull BiPredicate<String, String> predicate) {
+        return filter == null || filter.isEmpty()
+                ? Collections.unmodifiableList(list)
+                : list.stream().filter(element -> predicate.test(element, filter)).collect(Collectors.toList());
+    }
+
+    /**
+     * Get a filtered string list from a list.
+     * The filter is case-insensitive.
+     *
+     * @param list   the list
+     * @param filter the filter
+     * @return the filtered list
+     */
+    @NotNull
+    public static List<String> getFilteredList(@NotNull List<String> list, @Nullable String filter) {
+        return getFilteredList(list, filter, (s, s2) -> s.toLowerCase(Locale.ROOT).contains(s2.toLowerCase(Locale.ROOT)));
+    }
 }