Merge branch 'inventory-builder-improvements' into new-command-system-1.20.6-2

Author: WizardlyBump17

core/src/main/java/com/wizardlybump17/wlib/inventory/paginated/PaginatedInventoryBuilder.java

@@ -6,6 +6,7 @@
 import com.wizardlybump17.wlib.inventory.item.InventoryNavigator;
 import com.wizardlybump17.wlib.inventory.item.ItemButton;
 import com.wizardlybump17.wlib.inventory.listener.InventoryListener;
+import com.wizardlybump17.wlib.item.ItemBuilder;
 import com.wizardlybump17.wlib.util.MapUtils;
 import com.wizardlybump17.wlib.util.ObjectUtil;
 import lombok.AccessLevel;
@@ -105,14 +106,45 @@ public PaginatedInventoryBuilder shapeReplacements(@NonNull Map<Character, ItemB
         return this;
     }
 
-    public @NotNull PaginatedInventoryBuilder setReplacementActionByCustomData(@NotNull String key, @Nullable Object value, @NotNull ClickAction action) {
+    public @NotNull PaginatedInventoryBuilder setReplacementActionByCustomData(@NotNull Object key, @Nullable Object value, @NotNull ClickAction action) {
         for (ItemButton button : shapeReplacements.values()) {
             if (Objects.equals(button.getCustomData().get(key), value))
                 button.setClickAction(action);
         }
         return this;
     }
 
+    public @NotNull PaginatedInventoryBuilder setReplacementItemStackByCustomData(@NotNull Object key, @Nullable Object value, @NotNull Supplier<ItemStack> itemSupplier) {
+        for (ItemButton button : shapeReplacements.values())
+            if (Objects.equals(button.getCustomData().get(key), value))
+                button.setItem(itemSupplier);
+        return this;
+    }
+
+    public @NotNull PaginatedInventoryBuilder setReplacementItemStackByCustomData(@NotNull Object key, @Nullable Object value, @NotNull ItemStack item) {
+        Supplier<ItemStack> itemSupplier = () -> item;
+        for (ItemButton button : shapeReplacements.values())
+            if (Objects.equals(button.getCustomData().get(key), value))
+                button.setItem(itemSupplier);
+        return this;
+    }
+
+    public @NotNull PaginatedInventoryBuilder setReplacementItemByCustomData(@NotNull Object key, @Nullable Object value, @NotNull Supplier<ItemBuilder> itemSupplier) {
+        Supplier<ItemStack> itemStackSupplier = () -> itemSupplier.get().build();
+        for (ItemButton button : shapeReplacements.values())
+            if (Objects.equals(button.getCustomData().get(key), value))
+                button.setItem(itemStackSupplier);
+        return this;
+    }
+
+    public @NotNull PaginatedInventoryBuilder setReplacementItemByCustomData(@NotNull Object key, @Nullable Object value, @NotNull ItemBuilder item) {
+        Supplier<ItemStack> itemSupplier = item::build;
+        for (ItemButton button : shapeReplacements.values())
+            if (Objects.equals(button.getCustomData().get(key), value))
+                button.setItem(itemSupplier);
+        return this;
+    }
+
     public PaginatedInventoryBuilder content(@Nullable List<ItemButton> content) {
         this.content = content == null ? new ArrayList<>() : content;
         checkNullContent();