cleanup

Layers-of-Railways · Apr 20, 2024 · 8bac28d · 8bac28d
1 parent ed1f4a8
commit 8bac28d
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/...n/java/dev/ithundxr/createnumismatics/forge/mixin/self/VendorBlockEntityCapabilities.java b/...n/java/dev/ithundxr/createnumismatics/forge/mixin/self/VendorBlockEntityCapabilities.java
@@ -14,32 +14,39 @@
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Unique;
 
 @Mixin(VendorBlockEntity.class)
 public abstract class VendorBlockEntityCapabilities extends SmartBlockEntity implements ICapabilityProvider {
     public VendorBlockEntityCapabilities(BlockEntityType<?> type, BlockPos pos, BlockState state) {
         super(type, pos, state);
     }
 
-    private static Direction[] dirs = {Direction.UP, Direction.DOWN, Direction.NORTH};
-    LazyOptional<? extends IItemHandler>[] createNumismatics$handlers = SidedInvWrapper.create((WorldlyContainer) this, dirs);
+    @Unique private static final Direction[] numismatics$DIRECTIONS = {Direction.UP, Direction.DOWN, Direction.NORTH};
+    @Unique LazyOptional<? extends IItemHandler>[] numismatics$handlers = SidedInvWrapper.create((WorldlyContainer) this, numismatics$DIRECTIONS);
 
     @Override
     public <T> @NotNull LazyOptional<T> getCapability(@NotNull Capability<T> capability, @Nullable Direction facing) {
         if (remove || facing == null || capability != ForgeCapabilities.ITEM_HANDLER) {
             return super.getCapability(capability, facing);
         } else if (facing == Direction.UP) {
-            return createNumismatics$handlers[0].cast();
+            return numismatics$handlers[0].cast();
         } else {
-            return facing == Direction.DOWN ? createNumismatics$handlers[1].cast() : createNumismatics$handlers[2].cast();
+            return facing == Direction.DOWN ? numismatics$handlers[1].cast() : numismatics$handlers[2].cast();
         }
     }
 
+    @Override
+    public void reviveCaps() {
+        super.reviveCaps();
+        numismatics$handlers = SidedInvWrapper.create((WorldlyContainer) this, numismatics$DIRECTIONS);
+    }
+
     @Override
     public void invalidateCaps() {
         super.invalidateCaps();
 
-        for (LazyOptional<? extends IItemHandler> createNumismatics$handler : createNumismatics$handlers) {
+        for (LazyOptional<? extends IItemHandler> createNumismatics$handler : numismatics$handlers) {
             createNumismatics$handler.invalidate();
         }
     }