Refactor color tooltip

I can't test this as I don't know what it does and what to look for

2 files changed, 136 insertions, 0 deletions

diff --git a/src/main/java/de/hysky/skyblocker/utils/tooltip/ColorTooltip.java b/src/main/java/de/hysky/skyblocker/utils/tooltip/ColorTooltip.java
new file mode 100644
index 00000000..5908f4e8
--- /dev/null
+++ b/src/main/java/de/hysky/skyblocker/utils/tooltip/ColorTooltip.java
@@ -0,0 +1,135 @@
+package de.hysky.skyblocker.utils.tooltip;
+
+import de.hysky.skyblocker.config.SkyblockerConfigManager;
+import de.hysky.skyblocker.skyblock.item.tooltip.ItemTooltip;
+import de.hysky.skyblocker.skyblock.item.tooltip.TooltipInfoType;
+import de.hysky.skyblocker.utils.Constants;
+import de.hysky.skyblocker.utils.ItemUtils;
+import net.minecraft.component.DataComponentTypes;
+import net.minecraft.component.type.DyedColorComponent;
+import net.minecraft.item.ItemStack;
+import net.minecraft.nbt.NbtCompound;
+import net.minecraft.screen.slot.Slot;
+import net.minecraft.text.MutableText;
+import net.minecraft.text.Text;
+import net.minecraft.util.Formatting;
+import net.minecraft.util.StringIdentifiable;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.List;
+
+public class ColorTooltip extends TooltipAdder {
+	private static final Logger LOGGER = LoggerFactory.getLogger(ColorTooltip.class);
+
+	public ColorTooltip(int priority) {
+		super(priority);
+	}
+
+	@Override
+	public void addToTooltip(List<Text> lore, Slot focusedSlot) {
+		final ItemStack itemStack = focusedSlot.getStack();
+		final String internalID = ItemTooltip.getInternalNameFromNBT(itemStack, true);
+		if (TooltipInfoType.COLOR.isTooltipEnabledAndHasOrNullWarning(internalID) && itemStack.contains(DataComponentTypes.DYED_COLOR)) {
+			String uuid = ItemUtils.getItemUuid(itemStack);
+			boolean hasCustomDye = SkyblockerConfigManager.get().general.customDyeColors.containsKey(uuid) || SkyblockerConfigManager.get().general.customAnimatedDyes.containsKey(uuid);
+			//DyedColorComponent#getColor returns ARGB so we mask out the alpha bits
+			int dyeColor = DyedColorComponent.getColor(itemStack, 0);
+
+			// dyeColor will have alpha = 255 if it's dyed, and alpha = 0 if it's not dyed,
+			if (!hasCustomDye && dyeColor != 0) {
+				dyeColor = dyeColor & 0x00FFFFFF;
+				String colorHex = String.format("%06X", dyeColor);
+				String expectedHex = getExpectedHex(internalID);
+
+				boolean correctLine = false;
+				for (Text text : lore) {
+					String existingTooltip = text.getString() + " ";
+					if (existingTooltip.startsWith("Color: ")) {
+						correctLine = true;
+
+						addExoticTooltip(lore, internalID, ItemUtils.getCustomData(itemStack), colorHex, expectedHex, existingTooltip);
+						break;
+					}
+				}
+
+				if (!correctLine) {
+					addExoticTooltip(lore, internalID, ItemUtils.getCustomData(itemStack), colorHex, expectedHex, "");
+				}
+			}
+		}
+	}
+
+	private static void addExoticTooltip(List<Text> lines, String internalID, NbtCompound customData, String colorHex, String expectedHex, String existingTooltip) {
+		if (expectedHex != null && !colorHex.equalsIgnoreCase(expectedHex) && !isException(internalID, colorHex) && !intendedDyed(customData)) {
+			final DyeType type = checkDyeType(colorHex);
+			lines.add(1, Text.literal(existingTooltip + Formatting.DARK_GRAY + "(").append(type.getTranslatedText()).append(Formatting.DARK_GRAY + ")"));
+		}
+	}
+
+	public static String getExpectedHex(String id) {
+		String color = TooltipInfoType.COLOR.getData().get(id).getAsString();
+		if (color != null) {
+			String[] RGBValues = color.split(",");
+			return String.format("%02X%02X%02X", Integer.parseInt(RGBValues[0]), Integer.parseInt(RGBValues[1]), Integer.parseInt(RGBValues[2]));
+		} else {
+			LOGGER.warn("[Skyblocker Exotics] No expected color data found for id {}", id);
+			return null;
+		}
+	}
+
+	public static boolean isException(String id, String hex) {
+		return switch (id) {
+			case String it when it.startsWith("LEATHER") || it.equals("GHOST_BOOTS") || Constants.SEYMOUR_IDS.contains(it) -> true;
+			case String it when it.startsWith("RANCHER") -> Constants.RANCHERS.contains(hex);
+			case String it when it.contains("ADAPTIVE_CHESTPLATE") -> Constants.ADAPTIVE_CHEST.contains(hex);
+			case String it when it.contains("ADAPTIVE") -> Constants.ADAPTIVE.contains(hex);
+			case String it when it.contains("REAPER") -> Constants.REAPER.contains(hex);
+			case String it when it.contains("FAIRY") -> Constants.FAIRY_HEXES.contains(hex);
+			case String it when it.contains("CRYSTAL") -> Constants.CRYSTAL_HEXES.contains(hex);
+			case String it when it.contains("SPOOK") -> Constants.SPOOK.contains(hex);
+			default -> false;
+		};
+	}
+
+	public static DyeType checkDyeType(String hex) {
+		return switch (hex) {
+			case String it when Constants.CRYSTAL_HEXES.contains(it) -> DyeType.CRYSTAL;
+			case String it when Constants.FAIRY_HEXES.contains(it) -> DyeType.FAIRY;
+			case String it when Constants.OG_FAIRY_HEXES.contains(it) -> DyeType.OG_FAIRY;
+			case String it when Constants.SPOOK.contains(it) -> DyeType.SPOOK;
+			case String it when Constants.GLITCHED.contains(it) -> DyeType.GLITCHED;
+			default -> DyeType.EXOTIC;
+		};
+	}
+
+	public static boolean intendedDyed(NbtCompound customData) {
+		return customData.contains("dye_item");
+	}
+
+	public enum DyeType implements StringIdentifiable {
+		CRYSTAL("crystal", Formatting.AQUA),
+		FAIRY("fairy", Formatting.LIGHT_PURPLE),
+		OG_FAIRY("og_fairy", Formatting.DARK_PURPLE),
+		SPOOK("spook", Formatting.RED),
+		GLITCHED("glitched", Formatting.BLUE),
+		EXOTIC("exotic", Formatting.GOLD);
+		private final String name;
+		private final Formatting formatting;
+
+		DyeType(String name, Formatting formatting) {
+			this.name = name;
+			this.formatting = formatting;
+		}
+
+		@Override
+		public String asString() {
+			return name;
+		}
+
+		public MutableText getTranslatedText() {
+			return Text.translatable("skyblocker.exotic." + name).formatted(formatting);
+		}
+	}
+
+}
diff --git a/src/main/java/de/hysky/skyblocker/utils/tooltip/TooltipManager.java b/src/main/java/de/hysky/skyblocker/utils/tooltip/TooltipManager.java
index dcad236f..b5f44405 100644
--- a/src/main/java/de/hysky/skyblocker/utils/tooltip/TooltipManager.java
+++ b/src/main/java/de/hysky/skyblocker/utils/tooltip/TooltipManager.java
@@ -17,6 +17,7 @@ public class TooltipManager {
 			new DungeonQualityTooltip(0),
 			new ObtainedTooltip(3),
 			new MuseumTooltip(4),
+			new ColorTooltip(5),
 			new MotesTooltip(0),
 			new NpcPriceTooltip(1)
 	};