Fix folder for stable rustc (#797)

ecmascript/visit/Cargo.toml

@@ -13,5 +13,5 @@ edition = "2018"
 swc_atoms = { version = "0.2", path = "../../atoms" }
 swc_common = { version = "0.5.9", path = "../../common" }
 swc_ecma_ast = { version = "0.20.0", path ="../ast" }
-swc_ecma_visit_macros = { version = "0.3.0", path ="./macros" }
+swc_ecma_visit_macros = { version = "0.4.0", path ="./macros" }
 num-bigint = { version = "0.2", features = ["serde"] }

ecmascript/visit/macros/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "swc_ecma_visit_macros"
-version = "0.3.0"
+version = "0.4.0"
 authors = ["강동윤 <kdy1997.dev@gmail.com>"]
 license = "Apache-2.0/MIT"
 repository = "https://github.com/swc-project/swc.git"

ecmascript/visit/macros/src/lib.rs

@@ -778,7 +778,18 @@ fn create_method_body(mode: Mode, ty: &Type) -> Block {
                                         _ => {}
                                     }
 
-                                    return q!(
+                                    return match mode {
+                                        Mode::Folder => q!(
+                                            Vars { ident },
+                                            ({
+                                                match n {
+                                                    Some(n) => Some(_visitor.ident(n)),
+                                                    None => None,
+                                                }
+                                            })
+                                        )
+                                        .parse(),
+                                        Mode::Visitor => q!(
                                             Vars { ident },
                                             ({
                                                 match n {
@@ -787,7 +798,8 @@ fn create_method_body(mode: Mode, ty: &Type) -> Block {
                                                 }
                                             })
                                         )
-                                    .parse();
+                                        .parse(),
+                                    };
                                 }
                                 _ => unimplemented!("generic parameter other than type"),
                             }

ecmascript/visit/macros/tests/fold.rs

@@ -0,0 +1,29 @@
+use std::any::Any;
+use swc_ecma_visit_macros::define;
+
+/// Visitable nodes.
+pub trait Node: Any {}
+
+impl<T: ?Sized> Node for T where T: Any {}
+
+pub struct Item {
+    // pub field: usize,
+    // pub inner: Option<Box<Item>>,
+    pub opt_vec: Option<Vec<Item>>,
+    pub vec_opt: Vec<Option<Item>>,
+}
+pub enum Enum {
+    Item(Item),
+}
+
+define!({
+    pub struct Item {
+        // pub field: usize,
+        // pub inner: Option<Box<Item>>,
+        pub opt_vec: Option<Vec<Item>>,
+        pub vec_opt: Vec<Option<Item>>,
+    }
+    pub enum Enum {
+        Item(Item),
+    }
+});

ecmascript/visit/macros/tests/opt_vec.rs

@@ -0,0 +1,25 @@
+use std::any::Any;
+use swc_ecma_visit_macros::define;
+
+/// Visitable nodes.
+pub trait Node: Any {}
+
+impl<T: ?Sized> Node for T where T: Any {}
+
+pub struct Item {
+    pub opt_vec1: Option<Vec<Item>>,
+    pub opt_vec2: Option<Vec<Enum>>,
+}
+pub enum Enum {
+    Item(Item),
+}
+
+define!({
+    pub struct Item {
+        pub opt_vec1: Option<Vec<Item>>,
+        pub opt_vec2: Option<Vec<Enum>>,
+    }
+    pub enum Enum {
+        Item(Item),
+    }
+});

ecmascript/visit/macros/tests/vec_opt.rs

@@ -0,0 +1,25 @@
+use std::any::Any;
+use swc_ecma_visit_macros::define;
+
+/// Visitable nodes.
+pub trait Node: Any {}
+
+impl<T: ?Sized> Node for T where T: Any {}
+
+pub struct Item {
+    pub vec_opt1: Vec<Option<Item>>,
+    pub vec_opt2: Vec<Option<Enum>>,
+}
+pub enum Enum {
+    Item(Item),
+}
+
+define!({
+    pub struct Item {
+        pub vec_opt1: Vec<Option<Item>>,
+        pub vec_opt2: Vec<Option<Enum>>,
+    }
+    pub enum Enum {
+        Item(Item),
+    }
+});