123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154
// Copyright 2020-2021 postUI Lab. All rights reserved. MIT license.
#[macro_use]extern crate lazy_static;
mod aleph;mod error;mod fast_refresh;mod fixer;mod import_map;mod jsx;mod resolve;mod source_type;mod swc;
use import_map::ImportHashMap;use resolve::{DependencyDescriptor, InlineStyle, Resolver};use serde::{Deserialize, Serialize};use source_type::SourceType;use std::collections::HashMap;use std::{cell::RefCell, rc::Rc};use swc::{EmitOptions, ParsedModule};use swc_ecmascript::parser::JscTarget;use wasm_bindgen::prelude::{wasm_bindgen, JsValue};
#[derive(Clone, Debug, Default, Deserialize)]#[serde(deny_unknown_fields, rename_all = "camelCase")]pub struct Options {  pub url: String,
  #[serde(default)]  pub import_map: ImportHashMap,
  #[serde(default = "default_react_version")]  pub react_version: String,
  #[serde(default)]  pub swc_options: SWCOptions,
  #[serde(default)]  pub is_dev: bool,
  #[serde(default)]  pub bundle_mode: bool,
  #[serde(default)]  pub bundled_modules: Vec<String>,}
#[derive(Clone, Debug, Deserialize)]#[serde(deny_unknown_fields, rename_all = "camelCase")]pub struct SWCOptions {  #[serde(default = "default_target")]  pub target: JscTarget,
  #[serde(default = "default_pragma")]  pub jsx_factory: String,
  #[serde(default = "default_pragma_frag")]  pub jsx_fragment_factory: String,
  #[serde(default)]  pub source_type: String,
  #[serde(default)]  pub source_map: bool,}
impl Default for SWCOptions {  fn default() -> Self {    SWCOptions {      target: default_target(),      jsx_factory: default_pragma(),      jsx_fragment_factory: default_pragma_frag(),      source_type: "".into(),      source_map: false,    }  }}
fn default_target() -> JscTarget {  JscTarget::Es2020}
fn default_pragma() -> String {  "React.createElement".into()}
fn default_pragma_frag() -> String {  "React.Fragment".into()}
fn default_react_version() -> String {  "17.0.1".into()}
#[derive(Debug, Serialize)]#[serde(rename_all = "camelCase")]pub struct TransformOutput {  pub code: String,  #[serde(skip_serializing_if = "Option::is_none")]  pub map: Option<String>,  pub deps: Vec<DependencyDescriptor>,  pub inline_styles: HashMap<String, InlineStyle>,}
#[wasm_bindgen(js_name = "transformSync")]pub fn transform_sync(s: &str, opts: JsValue) -> Result<JsValue, JsValue> {  console_error_panic_hook::set_once();
  let opts: Options = opts    .into_serde()    .map_err(|err| format!("failed to parse options: {}", err))    .unwrap();  let resolver = Rc::new(RefCell::new(Resolver::new(    opts.url.as_str(),    opts.import_map,    Some(opts.react_version),    opts.bundle_mode,    opts.bundled_modules,  )));  let specify_source_type = match opts.swc_options.source_type.as_str() {    "js" => Some(SourceType::JavaScript),    "jsx" => Some(SourceType::JSX),    "ts" => Some(SourceType::TypeScript),    "tsx" => Some(SourceType::TSX),    _ => None,  };  let module =    ParsedModule::parse(opts.url.as_str(), s, specify_source_type).expect("could not parse module");  let (code, map) = module    .transpile(      resolver.clone(),      &EmitOptions {        target: opts.swc_options.target,        jsx_factory: opts.swc_options.jsx_factory.clone(),        jsx_fragment_factory: opts.swc_options.jsx_fragment_factory.clone(),        source_map: opts.swc_options.source_map,        is_dev: opts.is_dev,      },    )    .expect("could not transpile module");  let r = resolver.borrow_mut();  Ok(    JsValue::from_serde(&TransformOutput {      code,      map,      deps: r.dep_graph.clone(),      inline_styles: r.inline_styles.clone(),    })    .unwrap(),  )}