rspack_binding_api 0.102.2

Rspack shared binding API
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

use napi::Either;
use napi_derive::napi;
use rspack_core::{Compilation, CompilationId, ExportsInfo, ModuleGraph, RuntimeSpec};
use rspack_util::atom::Atom;

use crate::{runtime::JsRuntimeSpec, with_compilation, with_compilation_mut};

#[napi]
pub struct JsExportsInfo {
  exports_info: ExportsInfo,
  compilation_id: CompilationId,
}

impl JsExportsInfo {
  pub fn new(exports_info: ExportsInfo, compilation: &Compilation) -> Self {
    Self {
      exports_info,
      compilation_id: compilation.id(),
    }
  }

  fn with_compilation<R>(
    &self,
    f: impl FnOnce(&Compilation) -> napi::Result<R>,
  ) -> napi::Result<R> {
    with_compilation(self.compilation_id, f)
  }

  fn with_compilation_mut<R>(
    &mut self,
    f: impl FnOnce(&mut Compilation) -> napi::Result<R>,
  ) -> napi::Result<R> {
    with_compilation_mut(self.compilation_id, f)
  }
}

#[napi]
impl JsExportsInfo {
  #[napi(ts_args_type = "runtime: string | string[] | undefined")]
  pub fn is_used(&self, js_runtime: JsRuntimeSpec) -> napi::Result<bool> {
    let exports_info = self.exports_info;
    self.with_compilation(|compilation| {
      let runtime: Option<RuntimeSpec> = js_runtime.map(|js_rt| match js_rt {
        Either::A(str) => std::iter::once(str).map(Into::into).collect(),
        Either::B(vec) => vec.into_iter().map(Into::into).collect(),
      });
      Ok(
        exports_info
          .as_data(&compilation.exports_info_artifact)
          .is_used(runtime.as_ref()),
      )
    })
  }

  #[napi(ts_args_type = "runtime: string | string[] | undefined")]
  pub fn is_module_used(&self, js_runtime: JsRuntimeSpec) -> napi::Result<bool> {
    let exports_info = self.exports_info;
    self.with_compilation(|compilation| {
      let runtime: Option<RuntimeSpec> = js_runtime.map(|js_rt| match js_rt {
        Either::A(str) => std::iter::once(str).map(Into::into).collect(),
        Either::B(vec) => vec.into_iter().map(Into::into).collect(),
      });
      Ok(
        exports_info
          .as_data(&compilation.exports_info_artifact)
          .is_module_used(runtime.as_ref()),
      )
    })
  }

  #[napi(ts_args_type = "runtime: string | string[] | undefined")]
  pub fn set_used_in_unknown_way(&mut self, js_runtime: JsRuntimeSpec) -> napi::Result<bool> {
    let exports_info = self.exports_info;
    self.with_compilation_mut(|compilation| {
      let runtime: Option<RuntimeSpec> = js_runtime.map(|js_rt| match js_rt {
        Either::A(str) => std::iter::once(str).map(Into::into).collect(),
        Either::B(vec) => vec.into_iter().map(Into::into).collect(),
      });
      Ok(
        exports_info
          .as_data_mut(&mut compilation.exports_info_artifact)
          .set_used_in_unknown_way(runtime.as_ref()),
      )
    })
  }

  #[napi(
    ts_args_type = "name: string | string[], runtime: string | string[] | undefined",
    ts_return_type = " 0 | 1 | 2 | 3 | 4"
  )]
  pub fn get_used(
    &self,
    js_name: Either<String, Vec<String>>,
    js_runtime: JsRuntimeSpec,
  ) -> napi::Result<u32> {
    let exports_info = self.exports_info;
    self.with_compilation(|compilation| {
      let runtime: Option<RuntimeSpec> = js_runtime.map(|js_rt| match js_rt {
        Either::A(str) => std::iter::once(str).map(Into::into).collect(),
        Either::B(vec) => vec.into_iter().map(Into::into).collect(),
      });
      let names = match js_name {
        Either::A(s) => vec![Atom::from(s)],
        Either::B(v) => v.into_iter().map(Into::into).collect::<Vec<_>>(),
      };
      let used = exports_info
        .as_data(&compilation.exports_info_artifact)
        .get_used(&compilation.exports_info_artifact, &names, runtime.as_ref());
      Ok(used as u32)
    })
  }
}