Also handle item_name item data for component rewriting

Author: kennytv

common/src/main/java/com/viaversion/viaversion/rewriter/StructuredItemRewriter.java

@@ -69,14 +69,8 @@ public Item handleItemToClient(UserConnection connection, Item item) {
         final ComponentRewriter componentRewriter = protocol.getComponentRewriter();
         if (componentRewriter != null) {
             // Handle name and lore components
-            final StructuredData<Tag> customNameData = dataContainer.getNonEmpty(StructuredDataKey.CUSTOM_NAME);
-            if (customNameData != null) {
-                final Tag originalName = customNameData.value().copy();
-                componentRewriter.processTag(connection, customNameData.value());
-                if (!customNameData.value().equals(originalName)) {
-                    saveTag(createCustomTag(item), originalName, "Name");
-                }
-            }
+            updateComponent(connection, item, StructuredDataKey.ITEM_NAME, "item_name");
+            updateComponent(connection, item, StructuredDataKey.CUSTOM_NAME, "custom_name");
 
             final StructuredData<Tag[]> loreData = dataContainer.getNonEmpty(StructuredDataKey.LORE);
             if (loreData != null) {
@@ -164,6 +158,19 @@ protected void updateItemComponents(UserConnection connection, StructuredDataCon
         }
     }
 
+    private void updateComponent(final UserConnection connection, final Item item, final StructuredDataKey<Tag> key, final String backupKey) {
+        final StructuredData<Tag> name = item.dataContainer().getNonEmpty(key);
+        if (name == null) {
+            return;
+        }
+
+        final Tag originalName = name.value().copy();
+        protocol.getComponentRewriter().processTag(connection, name.value());
+        if (!name.value().equals(originalName)) {
+            saveTag(createCustomTag(item), originalName, backupKey);
+        }
+    }
+
     protected void restoreTextComponents(final Item item) {
         final StructuredDataContainer data = item.dataContainer();
         final StructuredData<CompoundTag> customData = data.getNonEmpty(StructuredDataKey.CUSTOM_DATA);
@@ -172,12 +179,17 @@ protected void restoreTextComponents(final Item item) {
         }
 
         // Remove custom name
-        if (customData.value().remove(nbtTagName("customName")) != null) {
+        if (customData.value().remove(nbtTagName("added_custom_name")) != null) {
             data.remove(StructuredDataKey.CUSTOM_NAME);
         } else {
-            final Tag name = removeBackupTag(customData.value(), "Name");
-            if (name != null) {
-                data.set(StructuredDataKey.CUSTOM_NAME, name);
+            final Tag customName = removeBackupTag(customData.value(), "custom_name");
+            if (customName != null) {
+                data.set(StructuredDataKey.CUSTOM_NAME, customName);
+            }
+
+            final Tag itemName = removeBackupTag(customData.value(), "item_name");
+            if (itemName != null) {
+                data.set(StructuredDataKey.ITEM_NAME, itemName);
             }
         }
     }