1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
use crate::cjs::CJSLexer;use crate::error::{DiagnosticBuffer, ErrorBuffer};
use indexmap::{IndexMap, IndexSet};use std::{path::Path, rc::Rc};use swc_common::{  comments::SingleThreadedComments,  errors::{Handler, HandlerFlags},  FileName, SourceMap,};use swc_ecma_ast::EsVersion;use swc_ecmascript::{  ast::{Module, Program},  parser::{lexer::Lexer, EsConfig, StringInput, Syntax},  visit::FoldWith,};
pub struct SWC {  pub specifier: String,  pub module: Module,  pub source_map: Rc<SourceMap>,  pub comments: SingleThreadedComments,}
impl SWC {  /// parse source code.  pub fn parse(specifier: &str, source: &str) -> Result<Self, anyhow::Error> {    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;    let error_buffer = ErrorBuffer::new(specifier);    let syntax = Syntax::Es(get_es_config());    let input = StringInput::from(&*source_file);    let comments = SingleThreadedComments::default();    let lexer = Lexer::new(syntax, EsVersion::Es2020, input, Some(&comments));    let mut parser = swc_ecmascript::parser::Parser::new_from(lexer);    let handler = Handler::with_emitter_and_flags(      Box::new(error_buffer.clone()),      HandlerFlags {        can_emit_warnings: true,        dont_buffer_diagnostics: true,        ..HandlerFlags::default()      },    );    let module = parser      .parse_module()      .map_err(move |err| {        let mut diagnostic = err.into_diagnostic(&handler);        diagnostic.emit();        DiagnosticBuffer::from_error_buffer(error_buffer, |span| sm.lookup_char_pos(span.lo))      })      .unwrap();
    Ok(SWC {      specifier: specifier.into(),      module,      source_map: Rc::new(source_map),      comments,    })  }
  /// parse export names in the cjs module.  pub fn parse_cjs_exports(    &self,    node_env: &str,    call_mode: bool,  ) -> Result<(Vec<String>, Vec<String>), anyhow::Error> {    let mut lexer = CJSLexer {      node_env: node_env.to_owned(),      call_mode: call_mode,      fn_returned: false,      idents: IndexMap::new(),      exports_alias: IndexSet::new(),      exports: IndexSet::new(),      reexports: IndexSet::new(),    };    let program = Program::Module(self.module.clone());    program.fold_with(&mut lexer);    Ok((      lexer.exports.into_iter().collect(),      lexer.reexports.into_iter().collect(),    ))  }}
fn get_es_config() -> EsConfig {  EsConfig {    import_attributes: true,    jsx: false,    ..EsConfig::default()  }}