make swc transform behave the same as oxc (#5)

Author: Boshen

Diff for: Cargo.lock

@@ -23,6 +23,7 @@ swc = "0.276.0"
 swc_common = "0.33.26"
 swc_ecma_ast = "0.113.7"
 swc_ecma_parser = { version = "0.144.2", features = ["typescript"] }
+swc_ecma_transforms = "0.230.1"
 swc_ecma_transforms_react = "0.184.1"
 swc_ecma_transforms_typescript = "0.189.1"
 swc_ecma_visit = "0.99.1"

Diff for: Cargo.toml

@@ -54,6 +54,17 @@ pnpm run table
 
 ## Maximum Resident Set Size
 
+```
+./memory.sh
+
+./files/cal.com.tsx
+oxc 41.7 mb
+swc 31.1 mb
+
+./files/typescript.js
+oxc 224.4 mb
+swc 160.1 mb
+```
 
 ## Setup
 

Diff for: README.md

@@ -37,7 +37,7 @@ trait TheBencher {
 struct OxcBencher;
 
 impl TheBencher for OxcBencher {
-    type RunOutput = oxc::allocator::Allocator;
+    type RunOutput = (oxc::allocator::Allocator, String);
 
     const ID: &'static str = "oxc";
 
@@ -49,7 +49,7 @@ impl TheBencher for OxcBencher {
 struct SwcBencher;
 
 impl TheBencher for SwcBencher {
-    type RunOutput = swc_ecma_ast::Program;
+    type RunOutput = (swc_ecma_ast::Program, String);
 
     const ID: &'static str = "swc";
 

Diff for: benches/transformer.rs

@@ -9,10 +9,10 @@ pub mod oxc {
         transformer::{ReactOptions, TransformOptions, Transformer, TypeScriptOptions},
     };
 
-    pub fn transform(path: &Path, source_text: &str) -> Allocator {
+    pub fn transform(path: &Path, source_text: &str) -> (Allocator, String) {
         let allocator = Allocator::default();
         let source_type = SourceType::from_path(path).unwrap();
-        {
+        let printed = {
             let ret = Parser::new(&allocator, source_text, source_type).parse();
             let trivias = ret.trivias;
             let mut program = ret.program;
@@ -31,27 +31,29 @@ pub mod oxc {
             )
             .build(&mut program)
             .unwrap();
-            let _transformed_text =
-                Codegen::<false>::new("", source_text, CodegenOptions::default(), None)
-                    .build(&program);
-        }
-        allocator
+            Codegen::<false>::new("", source_text, CodegenOptions::default(), None)
+                .build(&program)
+                .source_text
+        };
+
+        (allocator, printed)
     }
 }
 
 pub mod swc {
-    use std::path::Path;
+    use std::{path::Path, sync::Arc};
 
-    use std::sync::Arc;
     use swc::{Compiler, PrintArgs, SwcComments};
     use swc_common::{chain, source_map::SourceMap, sync::Lrc, Mark, GLOBALS};
     use swc_ecma_ast::Program;
     use swc_ecma_parser::{EsConfig, Parser, StringInput, Syntax, TsConfig};
-    use swc_ecma_transforms_react::{react, Options};
+    use swc_ecma_transforms::resolver;
+    use swc_ecma_transforms_react::{react, Options, Runtime};
     use swc_ecma_transforms_typescript::strip;
     use swc_ecma_visit::FoldWith;
+    use swc_ecma_visit::VisitMutWith;
 
-    pub fn transform(path: &Path, source_text: &str) -> Program {
+    pub fn transform(path: &Path, source_text: &str) -> (Program, String) {
         let cm = Lrc::new(SourceMap::new(swc_common::FilePathMapping::empty()));
         let compiler = Compiler::new(Arc::clone(&cm));
         let comments = SwcComments::default();
@@ -66,29 +68,39 @@ pub mod swc {
 
         GLOBALS.set(&Default::default(), || {
             let input = StringInput::new(source_text, Default::default(), Default::default());
-            let program = Parser::new(syntax, input, Some(&comments))
+            let mut program = Parser::new(syntax, input, Some(&comments))
                 .parse_program()
                 .unwrap();
 
             let top_level_mark = Mark::new();
             let unresolved_mark = Mark::new();
+
+            program.visit_mut_with(&mut resolver(
+                unresolved_mark,
+                top_level_mark,
+                syntax.typescript(),
+            ));
+
             let mut ast_pass = chain!(
                 strip(top_level_mark),
                 react(
                     Arc::clone(&cm),
                     Some(comments),
-                    Options::default(),
+                    Options {
+                        runtime: Some(Runtime::Automatic),
+                        ..Options::default()
+                    },
                     top_level_mark,
                     unresolved_mark
                 ),
             );
             let program = program.fold_with(&mut ast_pass);
 
-            let _ret = compiler
+            let printed = compiler
                 .print(&program, PrintArgs::default())
                 .expect("print failed");
 
-            program
+            (program, printed.code)
         })
     }
 }

Diff for: src/lib.rs

@@ -9,5 +9,6 @@ pub fn main() {
     let path = env::args().nth(1).unwrap();
     let path = Path::new(&path);
     let source_text = fs::read_to_string(path).unwrap();
-    let _ = oxc::transform(path, &source_text);
+    let _output = oxc::transform(path, &source_text);
+    // println!("{}", output.1);
 }

Diff for: src/oxc.rs

@@ -9,5 +9,6 @@ pub fn main() {
     let path = env::args().nth(1).unwrap();
     let path = Path::new(&path);
     let source_text = fs::read_to_string(path).unwrap();
-    let _ = swc::transform(path, &source_text);
+    let _output = swc::transform(path, &source_text);
+    // println!("{}", output.1);
 }