use std::sync::LazyLock;
use rspack_core::{
Compilation, RuntimeGlobals, RuntimeModule, RuntimeModuleGenerateContext, RuntimeTemplate,
impl_runtime_module,
};
use crate::extract_runtime_module_variables_from_ejs;
static ON_CHUNK_LOADED_TEMPLATE: &str = include_str!("runtime/on_chunk_loaded.ejs");
static RUNTIME_MODULE_VARIABLES: LazyLock<Vec<&'static str>> =
LazyLock::new(|| extract_runtime_module_variables_from_ejs(&[ON_CHUNK_LOADED_TEMPLATE]));
#[impl_runtime_module]
#[derive(Debug)]
pub struct OnChunkLoadedRuntimeModule {}
impl OnChunkLoadedRuntimeModule {
pub fn new(runtime_template: &RuntimeTemplate) -> Self {
Self::with_default(runtime_template)
}
}
#[async_trait::async_trait]
impl RuntimeModule for OnChunkLoadedRuntimeModule {
fn runtime_module_variables() -> &'static [&'static str] {
RUNTIME_MODULE_VARIABLES.as_slice()
}
fn runtime_requirements(
&self,
_compilation: &Compilation,
) -> rspack_core::RuntimeModuleRuntimeRequirements {
rspack_core::RuntimeModuleRuntimeRequirements {
define: { RuntimeGlobals::ON_CHUNKS_LOADED },
..Default::default()
}
}
fn template(&self) -> Vec<(String, String)> {
vec![(self.id().to_string(), ON_CHUNK_LOADED_TEMPLATE.to_string())]
}
async fn generate(
&self,
context: &RuntimeModuleGenerateContext<'_>,
) -> rspack_error::Result<String> {
let source = context.runtime_template.render(self.id(), None)?;
Ok(source)
}
}