Add (un)register_class hooks in src/_pythonscript_extension_class_script.pxi & src/_pythonscript_extension_class_language.pxi

touilleMan · touilleMan · commit ce6a76c743e9 · 2024-12-26T19:12:51.000+01:00
diff --git a/src/_pythonscript_extension_class_language.pxi b/src/_pythonscript_extension_class_language.pxi
@@ -1,8 +1,8 @@
 from godot.classes cimport ScriptLanguageExtensionProfilingInfo
 
 
-cdef StringName gdname_scriptlanguageextension = StringName("ScriptLanguageExtension")
-cdef StringName gdname_pythonscriptlanguage = StringName("PythonScriptLanguage")
+cdef gd_string_name_t gdname_scriptlanguageextension
+cdef gd_string_name_t gdname_pythonscriptlanguage
 
 
 debug_spy = False
@@ -17,8 +17,22 @@ cdef class PythonScriptLanguage:
     cdef gd_object_t _gd_ptr
 
     def __cinit__(self):
-        self._gd_ptr = pythonscript_gdextension.classdb_construct_object(&gdname_scriptlanguageextension._gd_data)
-        pythonscript_gdextension.object_set_instance(self._gd_ptr, &gdname_pythonscriptlanguage._gd_data, <PyObject*>self)
+        self._gd_ptr = pythonscript_gdextension.classdb_construct_object(&gdname_scriptlanguageextension)
+        pythonscript_gdextension.object_set_instance(self._gd_ptr, &gdname_pythonscriptlanguage, <PyObject*>self)
+
+    # godot_extension: register_class_hook()
+    @staticmethod
+    cdef inline void _register_class_hook():
+        global gdname_scriptlanguageextension, gdname_pythonscriptlanguage
+        gdname_scriptlanguageextension = gd_string_name_from_unchecked_pystr("ScriptLanguageExtension")
+        gdname_pythonscriptlanguage = gd_string_name_from_unchecked_pystr("PythonScriptLanguage")
+
+    # godot_extension: unregister_class_hook()
+    @staticmethod
+    cdef inline void _unregister_class_hook():
+        global gdname_scriptlanguageextension, gdname_pythonscriptlanguage
+        gd_string_name_del(&gdname_scriptlanguageextension)
+        gd_string_name_del(&gdname_pythonscriptlanguage)
 
     # godot_extension: generate_code()
 
diff --git a/src/_pythonscript_extension_class_script.pxi b/src/_pythonscript_extension_class_script.pxi
@@ -1,14 +1,28 @@
-cdef StringName gdname_scriptextension = StringName("ScriptExtension")
-cdef StringName gdname_pythonscript = StringName("PythonScript")
+cdef gd_string_name_t gdname_scriptextension
+cdef gd_string_name_t gdname_pythonscript
 
 # godot_extension: class(parent="ScriptExtension")
 @cython.final
 cdef class PythonScript:
     cdef gd_object_t _gd_ptr
 
     def __cinit__(self):
-        self._gd_ptr = pythonscript_gdextension.classdb_construct_object(&gdname_scriptextension._gd_data)
-        pythonscript_gdextension.object_set_instance(self._gd_ptr, &gdname_pythonscript._gd_data, <PyObject*>self)
+        self._gd_ptr = pythonscript_gdextension.classdb_construct_object(&gdname_scriptextension)
+        pythonscript_gdextension.object_set_instance(self._gd_ptr, &gdname_pythonscript, <PyObject*>self)
+
+    # godot_extension: register_class_hook()
+    @staticmethod
+    cdef inline void _register_class_hook():
+        global gdname_scriptextension, gdname_pythonscript
+        gdname_scriptextension = gd_string_name_from_unchecked_pystr("ScriptExtension")
+        gdname_pythonscript = gd_string_name_from_unchecked_pystr("PythonScript")
+
+    # godot_extension: unregister_class_hook()
+    @staticmethod
+    cdef inline void _unregister_class_hook():
+        global gdname_scriptextension, gdname_pythonscript
+        gd_string_name_del(&gdname_scriptextension)
+        gd_string_name_del(&gdname_pythonscript)
 
     # godot_extension: generate_code()
 
