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

refactor(transformer): move and simplify TS enum transform entry point (#3760)

Browse source 
Move logic for TS enum transform into `enum.rs`. Also simplify the logic, using `Statement` enum variants directly.
This commit is contained in:
overlookmotel 2024-06-19 12:16:30 +00:00
parent bd534375a1
commit 512740d33d
2 changed files with 21 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
crates/oxc_transformer/src/typescript/enum.rs
View file

@ -20,6 +20,26 @@ impl<'a> TypeScriptEnum<'a> {
Self { ctx, enums: FxHashMap::default() }
}
pub fn transform_statement(&mut self, stmt: &mut Statement<'a>, ctx: &TraverseCtx<'a>) {
let new_stmt = match stmt {
Statement::TSEnumDeclaration(ts_enum_decl) => {
self.transform_ts_enum(ts_enum_decl, false, ctx)
}
Statement::ExportNamedDeclaration(decl) => {
if let Some(Declaration::TSEnumDeclaration(ts_enum_decl)) = &decl.declaration {
self.transform_ts_enum(ts_enum_decl, true, ctx)
} else {
None
}
}
_ => None,
};
if let Some(new_stmt) = new_stmt {
*stmt = new_stmt;
}
}
/// ```TypeScript
/// enum Foo {
/// X = 1,

28
crates/oxc_transformer/src/typescript/mod.rs
View file

@ -152,33 +152,7 @@ impl<'a> TypeScript<'a> {
}
pub fn transform_statement(&mut self, stmt: &mut Statement<'a>, ctx: &TraverseCtx<'a>) {
let new_stmt = match stmt {
match_declaration!(Statement) => {
if let Declaration::TSEnumDeclaration(ts_enum_decl) = &stmt.to_declaration() {
self.r#enum.transform_ts_enum(ts_enum_decl, false, ctx)
} else {
None
}
}
match_module_declaration!(Statement) => {
if let ModuleDeclaration::ExportNamedDeclaration(decl) =
stmt.to_module_declaration_mut()
{
if let Some(Declaration::TSEnumDeclaration(ts_enum_decl)) = &decl.declaration {
self.r#enum.transform_ts_enum(ts_enum_decl, true, ctx)
} else {
None
}
} else {
None
}
}
_ => None,
};
if let Some(new_stmt) = new_stmt {
*stmt = new_stmt;
}
self.r#enum.transform_statement(stmt, ctx);
}
pub fn transform_if_statement(&mut self, stmt: &mut IfStatement<'a>) {