From 13cac25abd2007f89fb036c3e3e17b881674fa9c Mon Sep 17 00:00:00 2001
From: DBotThePony <dbotthepony@yandex.ru>
Date: Sat, 13 Jan 2024 12:04:26 +0700
Subject: [PATCH] Also remove player from player2ResubCount when he disconnects

---
 .../ru/dbotthepony/mc/otm/block/entity/MatteryBlockEntity.kt  | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/main/kotlin/ru/dbotthepony/mc/otm/block/entity/MatteryBlockEntity.kt b/src/main/kotlin/ru/dbotthepony/mc/otm/block/entity/MatteryBlockEntity.kt
index 5f88105b4..94711d14d 100644
--- a/src/main/kotlin/ru/dbotthepony/mc/otm/block/entity/MatteryBlockEntity.kt
+++ b/src/main/kotlin/ru/dbotthepony/mc/otm/block/entity/MatteryBlockEntity.kt
@@ -804,11 +804,13 @@ abstract class MatteryBlockEntity(p_155228_: BlockEntityType<*>, p_155229_: Bloc
 			val level = event.level as? ServerLevel ?: return
 			playerMap.remove(level)
 			tickingMap.remove(level)
+			player2ResubCount.clear()
 		}
 
 		fun onServerStopping(event: ServerStoppingEvent) {
 			playerMap.clear()
 			tickingMap.clear()
+			player2ResubCount.clear()
 		}
 
 		fun postLevelTick(event: LevelTickEvent) {
@@ -849,6 +851,8 @@ abstract class MatteryBlockEntity(p_155228_: BlockEntityType<*>, p_155229_: Bloc
 		}
 
 		fun playerDisconnected(event: PlayerEvent.PlayerLoggedOutEvent) {
+			player2ResubCount.removeInt(event.entity as ServerPlayer)
+
 			for (tree in playerMap.values) {
 				tree.values.removeIf {
 					it.unsubscribe(event.entity as ServerPlayer)