dan/oxc
1
0
Fork 0
mirror of https://github.com/danbulant/oxc synced 2026-05-19 04:08:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(xtask): impl as_ast_kind method for each variant (#5491)

Browse source
This commit is contained in:
IWANABETHATGUY 2024-09-05 16:17:08 +00:00
parent a96866d951
commit cedf7a4daa
2 changed files with 1374 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1348
crates/oxc_ast/src/generated/ast_kind.rs
View file

File diff suppressed because it is too large Load diff

28
tasks/ast_tools/src/generators/ast_kind.rs
View file

@ -1,6 +1,7 @@
use convert_case::{Case, Casing};
use itertools::Itertools;
use quote::quote;
use syn::{parse_quote, Arm, Ident, Type, Variant};
use quote::{format_ident, quote};
use syn::{parse_quote, Arm, Ident, ImplItemFn, Type, Variant};
use crate::{
codegen::{generated_header, LateCtx},
@ -154,6 +155,24 @@ impl Generator for AstKindGenerator {
.map(|(ident, _)| parse_quote!(Self :: #ident(it) => it.span()))
.collect_vec();
let as_ast_kind_impls: Vec<ImplItemFn> = have_kinds
.iter()
.map(|(ident, typ)| {
let snake_case_name =
format_ident!("as_{}", ident.to_string().to_case(Case::Snake));
parse_quote!(
///@@line_break
pub fn #snake_case_name(&self) -> Option<&#typ> {
if let Self::#ident(v) = self {
Some(*v)
} else {
None
}
}
)
})
.collect_vec();
let header = generated_header!();
GeneratorOutput(
@ -189,6 +208,11 @@ impl Generator for AstKindGenerator {
}
}
}
///@@line_break
impl<'a> AstKind<'a> {
#(#as_ast_kind_impls)*
}
},
)
}