[1.19.x] Add simple search bar to waystone screen

shared/src/main/java/net/blay09/mods/waystones/client/gui/screen/WaystoneSelectionScreenBase.java

@@ -20,6 +20,7 @@
 import net.minecraft.client.Minecraft;
 import net.minecraft.client.gui.Font;
 import net.minecraft.client.gui.components.Button;
+import net.minecraft.client.gui.components.EditBox;
 import net.minecraft.client.gui.components.Widget;
 import net.minecraft.client.gui.components.events.GuiEventListener;
 import net.minecraft.client.gui.narration.NarratableEntry;
@@ -32,32 +33,37 @@
 import net.minecraft.world.entity.player.Player;
 import net.minecraft.world.item.ItemStack;
 import net.minecraft.world.item.Items;
+import org.lwjgl.glfw.GLFW;
 
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Objects;
 import java.util.Optional;
 import java.util.function.Predicate;
+import java.util.stream.Collectors;
 
 public abstract class WaystoneSelectionScreenBase extends AbstractContainerScreen<WaystoneSelectionMenu> {
 
     private final List<IWaystone> waystones;
+    private List<IWaystone> filteredWaystones;
     private final List<ITooltipProvider> tooltipProviders = new ArrayList<>();
 
     private Button btnPrevPage;
     private Button btnNextPage;
+    private EditBox searchBox;
     private int pageOffset;
     private int headerY;
     private boolean isLocationHeaderHovered;
     private int buttonsPerPage;
 
-    private static final int headerHeight = 40;
+    private static final int headerHeight = 64;
     private static final int footerHeight = 25;
     private static final int entryHeight = 25;
 
     public WaystoneSelectionScreenBase(WaystoneSelectionMenu container, Inventory playerInventory, Component title) {
         super(container, playerInventory, title);
         waystones = container.getWaystones();
+        filteredWaystones = new ArrayList<>(waystones);
         imageWidth = 270;
         imageHeight = 200;
     }
@@ -89,6 +95,14 @@ public void init() {
         addRenderableWidget(btnNextPage);
 
         updateList();
+
+
+        searchBox = new EditBox(font, width / 2 - 99, topPos + headerHeight - 24, 198, 20, Component.empty());
+        searchBox.setResponder(text -> {
+           searchWaystones();
+        });
+
+        addRenderableWidget(searchBox);
     }
 
     @Override
@@ -103,7 +117,7 @@ private void updateList() {
         headerY = 0;
 
         btnPrevPage.active = pageOffset > 0;
-        btnNextPage.active = pageOffset < (waystones.size() - 1) / buttonsPerPage;
+        btnNextPage.active = pageOffset < (filteredWaystones.size() - 1) / buttonsPerPage;
 
         tooltipProviders.clear();
 
@@ -115,8 +129,8 @@ private void updateList() {
         int y = topPos + headerHeight + headerY;
         for (int i = 0; i < buttonsPerPage; i++) {
             int entryIndex = pageOffset * buttonsPerPage + i;
-            if (entryIndex >= 0 && entryIndex < waystones.size()) {
-                IWaystone waystone = waystones.get(entryIndex);
+            if (entryIndex >= 0 && entryIndex < filteredWaystones.size()) {
+                IWaystone waystone = filteredWaystones.get(entryIndex);
 
                 addRenderableWidget(createWaystoneButton(y, waystone));
 
@@ -128,7 +142,7 @@ private void updateList() {
                     addRenderableWidget(sortUpButton);
 
                     SortWaystoneButton sortDownButton = new SortWaystoneButton(width / 2 + 108, y + 13, 1, y, 20, it -> sortWaystone(entryIndex, 1));
-                    if (entryIndex == waystones.size() - 1) {
+                    if (entryIndex == filteredWaystones.size() - 1) {
                         sortDownButton.active = false;
                     }
                     addRenderableWidget(sortDownButton);
@@ -151,8 +165,8 @@ private void updateList() {
             }
         }
 
-        btnPrevPage.y = topPos + headerY + headerHeight + buttonsPerPage * 22 + (waystones.size() > 0 ? 10 : 0);
-        btnNextPage.y = topPos + headerY + headerHeight + buttonsPerPage * 22 + (waystones.size() > 0 ? 10 : 0);
+        btnPrevPage.y = topPos + headerY + headerHeight + buttonsPerPage * 22 + (filteredWaystones.size() > 0 ? 10 : 0);
+        btnNextPage.y = topPos + headerY + headerHeight + buttonsPerPage * 22 + (filteredWaystones.size() > 0 ? 10 : 0);
     }
 
     private WaystoneButton createWaystoneButton(int y, final IWaystone waystone) {
@@ -187,6 +201,16 @@ private void sortWaystone(int index, int sortDir) {
 
         PlayerWaystoneManager.swapWaystoneSorting(Minecraft.getInstance().player, index, otherIndex);
         Balm.getNetworking().sendToServer(new SortWaystoneMessage(index, otherIndex));
+        searchWaystones();
+    }
+
+    private void searchWaystones() {
+        filteredWaystones = waystones.stream().filter((waystone) -> waystone.getName().toLowerCase().contains(searchBox.getValue().toLowerCase())).collect(Collectors.toList());
+        int calculatedPageOffset = (filteredWaystones.size() - 1) / buttonsPerPage;
+        if (!searchBox.getValue().isEmpty() && calculatedPageOffset < pageOffset) {
+            pageOffset = calculatedPageOffset;
+        }
+
         updateList();
     }
 
@@ -281,4 +305,13 @@ protected boolean allowSorting() {
     protected boolean allowDeletion() {
         return true;
     }
+
+    @Override
+    public boolean keyPressed(int key, int scanCode, int modifiers) {
+        if (!this.searchBox.isFocused() || (key == GLFW.GLFW_KEY_ESCAPE && this.shouldCloseOnEsc())) {
+            return super.keyPressed(key, scanCode, modifiers);
+        }
+
+        return this.searchBox.keyPressed(key, scanCode, modifiers);
+    }
 }