use rspack_core::{
Compilation, RuntimeGlobals, RuntimeModule, RuntimeModuleGenerateContext, RuntimeTemplate,
impl_runtime_module,
};
#[impl_runtime_module]
#[derive(Debug)]
pub struct ChunkNameRuntimeModule {}
impl ChunkNameRuntimeModule {
pub fn new(runtime_template: &RuntimeTemplate) -> Self {
Self::with_default(runtime_template)
}
}
#[async_trait::async_trait]
impl RuntimeModule for ChunkNameRuntimeModule {
fn runtime_requirements(
&self,
_compilation: &Compilation,
) -> rspack_core::RuntimeModuleRuntimeRequirements {
rspack_core::RuntimeModuleRuntimeRequirements {
write: { RuntimeGlobals::CHUNK_NAME },
..Default::default()
}
}
async fn generate(
&self,
context: &RuntimeModuleGenerateContext<'_>,
) -> rspack_error::Result<String> {
let compilation = context.compilation;
if let Some(chunk_ukey) = self.chunk() {
let chunk = compilation
.build_chunk_graph_artifact
.chunk_by_ukey
.expect_get(&chunk_ukey);
Ok(format!(
"{} = {};",
context
.runtime_template
.render_runtime_globals(&RuntimeGlobals::CHUNK_NAME),
chunk
.name()
.map_or_else(|| "null".to_string(), rspack_util::json_stringify_str)
))
} else {
unreachable!("should attach chunk for css_loading")
}
}
}