Add warp to diana guess keybind

1 files changed, 77 insertions, 0 deletions

diff --git a/src/main/kotlin/moe/nea/firmament/util/WarpUtil.kt b/src/main/kotlin/moe/nea/firmament/util/WarpUtil.kt
new file mode 100644
index 0000000..25cfdb0
--- /dev/null
+++ b/src/main/kotlin/moe/nea/firmament/util/WarpUtil.kt
@@ -0,0 +1,77 @@
+/*
+ * SPDX-FileCopyrightText: 2024 Linnea Gräf <nea@nea.moe>
+ *
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ */
+
+package moe.nea.firmament.util
+
+import io.github.moulberry.repo.constants.Islands
+import io.github.moulberry.repo.constants.Islands.Warp
+import kotlinx.serialization.Serializable
+import kotlinx.serialization.serializer
+import kotlin.math.sqrt
+import kotlin.time.Duration.Companion.seconds
+import net.minecraft.text.Text
+import net.minecraft.util.math.Position
+import moe.nea.firmament.events.ProcessChatEvent
+import moe.nea.firmament.repo.RepoManager
+import moe.nea.firmament.util.data.ProfileSpecificDataHolder
+
+object WarpUtil {
+    val warps: List<Islands.Warp> get() = RepoManager.neuRepo.constants.islands.warps
+
+    @Serializable
+    data class Data(
+        val excludedWarps: MutableSet<String> = mutableSetOf(),
+    )
+
+    object DConfig : ProfileSpecificDataHolder<Data>(serializer(), "warp-util", ::Data)
+
+    private var lastAttemptedWarp = ""
+    private var lastWarpAttempt = TimeMark.farPast()
+    fun findNearestWarp(locrawMode: String, pos: Position): Islands.Warp? {
+        return warps.minByOrNull {
+            if (locrawMode != it.mode || (DConfig.data?.excludedWarps?.contains(it.warp) == true)) {
+                return@minByOrNull Double.MAX_VALUE
+            } else {
+                return@minByOrNull squaredDist(pos, it)
+            }
+        }
+    }
+
+    private fun squaredDist(pos: Position, warp: Warp): Double {
+        val dx = pos.x - warp.x
+        val dy = pos.y - warp.y
+        val dz = pos.z - warp.z
+        return dx * dx + dy * dy + dz * dz
+    }
+
+    fun teleportToNearestWarp(locrawMode: String, pos: Position) {
+        val nearestWarp = findNearestWarp(locrawMode, pos) ?: return
+
+        if (locrawMode == SBData.skyblockLocation
+            && sqrt(squaredDist(pos, nearestWarp)) > 1.1 * sqrt(squaredDist((MC.player ?: return).pos, nearestWarp))
+        ) {
+            return
+        }
+        MC.sendServerCommand("warp ${nearestWarp.warp}")
+    }
+
+    init {
+        ProcessChatEvent.subscribe {
+            if (it.unformattedString == "You haven't unlocked this fast travel destination!"
+                && lastWarpAttempt.passedTime() < 2.seconds
+            ) {
+                DConfig.data?.excludedWarps?.add(lastAttemptedWarp)
+                DConfig.markDirty()
+                MC.sendChat(Text.translatable("firmament.warp-util.mark-excluded", lastAttemptedWarp))
+                lastWarpAttempt = TimeMark.farPast()
+            }
+            if (it.unformattedString == "You may now fast travel to") {
+                DConfig.data?.excludedWarps?.clear()
+                DConfig.markDirty()
+            }
+        }
+    }
+}