Fix unbreakable upgrade

Author: Mrbysco

src/main/java/mrbysco/constructionstick/items/stick/ItemStick.java

@@ -98,7 +98,7 @@ public int remainingDurability(ItemStack stack) {
 	}
 
 	public void hurtItem(ItemStack stack, int amount, LivingEntity entity, EquipmentSlot slot) {
-		if (stack.has(ModDataComponents.BATTERY)) {
+		if (stack.has(ModDataComponents.BATTERY_ENABLED)) {
 			if (entity.hasInfiniteMaterials()) return;
 			IEnergyStorage storage = stack.getCapability(Capabilities.EnergyStorage.ITEM);
 			if (storage != null) {
@@ -150,7 +150,7 @@ public void appendHoverText(@NotNull ItemStack itemstack, TooltipContext context
 			StickUpgradesSelectable<IStickTemplate> upgrades = options.upgrades;
 			lines.add(Component.translatable(langTooltip + "blocks", limit).withStyle(ChatFormatting.GRAY));
 
-			if (itemstack.has(ModDataComponents.BATTERY)) {
+			if (itemstack.has(ModDataComponents.BATTERY_ENABLED)) {
 				IEnergyStorage storage = itemstack.getCapability(Capabilities.EnergyStorage.ITEM);
 				if (storage != null) {
 					int energy = storage.getEnergyStored();

src/main/java/mrbysco/constructionstick/items/stick/ItemStickBasic.java

@@ -28,7 +28,7 @@ public boolean isBarVisible(ItemStack stack) {
 
 	@Override
 	public int getBarWidth(ItemStack stack) {
-		if (stack.has(ModDataComponents.BATTERY)) {
+		if (stack.has(ModDataComponents.BATTERY_ENABLED)) {
 			IEnergyStorage storage = stack.getCapability(Capabilities.EnergyStorage.ITEM);
 			if (storage != null) {
 				return Math.round((13.0F / storage.getMaxEnergyStored() * storage.getEnergyStored()));
@@ -39,7 +39,7 @@ public int getBarWidth(ItemStack stack) {
 
 	@Override
 	public int getBarColor(ItemStack stack) {
-		if (stack.has(ModDataComponents.BATTERY)) {
+		if (stack.has(ModDataComponents.BATTERY_ENABLED)) {
 			return 0x971607;
 		}
 		return super.getBarColor(stack);
@@ -53,12 +53,14 @@ public int getMaxDamage(ItemStack stack) {
 	@Override
 	public int remainingDurability(ItemStack stack) {
 		if (stack.has(ModDataComponents.UNBREAKABLE)) {
+			return Integer.MAX_VALUE;
+		}
+		if (stack.has(ModDataComponents.BATTERY_ENABLED)) {
 			IEnergyStorage storage = stack.getCapability(Capabilities.EnergyStorage.ITEM);
 			if (storage != null) {
 				int usage = ConstructionConfig.getStickProperties(this).getBatteryUsage();
 				return storage.getEnergyStored() / usage;
 			}
-			return 0;
 		}
 		return stack.getMaxDamage() - stack.getDamageValue();
 	}