Feature: Pet Nametag (#1880)

Co-authored-by: Empa <42304516+ItsEmpa@users.noreply.github.com>
Co-authored-by: Cal <cwolfson58@gmail.com>
Co-authored-by: hannibal2 <24389977+hannibal00212@users.noreply.github.com>
Co-authored-by: CalMWolfs <94038482+CalMWolfs@users.noreply.github.com>

src/main/java/at/hannibal2/skyhanni/config/features/misc/pets/PetConfig.java

@@ -30,6 +30,11 @@ public class PetConfig {
     @Accordion
     public PetExperienceToolTipConfig petExperienceToolTip = new PetExperienceToolTipConfig();
 
+    @Expose
+    @ConfigOption(name = "Pet Nametag", desc = "")
+    @Accordion
+    public PetNametagConfig nametag = new PetNametagConfig();
+
     @Expose
     @ConfigOption(name = "Hide Autopet Messages", desc = "Hide the autopet messages from chat.\n" +
         "§eRequires the display to be enabled.")

src/main/java/at/hannibal2/skyhanni/config/features/misc/pets/PetNametagConfig.java

@@ -0,0 +1,19 @@
+package at.hannibal2.skyhanni.config.features.misc.pets;
+
+import com.google.gson.annotations.Expose;
+import io.github.notenoughupdates.moulconfig.annotations.ConfigEditorBoolean;
+import io.github.notenoughupdates.moulconfig.annotations.ConfigOption;
+
+public class PetNametagConfig {
+
+    @Expose
+    @ConfigOption(name = "Hide Pet Level", desc = "Hide the pet level above the pet.")
+    @ConfigEditorBoolean
+    public boolean hidePetLevel = false;
+
+    @Expose
+    @ConfigOption(name = "Hide Max Pet Level", desc = "Hide the pet level above the pet if it is max level.")
+    @ConfigEditorBoolean
+    public boolean hideMaxPetLevel = false;
+
+}

src/main/java/at/hannibal2/skyhanni/features/misc/CurrentPetDisplay.kt → src/main/java/at/hannibal2/skyhanni/features/misc/pets/CurrentPetDisplay.kt

@@ -1,4 +1,4 @@
-package at.hannibal2.skyhanni.features.misc
+package at.hannibal2.skyhanni.features.misc.pets
 
 import at.hannibal2.skyhanni.SkyHanniMod
 import at.hannibal2.skyhanni.config.ConfigUpdaterMigrator

src/main/java/at/hannibal2/skyhanni/features/misc/PetCandyUsedDisplay.kt → src/main/java/at/hannibal2/skyhanni/features/misc/pets/PetCandyUsedDisplay.kt

@@ -1,4 +1,4 @@
-package at.hannibal2.skyhanni.features.misc
+package at.hannibal2.skyhanni.features.misc.pets
 
 import at.hannibal2.skyhanni.SkyHanniMod
 import at.hannibal2.skyhanni.config.ConfigUpdaterMigrator

src/main/java/at/hannibal2/skyhanni/features/misc/PetExpTooltip.kt → src/main/java/at/hannibal2/skyhanni/features/misc/pets/PetExpTooltip.kt

@@ -1,4 +1,4 @@
-package at.hannibal2.skyhanni.features.misc
+package at.hannibal2.skyhanni.features.misc.pets
 
 import at.hannibal2.skyhanni.SkyHanniMod
 import at.hannibal2.skyhanni.config.ConfigUpdaterMigrator

src/main/java/at/hannibal2/skyhanni/features/misc/PetItemDisplay.kt → src/main/java/at/hannibal2/skyhanni/features/misc/pets/PetItemDisplay.kt

@@ -1,4 +1,4 @@
-package at.hannibal2.skyhanni.features.misc
+package at.hannibal2.skyhanni.features.misc.pets
 
 import at.hannibal2.skyhanni.SkyHanniMod
 import at.hannibal2.skyhanni.events.GuiRenderItemEvent

src/main/java/at/hannibal2/skyhanni/features/misc/pets/PetNametag.kt

@@ -0,0 +1,56 @@
+package at.hannibal2.skyhanni.features.misc.pets
+
+import at.hannibal2.skyhanni.SkyHanniMod
+import at.hannibal2.skyhanni.events.entity.EntityDisplayNameEvent
+import at.hannibal2.skyhanni.skyhannimodule.SkyHanniModule
+import at.hannibal2.skyhanni.utils.LorenzUtils
+import at.hannibal2.skyhanni.utils.NumberUtil.formatInt
+import at.hannibal2.skyhanni.utils.RegexUtils.groupOrNull
+import at.hannibal2.skyhanni.utils.RegexUtils.matchMatcher
+import at.hannibal2.skyhanni.utils.chat.Text.asComponent
+import at.hannibal2.skyhanni.utils.repopatterns.RepoPattern
+import net.minecraft.entity.item.EntityArmorStand
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+
+@SkyHanniModule
+object PetNametag {
+
+    private val config get() = SkyHanniMod.feature.misc.pets.nametag
+
+    /**
+     * REGEX-TEST: §8[§7Lv99§8] Ammonite
+     * REGEX-TEST: §8[§7Lv100§8] Endermite§5 ✦
+     */
+    private val petNametagPattern by RepoPattern.pattern(
+        "pet.nametag",
+        "(?<start>§8\\[§7Lv(?<lvl>\\d+)§8]) (?<rarity>§.)(?<pet>[\\w\\s]+)(?<skin>§. ✦)?",
+    )
+
+    @SubscribeEvent
+    fun onNameTagRender(event: EntityDisplayNameEvent) {
+        if (!isEnabled()) return
+        if (event.entity !is EntityArmorStand) return
+
+        petNametagPattern.matchMatcher(event.chatComponent.unformattedText) {
+            val start = group("start")
+            val lvl = group("lvl").formatInt()
+            val rarity = group("rarity")
+            val pet = group("pet")
+            val skin = groupOrNull("skin") ?: ""
+
+            val hideLevel = config.hidePetLevel
+            val hideMaxLevel = config.hideMaxPetLevel && (lvl == 100 || lvl == 200)
+
+            val text = buildString {
+                if (!hideLevel && !hideMaxLevel) {
+                    append(start)
+                }
+                append(rarity + pet + skin)
+            }
+
+            event.chatComponent = text.asComponent()
+        }
+    }
+
+    private fun isEnabled() = LorenzUtils.inSkyBlock && (config.hidePetLevel || config.hideMaxPetLevel)
+}