Changed the map creation and item name check to remove possibility of…

… index out of range error and key not found error.
MrGII · Jul 28, 2024 · 5e62f69 · 5e62f69
1 parent 6da6e1a
commit 5e62f69
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/github/mrgii/itemsstack/util/ItemsStackSizeModifier.java b/src/main/java/github/mrgii/itemsstack/util/ItemsStackSizeModifier.java
@@ -16,15 +16,15 @@ public static void ModifyPotionStackSize(List<String> itemNames, List<Integer> m
         ItemsStack.LOGGER.info("Changing stack size of minecraft potions.");
         for (Item item : Registries.ITEM) {
             String name = item.getName().getString();
-            if (itemNames.contains(name)) {
+            if (itemToMaxStackSize.containsKey(name)) {
                 ((ItemAccessor) item).setComponents(ComponentMap.of(item.getComponents(), ComponentMap.builder().add(DataComponentTypes.MAX_STACK_SIZE, itemToMaxStackSize.get(name)).build()));
             }
         }
     }
 
     private static Map<String, Integer> createItemToStackSizeMap(List<String> itemNames, List<Integer> maxStackSizes) {
         Map<String, Integer> itemToMaxStackSize = new java.util.HashMap<>(Map.of());
-        for (int i = 0; i < itemNames.size(); i++) {
+        for (int i = 0; i < Math.min(itemNames.size(), maxStackSizes.size()); i++) {
             itemToMaxStackSize.put(itemNames.get(i), maxStackSizes.get(i));
         }
         return itemToMaxStackSize;