From 862ef03345054353b133842d51a38f61cff28e60 Mon Sep 17 00:00:00 2001
From: libraryaddict <github@lib.co.nz>
Date: Sun, 21 Jul 2024 18:39:46 +1200
Subject: [PATCH] Try add custom model support

---
 .../params/types/custom/ParamInfoItemStack.java       | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/plugin/src/main/java/me/libraryaddict/disguise/utilities/params/types/custom/ParamInfoItemStack.java b/plugin/src/main/java/me/libraryaddict/disguise/utilities/params/types/custom/ParamInfoItemStack.java
index 5000c54e..313c7559 100644
--- a/plugin/src/main/java/me/libraryaddict/disguise/utilities/params/types/custom/ParamInfoItemStack.java
+++ b/plugin/src/main/java/me/libraryaddict/disguise/utilities/params/types/custom/ParamInfoItemStack.java
@@ -58,6 +58,10 @@ public class ParamInfoItemStack<I extends ItemStack> extends ParamInfoEnum<Objec
                 name += ":" + TranslateType.DISGUISE_OPTIONS_PARAMETERS.get("glow");
             }
 
+            if (NmsVersion.v1_14.isSupported() && item.getItemMeta().hasCustomModelData()) {
+                name += ":custom_model_" + item.getItemMeta().getCustomModelData();
+            }
+
             return name;
         }
 
@@ -172,6 +176,7 @@ public class ParamInfoItemStack<I extends ItemStack> extends ParamInfoEnum<Objec
 
         Integer amount = null;
         boolean enchanted = false;
+        Integer customModel = null;
 
         for (int i = 1; i < split.length; i++) {
             String s = split[i];
@@ -180,6 +185,8 @@ public class ParamInfoItemStack<I extends ItemStack> extends ParamInfoEnum<Objec
                 enchanted = true;
             } else if (s.matches("\\d+") && amount == null) {
                 amount = Integer.parseInt(s);
+            } else if (NmsVersion.v1_14.isSupported() && s.toLowerCase(Locale.ENGLISH).matches("^custom_model_-?\\d+$")) {
+                customModel = Integer.parseInt(s.split("_", 3)[2]);
             } else {
                 throw new IllegalArgumentException();
             }
@@ -191,6 +198,10 @@ public class ParamInfoItemStack<I extends ItemStack> extends ParamInfoEnum<Objec
             itemStack.addUnsafeEnchantment(DisguiseUtilities.getDurabilityEnchantment(), 1);
         }
 
+        if (customModel != null) {
+            itemStack.getItemMeta().setCustomModelData(customModel);
+        }
+
         return itemStack;
     }