swc_ecma_transforms_compat/es2015/
sticky_regex.rsuse swc_ecma_ast::*;
use swc_ecma_transforms_base::perf::Parallel;
use swc_ecma_utils::{quote_ident, ExprFactory};
use swc_ecma_visit::{as_folder, noop_visit_mut_type, Fold, VisitMut, VisitMutWith};
use swc_trace_macro::swc_trace;
#[tracing::instrument(level = "info", skip_all)]
pub fn sticky_regex() -> impl 'static + Fold + VisitMut {
as_folder(StickyRegex)
}
struct StickyRegex;
impl Parallel for StickyRegex {
fn merge(&mut self, _: Self) {}
fn create(&self) -> Self {
StickyRegex
}
}
#[swc_trace]
impl VisitMut for StickyRegex {
noop_visit_mut_type!();
fn visit_mut_expr(&mut self, e: &mut Expr) {
e.visit_mut_children_with(self);
if let Expr::Lit(Lit::Regex(Regex { exp, flags, span })) = e {
if flags.contains('y') {
*e = Expr::New(NewExpr {
span: *span,
callee: Box::new(quote_ident!(*span, "RegExp").into()),
args: Some(vec![exp.clone().as_arg(), flags.clone().as_arg()]),
type_args: Default::default(),
})
}
}
}
}
#[cfg(test)]
mod tests {
use swc_ecma_transforms_testing::test;
use super::*;
test!(
::swc_ecma_parser::Syntax::default(),
|_| sticky_regex(),
babel_basic,
"var re = /o\"'+/y;",
"var re = new RegExp(\"o\\\"'+\", \"y\");"
);
test!(
::swc_ecma_parser::Syntax::default(),
|_| sticky_regex(),
babel_ignore_non_sticky,
"var re = /o+/;",
"var re = /o+/;"
);
}