Simplify macro in fp::prime

I just learned this trick to capture a `tt` and reparse it as other types of ASTs and it's definitely a good one to know
JoeyBF · Aug 14, 2024 · b0a826b · b0a826b
1 parent bd831d5
commit b0a826b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/ext/crates/fp/src/prime/mod.rs b/ext/crates/fp/src/prime/mod.rs
@@ -232,8 +232,11 @@ macro_rules! impl_prime_ops {
 }
 
 macro_rules! impl_try_from {
+    ($p:tt) => {
+        impl_try_from!(@ $p, $p);
+    };
     // We need the type both as a type and as an expression.
-    ($pn:ty, $pn_ex:expr) => {
+    (@ $pn:ty, $pn_ex:expr) => {
         impl std::convert::TryFrom<u32> for $pn {
             type Error = PrimeError;
 

diff --git a/ext/crates/fp/src/prime/primes_2.rs b/ext/crates/fp/src/prime/primes_2.rs
@@ -4,7 +4,7 @@ use super::*;
 
 def_prime_static!(P2, 2);
 impl_prime_ops!(P2);
-impl_try_from!(P2, P2);
+impl_try_from!(P2);
 
 pub type ValidPrime = P2;
 

diff --git a/ext/crates/fp/src/prime/primes_generic.rs b/ext/crates/fp/src/prime/primes_generic.rs
@@ -12,10 +12,10 @@ impl_prime_ops!(P3);
 impl_prime_ops!(P5);
 impl_prime_ops!(P7);
 
-impl_try_from!(P2, P2);
-impl_try_from!(P3, P3);
-impl_try_from!(P5, P5);
-impl_try_from!(P7, P7);
+impl_try_from!(P2);
+impl_try_from!(P3);
+impl_try_from!(P5);
+impl_try_from!(P7);
 
 pub(crate) mod fp {
     use super::{P2, P3, P5, P7};