Adds macro to convert RadiantNode to NodeType

radiant-labs · Nov 5, 2023 · 5800549 · 5800549
1 parent 61981ec
commit 5800549
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 5 deletions.
diff --git a/crates/macros/src/lib.rs b/crates/macros/src/lib.rs
@@ -65,6 +65,14 @@ fn derive_node_internal(item: TokenStream2) -> syn::Result<TokenStream2> {
         .iter()
         .map(|variant| variant.ident.clone())
         .collect::<Vec<_>>();
+    let nodes = item.variants.iter().map(|variant| {
+        let fields = variant
+            .fields
+            .iter();
+        quote! {
+            #(#fields)*
+        }
+    });
 
     let res = quote! {
         impl RadiantNode for #name {
@@ -92,6 +100,14 @@ fn derive_node_internal(item: TokenStream2) -> syn::Result<TokenStream2> {
                 }
             }
         }
+
+        #(
+            impl From<#nodes> for #name {
+                fn from(node: #nodes) -> Self {
+                    Self::#node_names(node)
+                }
+            }
+        )*
     };
     Ok(res)
 }

diff --git a/examples/basic/src/main.rs b/examples/basic/src/main.rs
@@ -1,5 +1,5 @@
 use radiant_runtime::{
-    run_native, RadiantNodeType, RadiantPathNode, RadiantRectangleNode, RadiantResponse,
+    run_native, RadiantPathNode, RadiantRectangleNode, RadiantResponse,
     RadiantRuntime, Runtime,
 };
 
@@ -17,17 +17,17 @@ async fn run() {
     let mut runtime = RadiantRuntime::new().await;
     runtime
         .scene()
-        .add(RadiantNodeType::Rectangle(RadiantRectangleNode::new(
+        .add(RadiantRectangleNode::new(
             1,
             [100.0, 100.0],
             [200.0, 200.0],
-        )));
+        ).into());
     runtime
         .scene()
-        .add(RadiantNodeType::Path(RadiantPathNode::new(
+        .add(RadiantPathNode::new(
             2,
             [400.0, 400.0],
-        )));
+        ).into());
 
     run_native(runtime, handler);
 }