rspack_plugin_json 0.101.9

rspack json plugin
Documentation 
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

use json::JsonValue;
use rspack_cacheable::{cacheable, cacheable_dyn};
use rspack_core::{
  AsContextDependency, AsModuleDependency, Compilation, Dependency, DependencyCodeGeneration,
  DependencyId, ExportNameOrSpec, ExportSpec, ExportsInfoArtifact, ExportsOfExportsSpec,
  ExportsSpec, ModuleGraph, ModuleGraphCacheArtifact, RuntimeSpec,
};
use rspack_hash::{RspackHash, RspackHasher};
use rspack_util::itoa;

#[cacheable]
#[derive(Debug, Clone)]
pub struct JsonExportsDependency {
  id: DependencyId,
  exports_depth: u32,
}

impl JsonExportsDependency {
  pub fn new(exports_depth: u32) -> Self {
    Self {
      id: DependencyId::new(),
      exports_depth,
    }
  }

  fn data<'a>(&self, module_graph: &'a ModuleGraph) -> &'a JsonValue {
    module_graph
      .get_parent_module(&self.id)
      .and_then(|identifier| module_graph.module_by_identifier(identifier))
      .and_then(|module| module.build_info().json_data.as_ref())
      .expect("JSON export dependency should have parent JSON module data")
  }
}

#[cacheable_dyn]
impl Dependency for JsonExportsDependency {
  fn id(&self) -> &rspack_core::DependencyId {
    &self.id
  }

  fn get_exports(
    &self,
    module_graph: &ModuleGraph,
    _mg_cache: &ModuleGraphCacheArtifact,
    _exports_info_artifact: &ExportsInfoArtifact,
  ) -> Option<ExportsSpec> {
    Some(ExportsSpec {
      exports: get_exports_from_data(self.data(module_graph), self.exports_depth, 1)
        .map_or(ExportsOfExportsSpec::NoExports, ExportsOfExportsSpec::Names),
      ..Default::default()
    })
  }

  fn could_affect_referencing_module(&self) -> rspack_core::AffectType {
    rspack_core::AffectType::False
  }
}

impl AsModuleDependency for JsonExportsDependency {}
impl AsContextDependency for JsonExportsDependency {}

#[cacheable_dyn]
impl DependencyCodeGeneration for JsonExportsDependency {
  fn update_hash(
    &self,
    hasher: &mut RspackHasher,
    compilation: &Compilation,
    _runtime: Option<&RuntimeSpec>,
  ) {
    let module_graph = compilation.get_module_graph();
    self.data(module_graph).to_string().hash(hasher);
  }
}

fn get_exports_from_data(
  data: &JsonValue,
  exports_depth: u32,
  cur_depth: u32,
) -> Option<Vec<ExportNameOrSpec>> {
  if cur_depth > exports_depth {
    return None;
  }
  let ret = match data {
    JsonValue::Null
    | JsonValue::Short(_)
    | JsonValue::String(_)
    | JsonValue::Number(_)
    | JsonValue::Boolean(_) => {
      return None;
    }
    JsonValue::Object(obj) => obj
      .iter()
      .map(|(k, v)| {
        ExportNameOrSpec::ExportSpec(ExportSpec {
          name: k.into(),
          can_mangle: Some(true),
          exports: get_exports_from_data(v, exports_depth, cur_depth + 1),
          ..Default::default()
        })
      })
      .collect::<Vec<_>>(),
    JsonValue::Array(arr) => {
      if arr.len() > 100 {
        return None;
      }
      arr
        .iter()
        .enumerate()
        .map(|(i, item)| {
          let mut i_buffer = itoa::Buffer::new();
          let i_str = i_buffer.format(i);
          ExportNameOrSpec::ExportSpec(ExportSpec {
            name: i_str.into(),
            can_mangle: Some(true),
            exports: get_exports_from_data(item, exports_depth, cur_depth + 1),
            ..Default::default()
          })
        })
        .collect::<Vec<_>>()
    }
  };
  Some(ret)
}