rspack_loader_preact_refresh/
lib.rs

1mod plugin;
2
3pub use plugin::PreactRefreshLoaderPlugin;
4use rspack_cacheable::{cacheable, cacheable_dyn};
5use rspack_core::RunnerContext;
6use rspack_error::Result;
7use rspack_loader_runner::{Identifier, Loader, LoaderContext};
8
9#[cacheable]
10pub struct PreactRefreshLoader {
11  identifier: Identifier,
12}
13
14impl Default for PreactRefreshLoader {
15  fn default() -> Self {
16    Self {
17      identifier: PREACT_REFRESH_LOADER_IDENTIFIER.into(),
18    }
19  }
20}
21
22impl PreactRefreshLoader {
23  /// Panics:
24  /// Panics if `identifier` passed in is not starting with `builtin:preact-refresh-loader`.
25  pub fn with_identifier(mut self, identifier: Identifier) -> Self {
26    assert!(identifier.starts_with(PREACT_REFRESH_LOADER_IDENTIFIER));
27    self.identifier = identifier;
28    self
29  }
30}
31
32#[cacheable_dyn]
33#[async_trait::async_trait]
34impl Loader<RunnerContext> for PreactRefreshLoader {
35  fn identifier(&self) -> Identifier {
36    self.identifier
37  }
38
39  async fn run(&self, loader_context: &mut LoaderContext<RunnerContext>) -> Result<()> {
40    let Some(content) = loader_context.take_content() else {
41      return Ok(());
42    };
43    let mut source = content.try_into_string()?;
44    source += "\n";
45    source += include_str!("runtime.js");
46    let sm = loader_context.take_source_map();
47    loader_context.finish_with((source, sm));
48    Ok(())
49  }
50}
51
52pub const PREACT_REFRESH_LOADER_IDENTIFIER: &str = "builtin:preact-refresh-loader";