Enhanced shell synchronization

Kir-Antipov · Aug 6, 2022 · 1a27137 · 1a27137
1 parent d8d32bb
commit 1a27137
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main/java/dev/kir/sync/block/entity/AbstractShellContainerBlockEntity.java b/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);
  }