Support changing the displayed BlockState of PrimedTNT

avaruus1 · Feb 25, 2024 · cbc7b25 · cbc7b25
1 parent 744fd7e
commit cbc7b25
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/main/java/org/spongepowered/common/data/provider/entity/TNTData.java b/src/main/java/org/spongepowered/common/data/provider/entity/TNTData.java
@@ -26,6 +26,7 @@
 
 import net.minecraft.world.entity.LivingEntity;
 import net.minecraft.world.entity.item.PrimedTnt;
+import org.spongepowered.api.block.BlockState;
 import org.spongepowered.api.data.Keys;
 import org.spongepowered.api.entity.living.Living;
 import org.spongepowered.common.accessor.world.entity.item.PrimedTntAccessor;
@@ -44,7 +45,10 @@ public static void register(final DataProviderRegistrator registrator) {
                         .get(h -> (Living) h.getOwner())
                         .set((h, v) -> ((PrimedTntAccessor) h).accessor$owner((LivingEntity) v))
                     .create(Keys.IS_PRIMED)
-                        .get(h -> !h.isRemoved() && h.getFuse() > 0);
+                        .get(h -> !h.isRemoved() && h.getFuse() > 0)
+                    .create(Keys.BLOCK_STATE)
+                        .get(h -> (BlockState) h.getBlockState())
+                        .set((h, v) -> h.setBlockState((net.minecraft.world.level.block.state.BlockState) v));
     }
     // @formatter:on
 }