feat: rsdoctor native plugin get json size

Author: yifancong

crates/rspack_plugin_json/src/lib.rs

@@ -17,7 +17,7 @@ use rspack_core::{
   PrefetchExportsInfoMode, PrefetchedExportsInfoWrapper, RuntimeGlobals, RuntimeSpec, SourceType,
   UsageState, UsedNameItem,
   diagnostics::ModuleParseError,
-  rspack_sources::{BoxSource, RawStringSource, Source, SourceExt},
+  rspack_sources::{BoxSource, OriginalSource, RawStringSource, Source, SourceExt},
 };
 use rspack_error::{Error, IntoTWithDiagnosticArray, Result, TWithDiagnosticArray, error};
 use rspack_util::itoa;
@@ -42,6 +42,7 @@ impl ParserAndGenerator for JsonParserAndGenerator {
   }
 
   fn size(&self, module: &dyn Module, _source_type: Option<&SourceType>) -> f64 {
+    // dbg!(&module.build_info().json_data.as_ref());
     module
       .build_info()
       .json_data
@@ -229,7 +230,11 @@ impl ParserAndGenerator for JsonParserAndGenerator {
             .insert(RuntimeGlobals::MODULE);
           format!(r#"module.exports = {json_expr}"#)
         };
-        Ok(RawStringSource::from(content).boxed())
+        if module.get_source_map_kind().enabled() {
+          Ok(OriginalSource::new(content, module.identifier().as_str()).boxed())
+        } else {
+          Ok(RawStringSource::from(content).boxed())
+        }
       }
       _ => panic!(
         "Unsupported source type: {:?}",

crates/rspack_plugin_rsdoctor/src/data.rs

@@ -48,6 +48,7 @@ pub struct RsdoctorModule {
   pub belong_modules: HashSet<ModuleUkey>,
   pub issuer_path: Option<Vec<RsdoctorStatsModuleIssuer>>,
   pub bailout_reason: HashSet<String>,
+  pub size: i32,
 }
 
 #[derive(Debug, Default)]

crates/rspack_plugin_rsdoctor/src/module_graph.rs

@@ -21,6 +21,7 @@ pub fn collect_modules(
   module_graph: &ModuleGraph,
   chunk_graph: &ChunkGraph,
   context: &Context,
+  compilation: &Compilation,
 ) -> HashMap<Identifier, RsdoctorModule> {
   let module_ukey_counter: Arc<AtomicI32> = Arc::new(AtomicI32::new(0));
 
@@ -54,6 +55,20 @@ pub fn collect_modules(
             .collect::<HashSet<_>>()
         })
         .unwrap_or_default();
+
+      let mut size = 0;
+      let (map, result_map) = compilation.code_generation_results.inner();
+      if let Some(entry) = map.get(module_id)
+        && let Some(id) = entry.values().next()
+        && let Some(res) = result_map.get(id)
+      {
+        size = res.inner().values().map(|s| s.size() as i32).sum();
+      }
+
+      if true {
+        println!("Module: {}, Size: {}", module.identifier(), size);
+      }
+
       (
         module_id.to_owned(),
         RsdoctorModule {
@@ -74,6 +89,7 @@ pub fn collect_modules(
           chunks,
           issuer_path: None,
           bailout_reason: HashSet::default(),
+          size: 0,
         },
       )
     })
@@ -142,7 +158,8 @@ pub fn collect_module_original_sources(
       let module_ukey = module_ukeys.get(module_id)?;
       let resource = module.resource_resolved_data().resource().to_owned();
       let object_pool = tls.get_or(ObjectPool::default);
-      let source = module
+      // let source = module
+      let mut source = module
         .source()
         .and_then(|s| s.map(object_pool, &MapOptions::default()))
         .and_then(|s| {
@@ -164,6 +181,15 @@ pub fn collect_module_original_sources(
             source: content,
           })
         })?;
+
+      let (map, result_map) = compilation.code_generation_results.inner();
+      if let Some(entry) = map.get(module_id)
+        && let Some(id) = entry.values().next()
+        && let Some(res) = result_map.get(id)
+      {
+        source.size = res.inner().values().map(|s| s.size() as i32).sum();
+      }
+
       Some(source)
     })
     .collect::<Vec<_>>()