add enc/dec for bool

Signed-off-by: Dave Huseby <[email protected]>

Author: dhuseby

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "multitrait"
-version = "0.1.4"
+version = "0.1.5"
 edition = "2021"
 authors = ["Dave Huseby <[email protected]>"]
 description = "Multiformat traits"

src/enc_into.rs

@@ -6,6 +6,16 @@ pub trait EncodeInto {
     fn encode_into(&self) -> Vec<u8>;
 }
 
+/// Encode a bool into a compact varuint Vec<u8>
+impl EncodeInto for bool {
+    fn encode_into(&self) -> Vec<u8> {
+        match *self {
+            true => vec![1u8],
+            false => vec![0u8],
+        }
+    }
+}
+
 /// Encode a u8 into a compact varuint Vec<u8>
 impl EncodeInto for u8 {
     fn encode_into(&self) -> Vec<u8> {

src/lib.rs

@@ -28,6 +28,16 @@ pub mod prelude {
 mod test {
     use super::prelude::*;
 
+    #[test]
+    fn test_bool() {
+        let tbuf = true.encode_into();
+        let (tval, _) = bool::try_decode_from(&tbuf).unwrap();
+        assert_eq!(true, tval);
+        let fbuf = false.encode_into();
+        let (fval, _) = bool::try_decode_from(&fbuf).unwrap();
+        assert_eq!(false, fval);
+    }
+
     #[test]
     fn test_u8() {
         let buf = 0xff_u8.encode_into();

src/try_decode_from.rs

@@ -12,6 +12,16 @@ pub trait TryDecodeFrom<'a>: Sized {
     fn try_decode_from(bytes: &'a [u8]) -> Result<(Self, &'a [u8]), Self::Error>;
 }
 
+/// Try to decode a varuint encoded bool
+impl<'a> TryDecodeFrom<'a> for bool {
+    type Error = Error;
+
+    fn try_decode_from(bytes: &'a [u8]) -> Result<(Self, &'a [u8]), Self::Error> {
+        let (v, ptr) = decode::u8(bytes).map_err(|e| Self::Error::UnsignedVarintDecode(e))?;
+        Ok(((v != 0), ptr))
+    }
+}
+
 /// Try to decode a varuint encoded u8
 impl<'a> TryDecodeFrom<'a> for u8 {
     type Error = Error;