Enhanced shell synchronization

Author: Kira-NT

src/main/java/dev/kir/sync/block/entity/AbstractShellContainerBlockEntity.java

@@ -63,6 +63,9 @@ public AbstractShellContainerBlockEntity(BlockEntityType<?> type, BlockPos pos,
     @Override
     public void setShellState(ShellState shell) {
         this.shell = shell;
+        if (this.world != null && !this.world.isClient && this.pos != null && this.getCachedState() != null) {
+            this.checkShellState(this.world, this.pos, this.getCachedState());
+        }
     }
 
     @Override
@@ -97,6 +100,10 @@ protected Optional<AbstractShellContainerBlockEntity> getBottomPart() {
 
     @Override
     public void onServerTick(World world, BlockPos pos, BlockState state) {
+        this.checkShellState(world, pos, state);
+    }
+
+    private void checkShellState(World world, BlockPos pos, BlockState state) {
         if (this.shell != null && this.shell.getColor() != this.color) {
             this.shell.setColor(this.color);
         }