Egg Finder initial commit

Current issues:
- Rendered waypoints do not turn off, can't figure out why
- Entity equipment is sent before the chat message for new eggs, creating situations where the "found" message is repeated twice

4 files changed, 156 insertions, 0 deletions

diff --git a/src/main/java/de/hysky/skyblocker/SkyblockerMod.java b/src/main/java/de/hysky/skyblocker/SkyblockerMod.java
index e6b2d25a..e0e30b3c 100644
--- a/src/main/java/de/hysky/skyblocker/SkyblockerMod.java
+++ b/src/main/java/de/hysky/skyblocker/SkyblockerMod.java
@@ -11,6 +11,7 @@ import de.hysky.skyblocker.skyblock.*;
 import de.hysky.skyblocker.skyblock.calculators.CalculatorCommand;
 import de.hysky.skyblocker.skyblock.chat.ChatRuleAnnouncementScreen;
 import de.hysky.skyblocker.skyblock.chat.ChatRulesHandler;
+import de.hysky.skyblocker.skyblock.chocolatefactory.EggFinder;
 import de.hysky.skyblocker.skyblock.crimson.kuudra.Kuudra;
 import de.hysky.skyblocker.skyblock.dungeon.*;
 import de.hysky.skyblocker.skyblock.dungeon.partyfinder.PartyFinderScreen;
@@ -183,6 +184,7 @@ public class SkyblockerMod implements ClientModInitializer {
         BeaconHighlighter.init();
         WarpAutocomplete.init();
         MobBoundingBoxes.init();
+        EggFinder.init();
 
         Scheduler.INSTANCE.scheduleCyclic(Utils::update, 20);
         Scheduler.INSTANCE.scheduleCyclic(DiscordRPCManager::updateDataAndPresence, 200);
diff --git a/src/main/java/de/hysky/skyblocker/mixins/ClientPlayNetworkHandlerMixin.java b/src/main/java/de/hysky/skyblocker/mixins/ClientPlayNetworkHandlerMixin.java
index 2c2c1376..34e8264d 100644
--- a/src/main/java/de/hysky/skyblocker/mixins/ClientPlayNetworkHandlerMixin.java
+++ b/src/main/java/de/hysky/skyblocker/mixins/ClientPlayNetworkHandlerMixin.java
@@ -6,6 +6,7 @@ import com.llamalad7.mixinextras.sugar.Local;
 import de.hysky.skyblocker.config.SkyblockerConfigManager;
 import de.hysky.skyblocker.skyblock.CompactDamage;
 import de.hysky.skyblocker.skyblock.FishingHelper;
+import de.hysky.skyblocker.skyblock.chocolatefactory.EggFinder;
 import de.hysky.skyblocker.skyblock.dungeon.DungeonScore;
 import de.hysky.skyblocker.skyblock.dungeon.secrets.DungeonManager;
 import de.hysky.skyblocker.skyblock.end.BeaconHighlighter;
@@ -118,4 +119,9 @@ public abstract class ClientPlayNetworkHandlerMixin {
             LOGGER.error("[Skyblocker Compact Damage] Failed to compact damage number", e);
         }
     }
+
+    @Inject(method = "onEntityEquipmentUpdate", at = @At(value = "TAIL"))
+    private void skyblocker$onEntityEquip(EntityEquipmentUpdateS2CPacket packet, CallbackInfo ci, @Local Entity entity) {
+        EggFinder.checkIfEgg(entity);
+    }
 }
