Merge pull request #46 from D0gmaDev/patch-1

Add more item methods

Author: NichtStudioCode

inventoryaccess/inventory-access/src/main/java/xyz/xenondevs/inventoryaccess/component/AdventureComponentWrapper.java

@@ -29,7 +29,7 @@ public AdventureComponentWrapper(Component component) {
     }
 
     @Override
-    public @NotNull ComponentWrapper withoutPreFormatting() {
+    public @NotNull AdventureComponentWrapper withoutPreFormatting() {
         return new AdventureComponentWrapper(AdventureComponentUtils.withoutPreFormatting(component));
     }
 

invui-core/src/main/java/xyz/xenondevs/invui/item/builder/AbstractItemBuilder.java

@@ -32,6 +32,7 @@ public abstract class AbstractItemBuilder<S> implements ItemProvider {
     protected int amount = 1;
     protected int damage;
     protected int customModelData;
+    protected Boolean unbreakable;
     protected ComponentWrapper displayName;
     protected List<ComponentWrapper> lore;
     protected List<ItemFlag> itemFlags;
@@ -116,6 +117,10 @@ public AbstractItemBuilder(@NotNull ItemStack base) {
             if (customModelData != 0)
                 itemMeta.setCustomModelData(customModelData);
 
+            // unbreakable
+            if (unbreakable != null)
+                itemMeta.setUnbreakable(unbreakable);
+            
             // enchantments
             if (enchantments != null) {
                 if (base != null)
@@ -201,6 +206,16 @@ public int getCustomModelData() {
         return (S) this;
     }
 
+    public @Nullable Boolean isUnbreakable() {
+        return unbreakable;
+    }
+    
+    @Contract("_ -> this")
+    public @NotNull S setUnbreakable(boolean unbreakable) {
+        this.unbreakable = unbreakable;
+        return (S) this;
+    }
+    
     public @Nullable ComponentWrapper getDisplayName() {
         return displayName;
     }
@@ -229,15 +244,15 @@ public int getCustomModelData() {
     }
 
     @Contract("_ -> this")
-    public @NotNull S setLore(List<ComponentWrapper> lore) {
+    public @NotNull S setLore(@NotNull List<@NotNull ComponentWrapper> lore) {
         this.lore = lore.stream()
             .map(ComponentWrapper::withoutPreFormatting)
             .collect(Collectors.toList());
         return (S) this;
     }
 
     @Contract("_ -> this")
-    public @NotNull S setLegacyLore(@NotNull List<String> lore) {
+    public @NotNull S setLegacyLore(@NotNull List<@NotNull String> lore) {
         this.lore = lore.stream()
             .map(line -> new BungeeComponentWrapper(TextComponent.fromLegacyText(line)).withoutPreFormatting())
             .collect(Collectors.toList());
@@ -273,6 +288,26 @@ public int getCustomModelData() {
 
         return (S) this;
     }
+    
+    @Contract("_ -> this")
+    public @NotNull S addLoreLines(@NotNull List<@NotNull ComponentWrapper> lines) {
+        if (lore == null) lore = new ArrayList<>();
+        
+        for (ComponentWrapper line : lines)
+            lore.add(line.withoutPreFormatting());
+        
+        return (S) this;
+    }
+    
+    @Contract("_ -> this")
+    public @NotNull S addLegacyLoreLines(@NotNull List<@NotNull String> lines) {
+        if (lore == null) lore = new ArrayList<>();
+        
+        for (String line : lines)
+            lore.add(new BungeeComponentWrapper(TextComponent.fromLegacyText(line)).withoutPreFormatting());
+        
+        return (S) this;
+    }
     //</editor-fold>
 
     //<editor-fold desc="item flags">
@@ -293,6 +328,12 @@ public int getCustomModelData() {
         return (S) this;
     }
 
+    @Contract("-> this")
+    public @NotNull S addAllItemFlags() {
+        this.itemFlags = new ArrayList<>(Arrays.asList(ItemFlag.values()));
+        return (S) this;
+    }
+    
     @Contract("_ -> this")
     public @NotNull S removeItemFlags(@NotNull ItemFlag... itemFlags) {
         if (this.itemFlags != null)
@@ -327,8 +368,7 @@ public int getCustomModelData() {
 
     @Contract("_ -> this")
     public @NotNull S removeEnchantment(Enchantment enchantment) {
-        if (enchantments == null) enchantments = new HashMap<>();
-        enchantments.remove(enchantment);
+        if (enchantments != null) enchantments.remove(enchantment);
         return (S) this;
     }
 

invui-core/src/main/java/xyz/xenondevs/invui/item/builder/FireworkBuilder.java

@@ -7,12 +7,14 @@
 import org.jetbrains.annotations.Contract;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
+import org.jetbrains.annotations.Range;
 
 import java.util.ArrayList;
 import java.util.List;
 
 public final class FireworkBuilder extends AbstractItemBuilder<FireworkBuilder> {
 
+    private int power = -1;
     private List<FireworkEffect> effects = new ArrayList<>();
 
     public FireworkBuilder() {
@@ -27,6 +29,12 @@ public FireworkBuilder(@NotNull ItemStack base) {
         super(base);
     }
 
+    @Contract("_ -> this")
+    public @NotNull FireworkBuilder setPower(@Range(from = 0, to = 127) int power) {
+        this.power = power;
+        return this;
+    }
+    
     @Contract("_ -> this")
     public @NotNull FireworkBuilder addFireworkEffect(@NotNull FireworkEffect effect) {
         effects.add(effect);
@@ -57,6 +65,7 @@ public FireworkBuilder(@NotNull ItemStack base) {
         ItemStack item = super.get(lang);
         FireworkMeta meta = (FireworkMeta) item.getItemMeta();
 
+        if (power != -1) meta.setPower(power);
         meta.clearEffects();
         meta.addEffects(effects);