diff --git a/src/main/java/com/atsuishio/superbwarfare/init/ModTabs.java b/src/main/java/com/atsuishio/superbwarfare/init/ModTabs.java index 8599a8912..628f996c0 100644 --- a/src/main/java/com/atsuishio/superbwarfare/init/ModTabs.java +++ b/src/main/java/com/atsuishio/superbwarfare/init/ModTabs.java @@ -2,6 +2,7 @@ package com.atsuishio.superbwarfare.init; import com.atsuishio.superbwarfare.ModUtils; import com.atsuishio.superbwarfare.item.ArmorPlate; +import com.atsuishio.superbwarfare.item.BatteryItem; import com.atsuishio.superbwarfare.item.gun.handgun.*; import com.atsuishio.superbwarfare.item.gun.heavy.Ntw20Item; import com.atsuishio.superbwarfare.item.gun.launcher.JavelinItem; @@ -127,6 +128,9 @@ public class ModTabs { if (registryObject.get() == ModItems.ARMOR_PLATE.get()) { output.accept(ArmorPlate.getInfiniteInstance()); } + if (registryObject.get() instanceof BatteryItem batteryItem) { + output.accept(batteryItem.makeFullEnergyStack()); + } } })) .build()); diff --git a/src/main/java/com/atsuishio/superbwarfare/item/BatteryItem.java b/src/main/java/com/atsuishio/superbwarfare/item/BatteryItem.java index 7a453f245..15b8feec7 100644 --- a/src/main/java/com/atsuishio/superbwarfare/item/BatteryItem.java +++ b/src/main/java/com/atsuishio/superbwarfare/item/BatteryItem.java @@ -62,4 +62,12 @@ public class BatteryItem extends Item { public @NotNull Optional getTooltipImage(@NotNull ItemStack pStack) { return Optional.of(new CellImageComponent(pStack)); } + + public ItemStack makeFullEnergyStack() { + ItemStack stack = new ItemStack(this); + stack.getCapability(ForgeCapabilities.ENERGY).ifPresent( + e -> e.receiveEnergy(maxEnergy, false) + ); + return stack; + } }