diff --git a/src/main/java/de/hysky/skyblocker/skyblock/chocolatefactory/EggFinder.java b/src/main/java/de/hysky/skyblocker/skyblock/chocolatefactory/EggFinder.java
new file mode 100644
index 00000000..d5e1a540
--- /dev/null
+++ b/src/main/java/de/hysky/skyblocker/skyblock/chocolatefactory/EggFinder.java
@@ -0,0 +1,132 @@
+package de.hysky.skyblocker.skyblock.chocolatefactory;
+
+import de.hysky.skyblocker.utils.ColorUtils;
+import de.hysky.skyblocker.utils.Constants;
+import de.hysky.skyblocker.utils.waypoint.Waypoint;
+import net.fabricmc.fabric.api.client.message.v1.ClientReceiveMessageEvents;
+import net.fabricmc.fabric.api.client.networking.v1.ClientPlayConnectionEvents;
+import net.fabricmc.fabric.api.client.rendering.v1.WorldRenderContext;
+import net.fabricmc.fabric.api.client.rendering.v1.WorldRenderEvents;
+import net.minecraft.client.MinecraftClient;
+import net.minecraft.component.DataComponentTypes;
+import net.minecraft.entity.Entity;
+import net.minecraft.entity.decoration.ArmorStandEntity;
+import net.minecraft.item.ItemStack;
+import net.minecraft.item.Items;
+import net.minecraft.text.Text;
+import org.apache.commons.lang3.mutable.MutableObject;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+public class EggFinder {
+	private static final Pattern eggFoundPattern = Pattern.compile("^(?:HOPPITY'S HUNT You found a Chocolate|You have already collected this Chocolate) (Breakfast|Lunch|Dinner)");
+	private static final Pattern newEggPattern = Pattern.compile("^HOPPITY'S HUNT A Chocolate (Breakfast|Lunch|Dinner) Egg has appeared!$");
+	private static final MutableObject<Egg> breakfastEgg = new MutableObject<>(null);
+	private static final MutableObject<Egg> lunchEgg = new MutableObject<>(null);
+	private static final MutableObject<Egg> dinnerEgg = new MutableObject<>(null);
+	private static final Logger logger = LoggerFactory.getLogger("Skyblocker Egg Finder");
+
+	private EggFinder() {
+	}
+
+	public static void init() {
+		ClientPlayConnectionEvents.JOIN.register((ignored, ignored2, ignored3) -> {
+			breakfastEgg.setValue(null);
+			lunchEgg.setValue(null);
+			dinnerEgg.setValue(null);
+		});
+		ClientReceiveMessageEvents.GAME.register(EggFinder::onChatMessage);
+		WorldRenderEvents.AFTER_TRANSLUCENT.register(EggFinder::renderWaypoints);
+	}
+
+	public static void checkIfEgg(Entity entity) {
+		if (breakfastEgg.getValue() != null && dinnerEgg.getValue() != null && lunchEgg.getValue() != null) return; //Don't check for eggs if we already found all of them
+		if (!(entity instanceof ArmorStandEntity armorStand) || armorStand.hasCustomName() || !armorStand.isInvisible() || !armorStand.shouldHideBasePlate()) return;
+		for (ItemStack itemStack : armorStand.getArmorItems()) {
+			try {
+				if (!itemStack.isEmpty() || itemStack.getItem() == Items.PLAYER_HEAD) {
+					String texture = itemStack.getComponents()
+					                          .get(DataComponentTypes.PROFILE)
+					                          .properties()
+					                          .get("textures")
+					                          .iterator()
+					                          .next()
+					                          .value();
+
+					//Don't turn this into a switch statement, it's unreadable
+					if (texture.equals("ewogICJ0aW1lc3RhbXAiIDogMTcxMTQ2MjY3MzE0OSwKICAicHJvZmlsZUlkIiA6ICJiN2I4ZTlhZjEwZGE0NjFmOTY2YTQxM2RmOWJiM2U4OCIsCiAgInByb2ZpbGVOYW1lIiA6ICJBbmFiYW5hbmFZZzciLAogICJzaWduYXR1cmVSZXF1aXJlZCIgOiB0cnVlLAogICJ0ZXh0dXJlcyIgOiB7CiAgICAiU0tJTiIgOiB7CiAgICAgICJ1cmwiIDogImh0dHA6Ly90ZXh0dXJlcy5taW5lY3JhZnQubmV0L3RleHR1cmUvYTQ5MzMzZDg1YjhhMzE1ZDAzMzZlYjJkZjM3ZDhhNzE0Y2EyNGM1MWI4YzYwNzRmMWI1YjkyN2RlYjUxNmMyNCIKICAgIH0KICB9Cn0")) {
+						handleFoundEgg(armorStand, EggType.BREAKFAST);
+					} else if (texture.equals("ewogICJ0aW1lc3RhbXAiIDogMTcxMTQ2MjY0OTcwMSwKICAicHJvZmlsZUlkIiA6ICI3NGEwMzQxNWY1OTI0ZTA4YjMyMGM2MmU1NGE3ZjJhYiIsCiAgInByb2ZpbGVOYW1lIiA6ICJNZXp6aXIiLAogICJzaWduYXR1cmVSZXF1aXJlZCIgOiB0cnVlLAogICJ0ZXh0dXJlcyIgOiB7CiAgICAiU0tJTiIgOiB7CiAgICAgICJ1cmwiIDogImh0dHA6Ly90ZXh0dXJlcy5taW5lY3JhZnQubmV0L3RleHR1cmUvZTVlMzYxNjU4MTlmZDI4NTBmOTg1NTJlZGNkNzYzZmY5ODYzMTMxMTkyODNjMTI2YWNlMGM0Y2M0OTVlNzZhOCIKICAgIH0KICB9Cn0")) {
+						handleFoundEgg(armorStand, EggType.DINNER);
+					} else if (texture.equals("ewogICJ0aW1lc3RhbXAiIDogMTcxMTQ2MjU2ODExMiwKICAicHJvZmlsZUlkIiA6ICI3NzUwYzFhNTM5M2Q0ZWQ0Yjc2NmQ4ZGUwOWY4MjU0NiIsCiAgInByb2ZpbGVOYW1lIiA6ICJSZWVkcmVsIiwKICAic2lnbmF0dXJlUmVxdWlyZWQiIDogdHJ1ZSwKICAidGV4dHVyZXMiIDogewogICAgIlNLSU4iIDogewogICAgICAidXJsIiA6ICJodHRwOi8vdGV4dHVyZXMubWluZWNyYWZ0Lm5ldC90ZXh0dXJlLzdhZTZkMmQzMWQ4MTY3YmNhZjk1MjkzYjY4YTRhY2Q4NzJkNjZlNzUxZGI1YTM0ZjJjYmM2NzY2YTAzNTZkMGEiCiAgICB9CiAgfQp9")) {
+						handleFoundEgg(armorStand, EggType.LUNCH);
+					}
+				}
+			} catch (Exception e) {
+				//Ignored. This simply exists to make the code cleaner without a bunch of if statements to check the existence of each key.
+			}
+		}
+	}
+
+	private static void handleFoundEgg(ArmorStandEntity entity, EggType eggType) {
+		eggType.egg.setValue(new Egg(entity, new Waypoint(entity.getBlockPos().up(2), () -> Waypoint.Type.WAYPOINT, ColorUtils.getFloatComponents(eggType.color))));
+		MinecraftClient.getInstance().player.sendMessage(Constants.PREFIX.get().append("Found a ").append(Text.literal("Chocolate " + eggType + " Egg").withColor(eggType.color)).append(" at " + entity.getBlockPos().up(2).toShortString() + "!"));
+	}
+
+	private static void renderWaypoints(WorldRenderContext context) {
+		if (breakfastEgg.getValue() != null && breakfastEgg.getValue().waypoint.shouldRender()) breakfastEgg.getValue().waypoint.render(context);
+		if (lunchEgg.getValue() != null && lunchEgg.getValue().waypoint.shouldRender()) lunchEgg.getValue().waypoint.render(context);
+		if (dinnerEgg.getValue() != null && dinnerEgg.getValue().waypoint.shouldRender()) dinnerEgg.getValue().waypoint.render(context);
+	}
+
+	private static void onChatMessage(Text text, boolean overlay) {
+		if (overlay) return;
+		Matcher matcher = eggFoundPattern.matcher(text.getString());
+		if (matcher.matches()) {
+			try {
+				Egg egg = EggType.valueOf(matcher.group(1).toUpperCase()).egg.getValue();
+				if (egg != null) egg.waypoint.setFound();
+			} catch (IllegalArgumentException e) {
+				logger.error("Failed to find egg type for egg found message. Tried to match against: " + matcher.group(0), e);
+			}
+		}
+
+		//There's only one egg of the same type at any given time, so we can set the changed egg to null
+		matcher = newEggPattern.matcher(text.getString());
+		if (matcher.matches()) {
+			try {
+				EggType.valueOf(matcher.group(1).toUpperCase()).egg.setValue(null);
+			} catch (IllegalArgumentException e) {
+				logger.error("Failed to find egg type for egg spawn message. Tried to match against: " + matcher.group(0), e);
+			}
+		}
+	}
+
+	private record Egg(ArmorStandEntity entity, Waypoint waypoint) { }
+
+	private enum EggType {
+		LUNCH(lunchEgg, 0x5555FF),
+		DINNER(dinnerEgg, 0x55FF55),
+		BREAKFAST(breakfastEgg, 0xFFAA00);
+
+		public final MutableObject<Egg> egg;
+		public final int color;
+
+		EggType(MutableObject<Egg> egg, int color) {
+			this.egg = egg;
+			this.color = color;
+		}
+
+		@Override
+		public String toString() {
+			return switch (this) {
+				case LUNCH -> "Lunch";
+				case DINNER -> "Dinner";
+				case BREAKFAST -> "Breakfast";
+			};
+		}
+	}
+}
diff --git a/src/main/java/de/hysky/skyblocker/utils/ColorUtils.java b/src/main/java/de/hysky/skyblocker/utils/ColorUtils.java
new file mode 100644
index 00000000..0196edf2
--- /dev/null
+++ b/src/main/java/de/hysky/skyblocker/utils/ColorUtils.java
@@ -0,0 +1,16 @@
+package de.hysky.skyblocker.utils;
+
+public class ColorUtils {
+	/**
+	 * Takes an RGB color as an integer and returns an array of the color's components as floats, in RGB format.
+	 * @param color The color to get the components of.
+	 * @return An array of the color's components as floats.
+	 */
+	public static float[] getFloatComponents(int color) {
+		return new float[] {
+				((color >> 16) & 0xFF) / 255f,
+				((color >> 8) & 0xFF) / 255f,
+				(color & 0xFF) / 255f
+		};
+	}
+}