Add Potions to ItemResolutionQuery (#1067)

1 files changed, 9 insertions, 0 deletions

diff --git a/src/main/java/io/github/moulberry/notenoughupdates/util/ItemResolutionQuery.java b/src/main/java/io/github/moulberry/notenoughupdates/util/ItemResolutionQuery.java
index 2cccce72..9d279824 100644
--- a/src/main/java/io/github/moulberry/notenoughupdates/util/ItemResolutionQuery.java
+++ b/src/main/java/io/github/moulberry/notenoughupdates/util/ItemResolutionQuery.java
@@ -126,6 +126,9 @@ public class ItemResolutionQuery {
 				case "PARTY_HAT_SLOTH":
 					resolvedName = resolveSlothHatName();
 					break;
+				case "POTION":
+					resolvedName = resolvePotionName();
+					break;
 			}
 		}
 
@@ -337,6 +340,12 @@ public class ItemResolutionQuery {
 		}
 	}
 
+	private String resolvePotionName() {
+		String potion = getExtraAttributes().getString("potion");
+		int potionLvl = getExtraAttributes().getInteger("potion_level");
+		return "POTION_" + potion.toUpperCase(Locale.ROOT) + ";" + potionLvl;
+	}
+
 	private NBTTagCompound getExtraAttributes() {
 		if (compound == null) return new NBTTagCompound();
 		return compound.getCompoundTag(EXTRA_ATTRIBUTES);