1 files changed, 17 insertions, 14 deletions
diff --git a/src/main/java/los/numeros/utils/GiveItem.java b/src/main/java/los/numeros/utils/GiveItem.java
index d42207d..aaaf35c 100644
--- a/src/main/java/los/numeros/utils/GiveItem.java
+++ b/src/main/java/los/numeros/utils/GiveItem.java
@@ -1,13 +1,7 @@
 package los.numeros.utils;
 
-import java.text.AttributedCharacterIterator.Attribute;
+import org.jetbrains.annotations.Nullable;
 
-import net.minecraft.enchantment.Enchantment;
-import net.minecraft.enchantment.Enchantments;
-import net.minecraft.enchantment.UnbreakingEnchantment;
-import net.minecraft.entity.attribute.EntityAttribute;
-import net.minecraft.entity.attribute.EntityAttributeModifier;
-import net.minecraft.item.Item;
 import net.minecraft.item.ItemStack;
 import net.minecraft.server.network.ServerPlayerEntity;
 import net.minecraft.util.Identifier;
@@ -19,18 +13,27 @@ public class GiveItem {
   public static final int ARMOR_LEGS = 101;
   public static final int ARMOR_BOOTS = 100;
 
-  public static void givePlayerItem(ServerPlayerEntity player, String itemID) {
-    ItemStack item = new ItemStack(Registry.ITEM.get(new Identifier(itemID)));
-    player.inventory.insertStack(item);
+  public static void givePlayerItem(ServerPlayerEntity player, String itemID, @Nullable Integer slot) {
+    ItemStack item = getItem(itemID);
+    if (slot == null) {
+      player.inventory.insertStack(item);
+    } else {
+      player.equip(slot, item);
+    }
   }
 
-  public static ItemStack givePlayerSpecialItem(ServerPlayerEntity player, String itemID) {
+  public static ItemStack getItem(String itemID) {
     ItemStack item = new ItemStack(Registry.ITEM.get(new Identifier(itemID)));
     return item;
   }
 
-  public static void setPlayerSlot(ServerPlayerEntity player, String itemID, int slot) {
-    ItemStack item = new ItemStack(Registry.ITEM.get(new Identifier(itemID)));
-    player.equip(slot, item);
+  public static void giveMultipleItem(ServerPlayerEntity player, String itemID, @Nullable Integer slot, int quantity) {
+    ItemStack item = getItem(itemID);
+    item.increment(quantity - 1);
+    if (slot == null) {
+      player.inventory.insertStack(item);
+    } else {
+      player.equip(slot, item);
+    }
   }
 }