1 files changed, 36 insertions, 24 deletions
diff --git a/src/main/java/at/hannibal2/skyhanni/utils/tracker/SkyHanniTracker.kt b/src/main/java/at/hannibal2/skyhanni/utils/tracker/SkyHanniTracker.kt
index d1b783f7e..e80f93edf 100644
--- a/src/main/java/at/hannibal2/skyhanni/utils/tracker/SkyHanniTracker.kt
+++ b/src/main/java/at/hannibal2/skyhanni/utils/tracker/SkyHanniTracker.kt
@@ -6,9 +6,11 @@ import at.hannibal2.skyhanni.data.ProfileStorageData
 import at.hannibal2.skyhanni.utils.LorenzUtils
 import at.hannibal2.skyhanni.utils.LorenzUtils.addAsSingletonList
 import at.hannibal2.skyhanni.utils.RenderUtils.renderStringsAndItems
+import at.hannibal2.skyhanni.utils.SimpleTimeMark
 import at.hannibal2.skyhanni.utils.renderables.Renderable
 import net.minecraft.client.Minecraft
 import net.minecraft.client.gui.inventory.GuiInventory
+import kotlin.time.Duration.Companion.seconds
 
 class SkyHanniTracker<Data : TrackerData>(
     private val name: String,
@@ -20,6 +22,7 @@ class SkyHanniTracker<Data : TrackerData>(
     private var displayMode = DisplayMode.TOTAL
     private val currentSessions = mutableMapOf<Storage.ProfileSpecific, Data>()
     private var display = emptyList<List<Any>>()
+    private var sessionResetTime = SimpleTimeMark.farPast()
 
     fun isInventoryOpen() = inventoryOpen
 
@@ -51,37 +54,46 @@ class SkyHanniTracker<Data : TrackerData>(
     }
 
     fun update() {
-        display = currentDisplay()?.let {
-            val list = drawDisplay(it).toMutableList()
+        display = buildFinalDisplay()
+    }
+
+    private fun buildFinalDisplay(): List<List<Any>> {
+        val data = currentDisplay() ?: return emptyList()
+
+        return drawDisplay(data).toMutableList().also {
             if (inventoryOpen) {
-                list.add(1, LorenzUtils.buildSelector<DisplayMode>(
-                    "§7Display Mode: ",
-                    getName = { type -> type.displayName },
-                    isCurrent = { it == displayMode },
-                    onChange = {
-                        displayMode = it
-                        update()
-                    }
-                ))
+                it.add(1, buildDisplayModeView())
             }
             if (inventoryOpen && displayMode == DisplayMode.SESSION) {
-                list.addAsSingletonList(
-                    Renderable.clickAndHover(
-                        "§cReset session!",
-                        listOf(
-                            "§cThis will reset your",
-                            "§ccurrent session of",
-                            "§c$name"
-                        ),
-                    ) {
-                        reset(DisplayMode.SESSION, "§e[SkyHanni] Reset this session of $name!")
-                    })
+                it.addAsSingletonList(buildSessionResetButton())
             }
+        }
+    }
 
-            list
-        } ?: emptyList()
+    private fun buildSessionResetButton() = Renderable.clickAndHover(
+        "§cReset session!",
+        listOf(
+            "§cThis will reset your",
+            "§ccurrent session of",
+            "§c$name"
+        ),
+    ) {
+        if (sessionResetTime.passedSince() > 3.seconds) {
+            reset(DisplayMode.SESSION, "§e[SkyHanni] Reset this session of $name!")
+            sessionResetTime = SimpleTimeMark.now()
+        }
     }
 
+    private fun buildDisplayModeView() = LorenzUtils.buildSelector<DisplayMode>(
+        "§7Display Mode: ",
+        getName = { type -> type.displayName },
+        isCurrent = { it == displayMode },
+        onChange = {
+            displayMode = it
+            update()
+        }
+    )
+
     private fun currentDisplay() = getSharedTracker()?.get(displayMode)
 
     private fun getSharedTracker(): SharedTracker<Data>? {