use rspack_collections::Identifier;
use rspack_core::{
ChunkUkey, Compilation, RuntimeGlobals, RuntimeModule, RuntimeTemplate, impl_runtime_module,
};
#[impl_runtime_module]
#[derive(Debug)]
pub struct BaseUriRuntimeModule {
id: Identifier,
chunk: Option<ChunkUkey>,
}
impl BaseUriRuntimeModule {
pub fn new(runtime_template: &RuntimeTemplate) -> Self {
Self::with_default(
Identifier::from(format!(
"{}base_uri",
runtime_template.runtime_module_prefix()
)),
None,
)
}
}
#[async_trait::async_trait]
impl RuntimeModule for BaseUriRuntimeModule {
async fn generate(&self, compilation: &Compilation) -> rspack_error::Result<String> {
let base_uri = self
.chunk
.and_then(|ukey| compilation.chunk_by_ukey.get(&ukey))
.and_then(|chunk| chunk.get_entry_options(&compilation.chunk_group_by_ukey))
.and_then(|options| options.base_uri.as_ref())
.and_then(|base_uri| serde_json::to_string(base_uri).ok())
.unwrap_or_else(|| "undefined".to_string());
Ok(format!(
"{} = {};\n",
compilation
.runtime_template
.render_runtime_globals(&RuntimeGlobals::BASE_URI),
base_uri
))
}
fn name(&self) -> Identifier {
self.id
}
fn attach(&mut self, chunk: ChunkUkey) {
self.chunk = Some(chunk);
}
}