From 8f7e85ef5ed68e1cd401927b54ad39ba97441fb4 Mon Sep 17 00:00:00 2001
From: Dominik Nakamura <dnaka91@gmail.com>
Date: Sat, 14 Oct 2023 18:28:15 +0900
Subject: [PATCH] feat: implement Decode for non-zero types

---
 crates/stef/src/buf/decode.rs | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/crates/stef/src/buf/decode.rs b/crates/stef/src/buf/decode.rs
index 0e35f92..a09f655 100644
--- a/crates/stef/src/buf/decode.rs
+++ b/crates/stef/src/buf/decode.rs
@@ -19,6 +19,7 @@ pub enum Error {
     NonUtf8(std::string::FromUtf8Error),
     MissingField { id: u32, name: Option<&'static str> },
     UnknownVariant(u32),
+    Zero,
 }
 
 impl From<varint::DecodeIntError> for Error {
@@ -273,9 +274,24 @@ where
     }
 }
 
-// -----------------------------
-// TODO: NonZero
-// -----------------------------
+macro_rules! non_zero {
+    ($ty:ty) => {
+        paste::paste! {
+            impl Decode for std::num::[<NonZero $ty:upper>] {
+                #[inline(always)]
+                fn decode(r: &mut impl Buf) -> Result<Self> {
+                    Self::new([<decode_ $ty>](r)?).ok_or_else(|| Error::Zero)
+                }
+            }
+        }
+    };
+    ($($ty:ty),+ $(,)?) => {
+        $(non_zero!($ty);)+
+    };
+}
+
+non_zero!(u8, u16, u32, u64, u128);
+non_zero!(i8, i16, i32, i64, i128);
 
 impl<'a, T> Decode for std::borrow::Cow<'a, T>
 where