fix init in server.

way-zer · Aug 4, 2024 · ef8172b · ef8172b
1 parent 39bfedd
commit ef8172b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 10 deletions.
diff --git a/src/main/kotlin/cf/wayzer/contentsTweaker/CTNode.kt b/src/main/kotlin/cf/wayzer/contentsTweaker/CTNode.kt
@@ -162,7 +162,7 @@ class CTNode private constructor() : ExtendableClass<CTExtInfo>() {
         }
 
         val resetHandlers = mutableSetOf<Resettable>()
-        private val afterHandlers = mutableSetOf<AfterHandler>()
+        internal val afterHandlers = mutableSetOf<AfterHandler>()
 
         fun modified(modifiable: Modifiable<*>) {
             resetHandlers.add(modifiable)

diff --git a/src/main/kotlin/cf/wayzer/contentsTweaker/ContentsTweaker.kt b/src/main/kotlin/cf/wayzer/contentsTweaker/ContentsTweaker.kt
@@ -34,6 +34,7 @@ object ContentsTweaker {
     )
 
     fun afterHandle() {
+        if (CTNode.PatchHandler.afterHandlers.isEmpty()) return
         val time = measureTimeMillis {
             CTNode.PatchHandler.doAfterHandle()
         }

diff --git a/src/main/kotlin/cf/wayzer/contentsTweaker/ModMain.kt b/src/main/kotlin/cf/wayzer/contentsTweaker/ModMain.kt
@@ -33,15 +33,17 @@ class ModMain : Mod() {
             }
             ContentsTweaker.afterHandle()
         }
-        Vars.netClient.addPacketHandler("ContentsLoader|newPatch") {
-            val (name, content) = it.split('\n', limit = 2)
-            ContentsTweaker.loadPatch(name, content)
-        }
-        //TODO: Deprecated
-        Vars.netClient.addPacketHandler("ContentsLoader|loadPatch") { name ->
-            val patch = Vars.state.map.tags.get("CT@$name")
-                ?: return@addPacketHandler Call.serverPacketReliable("ContentsLoader|requestPatch", name)
-            ContentsTweaker.loadPatch(name, patch)
+        if (Vars.netClient != null) {
+            Vars.netClient.addPacketHandler("ContentsLoader|newPatch") {
+                val (name, content) = it.split('\n', limit = 2)
+                ContentsTweaker.loadPatch(name, content)
+            }
+            //TODO: Deprecated
+            Vars.netClient.addPacketHandler("ContentsLoader|loadPatch") { name ->
+                val patch = Vars.state.map.tags.get("CT@$name")
+                    ?: return@addPacketHandler Call.serverPacketReliable("ContentsLoader|requestPatch", name)
+                ContentsTweaker.loadPatch(name, patch)
+            }
         }
         Vars.mods.scripts.scope.apply {
             put("CT", this, ContentsTweaker)