Use notification callback for when CasingComponent updates

Author: Swedz

src/main/java/aztech/modern_industrialization/machines/components/CasingComponent.java

@@ -24,7 +24,6 @@
 package aztech.modern_industrialization.machines.components;
 
 import aztech.modern_industrialization.api.energy.CableTier;
-import aztech.modern_industrialization.api.energy.CableTierHolder;
 import aztech.modern_industrialization.machines.IComponent;
 import aztech.modern_industrialization.machines.MachineBlockEntity;
 import aztech.modern_industrialization.machines.models.MachineCasing;
@@ -43,15 +42,31 @@
 import net.minecraft.world.level.block.Blocks;
 import org.jetbrains.annotations.Nullable;
 
-public class CasingComponent implements IComponent, DropableComponent, CableTierHolder {
+public class CasingComponent implements IComponent, DropableComponent {
 
-    protected ItemStack casingStack = ItemStack.EMPTY;
-    protected CableTier currentTier = CableTier.LV;
+    @FunctionalInterface
+    public interface UpdatedCallback {
+        void onUpdated(CableTier from, CableTier to);
+    }
+
+    @Nullable
+    private final UpdatedCallback callback;
+
+    private ItemStack casingStack = ItemStack.EMPTY;
+    private CableTier currentTier = CableTier.LV;
+
+    public CasingComponent(@Nullable UpdatedCallback callback) {
+        this.callback = callback;
+    }
+
+    public CasingComponent() {
+        this(null);
+    }
 
     /**
      * Sets the current casing stack and update {@link #currentTier} accordingly.
      */
-    protected void setCasingStack(ItemStack stack) {
+    private void setCasingStack(ItemStack stack) {
         casingStack = stack;
 
         // Compute tier
@@ -81,7 +96,6 @@ public void readClientNbt(CompoundTag tag, HolderLookup.Provider registries) {
         currentTier = CableTier.getTier(tag.getString("casing"));
     }
 
-    @Override
     public CableTier getCableTier() {
         return currentTier;
     }
@@ -93,6 +107,7 @@ public void dropCasing(Level world, BlockPos pos) {
     public ItemInteractionResult onUse(MachineBlockEntity be, Player player, InteractionHand hand) {
         ItemStack stackInHand = player.getItemInHand(hand);
         if (stackInHand.getCount() >= 1) {
+            var previousTier = currentTier;
             var newTier = getCasingTier(stackInHand.getItem());
             if (newTier != null && newTier != currentTier) {
                 if (currentTier != CableTier.LV) {
@@ -109,13 +124,16 @@ public ItemInteractionResult onUse(MachineBlockEntity be, Player player, Interac
                 be.getLevel().updateNeighborsAt(be.getBlockPos(), Blocks.AIR);
                 // Play a nice sound :)
                 playCasingPlaceSound(be);
+                if (callback != null) {
+                    callback.onUpdated(previousTier, currentTier);
+                }
                 return ItemInteractionResult.sidedSuccess(be.getLevel().isClientSide);
             }
         }
         return ItemInteractionResult.PASS_TO_DEFAULT_BLOCK_INTERACTION;
     }
 
-    protected void playCasingPlaceSound(MachineBlockEntity be) {
+    private void playCasingPlaceSound(MachineBlockEntity be) {
         var blockKey = currentTier.itemKey;
         if (blockKey == null) {
             return; // no sound for LV
@@ -148,11 +166,15 @@ public ItemStack getDrop() {
     }
 
     public void setCasingServer(MachineBlockEntity be, ItemStack casing) {
+        var previousTier = currentTier;
         setCasingStack(casing);
         be.setChanged();
         be.sync();
         be.getLevel().updateNeighborsAt(be.getBlockPos(), Blocks.AIR);
         playCasingPlaceSound(be);
+        if (callback != null && previousTier != currentTier) {
+            callback.onUpdated(previousTier, currentTier);
+        }
     }
 
     public MachineCasing getCasing() {