rspack_plugin_runtime 0.100.1

use std::{hash::Hash, sync::LazyLock};

use itertools::Itertools;
use regex::Regex;
use rspack_collections::IdentifierLinkedMap;
use rspack_core::{
  Chunk, ChunkGraph, ChunkGroupByUkey, ChunkGroupUkey, ChunkLoading, ChunkLoadingType, ChunkUkey,
  Compilation, PathData, RuntimeCodeTemplate, RuntimeGlobals, RuntimeVariable, SourceType,
  chunk_graph_chunk::ChunkIdSet,
  get_js_chunk_filename_template,
  rspack_sources::{BoxSource, RawStringSource, SourceExt},
};
use rspack_error::{Result, error};
use rspack_hash::RspackHash;
use rspack_plugin_javascript::runtime::stringify_chunks_to_array;
use rspack_util::fx_hash::FxIndexSet;
use rustc_hash::FxHashSet as HashSet;

use crate::runtime_module::is_enabled_for_chunk;

pub fn should_export_webpack_require_for_module_chunk_loading(
  chunk_ukey: &ChunkUkey,
  compilation: &Compilation,
) -> bool {
  let chunk_loading = ChunkLoading::Enable(ChunkLoadingType::Import);
  is_enabled_for_chunk(chunk_ukey, &chunk_loading, compilation)
    && compilation
      .build_chunk_graph_artifact
      .chunk_graph
      .has_chunk_entry_dependent_chunks(
        chunk_ukey,
        &compilation.build_chunk_graph_artifact.chunk_group_by_ukey,
      )
}

pub fn update_hash_for_entry_startup(
  hasher: &mut RspackHash,
  compilation: &Compilation,
  entries: &IdentifierLinkedMap<ChunkGroupUkey>,
  chunk: &ChunkUkey,
) {
  for (module, entry) in entries {
    if let Some(module_id) = compilation
      .get_module_graph()
      .module_graph_module_by_identifier(module)
      .and_then(|module| {
        ChunkGraph::get_module_id(&compilation.module_ids_artifact, module.module_identifier)
      })
    {
      module_id.hash(hasher);
    }

    if let Some(runtime_chunk) = compilation
      .build_chunk_graph_artifact
      .chunk_group_by_ukey
      .get(entry)
      .map(|e| e.get_runtime_chunk(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey))
    {
      for chunk_ukey in get_all_chunks(
        entry,
        chunk,
        Some(&runtime_chunk),
        &compilation.build_chunk_graph_artifact.chunk_group_by_ukey,
      ) {
        if let Some(chunk) = compilation
          .build_chunk_graph_artifact
          .chunk_by_ukey
          .get(&chunk_ukey)
        {
          chunk.id().hash(hasher);
        }
      }
    }
  }
}

pub fn get_all_chunks(
  entrypoint: &ChunkGroupUkey,
  exclude_chunk1: &ChunkUkey,
  exclude_chunk2: Option<&ChunkUkey>,
  chunk_group_by_ukey: &ChunkGroupByUkey,
) -> FxIndexSet<ChunkUkey> {
  fn add_chunks(
    chunk_group_by_ukey: &ChunkGroupByUkey,
    chunks: &mut FxIndexSet<ChunkUkey>,
    entrypoint_ukey: &ChunkGroupUkey,
    exclude_chunk1: &ChunkUkey,
    exclude_chunk2: Option<&ChunkUkey>,
    visit_chunk_groups: &mut FxIndexSet<ChunkGroupUkey>,
  ) {
    if let Some(entrypoint) = chunk_group_by_ukey.get(entrypoint_ukey) {
      for chunk in &entrypoint.chunks {
        if chunk == exclude_chunk1 {
          continue;
        }
        if let Some(exclude_chunk2) = exclude_chunk2
          && chunk == exclude_chunk2
        {
          continue;
        }
        chunks.insert(*chunk);
      }

      for parent in entrypoint.parents_iterable() {
        if visit_chunk_groups.contains(parent) {
          continue;
        }
        visit_chunk_groups.insert(*parent);
        if let Some(chunk_group) = chunk_group_by_ukey.get(parent)
          && chunk_group.is_initial()
        {
          add_chunks(
            chunk_group_by_ukey,
            chunks,
            &chunk_group.ukey,
            exclude_chunk1,
            exclude_chunk2,
            visit_chunk_groups,
          );
        }
      }
    }
  }

  let mut chunks = FxIndexSet::default();
  let mut visit_chunk_groups = FxIndexSet::default();

  add_chunks(
    chunk_group_by_ukey,
    &mut chunks,
    entrypoint,
    exclude_chunk1,
    exclude_chunk2,
    &mut visit_chunk_groups,
  );

  chunks
}

