rspack_loader_react_refresh 0.100.0

rspack react refresh loader
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

mod plugin;

pub use plugin::ReactRefreshLoaderPlugin;
use rspack_cacheable::{cacheable, cacheable_dyn};
use rspack_core::RunnerContext;
use rspack_error::Result;
use rspack_loader_runner::{Identifier, Loader, LoaderContext};

#[cacheable]
pub struct ReactRefreshLoader {
  identifier: Identifier,
}

impl Default for ReactRefreshLoader {
  fn default() -> Self {
    Self {
      identifier: REACT_REFRESH_LOADER_IDENTIFIER.into(),
    }
  }
}

impl ReactRefreshLoader {
  /// Panics:
  /// Panics if `identifier` passed in is not starting with `builtin:react-refresh-loader`.
  pub fn with_identifier(mut self, identifier: Identifier) -> Self {
    debug_assert!(identifier.starts_with(REACT_REFRESH_LOADER_IDENTIFIER));
    self.identifier = identifier;
    self
  }
}

#[cacheable_dyn]
#[async_trait::async_trait]
impl Loader<RunnerContext> for ReactRefreshLoader {
  fn identifier(&self) -> Identifier {
    self.identifier
  }

  async fn run(&self, loader_context: &mut LoaderContext<RunnerContext>) -> Result<()> {
    let Some(content) = loader_context.take_content() else {
      return Ok(());
    };
    let supports_arrow_function = loader_context
      .context
      .options
      .output
      .environment
      .supports_arrow_function();

    let mut source = content.try_into_string()?;

    if supports_arrow_function {
      source += r#"
function $RefreshSig$() { return $ReactRefreshRuntime$.createSignatureFunctionForTransform() }
function $RefreshReg$(type, id) { $ReactRefreshRuntime$.register(type, __webpack_module__.id + "_" + id) }
Promise.resolve().then(() => { $ReactRefreshRuntime$.refresh(__webpack_module__.id, __webpack_module__.hot) });
"#;
    } else {
      source += r#"
function $RefreshSig$() { return $ReactRefreshRuntime$.createSignatureFunctionForTransform() }
function $RefreshReg$(type, id) { $ReactRefreshRuntime$.register(type, __webpack_module__.id + "_" + id) }
Promise.resolve().then(function() { $ReactRefreshRuntime$.refresh(__webpack_module__.id, __webpack_module__.hot) });
"#;
    }
    let sm = loader_context.take_source_map();
    let additional_data = loader_context.take_additional_data();
    loader_context.finish_with((source, sm, additional_data));
    Ok(())
  }
}

pub const REACT_REFRESH_LOADER_IDENTIFIER: &str = "builtin:react-refresh-loader";