Feature: Pet Nametag (hannibal002#1880)

Co-authored-by: Empa <[email protected]> Co-authored-by: Cal <[email protected]> Co-authored-by: hannibal2 <[email protected]> Co-authored-by: CalMWolfs <[email protected]>
j10a1n15 · Sep 11, 2024 · c48d82d · c48d82d
1 parent 7aea6aa
commit c48d82d
Show file tree

Hide file tree

Showing 7 changed files with 84 additions and 4 deletions.
diff --git a/src/main/java/at/hannibal2/skyhanni/config/features/misc/pets/PetConfig.java b/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.")

diff --git a/src/main/java/at/hannibal2/skyhanni/config/features/misc/pets/PetNametagConfig.java b/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;
+
+}
diff --git a/...yhanni/features/misc/CurrentPetDisplay.kt → ...i/features/misc/pets/CurrentPetDisplay.kt b/...yhanni/features/misc/CurrentPetDisplay.kt → ...i/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

diff --git a/...anni/features/misc/PetCandyUsedDisplay.kt → ...features/misc/pets/PetCandyUsedDisplay.kt b/...anni/features/misc/PetCandyUsedDisplay.kt → ...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

diff --git a/...2/skyhanni/features/misc/PetExpTooltip.kt → ...hanni/features/misc/pets/PetExpTooltip.kt b/...2/skyhanni/features/misc/PetExpTooltip.kt → ...hanni/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

diff --git a/.../skyhanni/features/misc/PetItemDisplay.kt → ...anni/features/misc/pets/PetItemDisplay.kt b/.../skyhanni/features/misc/PetItemDisplay.kt → ...anni/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

diff --git a/src/main/java/at/hannibal2/skyhanni/features/misc/pets/PetNametag.kt b/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)
+}