Allow exporting RsRef<T> (#71)

Author: TitanNano

rust-script/src/interface.rs

@@ -13,7 +13,7 @@ use std::{collections::HashMap, fmt::Debug};
 
 use godot::meta::{FromGodot, GodotConvert, ToGodot};
 use godot::obj::Inherits;
-use godot::prelude::{Gd, Object, StringName, Variant};
+use godot::prelude::{ConvertError, Gd, Object, StringName, Variant};
 
 pub use crate::runtime::Context;
 
@@ -110,6 +110,30 @@ impl<T: GodotScript> Clone for RsRef<T> {
     }
 }
 
+impl<T: GodotScript> GodotConvert for RsRef<T> {
+    type Via = Gd<T::Base>;
+}
+
+impl<T: GodotScript> FromGodot for RsRef<T>
+where
+    T::Base: Inherits<T::Base>,
+{
+    fn try_from_godot(via: Self::Via) -> Result<Self, godot::prelude::ConvertError> {
+        via.try_to_script().map_err(ConvertError::with_error)
+    }
+}
+
+impl<T: GodotScript> ToGodot for RsRef<T> {
+    type ToVia<'v>
+        = Gd<T::Base>
+    where
+        Self: 'v;
+
+    fn to_godot(&self) -> Self::ToVia<'_> {
+        self.deref().clone()
+    }
+}
+
 #[derive(thiserror::Error, Debug)]
 pub enum GodotScriptCastError {
     #[error("Object has no script attached!")]

rust-script/src/interface/export.rs

@@ -20,6 +20,8 @@ use godot::obj::{EngineEnum, Gd};
 use godot::prelude::GodotClass;
 use godot::sys::GodotFfi;
 
+use super::{GodotScript, RsRef};
+
 pub trait GodotScriptExport: GodotConvert + FromGodot + ToGodot {
     fn hint_string(custom_hint: Option<PropertyHint>, custom_string: Option<String>) -> String;
 
@@ -50,6 +52,30 @@ impl<T: GodotClass> GodotScriptExport for Gd<T> {
     }
 }
 
+impl<T: GodotScript> GodotScriptExport for RsRef<T> {
+    fn hint_string(_custom_hint: Option<PropertyHint>, custom_string: Option<String>) -> String {
+        if let Some(custom) = custom_string {
+            return custom;
+        }
+
+        T::CLASS_NAME.to_string()
+    }
+
+    fn hint(custom: Option<PropertyHint>) -> PropertyHint {
+        if let Some(custom) = custom {
+            return custom;
+        }
+
+        if T::Base::inherits::<Node>() {
+            PropertyHint::NODE_TYPE
+        } else if T::Base::inherits::<Resource>() {
+            PropertyHint::RESOURCE_TYPE
+        } else {
+            PropertyHint::NONE
+        }
+    }
+}
+
 impl<T: GodotScriptExport> GodotScriptExport for Option<T>
 where
     for<'v> T: 'v,