🐛 fix player quit

XAS-Dev · Aug 7, 2024 · 8ae96cb · 8ae96cb
1 parent a137057
commit 8ae96cb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/main/kotlin/xyz/xasmc/hashbook/listener/BookshelfListener.kt b/src/main/kotlin/xyz/xasmc/hashbook/listener/BookshelfListener.kt
@@ -5,6 +5,7 @@ import org.bukkit.block.ChiseledBookshelf
 import org.bukkit.event.EventHandler
 import org.bukkit.event.Listener
 import org.bukkit.event.player.PlayerMoveEvent
+import org.bukkit.event.player.PlayerQuitEvent
 import org.bukkit.inventory.meta.BookMeta
 import org.bukkit.inventory.meta.BookMeta.Generation.*
 import xyz.xasmc.hashbook.util.MarkUtil
@@ -58,4 +59,9 @@ class BookshelfListener : Listener {
         }
         MarkUtil.updateMark(player, markLocation.toLocation(world), nameSb.toString())
     }
+
+    @EventHandler
+    fun onPlayerQuit(event: PlayerQuitEvent) {
+        MarkUtil.removeMark(event.player)
+    }
 }
diff --git a/src/main/kotlin/xyz/xasmc/hashbook/util/MarkUtil.kt b/src/main/kotlin/xyz/xasmc/hashbook/util/MarkUtil.kt
@@ -17,15 +17,12 @@ object MarkUtil {
         val count = textList.size
         val interval = 0.2
         val offset = -0.4
-
         val marks = playerMark[player] ?: mutableListOf<ArmorStand>().also { playerMark[player] = it }
-
         val currentCount = marks.size
         when {
             count > currentCount -> repeat(count - currentCount) { marks.add(createArmorStand(world)) }
             count < currentCount -> repeat(currentCount - count) { marks.removeLast()?.remove() }
         }
-
         val top = location.clone().add(0.0, interval * count / 2 + offset, 0.0)
         textList.forEachIndexed { i, it ->
             val mark = playerMark[player]!![i]