From a82a15641964c8233e1c3714d97b3c1ac7de259b Mon Sep 17 00:00:00 2001
From: Camotoy <20743703+Camotoy@users.noreply.github.com>
Date: Mon, 29 Apr 2024 01:03:18 -0400
Subject: [PATCH] Firework shapes

---
 .../org/geysermc/geyser/item/type/FireworkRocketItem.java    | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/core/src/main/java/org/geysermc/geyser/item/type/FireworkRocketItem.java b/core/src/main/java/org/geysermc/geyser/item/type/FireworkRocketItem.java
index 1265956da..bba13e753 100644
--- a/core/src/main/java/org/geysermc/geyser/item/type/FireworkRocketItem.java
+++ b/core/src/main/java/org/geysermc/geyser/item/type/FireworkRocketItem.java
@@ -92,10 +92,7 @@ public class FireworkRocketItem extends Item {
     static NbtMap translateExplosionToBedrock(Fireworks.FireworkExplosion explosion) {
         NbtMapBuilder newExplosionData = NbtMap.builder();
 
-//        if (explosion.get("Type") != null) {
-//            newExplosionData.put(new ByteTag("FireworkType", MathUtils.getNbtByte(explosion.get("Type").getValue())));
-//        }
-        //newExplosionData.putByte("FireworkType", explosion.get) //TODO???
+        newExplosionData.putByte("FireworkType", (byte) explosion.getShapeId());
 
         int[] oldColors = explosion.getColors();
         byte[] colors = new byte[oldColors.length];