feat: replace SymbolId with SymbolRef (#95)

hyf0 · web-flow · commit 9c90067a9db2 · 2023-10-25T09:22:18.000-05:00
diff --git a/crates/rolldown/src/bundler/chunk/de_conflict.rs b/crates/rolldown/src/bundler/chunk/de_conflict.rs
@@ -43,9 +43,8 @@ impl Chunk {
                 .iter()
                 .flat_map(|part| part.declared_symbols.iter().copied()),
             )
-            .for_each(|symbol_id| {
-              let canonical_ref =
-                graph.symbols.par_get_canonical_ref((module.id, symbol_id).into());
+            .for_each(|symbol_ref| {
+              let canonical_ref = graph.symbols.par_get_canonical_ref(symbol_ref);
 
               let original_name = graph.symbols.get_original_name(canonical_ref);
 
diff --git a/crates/rolldown/src/bundler/module/normal_module.rs b/crates/rolldown/src/bundler/module/normal_module.rs
@@ -103,7 +103,7 @@ impl NormalModule {
     if !self_linker_module.is_symbol_for_namespace_referenced {
       self_linker_module.is_symbol_for_namespace_referenced = true;
       self_linker_module.virtual_stmt_infos.push(VirtualStmtInfo {
-        declared_symbols: vec![self.namespace_symbol.symbol],
+        declared_symbols: vec![self.namespace_symbol],
         ..Default::default()
       });
     }
@@ -335,9 +335,10 @@ impl NormalModule {
       .into();
       let symbol = symbols.create_symbol(self.id, name).symbol;
       self_linker_module.wrap_symbol = Some((self.id, symbol).into());
-      self_linker_module
-        .virtual_stmt_infos
-        .push(VirtualStmtInfo { declared_symbols: vec![symbol], ..Default::default() });
+      self_linker_module.virtual_stmt_infos.push(VirtualStmtInfo {
+        declared_symbols: vec![(self.id, symbol).into()],
+        ..Default::default()
+      });
       self.initialize_namespace(self_linker_module);
     }
   }
@@ -365,7 +366,7 @@ impl NormalModule {
       // FIXME: should store the symbol in `used_symbols` instead of `declared_symbols`.
       // The deconflict for runtime symbols would be handled in the deconflict on cross-chunk-imported
       // symbols
-      declared_symbols: vec![local_symbol_ref.symbol],
+      declared_symbols: vec![local_symbol_ref],
       ..Default::default()
     });
     local_symbol_ref
diff --git a/crates/rolldown/src/bundler/visitors/scanner.rs b/crates/rolldown/src/bundler/visitors/scanner.rs
@@ -89,7 +89,7 @@ impl<'a> Scanner<'a> {
   }
 
   fn add_declared_id(&mut self, id: SymbolId) {
-    self.current_stmt_info.declared_symbols.push(id);
+    self.current_stmt_info.declared_symbols.push((self.idx, id).into());
   }
 
   fn get_root_binding(&self, name: &Atom) -> SymbolId {
@@ -203,7 +203,7 @@ impl<'a> Scanner<'a> {
           oxc::ast::ast::Declaration::FunctionDeclaration(fn_decl) => {
             let id = fn_decl.id.as_ref().unwrap();
             // FIXME: remove this line after https://github.com/web-infra-dev/oxc/pull/843 being merged.
-            self.current_stmt_info.declared_symbols.push(id.expect_symbol_id());
+            self.add_declared_id(id.expect_symbol_id());
             self.add_local_export(&id.name, id.expect_symbol_id());
           }
           oxc::ast::ast::Declaration::ClassDeclaration(cls_decl) => {
@@ -247,8 +247,7 @@ impl<'a> Scanner<'a> {
         SymbolFlags::None,
         self.scope.root_scope_id(),
       );
-
-      self.current_stmt_info.declared_symbols.push(sym_id);
+      self.add_declared_id(sym_id);
       sym_id
     });
     self.add_local_default_export(local);
diff --git a/crates/rolldown_common/src/stmt_info.rs b/crates/rolldown_common/src/stmt_info.rs
@@ -1,5 +1,3 @@
-use oxc::semantic::SymbolId;
-
 use crate::SymbolRef;
 
 index_vec::define_index_type! {
@@ -10,7 +8,7 @@ index_vec::define_index_type! {
 pub struct StmtInfo {
   pub stmt_idx: usize,
   // currently, we only store top level symbols
-  pub declared_symbols: Vec<SymbolId>,
+  pub declared_symbols: Vec<SymbolRef>,
   // We will add symbols of other modules to `referenced_symbols`, so we need `SymbolRef`
   // here instead of `SymbolId`.
   /// Top level symbols referenced by this statement.
@@ -20,6 +18,6 @@ pub struct StmtInfo {
 // Because we want declare symbols at linker, it shouldn't mutate the original `StmtInfo`.
 #[derive(Default, Debug)]
 pub struct VirtualStmtInfo {
-  pub declared_symbols: Vec<SymbolId>,
+  pub declared_symbols: Vec<SymbolRef>,
   pub referenced_symbols: Vec<SymbolRef>,
 }
