Implement AsArg for Instance + TInstance

Author: Bromeon

gdnative-core/src/object/as_arg.rs

@@ -1,5 +1,7 @@
+use crate::export::user_data::Map;
+use crate::export::NativeClass;
 use crate::object::ownership::{Ownership, Shared, Unique};
-use crate::object::{GodotObject, Null, Ref, SubClass, TRef};
+use crate::object::{GodotObject, Instance, Null, Ref, SubClass, TInstance, TRef};
 
 /// Trait for safe conversion from Godot object references into API method arguments. This is
 /// a sealed trait with no public interface.
@@ -41,10 +43,18 @@ mod private {
     pub trait Sealed {}
 }
 
+// Null
 impl<'a, T> private::Sealed for Null<T> {}
+
+// Temporary references (shared ownership)
 impl<'a, T: GodotObject> private::Sealed for TRef<'a, T, Shared> {}
-impl<T: GodotObject, Own: Ownership> private::Sealed for Ref<T, Own> {}
 impl<'a, T: GodotObject> private::Sealed for &'a Ref<T, Shared> {}
+impl<'a, T: NativeClass> private::Sealed for TInstance<'a, T, Shared> {}
+impl<'a, T: NativeClass> private::Sealed for &'a Instance<T, Shared> {}
+
+// Persistent references (any ownership)
+impl<T: GodotObject, Own: Ownership> private::Sealed for Ref<T, Own> {}
+impl<T: NativeClass, Own: Ownership> private::Sealed for Instance<T, Own> {}
 
 // ----------------------------------------------------------------------------------------------------------------------------------------------
 // Null
@@ -125,3 +135,48 @@ impl<T: GodotObject> AsVariant for Ref<T, Shared> {
 impl<'a, T: GodotObject> AsVariant for &'a Ref<T, Shared> {
     type Target = T;
 }
+
+// ----------------------------------------------------------------------------------------------------------------------------------------------
+// TInstance
+
+impl<'a, T, U> AsArg<U> for TInstance<'a, T, Shared>
+where
+    T: NativeClass,
+    T::Base: GodotObject + SubClass<U>,
+    T::UserData: Map,
+    U: GodotObject,
+{
+    #[inline]
+    fn as_arg_ptr(&self) -> *mut sys::godot_object {
+        self.as_base_ptr()
+    }
+}
+
+// ----------------------------------------------------------------------------------------------------------------------------------------------
+// Instance
+
+impl<T, U, Own: Ownership> AsArg<U> for Instance<T, Own>
+where
+    T: NativeClass,
+    T::Base: GodotObject + SubClass<U>,
+    T::UserData: Map,
+    U: GodotObject,
+{
+    #[inline]
+    fn as_arg_ptr(&self) -> *mut sys::godot_object {
+        self.as_base_ptr()
+    }
+}
+
+impl<'a, T, U> AsArg<U> for &'a Instance<T, Shared>
+where
+    T: NativeClass,
+    T::Base: GodotObject + SubClass<U>,
+    T::UserData: Map,
+    U: GodotObject,
+{
+    #[inline]
+    fn as_arg_ptr(&self) -> *mut sys::godot_object {
+        self.as_base_ptr()
+    }
+}

gdnative-core/src/object/instance.rs

@@ -229,6 +229,11 @@ impl<T: NativeClass, Own: Ownership> Instance<T, Own> {
     pub fn script(&self) -> &T::UserData {
         &self.script
     }
+
+    /// Convert to a nullable raw pointer. Used for AsArg.
+    pub(super) fn as_base_ptr(&self) -> *mut sys::godot_object {
+        self.owner.as_ptr()
+    }
 }
 
 impl<T: NativeClass, Own: Ownership> Instance<T, Own>
@@ -464,6 +469,11 @@ impl<'a, T: NativeClass, Own: Ownership> TInstance<'a, T, Own> {
         let script = T::UserData::clone_from_user_data_unchecked(user_data);
         TInstance { owner, script }
     }
+
+    /// Convert to a nullable raw pointer. Used for AsArg.
+    pub(super) fn as_base_ptr(&self) -> *mut sys::godot_object {
+        self.owner.as_ptr()
+    }
 }
 
 impl<'a, T: NativeClass, Own: NonUniqueOwnership> TInstance<'a, T, Own> {