feat: support wrappedContextRegExp (#8149)

crates/node_binding/binding.d.ts

@@ -1437,6 +1437,7 @@ export interface RawJavascriptParserOptions {
   url?: string
   exprContextCritical?: boolean
   wrappedContextCritical?: boolean
+  wrappedContextRegExp?: RegExp
   exportsPresence?: string
   importExportsPresence?: string
   reexportExportsPresence?: string

crates/rspack_binding_options/src/options/raw_module/mod.rs

@@ -21,6 +21,7 @@ use rspack_core::{
 use rspack_error::error;
 use rspack_napi::regexp::{JsRegExp, JsRegExpExt};
 use rspack_napi::threadsafe_function::ThreadsafeFunction;
+use rspack_regex::RspackRegex;
 use tokio::runtime::Handle;
 
 use crate::RawResolveOptions;
@@ -110,7 +111,7 @@ impl TryFrom<RawRuleSetCondition> for rspack_core::RuleSetCondition {
     let result = match x {
       RawRuleSetCondition::string(s) => Self::String(s),
       RawRuleSetCondition::regexp(r) => {
-        let reg = rspack_regex::RspackRegex::with_flags(&r.source, &r.flags)?;
+        let reg = RspackRegex::with_flags(&r.source, &r.flags)?;
         Self::Regexp(reg)
       }
       RawRuleSetCondition::logical(mut l) => {
@@ -265,6 +266,8 @@ pub struct RawJavascriptParserOptions {
   pub url: Option<String>,
   pub expr_context_critical: Option<bool>,
   pub wrapped_context_critical: Option<bool>,
+  #[napi(ts_type = "RegExp")]
+  pub wrapped_context_reg_exp: Option<JsRegExp>,
   pub exports_presence: Option<String>,
   pub import_exports_presence: Option<String>,
   pub reexport_exports_presence: Option<String>,
@@ -303,6 +306,9 @@ impl From<RawJavascriptParserOptions> for JavascriptParserOptions {
         .map(|x| DynamicImportFetchPriority::from(x.as_str())),
       url: value.url.map(|v| JavascriptParserUrl::from(v.as_str())),
       expr_context_critical: value.expr_context_critical,
+      wrapped_context_reg_exp: value
+        .wrapped_context_reg_exp
+        .map(|context_reg_exp| context_reg_exp.to_rspack_regex()),
       wrapped_context_critical: value.wrapped_context_critical,
       exports_presence: value
         .exports_presence

crates/rspack_core/src/options/module.rs

@@ -228,6 +228,7 @@ pub struct JavascriptParserOptions {
   pub url: Option<JavascriptParserUrl>,
   pub expr_context_critical: Option<bool>,
   pub wrapped_context_critical: Option<bool>,
+  pub wrapped_context_reg_exp: Option<RspackRegex>,
   pub exports_presence: Option<ExportPresenceMode>,
   pub import_exports_presence: Option<ExportPresenceMode>,
   pub reexport_exports_presence: Option<ExportPresenceMode>,

crates/rspack_plugin_javascript/src/visitors/dependency/context_dependency_helper.rs

@@ -11,15 +11,18 @@ use swc_core::ecma::ast::Expr;
 use super::create_traceable_error;
 use crate::utils::eval::{BasicEvaluatedExpression, TemplateStringKind};
 
-// FIXME: delete this after `parserOptions.wrappedContextRegExp.source`
-const DEFAULT_WRAPPED_CONTEXT_REGEXP: &str = ".*";
-
 pub fn create_context_dependency(
   param: &BasicEvaluatedExpression,
   expr: &Expr,
   parser: &mut crate::visitors::JavascriptParser,
 ) -> ContextModuleScanResult {
   let mut critical = None;
+  let wrapped_context_reg_exp = parser
+    .javascript_options
+    .wrapped_context_reg_exp
+    .as_ref()
+    .expect("should have wrapped_context_reg_exp")
+    .source();
 
   if param.is_template_string() {
     let quasis = param.quasis();
@@ -47,10 +50,10 @@ pub fn create_context_dependency(
     let reg = format!(
       "^{}{}{}{}$",
       quote_meta(&prefix),
-      DEFAULT_WRAPPED_CONTEXT_REGEXP,
+      wrapped_context_reg_exp,
       quasis[1..quasis.len() - 1]
         .iter()
-        .map(|q| quote_meta(q.string().as_str()) + DEFAULT_WRAPPED_CONTEXT_REGEXP)
+        .map(|q| quote_meta(q.string().as_str()) + wrapped_context_reg_exp)
         .join(""),
       quote_meta(&postfix)
     );
@@ -148,7 +151,7 @@ pub fn create_context_dependency(
     };
 
     let reg = format!(
-      "^{}{DEFAULT_WRAPPED_CONTEXT_REGEXP}{}$",
+      "^{}{wrapped_context_reg_exp}{}$",
       quote_meta(&prefix),
       quote_meta(&postfix)
     );

crates/rspack_util/src/merge.rs

@@ -1,3 +1,5 @@
+use rspack_regex::RspackRegex;
+
 use crate::atom::Atom;
 
 pub trait MergeFrom: Clone {
@@ -42,6 +44,7 @@ impl_merge_from!(u8, u16, u32, u64, u128);
 impl_merge_from!(bool);
 impl_merge_from!(String);
 impl_merge_from!(Atom);
+impl_merge_from!(RspackRegex);
 
 pub fn merge_from_optional_with<T: MergeFrom>(
   base: Option<T>,

packages/rspack-test-tools/tests/__snapshots__/Defaults.test.js.snap

@@ -182,6 +182,7 @@ Object {
           ...,
         ],
         wrappedContextCritical: false,
+        wrappedContextRegExp: /\\.\\*/,
       },
     },
     rules: Array [],

packages/rspack-test-tools/tests/configCases/context-module/wrapped-context-reg-exp/index.js

@@ -0,0 +1,6 @@
+it("should not include foo.js", async () => {
+	let a1 = 'a1';
+	let a2 = 'a2';
+	expect(require('./sub/' + a1)).toBe("a1");
+	expect(() => require('./sub/' + a2)).toThrow();
+});

packages/rspack-test-tools/tests/configCases/context-module/wrapped-context-reg-exp/rspack.config.js

@@ -0,0 +1,11 @@
+/** @type {import("@rspack/core").Configuration} */
+module.exports = {
+	module: {
+	  parser: {
+		javascript: {
+		  wrappedContextRegExp: /.*1/,
+		}
+	  }
+	},
+  }
+

packages/rspack-test-tools/tests/configCases/context-module/wrapped-context-reg-exp/sub/a1.js

@@ -0,0 +1 @@
+module.exports = "a1";

packages/rspack-test-tools/tests/configCases/context-module/wrapped-context-reg-exp/sub/a2.js

@@ -0,0 +1 @@
+module.exports = "a2";