Derive Num for newtypes

rust-num · Sep 29, 2018 · 4bb4c8d · 4bb4c8d
1 parent 86ab909
commit 4bb4c8d
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/src/lib.rs b/src/lib.rs
@@ -499,3 +499,19 @@ pub fn one(input: TokenStream) -> TokenStream {
         }
     }).into()
 }
+
+#[proc_macro_derive(Num)]
+pub fn num(input: TokenStream) -> TokenStream {
+    let ast: syn::DeriveInput = syn::parse(input).unwrap();
+    let name = &ast.ident;
+    let inner_ty = newtype_inner(&ast.data).expect(NEWTYPE_ONLY);
+    dummy_const_trick("Num", &name, quote! {
+        extern crate num_traits as _num_traits;
+        impl _num_traits::Num for #name {
+            type FromStrRadixErr = <#inner_ty as _num_traits::Num>::FromStrRadixErr;
+            fn from_str_radix(s: &str, radix: u32) -> Result<Self, Self::FromStrRadixErr> {
+                <#inner_ty as _num_traits::Num>::from_str_radix(s, radix).map(#name)
+            }
+        }
+    }).into()
+}
diff --git a/tests/newtype.rs b/tests/newtype.rs
@@ -2,7 +2,7 @@ extern crate num as num_renamed;
 #[macro_use]
 extern crate num_derive;
 
-use num_renamed::{FromPrimitive, ToPrimitive, NumCast, One, Zero};
+use num_renamed::{FromPrimitive, ToPrimitive, NumCast, One, Zero, Num};
 
 #[derive(
     Debug,
@@ -16,6 +16,7 @@ use num_renamed::{FromPrimitive, ToPrimitive, NumCast, One, Zero};
     NumCast,
     One,
     Zero,
+    Num,
 )]
 struct MyFloat(f64);
 
@@ -52,3 +53,8 @@ fn test_zero() {
 fn test_one() {
     assert_eq!(MyFloat::one(), MyFloat(1.0));
 }
+
+#[test]
+fn test_num() {
+    assert_eq!(MyFloat::from_str_radix("25", 10).ok(), Some(MyFloat(25.0)));
+}