From 12883de4872437b5e53eea2ae7db104e3bf69b8f Mon Sep 17 00:00:00 2001
From: Daan Vanden Bosch <daan.v.d.bosch@gmail.com>
Date: Tue, 11 May 2021 19:35:53 +0200
Subject: [PATCH] Fixed two bugs in Throttle.

---
 .../kotlin/world/phantasmal/web/shared/Throttle.kt       | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/web/shared/src/jsMain/kotlin/world/phantasmal/web/shared/Throttle.kt b/web/shared/src/jsMain/kotlin/world/phantasmal/web/shared/Throttle.kt
index 2ad0fc1e..e00e0c45 100644
--- a/web/shared/src/jsMain/kotlin/world/phantasmal/web/shared/Throttle.kt
+++ b/web/shared/src/jsMain/kotlin/world/phantasmal/web/shared/Throttle.kt
@@ -16,16 +16,21 @@ class Throttle(
 ) {
     private var timeout: Int? = null
     private var invokeOnTimeout = false
+    private var function: () -> Unit = {}
 
     operator fun invoke(f: () -> Unit) {
+        function = f
+
         if (timeout == null) {
             if (leading) {
-                f()
+                function()
+            } else if (trailing) {
+                invokeOnTimeout = true
             }
 
             timeout = self.setTimeout({
                 if (invokeOnTimeout) {
-                    f()
+                    function()
                 }
 
                 timeout = null