GDExtension: Call startup callback only after reload is fully finished

Author: dsnopek

core/extension/gdextension_manager.cpp

@@ -55,6 +55,16 @@ GDExtensionManager::LoadStatus GDExtensionManager::_load_extension_internal(cons
 		gdextension_class_icon_paths[kv.key] = kv.value;
 	}
 
+	if (p_first_load) {
+		// On first load, we automatically do everything at once. For reload, we need to
+		// defer this until a little later.
+		_finish_load_extension(p_extension);
+	}
+
+	return LOAD_STATUS_OK;
+}
+
+void GDExtensionManager::_finish_load_extension(const Ref<GDExtension> &p_extension) {
 #ifdef TOOLS_ENABLED
 	// Signals that a new extension is loaded so GDScript can register new class names.
 	emit_signal("extension_loaded", p_extension);
@@ -67,8 +77,6 @@ GDExtensionManager::LoadStatus GDExtensionManager::_load_extension_internal(cons
 			p_extension->startup_callback();
 		}
 	}
-
-	return LOAD_STATUS_OK;
 }
 
 GDExtensionManager::LoadStatus GDExtensionManager::_unload_extension_internal(const Ref<GDExtension> &p_extension) {
@@ -186,6 +194,8 @@ GDExtensionManager::LoadStatus GDExtensionManager::reload_extension(const String
 
 	extension->finish_reload();
 
+	_finish_load_extension(extension);
+
 	return LOAD_STATUS_OK;
 #endif
 }

core/extension/gdextension_manager.h

@@ -57,6 +57,7 @@ class GDExtensionManager : public Object {
 
 private:
 	LoadStatus _load_extension_internal(const Ref<GDExtension> &p_extension, bool p_first_load);
+	void _finish_load_extension(const Ref<GDExtension> &p_extension);
 	LoadStatus _unload_extension_internal(const Ref<GDExtension> &p_extension);
 
 #ifdef TOOLS_ENABLED