Add Item.Builder#setCyclingItemProvider

Author: NichtStudioCode

invui/src/main/java/xyz/xenondevs/invui/internal/util/ArrayUtils.java

@@ -5,6 +5,7 @@
 
 import java.util.HashMap;
 import java.util.Map;
+import java.util.function.IntFunction;
 import java.util.function.Predicate;
 
 public class ArrayUtils {
@@ -29,9 +30,8 @@ public static <T> Map<Integer, T> findAllOccurrences(@Nullable T[] array, Predic
     }
 
     @NullUnmarked
-    @SuppressWarnings("unchecked")
-    public static <T> T[] concat(T first, T[] rest) {
-        T[] result = (T[]) new Object[rest.length + 1];
+    public static <T> T[] concat(IntFunction<T[]> newArray, T first, T[] rest) {
+        T[] result = newArray.apply(rest.length + 1);
         result[0] = first;
         System.arraycopy(rest, 0, result, 1, rest.length);
         return result;

invui/src/main/java/xyz/xenondevs/invui/inventory/CompositeInventory.java

@@ -26,7 +26,7 @@ public final class CompositeInventory extends Inventory {
      */
     public CompositeInventory(Inventory first, Inventory... other) {
         super(calculateSize(first, other));
-        this.inventories = ArrayUtils.concat(first, other);
+        this.inventories = ArrayUtils.concat(Inventory[]::new, first, other);
     }
 
     /**

invui/src/main/java/xyz/xenondevs/invui/inventory/Inventory.java

@@ -904,7 +904,7 @@ public int[] simulateAdd(ItemStack first, ItemStack... rest) {
         if (rest.length == 0) {
             return new int[] {simulateSingleAdd(first)};
         } else {
-            ItemStack[] allStacks = ArrayUtils.concat(first, rest);
+            ItemStack[] allStacks = ArrayUtils.concat(ItemStack[]::new, first, rest);
             return simulateMultiAdd(Arrays.asList(allStacks));
         }
     }
@@ -940,7 +940,7 @@ public boolean canHold(ItemStack first, ItemStack... rest) {
         if (rest.length == 0) {
             return simulateSingleAdd(first) == 0;
         } else {
-            ItemStack[] allStacks = ArrayUtils.concat(first, rest);
+            ItemStack[] allStacks = ArrayUtils.concat(ItemStack[]::new, first, rest);
             return Arrays.stream(simulateMultiAdd(Arrays.asList(allStacks))).allMatch(i -> i == 0);
         }
     }

invui/src/main/java/xyz/xenondevs/invui/item/CustomBoundItem.java

@@ -14,6 +14,7 @@
 import xyz.xenondevs.invui.util.TriConsumer;
 import xyz.xenondevs.invui.window.AbstractWindow;
 
+import java.util.List;
 import java.util.function.*;
 
 class CustomBoundItem<G extends Gui> extends AbstractBoundItem {
@@ -104,6 +105,23 @@ public Builder<G> setItemProvider(BiFunction<Player, G, ItemProvider> itemProvid
             return this;
         }
 
+        @Override
+        public BoundItem.Builder<G> setCyclingItemProvider(int period, List<? extends ItemProvider> itemProviders) {
+            if (itemProviders.isEmpty())
+                throw new IllegalArgumentException("itemProviders must not be empty");
+            
+            if (itemProviders.size() > 1) {
+                updatePeriodically(period);
+                this.itemProviderFn = (viewer, gui) -> {
+                    int i = (Bukkit.getCurrentTick() / period) % itemProviders.size();
+                    return itemProviders.get(i);
+                };
+            } else {
+                setItemProvider(itemProviders.getFirst());
+            }
+            return this;
+        }
+        
         @Override
         public Builder<G> async(ItemProvider placeholder) {
             this.asyncPlaceholder = placeholder;

invui/src/main/java/xyz/xenondevs/invui/item/CustomItem.java

@@ -8,6 +8,7 @@
 import xyz.xenondevs.invui.InvUI;
 import xyz.xenondevs.invui.window.AbstractWindow;
 
+import java.util.List;
 import java.util.function.BiConsumer;
 import java.util.function.Consumer;
 import java.util.function.Function;
@@ -78,6 +79,23 @@ public Builder setItemProvider(Function<Player, ItemProvider> itemProvider) {
             return this;
         }
 
+        @Override
+        public Builder setCyclingItemProvider(int period, List<? extends ItemProvider> itemProviders) {
+            if (itemProviders.isEmpty())
+                throw new IllegalArgumentException("itemProviders must not be empty");
+            
+            if (itemProviders.size() > 1) {
+                updatePeriodically(period);
+                this.itemProviderFn = viewer -> {
+                    int i = (Bukkit.getCurrentTick() / period) % itemProviders.size();
+                    return itemProviders.get(i);
+                };
+            } else {
+                setItemProvider(itemProviders.getFirst());
+            }
+            return this;
+        }
+        
         @Override
         public Builder async(ItemProvider placeholder) {
             this.asyncPlaceholder = placeholder;

invui/src/main/java/xyz/xenondevs/invui/item/Item.java

@@ -4,8 +4,10 @@
 import org.bukkit.event.inventory.ClickType;
 import org.bukkit.inventory.ItemStack;
 import xyz.xenondevs.invui.gui.Gui;
+import xyz.xenondevs.invui.internal.util.ArrayUtils;
 import xyz.xenondevs.invui.window.Window;
 
+import java.util.List;
 import java.util.function.BiConsumer;
 import java.util.function.Consumer;
 import java.util.function.Function;
@@ -65,6 +67,12 @@ interface Builder<S extends Builder<S>> {
 
         S setItemProvider(Function<Player, ItemProvider> itemProvider);
 
+        default S setCyclingItemProvider(int period, ItemProvider itemProvider, ItemProvider... itemProviders) {
+            return setCyclingItemProvider(period, List.of(ArrayUtils.concat(ItemProvider[]::new, itemProvider, itemProviders)));
+        }
+        
+        S setCyclingItemProvider(int period, List<? extends ItemProvider> itemProviders);
+        
         S async(ItemProvider placeholder);
 
         S updatePeriodically(long period);