replacing items based on ItemButtons

Author: WizardlyBump17

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

@@ -170,6 +170,36 @@ public PaginatedInventoryBuilder shapeReplacements(@NonNull Map<Character, ItemB
         return this;
     }
 
+    //ItemButton
+
+    public @NotNull PaginatedInventoryBuilder setReplacementItemButtonByCustomData(@NotNull Object key, @Nullable Object value, @NotNull Supplier<ItemButton> itemSupplier) {
+        ItemButton newButton = itemSupplier.get();
+        for (Map.Entry<Character, ItemButton> entry : shapeReplacements.entrySet()) {
+            ItemButton button = entry.getValue();
+            if (Objects.equals(button.getCustomData().get(key), value))
+                entry.setValue(newButton);
+        }
+        return this;
+    }
+
+    public @NotNull PaginatedInventoryBuilder setReplacementItemButtonByCustomData(@NotNull Object key, @Nullable Object value, @NotNull ItemButton item) {
+        for (Map.Entry<Character, ItemButton> entry : shapeReplacements.entrySet()) {
+            ItemButton button = entry.getValue();
+            if (Objects.equals(button.getCustomData().get(key), value))
+                entry.setValue(item);
+        }
+        return this;
+    }
+
+    public @NotNull PaginatedInventoryBuilder setReplacementItemButtonByCustomData(@NotNull Object key, @Nullable Object value, @NotNull Function<ItemButton, ItemButton> replacer) {
+        for (Map.Entry<Character, ItemButton> entry : shapeReplacements.entrySet()) {
+            ItemButton button = entry.getValue();
+            if (Objects.equals(button.getCustomData().get(key), value))
+                entry.setValue(replacer.apply(button));
+        }
+        return this;
+    }
+
     //rest of the code
 
     public PaginatedInventoryBuilder content(@Nullable List<ItemButton> content) {