It's almost not worth it here, but factor out preparation from token …

…stream generation.
davidcole1340 · Oct 22, 2024 · c8a363d · c8a363d
1 parent 6e804b6
commit c8a363d
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/crates/macros/src/constant.rs b/crates/macros/src/constant.rs
@@ -15,6 +15,13 @@ pub struct Constant {
     pub value: String,
 }
 
+fn prepare(input: ItemConst) -> Result<TokenStream> {
+    Ok(quote! {
+        #[allow(dead_code)]
+        #input
+    })
+}
+
 pub fn parser(input: ItemConst) -> Result<TokenStream> {
     let mut state = STATE.lock();
 
@@ -28,10 +35,7 @@ pub fn parser(input: ItemConst) -> Result<TokenStream> {
         value: input.expr.to_token_stream().to_string(),
     });
 
-    Ok(quote! {
-        #[allow(dead_code)]
-        #input
-    })
+    prepare(input)
 }
 
 impl Constant {