Fix POIs not getting updated with assembly

Author: ryanhcode

common/src/main/java/dev/ryanhcode/sable/api/SubLevelAssemblyHelper.java

@@ -290,6 +290,7 @@ public static void moveBlocks(final ServerLevel level, final AssemblyTransform t
 
         final LevelAccelerator accelerator = new LevelAccelerator(level);
         final LevelAccelerator resultingAccelerator = new LevelAccelerator(resultingLevel);
+        final BlockState airState = Blocks.AIR.defaultBlockState();
 
         final List<BlockState> states = new ArrayList<>();
 
@@ -378,6 +379,8 @@ public static void moveBlocks(final ServerLevel level, final AssemblyTransform t
                 if (state.getBlock() instanceof final BlockSubLevelAssemblyListener listener) {
                     listener.afterMove(level, resultingLevel, state, block, newPos);
                 }
+
+                level.onBlockStateChange(newPos, airState, state);
             } catch (final Exception e) {
                 Sable.LOGGER.error("Failed to move block {} at {} to {}", state, block, newPos, e);
             }
@@ -391,7 +394,7 @@ public static void moveBlocks(final ServerLevel level, final AssemblyTransform t
             try {
                 final LevelChunk levelchunk = resultingAccelerator.getChunk(SectionPos.blockToSectionCoord(pos.getX()), SectionPos.blockToSectionCoord(pos.getZ()));
                 final BlockState subLevelState = states.get(i);
-                SubLevelAssemblyHelper.markAndNotifyBlock(resultingLevel, pos, levelchunk, Blocks.AIR.defaultBlockState(), subLevelState, 3, 512);
+                SubLevelAssemblyHelper.markAndNotifyBlock(resultingLevel, pos, levelchunk, airState, subLevelState, 3, 512);
             } catch (final Exception e) {
                 Sable.LOGGER.error("Failed to mark & notify block {} (untransformed = {})", pos, untransformed, e);
             }
@@ -402,21 +405,20 @@ public static void moveBlocks(final ServerLevel level, final AssemblyTransform t
         SableAssemblyPlatform.INSTANCE.setIgnoreOnPlace(resultingLevel, true);
         // destroy all the old blocks
         for (final BlockPos block : blocks) {
-            final BlockState subLevelState = Blocks.AIR.defaultBlockState();
-
             try {
                 final LevelChunk chunk = accelerator.getChunk(SectionPos.blockToSectionCoord(block.getX()),
                         SectionPos.blockToSectionCoord(block.getZ()));
 
-                chunk.setBlockState(block, subLevelState, true);
+                level.onBlockStateChange(block, chunk.getBlockState(block), airState);
+                chunk.setBlockState(block, airState, true);
             } catch (final Exception e) {
                 Sable.LOGGER.error("Failed to destroy old block during assembly {}", block, e);
             }
         }
         SableAssemblyPlatform.INSTANCE.setIgnoreOnPlace(resultingLevel, false);
 
         for (final BlockPos block : blocks) {
-            final BlockState subLevelState = Blocks.AIR.defaultBlockState();
+            final BlockState subLevelState = airState;
             resultingLevel.sendBlockUpdated(block, Blocks.STONE.defaultBlockState(), subLevelState, 3);
         }
     }

common/src/main/java/dev/ryanhcode/sable/util/SubLevelInclusiveLevelEntityGetter.java

@@ -1,7 +1,6 @@
 package dev.ryanhcode.sable.util;
 
 import dev.ryanhcode.sable.Sable;
-import dev.ryanhcode.sable.api.SubLevelHelper;
 import dev.ryanhcode.sable.companion.math.BoundingBox3d;
 import dev.ryanhcode.sable.sublevel.SubLevel;
 import net.minecraft.util.AbortableIterationConsumer;
@@ -90,7 +89,7 @@ public void get(AABB aABB, final Consumer<T> consumer) {
     }
 
     @Override
-    public <U extends T> void get(final EntityTypeTest<T, U> entityTypeTest, AABB aABB, final AbortableIterationConsumer<U> abortableIterationConsumer) {
+    public <U extends T> void get(final @NotNull EntityTypeTest<T, U> entityTypeTest, AABB aABB, final AbortableIterationConsumer<U> abortableIterationConsumer) {
         if (aABB.getSize() > MAX_GET_SIDE_LENGTH) {
             logError(aABB);
             return;