rspack_plugin_javascript 0.100.0-rc.3

rspack javascript plugin
Documentation
use rspack_cacheable::{cacheable, cacheable_dyn};
use rspack_core::{
  AsContextDependency, Dependency, DependencyCodeGeneration, DependencyId, DependencyRange,
  DependencyTemplate, DependencyTemplateType, DependencyType, ExportsInfoArtifact,
  ExtendedReferencedExport, FactorizeInfo, ModuleDependency, ModuleGraph, ModuleGraphCacheArtifact,
  RuntimeSpec, TemplateContext, TemplateReplaceSource,
};

#[cacheable]
#[derive(Debug, Clone)]
pub struct IsIncludeDependency {
  pub range: DependencyRange,
  pub id: DependencyId,
  pub request: String,
  factorize_info: FactorizeInfo,
}

impl IsIncludeDependency {
  pub fn new(range: DependencyRange, request: String) -> Self {
    Self {
      range,
      id: DependencyId::default(),
      request,
      factorize_info: Default::default(),
    }
  }
}

impl AsContextDependency for IsIncludeDependency {}

#[cacheable_dyn]
impl Dependency for IsIncludeDependency {
  fn dependency_type(&self) -> &DependencyType {
    &DependencyType::IsIncluded
  }

  fn range(&self) -> Option<DependencyRange> {
    Some(self.range)
  }

  fn id(&self) -> &DependencyId {
    &self.id
  }

  fn get_referenced_exports(
    &self,
    _module_graph: &ModuleGraph,
    _module_graph_cache: &ModuleGraphCacheArtifact,
    _exports_info_artifact: &ExportsInfoArtifact,
    _runtime: Option<&RuntimeSpec>,
  ) -> Vec<ExtendedReferencedExport> {
    vec![]
  }

  fn could_affect_referencing_module(&self) -> rspack_core::AffectType {
    rspack_core::AffectType::True
  }
}

#[cacheable_dyn]
impl ModuleDependency for IsIncludeDependency {
  fn weak(&self) -> bool {
    true
  }

  fn request(&self) -> &str {
    &self.request
  }

  fn factorize_info(&self) -> &FactorizeInfo {
    &self.factorize_info
  }

  fn factorize_info_mut(&mut self) -> &mut FactorizeInfo {
    &mut self.factorize_info
  }
}

#[cacheable_dyn]
impl DependencyCodeGeneration for IsIncludeDependency {
  fn dependency_template(&self) -> Option<DependencyTemplateType> {
    Some(IsIncludedDependencyTemplate::template_type())
  }
}

#[cacheable]
#[derive(Debug, Clone, Default)]
pub struct IsIncludedDependencyTemplate;

impl IsIncludedDependencyTemplate {
  pub fn template_type() -> DependencyTemplateType {
    DependencyTemplateType::Dependency(DependencyType::IsIncluded)
  }
}

impl DependencyTemplate for IsIncludedDependencyTemplate {
  fn render(
    &self,
    dep: &dyn DependencyCodeGeneration,
    source: &mut TemplateReplaceSource,
    code_generatable_context: &mut TemplateContext,
  ) {
    let dep = dep
      .as_any()
      .downcast_ref::<IsIncludeDependency>()
      .expect("IsIncludedDependencyTemplate should be used for IsIncludeDependency");
    let TemplateContext { compilation, .. } = code_generatable_context;

    let included = compilation
      .get_module_graph()
      .connection_by_dependency_id(&dep.id)
      .is_some_and(|connection| {
        compilation
          .build_chunk_graph_artifact
          .chunk_graph
          .get_number_of_module_chunks(*connection.module_identifier())
          > 0
      });

    source.replace(dep.range.start, dep.range.end, included.to_string(), None);
  }
}