Merge pull request #116 from hug-dev/deref

Implement Deref(Mut) on bodies

Author: hug-dev

src/requests/request/request_body.rs

@@ -4,6 +4,7 @@ use crate::requests::Result;
 #[cfg(feature = "fuzz")]
 use arbitrary::Arbitrary;
 use std::io::{Read, Write};
+use std::ops::{Deref, DerefMut};
 use zeroize::Zeroize;
 
 /// Wrapper around the body of a request.
@@ -16,6 +17,20 @@ pub struct RequestBody {
     buffer: Vec<u8>,
 }
 
+impl Deref for RequestBody {
+    type Target = [u8];
+
+    fn deref(&self) -> &Self::Target {
+        &self.buffer
+    }
+}
+
+impl DerefMut for RequestBody {
+    fn deref_mut(&mut self) -> &mut Self::Target {
+        &mut self.buffer
+    }
+}
+
 impl RequestBody {
     /// Create a new, emtpy request body field.
     /// Available for testing only.

src/requests/response/response_body.rs

@@ -3,6 +3,7 @@
 use super::Result;
 use serde::{Deserialize, Serialize};
 use std::io::{Read, Write};
+use std::ops::{Deref, DerefMut};
 use zeroize::Zeroize;
 
 /// Wrapper around the body of a response.
@@ -14,6 +15,20 @@ pub struct ResponseBody {
     buffer: Vec<u8>,
 }
 
+impl Deref for ResponseBody {
+    type Target = [u8];
+
+    fn deref(&self) -> &Self::Target {
+        &self.buffer
+    }
+}
+
+impl DerefMut for ResponseBody {
+    fn deref_mut(&mut self) -> &mut Self::Target {
+        &mut self.buffer
+    }
+}
+
 impl ResponseBody {
     /// Create a new empty response body.
     pub(crate) fn new() -> ResponseBody {