pub async fn get_runtime_chunk_output_name(
  compilation: &Compilation,
  chunk_ukey: &ChunkUkey,
) -> Result<String> {
  let entry_point = {
    let entry_points = compilation
      .build_chunk_graph_artifact
      .chunk_graph
      .get_chunk_entry_modules_with_chunk_group_iterable(chunk_ukey);

    let (_, entry_point_ukey) = entry_points
      .iter()
      .next()
      .ok_or_else(|| error!("should has entry point ukey"))?;

    compilation
      .build_chunk_graph_artifact
      .chunk_group_by_ukey
      .expect_get(entry_point_ukey)
  };

  let runtime_chunk = compilation
    .build_chunk_graph_artifact
    .chunk_by_ukey
    .expect_get(
      &entry_point.get_runtime_chunk(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey),
    );

  get_chunk_output_name(runtime_chunk, compilation).await
}

pub fn runtime_chunk_has_hash(compilation: &Compilation, chunk_ukey: &ChunkUkey) -> Result<bool> {
  let entry_point = {
    let entry_points = compilation
      .build_chunk_graph_artifact
      .chunk_graph
      .get_chunk_entry_modules_with_chunk_group_iterable(chunk_ukey);

    let (_, entry_point_ukey) = entry_points
      .iter()
      .next()
      .ok_or_else(|| error!("should has entry point ukey"))?;

    compilation
      .build_chunk_graph_artifact
      .chunk_group_by_ukey
      .expect_get(entry_point_ukey)
  };

  let runtime_chunk_ukey =
    entry_point.get_runtime_chunk(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey);
  let runtime_chunk = compilation
    .build_chunk_graph_artifact
    .chunk_by_ukey
    .expect_get(&runtime_chunk_ukey);

  let filename = get_js_chunk_filename_template(
    runtime_chunk,
    &compilation.options.output,
    &compilation.build_chunk_graph_artifact.chunk_group_by_ukey,
  );

  if filename.has_hash_placeholder() {
    return Ok(true);
  }

  if filename.has_content_hash_placeholder()
    && (compilation
      .build_chunk_graph_artifact
      .chunk_graph
      .has_chunk_full_hash_modules(&runtime_chunk_ukey, &compilation.runtime_modules)
      || compilation
        .build_chunk_graph_artifact
        .chunk_graph
        .has_chunk_dependent_hash_modules(&runtime_chunk_ukey, &compilation.runtime_modules))
  {
    return Ok(true);
  }

  Ok(false)
}

pub fn generate_entry_startup(
  compilation: &Compilation,
  chunk: &ChunkUkey,
  entries: &IdentifierLinkedMap<ChunkGroupUkey>,
  passive: bool,
  runtime_template: &RuntimeCodeTemplate<'_>,
) -> BoxSource {
  let mut module_id_exprs = vec![];
  let mut chunks_ids = ChunkIdSet::default();
  let module_graph = compilation.get_module_graph();
  for (module, entry) in entries {
    if let Some(module_id) = module_graph
      .module_by_identifier(module)
      .filter(|module| {
        module
          .source_types(module_graph)
          .contains(&SourceType::JavaScript)
      })
      .and_then(|module| {
        ChunkGraph::get_module_id(&compilation.module_ids_artifact, module.identifier())
      })
    {
      let module_id_expr = rspack_util::json_stringify(module_id);
      module_id_exprs.push(module_id_expr);
    } else {
      continue;
    }

    if let Some(runtime_chunk) = compilation
      .build_chunk_graph_artifact
      .chunk_group_by_ukey
      .get(entry)
      .map(|e| e.get_runtime_chunk(&compilation.build_chunk_graph_artifact.chunk_group_by_ukey))
    {
      let chunks = get_all_chunks(
        entry,
        chunk,
        Some(&runtime_chunk),
        &compilation.build_chunk_graph_artifact.chunk_group_by_ukey,
      );
      chunks_ids.extend(
        chunks
          .iter()
          .map(|chunk_ukey| {
            let chunk = compilation
              .build_chunk_graph_artifact
              .chunk_by_ukey
              .expect_get(chunk_ukey);
            chunk.expect_id().clone()
          })
          .collect::<HashSet<_>>(),
      );
    }
  }

  let mut source = String::default();
  source.push_str(&format!(
    "var {} = function(moduleId) {{ return {}({} = moduleId) }}\n",
    runtime_template.render_runtime_variable(&RuntimeVariable::StartupExec),
    runtime_template.render_runtime_globals(&RuntimeGlobals::REQUIRE),
    runtime_template.render_runtime_globals(&RuntimeGlobals::ENTRY_MODULE_ID)
  ));

  let exports_name = runtime_template.render_runtime_variable(&RuntimeVariable::Exports);

  let module_ids_code = &module_id_exprs
    .iter()
    .map(|module_id_expr| {
      format!(
        "{}({module_id_expr})",
        runtime_template.render_runtime_variable(&RuntimeVariable::StartupExec)
      )
    })
    .collect::<Vec<_>>()
    .join(", ");
  if chunks_ids.is_empty() {
    if !module_ids_code.is_empty() {
      if passive {
        source.push_str(&format!("var {exports_name} = ("));
        source.push_str(module_ids_code);
        source.push_str(");\n");
      } else {
        // Ensure STARTUP_ENTRYPOINT wrappers run even without dependent chunks (e.g. async startup).
        source.push_str(&format!(
          "var {exports_name} = {}(0, [], function() {{\n        return {};\n      }});\n",
          runtime_template.render_runtime_globals(&RuntimeGlobals::STARTUP_ENTRYPOINT),
          module_ids_code
        ));
      }
    }
  } else {
    if !passive {
      source.push_str(&format!("var {exports_name} = "));
    }
    source.push_str(&format!(
      "{}(0, {}, function() {{
        return {};
      }});\n",
      if passive {
        runtime_template.render_runtime_globals(&RuntimeGlobals::ON_CHUNKS_LOADED)
      } else {
        runtime_template.render_runtime_globals(&RuntimeGlobals::STARTUP_ENTRYPOINT)
      },
      stringify_chunks_to_array(&chunks_ids),
      module_ids_code
    ));
    if passive {
      source.push_str(&format!(
        "var {exports_name} = {}();\n",
        runtime_template.render_runtime_globals(&RuntimeGlobals::ON_CHUNKS_LOADED)
      ));
    }
  }

  RawStringSource::from(source).boxed()
}

