definition_named_data_type use Type::definition + `reference::cus…

…tom`
specta-rs · Dec 4, 2023 · 01b6f2c · 01b6f2c
1 parent aa6676c
commit 01b6f2c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/macros/src/type/mod.rs b/macros/src/type/mod.rs
@@ -167,9 +167,13 @@ pub fn derive(input: proc_macro::TokenStream) -> syn::Result<proc_macro::TokenSt
                 }
 
                 fn definition_named_data_type(type_map: &mut #crate_ref::TypeMap) -> #crate_ref::NamedDataType {
-                    Self::named_data_type(
-                        type_map,
-                        &DEFINITION_GENERICS
+                    #crate_ref::internal::construct::named_data_type(
+                        #name.into(),
+                        #comments.into(),
+                        #deprecated,
+                        SID,
+                        IMPL_LOCATION,
+                        <Self as #crate_ref::Type>::definition(type_map)
                     )
                 }
             }

diff --git a/src/datatype/reference.rs b/src/datatype/reference.rs
@@ -19,7 +19,6 @@ pub fn inline<T: Type + ?Sized>(type_map: &mut TypeMap, generics: &[DataType]) -
 pub fn reference<T: NamedType>(type_map: &mut TypeMap, reference: DataTypeReference) -> Reference {
     if type_map.map.get(&T::SID).is_none() {
         type_map.map.entry(T::SID).or_insert(None);
-
         let dt = T::definition_named_data_type(type_map);
         type_map.map.insert(T::SID, Some(dt));
     }
@@ -28,3 +27,10 @@ pub fn reference<T: NamedType>(type_map: &mut TypeMap, reference: DataTypeRefere
         inner: DataType::Reference(reference),
     }
 }
+
+/// Construct a reference from a custom [DataType].
+///
+/// This function is advanced and should only be used if you know what you're doing.
+pub fn custom(inner: DataType) -> Reference {
+    Reference { inner }
+}