rspack_loader_swc 0.102.3

use rspack_core::{CollectedTypeScriptInfo, EvaluatedInlinableValue, TSEnumValue};
use rspack_intern::Atom as RspackAtom;
use rspack_swc_plugin_ts_collector::{
  EnumMemberValue, ExportedEnumCollector, TypeExportsCollector,
};
use rustc_hash::FxHashMap;
use swc::atoms::{Atom, Wtf8Atom};
use swc_core::{
  common::SyntaxContext,
  ecma::{ast::Program, visit::VisitWith},
};

use crate::options::{CollectTypeScriptInfoOptions, CollectingEnumKind};

pub fn collect_typescript_info(
  program: &Program,
  unresolved_ctxt: SyntaxContext,
  options: &CollectTypeScriptInfoOptions,
) -> CollectedTypeScriptInfo {
  let mut type_exports = Default::default();
  if options.type_exports.unwrap_or_default() {
    program.visit_with(&mut TypeExportsCollector::new(&mut type_exports));
  }
  // `ExportedEnumCollector` is SWC-facing and writes SWC atoms. Convert them
  // once when constructing Rspack's metadata below.
  let mut swc_exported_enums: FxHashMap<Atom, FxHashMap<Wtf8Atom, EnumMemberValue>> =
    Default::default();
  if let Some(kind) = &options.exported_enum {
    program.visit_with(&mut ExportedEnumCollector::new(
      matches!(kind, CollectingEnumKind::ConstOnly),
      unresolved_ctxt,
      &mut swc_exported_enums,
    ));
  }
  CollectedTypeScriptInfo {
    type_exports: type_exports.into_iter().map(RspackAtom::from).collect(),
    exported_enums: swc_exported_enums
      .into_iter()
      .map(|(k, members)| {
        let value = TSEnumValue::new(
          members
            .into_iter()
            .map(|(id, v)| {
              let value = match v {
                EnumMemberValue::Number(n) => Some(EvaluatedInlinableValue::new_number(n)),
                EnumMemberValue::String(s) => Some(EvaluatedInlinableValue::new_string(
                  RspackAtom::from(s.to_string_lossy().as_ref()),
                )),
                EnumMemberValue::Unknown => None,
              };
              (RspackAtom::from(id.to_string_lossy().as_ref()), value)
            })
            .collect(),
        );
        (RspackAtom::from(k), value)
      })
      .collect(),
  }
}