refactor(ast_tools): shorten code

oxc-project · Oct 19, 2024 · 66ed87a · 66ed87a
1 parent 85443d4
commit 66ed87a
Showing 1 changed file with 8 additions and 21 deletions.
diff --git a/tasks/ast_tools/src/derives/estree.rs b/tasks/ast_tools/src/derives/estree.rs
@@ -48,30 +48,17 @@ impl Derive for DeriveESTree {
         };
         let ident = def.ident();
 
-        if def.has_lifetime() {
-            quote! {
-                impl<'a> Serialize for #ident<'a> {
-                    #[allow(clippy::match_same_arms, unused_mut)]
-                    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
-                    where S: Serializer,
-                    {
-                        #body
-                    }
-                }
-                #ts_type_def
-            }
-        } else {
-            quote! {
-                impl Serialize for #ident {
-                    #[allow(clippy::match_same_arms, unused_mut)]
-                    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+        let lifetime = if def.has_lifetime() { quote!(<'a>) } else { TokenStream::new() };
+        quote! {
+            impl #lifetime Serialize for #ident #lifetime {
+                #[allow(clippy::match_same_arms, unused_mut)]
+                fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
                     where S: Serializer,
-                    {
-                        #body
-                    }
+                {
+                    #body
                 }
-                #ts_type_def
             }
+            #ts_type_def
         }
     }