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 additional_write_runtime_requirements(&self, _compilation: &Compilation) -> RuntimeGlobals {
RuntimeGlobals::CHUNK_NAME
}
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")
}
}
}