Only remove the callback value from our shared array when the last ex…

…ntesionLevel has been processed
migueldeicaza · Jun 16, 2024 · 522cd18 · 522cd18
1 parent 9ef16a2
commit 522cd18
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Sources/SwiftGodot/EntryPoint.swift b/Sources/SwiftGodot/EntryPoint.swift
@@ -53,9 +53,13 @@ func extension_deinitialize (userData: UnsafeMutableRawPointer?, l: GDExtensionI
     //print ("SWIFT: extension_deinitialize")
     guard let userData else { return }
     let key = OpaquePointer(userData)
-    guard let callback = extensionDeInitCallbacks.removeValue(forKey: key) else { return }
+    guard let callback = extensionDeInitCallbacks [key] else { return }
     guard let level = GDExtension.InitializationLevel(rawValue: Int64 (exactly: l.rawValue)!) else { return }
     callback (level)
+    if level == .core {
+        // Last one, remove 
+        extensionDeInitCallbacks.removeValue(forKey: key)
+    }
 }
 
 /// Error types returned by Godot when invoking a method