use rspack_core::{
Compilation, RuntimeGlobals, RuntimeModule, RuntimeModuleGenerateContext, RuntimeTemplate,
impl_runtime_module,
};
#[impl_runtime_module]
#[derive(Debug)]
pub struct StartupEntrypointRuntimeModule {
async_chunk_loading: bool,
}
impl StartupEntrypointRuntimeModule {
pub fn new(runtime_template: &RuntimeTemplate, async_chunk_loading: bool) -> Self {
Self::with_default(runtime_template, async_chunk_loading)
}
}
#[async_trait::async_trait]
impl RuntimeModule for StartupEntrypointRuntimeModule {
fn template(&self) -> Vec<(String, String)> {
vec![(
self.id().to_string(),
if self.async_chunk_loading {
include_str!("runtime/startup_entrypoint_with_async.ejs").to_string()
} else {
include_str!("runtime/startup_entrypoint.ejs").to_string()
},
)]
}
async fn generate(
&self,
context: &RuntimeModuleGenerateContext<'_>,
) -> rspack_error::Result<String> {
context.runtime_template.render(self.id(), None)
}
fn runtime_requirements(
&self,
_compilation: &Compilation,
) -> rspack_core::RuntimeModuleRuntimeRequirements {
rspack_core::RuntimeModuleRuntimeRequirements {
dependencies: RuntimeGlobals::REQUIRE
| RuntimeGlobals::ENSURE_CHUNK
| RuntimeGlobals::ENSURE_CHUNK_INCLUDE_ENTRIES,
write: { RuntimeGlobals::STARTUP_ENTRYPOINT },
..Default::default()
}
}
}