use rspack_core::{
Compilation, PathData, RuntimeGlobals, RuntimeModule, RuntimeModuleGenerateContext,
RuntimeTemplate, SourceType, has_hash_placeholder, impl_runtime_module,
};
#[impl_runtime_module]
#[derive(Debug)]
pub struct GetChunkUpdateFilenameRuntimeModule {}
impl GetChunkUpdateFilenameRuntimeModule {
pub fn new(runtime_template: &RuntimeTemplate) -> Self {
Self::with_default(runtime_template)
}
}
#[async_trait::async_trait]
impl RuntimeModule for GetChunkUpdateFilenameRuntimeModule {
fn runtime_requirements(
&self,
compilation: &Compilation,
) -> rspack_core::RuntimeModuleRuntimeRequirements {
rspack_core::RuntimeModuleRuntimeRequirements {
dependencies: {
if has_hash_placeholder(
compilation
.options
.output
.hot_update_chunk_filename
.as_str(),
) {
RuntimeGlobals::GET_FULL_HASH
} else {
RuntimeGlobals::default()
}
},
write: { RuntimeGlobals::GET_CHUNK_UPDATE_SCRIPT_FILENAME },
..Default::default()
}
}
fn template(&self) -> Vec<(String, String)> {
vec![(
self.id().to_string(),
include_str!("runtime/get_chunk_update_filename.ejs").to_string(),
)]
}
async fn generate(
&self,
context: &RuntimeModuleGenerateContext<'_>,
) -> rspack_error::Result<String> {
let compilation = context.compilation;
let runtime_template = context.runtime_template;
if let Some(chunk_ukey) = self.chunk() {
let chunk = compilation
.build_chunk_graph_artifact
.chunk_by_ukey
.expect_get(&chunk_ukey);
let filename = compilation
.get_path(
&compilation.options.output.hot_update_chunk_filename,
PathData::default()
.chunk_hash_optional(chunk.rendered_hash(
&compilation.chunk_hashes_artifact,
compilation.options.output.hash_digest_length,
))
.chunk_name_optional(chunk.name_for_filename_template())
.content_hash_optional(chunk.rendered_content_hash_by_source_type(
&compilation.chunk_hashes_artifact,
&SourceType::JavaScript,
compilation.options.output.hash_digest_length,
))
.hash(
format!(
"' + {}() + '",
runtime_template.render_runtime_globals(&RuntimeGlobals::GET_FULL_HASH)
)
.as_str(),
)
.id("' + chunkId + '")
.runtime(chunk.runtime().as_str()),
)
.await?;
let source = runtime_template.render(
self.id(),
Some(serde_json::json!({
"_filename": format!("'{}'", filename),
})),
)?;
Ok(source)
} else {
unreachable!("should attach chunk for get_main_filename")
}
}
}