rspack_loader_swc 0.101.8

use rspack_core::{CollectedTypeScriptInfo, EvaluatedInlinableValue, TSEnumValue};
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));
  }
  let mut 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 exported_enums,
    ));
  }
  CollectedTypeScriptInfo {
    type_exports,
    exported_enums: 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(
                  s.to_atom_lossy().into_owned(),
                )),
                EnumMemberValue::Unknown => None,
              };
              (id.to_atom_lossy().into_owned(), value)
            })
            .collect(),
        );
        (k, value)
      })
      .collect(),
  }
}