Support enum exports (#42)

TitanNano · Jul 18, 2024 · 783d798 · 783d798
1 parent d613594
commit 783d798
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/derive/src/attribute_ops.rs b/derive/src/attribute_ops.rs
@@ -19,6 +19,7 @@ pub struct FieldExportOps {
     dir: bool,
     exp_easing: Option<syn::ExprArray>,
     file: Option<syn::ExprArray>,
+    enum_options: Option<syn::ExprArray>,
     flags: Option<syn::ExprArray>,
     #[darling(default)]
     global_dir: bool,
@@ -101,6 +102,23 @@ impl FieldExportOps {
             result = Some((field, quote!(#property_hints::FILE), filters.join(",")));
         }
 
+        if let Some(list) = self.enum_options.as_ref() {
+            let field = "enum";
+
+            if let Some((active_field, _, _)) = result {
+                return Self::error(span, active_field, field);
+            }
+
+            let flags = list
+                .elems
+                .iter()
+                .map(String::from_expr)
+                .collect::<Result<Vec<_>, _>>()
+                .map_err(|err| err.write_errors())?;
+
+            result = Some((field, quote!(#property_hints::ENUM), flags.join(",")));
+        }
+
         if let Some(list) = self.flags.as_ref() {
             let field = "flags";
 

diff --git a/rust-script/tests/script_derive.rs b/rust-script/tests/script_derive.rs
@@ -16,6 +16,9 @@ struct TestScript {
     #[export]
     pub editor_prop: u16,
 
+    #[export(enum_options = ["inactive", "water", "teargas"])]
+    pub enum_prop: u8,
+
     base: Gd<Node>,
 }