add enigma soul waypoints

2 files changed, 126 insertions, 0 deletions

diff --git a/src/main/java/at/hannibal2/skyhanni/SkyHanniMod.kt b/src/main/java/at/hannibal2/skyhanni/SkyHanniMod.kt
index 92b2e9f08..2ba491953 100644
--- a/src/main/java/at/hannibal2/skyhanni/SkyHanniMod.kt
+++ b/src/main/java/at/hannibal2/skyhanni/SkyHanniMod.kt
@@ -309,6 +309,7 @@ class SkyHanniMod {
         loadModule(RiftLarva())
         loadModule(RiftOdonata())
         loadModule(RiftAgaricusCap())
+        loadModule(EnigmaSoulWaypoints())
 
         init()
 
diff --git a/src/main/java/at/hannibal2/skyhanni/features/rift/EnigmaSoulWaypoints.kt b/src/main/java/at/hannibal2/skyhanni/features/rift/EnigmaSoulWaypoints.kt
new file mode 100644
index 000000000..7f5c901a6
--- /dev/null
+++ b/src/main/java/at/hannibal2/skyhanni/features/rift/EnigmaSoulWaypoints.kt
@@ -0,0 +1,125 @@
+package at.hannibal2.skyhanni.features.rift
+
+import at.hannibal2.skyhanni.events.*
+import at.hannibal2.skyhanni.test.GriffinUtils.drawWaypointFilled
+import at.hannibal2.skyhanni.utils.LocationUtils.distanceToPlayer
+import at.hannibal2.skyhanni.utils.LorenzColor
+import at.hannibal2.skyhanni.utils.LorenzUtils
+import at.hannibal2.skyhanni.utils.LorenzVec
+import at.hannibal2.skyhanni.utils.RenderUtils.drawDynamicText
+import at.hannibal2.skyhanni.utils.RenderUtils.highlight
+import at.hannibal2.skyhanni.utils.StringUtils.removeColor
+import io.github.moulberry.notenoughupdates.events.SlotClickEvent
+import net.minecraft.client.gui.inventory.GuiChest
+import net.minecraft.inventory.ContainerChest
+import net.minecraftforge.client.event.RenderWorldLastEvent
+import net.minecraftforge.fml.common.eventhandler.EventPriority
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+
+class EnigmaSoulWaypoints {
+    private var inInventory = false
+    private val soulLocations = mutableMapOf<String, LorenzVec>()
+    private val trackedSouls = mutableListOf<String>()
+
+    @SubscribeEvent
+    fun onInventoryOpen(event: InventoryOpenEvent) {
+        inInventory = false
+        if (event.inventoryName.contains("Enigma Souls")) inInventory = true
+    }
+
+    @SubscribeEvent
+    fun onInventoryClose(event: InventoryCloseEvent) {
+        inInventory = false
+    }
+
+    @SubscribeEvent(priority = EventPriority.HIGH)
+    fun onSlotClick(event: SlotClickEvent) {
+        if (!inInventory || !RiftAPI.inRift()) return
+
+        val split = event.slot.stack.displayName.split("Enigma: ")
+        if (split.size == 2) {
+            event.isCanceled =  true // maybe change to a middle click
+            if (soulLocations.contains(split.last())) {
+                if (!trackedSouls.contains(split.last())) {
+                    LorenzUtils.chat("§5Tracking the ${split.last()} Enigma Soul!")
+                    trackedSouls.add(split.last())
+                } else {
+                    trackedSouls.remove(split.last())
+                    LorenzUtils.chat("§5No longer tracking the ${split.last()} Enigma Soul!")
+                }
+            }
+        }
+    }
+
+    @SubscribeEvent(priority = EventPriority.LOWEST)
+    fun onBackgroundDrawn(event: GuiContainerEvent.BackgroundDrawnEvent) {
+        if (!RiftAPI.inRift() || !inInventory) return
+
+        if (event.gui !is GuiChest) return
+        val guiChest = event.gui
+        val chest = guiChest.inventorySlots as ContainerChest
+
+        for (slot in chest.inventorySlots) {
+            if (slot == null) continue
+            val stack = slot.stack ?: continue
+
+            for (soul in trackedSouls) {
+                if (stack.displayName.removeColor().contains(soul)) {
+                    slot highlight LorenzColor.DARK_PURPLE
+                }
+            }
+        }
+    }
+
+    @SubscribeEvent
+    fun onRenderWorld(event: RenderWorldLastEvent) {
+        if (!RiftAPI.inRift()) return
+        for (soul in trackedSouls) {
+            soulLocations[soul]?.let { event.drawWaypointFilled(it, LorenzColor.DARK_PURPLE.toColor(), true, true) }
+            soulLocations[soul]?.let { event.drawDynamicText(it.add(0, 1, 0), "§5$soul Soul", 1.0) }
+        }
+    }
+
+    @SubscribeEvent
+    fun onRepoReload(event: RepositoryReloadEvent) {
+        val data = event.getConstant("EnigmaSouls") ?: return
+
+        for (area in data.entrySet()) {
+            val element = area.value.asJsonArray
+
+            for (i in 0 until element.size()) {
+                val itemObject = element[i].asJsonObject
+                val name = itemObject["name"].asString
+                val position = itemObject["position"].asString
+                val split = position.split(", ")
+                if (split.size == 3) {
+                    soulLocations[name] = LorenzVec(split[0].toDouble(), split[1].toDouble(), split[2].toDouble())
+                }
+            }
+        }
+    }
+
+    @SubscribeEvent
+    fun onChat(event: LorenzChatEvent) {
+        if (!RiftAPI.inRift()) return
+        val message = event.message.removeColor().trim()
+        // not sure how this works for buying the souls
+        if (message == "You have already found that Enigma Soul!" || message == "SOUL! You unlocked an Enigma Soul!") {
+            hideClosestSoul()
+        }
+    }
+
+    private fun hideClosestSoul() {
+        var closestSoul = ""
+        var closestDistance = 8.0
+
+        for (soul in soulLocations) {
+            if (soul.value.distanceToPlayer() < closestDistance) {
+                closestSoul = soul.key
+                closestDistance = soul.value.distanceToPlayer()
+            }
+        }
+        trackedSouls.remove(closestSoul)
+        LorenzUtils.chat("§5Found the $closestSoul Enigma Soul!")
+    }
+}
\ No newline at end of file