add support for nested $ methods

unyt-org · Nov 28, 2024 · d8a30b6 · d8a30b6
1 parent da50245
commit d8a30b6
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 7 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -900,4 +900,23 @@ test!(
         console.log(x);
     })
     "#
-);
+);
+
+
+
+test!(
+    Syntax::Es(EsSyntax {
+        jsx: true,
+        ..Default::default()
+    },),
+    |_| TransformVisitor,
+    t56,
+    r#"<div>
+        {
+            array
+                .filter(item => item > 0)
+                .map(item => <span>{item}</span>)
+        }
+    </div>
+    "#
+);
diff --git a/src/visitor.rs b/src/visitor.rs
@@ -499,7 +499,7 @@ impl TransformVisitor {
 
                 let mut args: Vec::<ExprOrSpread> = vec![
                     ExprOrSpread {
-                        expr: obj,
+                        expr: self.transform_expr_reactive(obj, "_$"),
                         spread: None
                     },
                     ExprOrSpread {

diff --git a/tests/__swc_snapshots__/src/lib.rs/t49.js b/tests/__swc_snapshots__/src/lib.rs/t49.js
@@ -18,10 +18,10 @@ call(function({ title, icon, children }) {
             2,
             3
         ])}
-            {..._$method([
-        1,
-        2,
-        3
-    ], "map", (item)=><span>{item}</span>)}
+            {..._$method(_$(()=>[
+            1,
+            2,
+            3
+        ]), "map", (item)=><span>{item}</span>)}
         </div>;
 });
diff --git a/tests/__swc_snapshots__/src/lib.rs/t56.js b/tests/__swc_snapshots__/src/lib.rs/t56.js
@@ -0,0 +1,3 @@
+<div>
+        {_$method(_$method(array, "filter", (item)=>item > 0), "map", (item)=><span>{item}</span>)}
+    </div>;