/**
 * This is ported from https://github.com/webpack/webpack/blob/87660921808566ef3b8796f8df61bd79fc026108/lib/esm/ModuleChunkFormatPlugin.js#L98
 */
pub fn get_relative_path(base_chunk_output_name: &str, other_chunk_output_name: &str) -> String {
  let mut base_chunk_output_name_arr = base_chunk_output_name.split('/').collect::<Vec<_>>();
  base_chunk_output_name_arr.pop();
  let mut other_chunk_output_name_arr = other_chunk_output_name.split('/').collect::<Vec<_>>();
  while !base_chunk_output_name_arr.is_empty()
    && !other_chunk_output_name_arr.is_empty()
    && base_chunk_output_name_arr[0] == other_chunk_output_name_arr[0]
  {
    base_chunk_output_name_arr.remove(0);
    other_chunk_output_name_arr.remove(0);
  }
  let path = if base_chunk_output_name_arr.is_empty() {
    "./".to_string()
  } else {
    "../".repeat(base_chunk_output_name_arr.len())
  };
  format!("{path}{}", other_chunk_output_name_arr.join("/"))
}

pub async fn get_chunk_output_name(chunk: &Chunk, compilation: &Compilation) -> Result<String> {
  let hash = chunk.rendered_hash(
    &compilation.chunk_hashes_artifact,
    compilation.options.output.hash_digest_length,
  );
  let filename = get_js_chunk_filename_template(
    chunk,
    &compilation.options.output,
    &compilation.build_chunk_graph_artifact.chunk_group_by_ukey,
  );
  compilation
    .get_path(
      &filename,
      PathData::default()
        .chunk_id_optional(chunk.id().map(|id| id.as_str()))
        .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_optional(hash),
    )
    .await
}

pub fn get_chunk_runtime_requirements<'a>(
  compilation: &'a Compilation,
  chunk_ukey: &ChunkUkey,
) -> &'a RuntimeGlobals {
  ChunkGraph::get_chunk_runtime_requirements(compilation, chunk_ukey)
}

/// Regex to match EJS `<%- RUNTIME_GLOBALS %>` where the identifier is uppercase (RuntimeGlobals enum style).
static EJS_RUNTIME_GLOBALS_RE: LazyLock<Regex> = LazyLock::new(|| {
  Regex::new(r"<%\-\s*([A-Z][A-Z0-9_]*)\s*%>").expect("invalid EJS runtime globals regex")
});

