Small update to Bytes/BytesImpl/Program conversions

Chia-Network · May 18, 2024 · 2d9f5d9 · 2d9f5d9
1 parent 88d1e07
commit 2d9f5d9
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/crates/chia-protocol/src/bytes.rs b/crates/chia-protocol/src/bytes.rs
@@ -39,6 +39,10 @@ impl Bytes {
         self.0.as_slice()
     }
 
+    pub fn to_vec(&self) -> Vec<u8> {
+        self.0.clone()
+    }
+
     pub fn into_inner(self) -> Vec<u8> {
         self.0
     }
@@ -124,6 +128,12 @@ impl From<&[u8]> for Bytes {
     }
 }
 
+impl<const N: usize> From<BytesImpl<N>> for Bytes {
+    fn from(value: BytesImpl<N>) -> Self {
+        Self(value.0.to_vec())
+    }
+}
+
 impl From<Vec<u8>> for Bytes {
     fn from(value: Vec<u8>) -> Self {
         Self(value)
@@ -159,6 +169,14 @@ impl<const N: usize> BytesImpl<N> {
         Self(bytes)
     }
 
+    pub const fn len(&self) -> usize {
+        N
+    }
+
+    pub const fn is_empty(&self) -> bool {
+        N == 0
+    }
+
     pub fn as_slice(&self) -> &[u8] {
         &self.0
     }
@@ -281,6 +299,22 @@ impl<const N: usize> TryFrom<&Vec<u8>> for BytesImpl<N> {
     }
 }
 
+impl<const N: usize> TryFrom<Bytes> for BytesImpl<N> {
+    type Error = TryFromSliceError;
+
+    fn try_from(value: Bytes) -> Result<Self, TryFromSliceError> {
+        value.0.as_slice().try_into()
+    }
+}
+
+impl<const N: usize> TryFrom<&Bytes> for BytesImpl<N> {
+    type Error = TryFromSliceError;
+
+    fn try_from(value: &Bytes) -> Result<Self, TryFromSliceError> {
+        value.0.as_slice().try_into()
+    }
+}
+
 impl<const N: usize> From<BytesImpl<N>> for Vec<u8> {
     fn from(value: BytesImpl<N>) -> Self {
         value.to_vec()