mirror of
https://github.com/danbulant/oxc
synced 2026-05-21 13:18:59 +00:00
2c7bb9f6c8
We have a strange workaround for `visit_function` where we pass in `ScopeFlags`, to support creating the scope inside `Function`, but setting different flags for `MethodDefinition`s. Previously `visit_function` took `Option<ScopeFlags>` and then did `flags.unwrap_or(ScopeFlags::empty()) | ScopeFlags::Function` to it. Personally, I found this confusing. When I was looking at `MethodDefinition`, I was wondering "It's a function, why doesn't it set Function flag too?" This changes makes it more explicit and clear what `ScopeFlags` everything has.
63 lines
1.6 KiB
Rust
63 lines
1.6 KiB
Rust
#![allow(clippy::print_stdout)]
|
|
use std::{env, path::Path};
|
|
|
|
use oxc_allocator::Allocator;
|
|
use oxc_ast::{
|
|
ast::{Class, Function, TSImportType},
|
|
visit::walk,
|
|
Visit,
|
|
};
|
|
use oxc_parser::Parser;
|
|
use oxc_span::SourceType;
|
|
use oxc_syntax::scope::ScopeFlags;
|
|
|
|
// Instruction:
|
|
// create a `test.js`,
|
|
// run `cargo run -p oxc_parser --example visitor`
|
|
// or `cargo watch -x "run -p oxc_parser --example visitor"`
|
|
|
|
fn main() -> std::io::Result<()> {
|
|
let name = env::args().nth(1).unwrap_or_else(|| "test.js".to_string());
|
|
let path = Path::new(&name);
|
|
let source_text = std::fs::read_to_string(path)?;
|
|
let allocator = Allocator::default();
|
|
let source_type = SourceType::from_path(path).unwrap();
|
|
let ret = Parser::new(&allocator, &source_text, source_type).parse();
|
|
|
|
for error in ret.errors {
|
|
let error = error.with_source_code(source_text.clone());
|
|
println!("{error:?}");
|
|
}
|
|
|
|
let program = ret.program;
|
|
|
|
let mut ast_pass = CountASTNodes::default();
|
|
ast_pass.visit_program(&program);
|
|
println!("{ast_pass:?}");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[derive(Debug, Default)]
|
|
struct CountASTNodes {
|
|
functions: usize,
|
|
classes: usize,
|
|
ts_import_types: usize,
|
|
}
|
|
|
|
impl<'a> Visit<'a> for CountASTNodes {
|
|
fn visit_function(&mut self, func: &Function<'a>, flags: ScopeFlags) {
|
|
self.functions += 1;
|
|
walk::walk_function(self, func, flags);
|
|
}
|
|
|
|
fn visit_class(&mut self, class: &Class<'a>) {
|
|
self.classes += 1;
|
|
walk::walk_class(self, class);
|
|
}
|
|
|
|
fn visit_ts_import_type(&mut self, ty: &TSImportType<'a>) {
|
|
self.ts_import_types += 1;
|
|
walk::walk_ts_import_type(self, ty);
|
|
}
|
|
}
|