/// Extracts all RuntimeGlobals references from an EJS template string.
///
/// Matches patterns like `<%- PUBLIC_PATH %>` or `<%- GET_CHUNK_SCRIPT_FILENAME %>`
/// where the identifier is uppercase letters, digits and underscores (RuntimeGlobals enum variant naming).
/// Does not match other EJS placeholders such as `<%- basicFunction(...) %>` or `<%- _modules %>`.
pub fn extract_runtime_globals_from_ejs(ejs_content: &str) -> RuntimeGlobals {
  let names = EJS_RUNTIME_GLOBALS_RE
    .captures_iter(ejs_content)
    .map(|cap| cap[1].to_string())
    // script nonce is always optional
    .filter(|name| name.as_str() != "SCRIPT_NONCE")
    .collect_vec();
  RuntimeGlobals::from_names(&names)
}

#[cfg(test)]
mod tests {
  use super::{RuntimeGlobals, extract_runtime_globals_from_ejs};

  fn expected_globals(names: &[&str]) -> RuntimeGlobals {
    let names: Vec<String> = names.iter().map(|s| (*s).to_string()).collect();
    RuntimeGlobals::from_names(&names)
  }

  #[test]
  fn test_extract_runtime_globals_empty() {
    assert!(extract_runtime_globals_from_ejs("").is_empty());
    assert!(extract_runtime_globals_from_ejs("plain text").is_empty());
  }

  #[test]
  fn test_extract_runtime_globals_single() {
    assert_eq!(
      extract_runtime_globals_from_ejs("<%- PUBLIC_PATH %>"),
      expected_globals(&["PUBLIC_PATH"])
    );
    assert_eq!(
      extract_runtime_globals_from_ejs("var x = <%- REQUIRE %>;"),
      expected_globals(&["REQUIRE"])
    );
  }

  #[test]
  fn test_extract_runtime_globals_multiple_unique() {
    let ejs = r#"link.href = <%- PUBLIC_PATH %> + <%- GET_CHUNK_SCRIPT_FILENAME %>(chunkId);
    if (<%- HAS_OWN_PROPERTY %>(installedChunks, chunkId)) {}"#;
    assert_eq!(
      extract_runtime_globals_from_ejs(ejs),
      expected_globals(&[
        "PUBLIC_PATH",
        "GET_CHUNK_SCRIPT_FILENAME",
        "HAS_OWN_PROPERTY"
      ])
    );
  }

  #[test]
  fn test_extract_runtime_globals_deduplicate() {
    let ejs = "<%- REQUIRE %>; <%- PUBLIC_PATH %>; <%- REQUIRE %>; <%- PUBLIC_PATH %>";
    assert_eq!(
      extract_runtime_globals_from_ejs(ejs),
      expected_globals(&["REQUIRE", "PUBLIC_PATH"])
    );
  }

  #[test]
  fn test_extract_runtime_globals_with_spaces() {
    assert_eq!(
      extract_runtime_globals_from_ejs("<%-  ENSURE_CHUNK  %>"),
      expected_globals(&["ENSURE_CHUNK"])
    );
  }

  #[test]
  fn test_extract_runtime_globals_ignore_non_globals() {
    // Should not match: lowercase, underscore prefix, function calls
    let ejs = r#"<%- basicFunction("resolve, reject")  %>
    <%- _modules %>
    <%- _cross_origin %>
    <%- MODULE_CACHE %>"#;
    assert_eq!(
      extract_runtime_globals_from_ejs(ejs),
      expected_globals(&["MODULE_CACHE"])
    );
  }

  #[test]
  fn test_extract_runtime_globals_uppercase_with_underscores() {
    let ejs = "<%- GET_CHUNK_UPDATE_SCRIPT_FILENAME %> <%- HMR_DOWNLOAD_UPDATE_HANDLERS %>";
    assert_eq!(
      extract_runtime_globals_from_ejs(ejs),
      expected_globals(&[
        "GET_CHUNK_UPDATE_SCRIPT_FILENAME",
        "HMR_DOWNLOAD_UPDATE_HANDLERS"
      ])
    );
  }

  #[test]
  fn test_extract_runtime_globals_real_ejs_snippet() {
    let ejs = r#"var installChunk = <%- basicFunction("data") %> {
    var <%- _modules %> = data.<%- _modules %>;
    for (moduleId in <%- _modules %>) {
        if (<%- HAS_OWN_PROPERTY %>(<%- _modules %>, moduleId)) {
            <%- MODULE_FACTORIES %>[moduleId] = <%- _modules %>[moduleId];
        }
    }
    if (__rspack_esm_runtime) __rspack_esm_runtime(<%- REQUIRE %>);
};"#;
    assert_eq!(
      extract_runtime_globals_from_ejs(ejs),
      expected_globals(&["HAS_OWN_PROPERTY", "MODULE_FACTORIES", "REQUIRE"])
    );
  }
}