Skip to main content

regular_expression/
regular_expression.rs

1#![expect(clippy::print_stdout)]
2use std::{env, fs, path::Path, sync::Arc};
3
4use oxc_allocator::Allocator;
5use oxc_ast::ast::*;
6use oxc_ast_visit::VisitJs;
7use oxc_parser::{ParseOptions, Parser};
8use oxc_regular_expression::{ConstructorParser as RegExpParser, Options as RegExpParserOptions};
9use oxc_span::SourceType;
10
11// `cargo run -p oxc_parser --example regular_expression`
12
13fn main() {
14    // 1. Get the file content and parse
15    let name = env::args().nth(1).unwrap_or_else(|| "test.js".to_string());
16    let path = Path::new(&name);
17
18    let source_text: Arc<str> = Arc::from(fs::read_to_string(path).unwrap());
19    let source_type = SourceType::from_path(path).unwrap();
20
21    let allocator = Allocator::default();
22    let options = ParseOptions { parse_regular_expression: true, ..ParseOptions::default() };
23
24    let parser_ret =
25        Parser::new(&allocator, source_text.as_ref(), source_type).with_options(options).parse();
26    if !parser_ret.diagnostics.is_empty() {
27        println!("Parsing failed:");
28        for error in parser_ret.diagnostics {
29            let error = error.with_source_code(Arc::clone(&source_text));
30            println!("{error:?}");
31        }
32        return;
33    }
34
35    // Parse regular expressions
36    // - RegExpLiteral
37    // - new RegExp() with string or template literal if static
38    RegularExpressionVisitor { source_text: Arc::clone(&source_text) }
39        .visit_program(&parser_ret.program);
40}
41
42struct RegularExpressionVisitor {
43    source_text: Arc<str>,
44}
45
46impl<'a> VisitJs<'a> for RegularExpressionVisitor {
47    fn visit_reg_exp_literal(&mut self, re: &RegExpLiteral<'a>) {
48        println!("🍀 {}", re.span.source_text(self.source_text.as_ref()));
49
50        println!("{re:#?}");
51        println!();
52    }
53
54    fn visit_new_expression(&mut self, new_expr: &NewExpression<'a>) {
55        if new_expr.callee.get_identifier_reference().is_some_and(|ident| ident.name == "RegExp") {
56            println!("🍀 {}", new_expr.span.source_text(&self.source_text));
57
58            let pattern_span = match new_expr.arguments.first() {
59                Some(Argument::StringLiteral(sl)) => sl.span,
60                _ => return,
61            };
62
63            let flags_span = match new_expr.arguments.get(1) {
64                Some(Argument::StringLiteral(sl)) => Some(sl.span),
65                _ => None,
66            };
67
68            let allocator = Allocator::default();
69            let parsed = RegExpParser::new(
70                &allocator,
71                pattern_span.source_text(&self.source_text),
72                flags_span.map(|span| span.source_text(&self.source_text)),
73                RegExpParserOptions {
74                    pattern_span_offset: pattern_span.start,
75                    flags_span_offset: flags_span.map_or(0, |span| span.start),
76                },
77            )
78            .parse();
79
80            if let Err(error) = parsed {
81                let error = error.with_source_code(Arc::clone(&self.source_text));
82                println!("{error:?}");
83                return;
84            }
85            println!("{parsed:#?}");
86            println!();
87        }
88    }
89}