Add jsx option

Author: ije

src/lib.rs

@@ -34,6 +34,7 @@ pub struct Options {
   pub resolve_remote_module: Option<bool>,
   pub is_dev: Option<bool>,
   pub source_map: Option<bool>,
+  pub jsx: Option<String>,
   pub jsx_pragma: Option<String>,
   pub jsx_pragma_frag: Option<String>,
   pub jsx_import_source: Option<String>,
@@ -116,6 +117,7 @@ pub fn transform(specifier: &str, code: &str, options: JsValue) -> Result<JsValu
       resolver.clone(),
       &EmitOptions {
         target,
+        jsx: options.jsx,
         jsx_pragma: options.jsx_pragma,
         jsx_pragma_frag: options.jsx_pragma_frag,
         jsx_import_source: options.jsx_import_source,

src/swc.rs

@@ -23,6 +23,7 @@ use swc_ecmascript::visit::{as_folder, Fold, FoldWith};
 #[derive(Clone)]
 pub struct EmitOptions {
   pub target: EsVersion,
+  pub jsx: Option<String>,
   pub jsx_pragma: Option<String>,
   pub jsx_pragma_frag: Option<String>,
   pub jsx_import_source: Option<String>,
@@ -36,6 +37,7 @@ impl Default for EmitOptions {
   fn default() -> Self {
     EmitOptions {
       target: EsVersion::Es2022,
+      jsx: None,
       jsx_pragma: None,
       jsx_pragma_frag: None,
       jsx_import_source: None,
@@ -58,7 +60,6 @@ pub struct SWC {
 impl SWC {
   /// parse source code.
   pub fn parse(specifier: &str, source: &str, target: EsVersion, lang: Option<String>) -> Result<Self, anyhow::Error> {
-    print!("--- {} {:?} {:?}\n", specifier, target, lang);
     let source_map = SourceMap::default();
     let source_file = source_map.new_source_file(FileName::Real(Path::new(specifier).to_path_buf()), source.into());
     let sm = &source_map;
@@ -112,14 +113,19 @@ impl SWC {
       let unresolved_mark = Mark::new();
       let top_level_mark = Mark::fresh(Mark::root());
       let specifier_is_remote = resolver.borrow().specifier_is_remote;
+      let extname = Path::new(&self.specifier)
+        .extension()
+        .unwrap_or_default()
+        .to_ascii_lowercase();
       let is_dev = resolver.borrow().is_dev;
-      let is_ts =
-        self.specifier.ends_with(".ts") || self.specifier.ends_with(".mts") || self.specifier.ends_with(".tsx");
-      let is_jsx = self.specifier.ends_with(".tsx") || self.specifier.ends_with(".jsx");
-      let react_options = if let Some(jsx_import_source) = &options.jsx_import_source {
+      let is_ts = extname == "ts" || extname == "mts" || extname == "tsx";
+      let jsxt = options.jsx.as_deref().unwrap_or("classic");
+      let is_jsx = jsxt != "preserve" && (extname == "jsx" || extname == "tsx");
+      let react_options = if jsxt == "automatic" {
         let mut resolver = resolver.borrow_mut();
+        let import_source = options.jsx_import_source.as_deref().unwrap_or("react");
         let runtime = if is_dev { "/jsx-dev-runtime" } else { "/jsx-runtime" };
-        let import_source = resolver.resolve(&(jsx_import_source.to_owned() + runtime), false, None);
+        let import_source = resolver.resolve(&(import_source.to_owned() + runtime), false, None);
         let import_source = import_source
           .split("?")
           .next()

src/tests.rs

@@ -146,6 +146,56 @@ fn import_resolving() {
   assert!(code.contains("new Worker(\"/-/esm.sh/asksomeonelse\")"));
 }
 
+#[test]
+fn jsx_preserve() {
+  let source = r#"
+    export default function App() {
+      return (
+        <>
+          <h1 className="title">Hello world!</h1>
+        </>
+      )
+    }
+  "#;
+  let (code, _) = transform(
+    "./app.tsx",
+    source,
+    false,
+    &EmitOptions {
+      jsx: Some("preserve".into()),
+      ..Default::default()
+    },
+  );
+  assert!(code.contains("<h1 className=\"title\">Hello world!</h1>"));
+  assert!(code.contains("<>"));
+  assert!(code.contains("</>"));
+}
+
+#[test]
+fn jsx_classic() {
+  let source = r#"
+    import React from "react"
+    export default function App() {
+      return (
+        <>
+          <h1 className="title">Hello world!</h1>
+        </>
+      )
+    }
+  "#;
+  let (code, _) = transform(
+    "./app.tsx",
+    source,
+    false,
+    &EmitOptions {
+      jsx: Some("classic".into()),
+      ..Default::default()
+    },
+  );
+  assert!(code.contains("React.createElement(\"h1\""));
+  assert!(code.contains("React.createElement(React.Fragment,"));
+}
+
 #[test]
 fn jsx_automtic() {
   let source = r#"
@@ -163,6 +213,7 @@ fn jsx_automtic() {
     source,
     false,
     &EmitOptions {
+      jsx: Some("automatic".into()),
       jsx_import_source: Some("https://esm.sh/react@18".to_owned()),
       ..Default::default()
     },
@@ -194,6 +245,7 @@ fn react_refresh() {
     true,
     &EmitOptions {
       react_refresh: true,
+      jsx: Some("automatic".into()),
       jsx_import_source: Some("https://esm.sh/react@18".to_owned()),
       ..Default::default()
     },
@@ -264,6 +316,7 @@ fn strip_data_export() {
     false,
     &EmitOptions {
       strip_data_export: true,
+      jsx: Some("automatic".into()),
       jsx_import_source: Some("https://esm.sh/react@18".to_owned()),
       ..Default::default()
     },

types.ts

@@ -20,6 +20,7 @@ export type TransformOptions = {
   isDev?: boolean;
   reactRefresh?: boolean;
   sourceMap?: boolean;
+  jsx?: "automatic" | "classic" | "preserve";
   jsxPragma?: string;
   jsxPragmaFrag?: string;
   jsxImportSource?: string;