Implement 1.16.100 anvil renaming support

bundabrg · Jan 12, 2021 · 6a9d4cd · 6a9d4cd
1 parent 4db9ec9
commit 6a9d4cd
Showing 1 changed file with 60 additions and 0 deletions.
diff --git a/...eysermc/connector/network/translators/inventory/translators/AnvilInventoryTranslator.java b/...eysermc/connector/network/translators/inventory/translators/AnvilInventoryTranslator.java
@@ -26,20 +26,80 @@
 package org.geysermc.connector.network.translators.inventory.translators;
 
 import com.github.steveice10.mc.protocol.data.game.window.WindowType;
+import com.github.steveice10.mc.protocol.packet.ingame.client.window.ClientRenameItemPacket;
+import com.nukkitx.nbt.NbtMap;
 import com.nukkitx.protocol.bedrock.data.inventory.ContainerSlotType;
 import com.nukkitx.protocol.bedrock.data.inventory.ContainerType;
+import com.nukkitx.protocol.bedrock.data.inventory.ItemData;
 import com.nukkitx.protocol.bedrock.data.inventory.StackRequestSlotInfoData;
+import com.nukkitx.protocol.bedrock.data.inventory.stackrequestactions.CraftResultsDeprecatedStackRequestActionData;
+import com.nukkitx.protocol.bedrock.data.inventory.stackrequestactions.StackRequestActionData;
+import com.nukkitx.protocol.bedrock.data.inventory.stackrequestactions.StackRequestActionType;
+import com.nukkitx.protocol.bedrock.packet.ItemStackRequestPacket;
+import com.nukkitx.protocol.bedrock.packet.ItemStackResponsePacket;
 import org.geysermc.connector.inventory.AnvilContainer;
+import org.geysermc.connector.inventory.GeyserItemStack;
 import org.geysermc.connector.inventory.Inventory;
 import org.geysermc.connector.inventory.PlayerInventory;
+import org.geysermc.connector.network.session.GeyserSession;
 import org.geysermc.connector.network.translators.inventory.BedrockContainerSlot;
 import org.geysermc.connector.network.translators.inventory.updater.UIInventoryUpdater;
+import org.geysermc.connector.network.translators.item.ItemTranslator;
 
 public class AnvilInventoryTranslator extends AbstractBlockInventoryTranslator {
     public AnvilInventoryTranslator() {
         super(3, "minecraft:anvil[facing=north]", ContainerType.ANVIL, UIInventoryUpdater.INSTANCE);
     }
 
+    /* 1.16.100 support start */
+    @Override
+    @Deprecated
+    public boolean shouldHandleRequestFirst(StackRequestActionData action, Inventory inventory) {
+        return action.getType() == StackRequestActionType.CRAFT_NON_IMPLEMENTED_DEPRECATED;
+    }
+
+    @Override
+    @Deprecated
+    public ItemStackResponsePacket.Response translateSpecialRequest(GeyserSession session, Inventory inventory, ItemStackRequestPacket.Request request) {
+        if (!(request.getActions()[1] instanceof CraftResultsDeprecatedStackRequestActionData)) {
+            // Just silently log an error
+            session.getConnector().getLogger().debug("Something isn't quite right with taking an item out of an anvil.");
+            return translateRequest(session, inventory, request);
+        }
+        CraftResultsDeprecatedStackRequestActionData actionData = (CraftResultsDeprecatedStackRequestActionData) request.getActions()[1];
+        ItemData resultItem = actionData.getResultItems()[0];
+        if (resultItem.getTag() != null) {
+            NbtMap displayTag = resultItem.getTag().getCompound("display");
+            if (displayTag != null && displayTag.containsKey("Name")) {
+                ItemData sourceSlot = inventory.getItem(0).getItemData(session);
+
+                if (sourceSlot.getTag() != null) {
+                    NbtMap oldDisplayTag = sourceSlot.getTag().getCompound("display");
+                    if (oldDisplayTag != null && oldDisplayTag.containsKey("Name")) {
+                        if (!displayTag.getString("Name").equals(oldDisplayTag.getString("Name"))) {
+                            // Name has changed
+                            sendRenamePacket(session, inventory, resultItem, displayTag.getString("Name"));
+                        }
+                    } else {
+                        // No display tag on the old item
+                        sendRenamePacket(session, inventory, resultItem, displayTag.getString("Name"));
+                    }
+                } else {
+                    // New NBT tag
+                    sendRenamePacket(session, inventory, resultItem, displayTag.getString("Name"));
+                }
+            }
+        }
+        return translateRequest(session, inventory, request);
+    }
+
+    private void sendRenamePacket(GeyserSession session, Inventory inventory, ItemData outputItem, String name) {
+        session.sendDownstreamPacket(new ClientRenameItemPacket(name));
+        inventory.setItem(2, GeyserItemStack.from(ItemTranslator.translateToJava(outputItem)), session);
+    }
+
+    /* 1.16.100 support end */
+
     @Override
     public int bedrockSlotToJava(StackRequestSlotInfoData slotInfoData) {
         if (slotInfoData.getContainer() == ContainerSlotType.ANVIL_INPUT) {