cooldown example fix

Matt-MX · May 6, 2024 · a418094 · a418094
1 parent f4e5479
commit a418094
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/api/src/main/kotlin/com/mattmx/ktgui/commands/SimpleCommandBuilder.kt b/api/src/main/kotlin/com/mattmx/ktgui/commands/SimpleCommandBuilder.kt
@@ -66,6 +66,9 @@ open class SimpleCommandBuilder(
     }
 
     fun cooldown(duration: Duration?) = apply {
+        cooldown.ifPresent {
+            ActionCoolDown.unregister(it)
+        }
         cooldown = Optional.ofNullable(duration?.let { ActionCoolDown(duration) })
     }
 

diff --git a/api/src/main/kotlin/com/mattmx/ktgui/cooldown/ActionCoolDown.kt b/api/src/main/kotlin/com/mattmx/ktgui/cooldown/ActionCoolDown.kt
@@ -58,6 +58,10 @@ class ActionCoolDown<T>(
             registeredCoolDowns.add(coolDown)
         }
 
+        fun unregister(coolDown: ActionCoolDown<*>) {
+            registeredCoolDowns.remove(coolDown)
+        }
+
         fun removeUsers(vararg user: Any) {
             registeredCoolDowns.forEach { it.removeAnyUsers(*user) }
         }

diff --git a/plugin/src/main/kotlin/com/mattmx/ktgui/KotlinGui.kt b/plugin/src/main/kotlin/com/mattmx/ktgui/KotlinGui.kt
@@ -4,6 +4,7 @@ import com.mattmx.ktgui.commands.rawCommand
 import com.mattmx.ktgui.commands.simpleCommand
 import com.mattmx.ktgui.examples.*
 import com.mattmx.ktgui.utils.not
+import com.mattmx.ktgui.utils.pretty
 import org.bukkit.Bukkit
 import org.bukkit.plugin.java.JavaPlugin
 import java.time.Duration
@@ -81,7 +82,16 @@ class KotlinGui : JavaPlugin() {
                 cooldown(Duration.ofSeconds(2))
 
                 executes {
-                    player.sendMessage(!"&aNot on cool-down!")
+                    source.sendMessage(!"&aNot on cool-down!")
+
+                    if (args.isNotEmpty()) {
+                        val newCoolDown = args.first().toLongOrNull()
+                            ?: return@executes
+
+                        val dur = Duration.ofMillis(newCoolDown)
+                        source.sendMessage(!"&aNew cool-down set: ${dur.pretty()}")
+                        cooldown(dur)
+                    }
                 }
 
                 onCooldown {