feat: support removal of simple form of assertion

twada · twada · commit 825d747b9e0b · 2024-02-19T14:48:04.000+09:00
diff --git a/src/lib.rs b/src/lib.rs
@@ -11,6 +11,11 @@ use swc_core::ecma::ast::{
     ImportNamedSpecifier,
     ModuleItem,
     ModuleDecl,
+    CallExpr,
+    Callee,
+    Expr,
+    Stmt,
+    ExprStmt,
 };
 use swc_core::ecma::visit::{
     as_folder,
@@ -71,6 +76,44 @@ impl VisitMut for TransformVisitor {
             }
         });
     }
+
+    fn visit_mut_call_expr(&mut self, n: &mut CallExpr) {
+        if self.target_variables.is_empty() {
+            n.visit_mut_children_with(self);
+            return;
+        }
+        let matched = match n.callee {
+            Callee::Expr(ref expr) => {
+                match expr.as_ref() {
+                    Expr::Ident(ref ident) => self.target_variables.contains(&ident.to_id()),
+                    _ => false
+                }
+            },
+            _ => false
+        };
+        if matched {
+            n.take();
+        } else {
+            n.visit_mut_children_with(self);
+        }
+    }
+
+    fn visit_mut_stmt(&mut self, n: &mut Stmt) {
+        n.visit_mut_children_with(self);
+        match n {
+            Stmt::Expr(ExprStmt{ expr, ..}) => {
+                match expr.as_ref() {
+                    Expr::Call(call_expr) => {
+                        if *call_expr == CallExpr::dummy() {
+                            n.take();
+                        }
+                    },
+                    _ => {}
+                }
+            },
+            _ => {}
+        }
+    }
 }
 
 /// An example plugin function with macro support.
diff --git a/tests/fixtures/simple/expected.mjs b/tests/fixtures/simple/expected.mjs
@@ -1,2 +1,3 @@
 const foo = 1;
 const bar = 2;
+;

Original file line number	Diff line number	Diff line change
`@@ -1,2 +1,3 @@`
`1`	`1`	`const foo = 1;`
`2`	`2`	`const bar = 2;`
	`3`	`+;`