feat(decompression-plz): added main decode logic to decompress function

Author: darkseid-is

decompression-plz/src/lib.rs

@@ -7,9 +7,10 @@ use crate::{
     content_length::add_body_and_update_cl,
     decompression::{
         multi::error::{MultiDecompressError, MultiDecompressErrorReason},
-        state::runner,
+        state::decompression_runner,
     },
     dtraits::DecompressTrait,
+    state::DecodeState,
 };
 mod content_length;
 mod decode_struct;
@@ -21,16 +22,23 @@ pub fn decompress<T>(
     mut message: T,
     mut extra_body: Option<BytesMut>,
     buf: &mut BytesMut,
-) -> Result<(), MultiDecompressError>
+)
+//-> Result<(), MultiDecompressError>
 where
     T: DecompressTrait,
 {
     let mut body = message.get_body().into_bytes().unwrap();
     let mut body_headers = message.body_headers_as_mut().take();
 
-    //
-    add_body_and_update_cl(&mut message, body);
-    Ok(())
+    let mut state = DecodeState::init(&mut message, extra_body, buf);
+    loop {
+        state = state.try_next();
+        if state.is_ended() {
+            break;
+        }
+    }
+
+    // Ok(())
 }
 
 // helper function for tests

decompression-plz/src/state.rs

@@ -35,7 +35,7 @@ where
         Self::Start(decode_struct)
     }
 
-    fn try_next(self) -> Self {
+    pub fn try_next(self) -> Self {
         match self {
             DecodeState::Start(mut decode_struct) => {
                 if decode_struct.transfer_encoding_is_some() {
@@ -91,14 +91,14 @@ where
                 if let Some(extra) = decode_struct.take_extra_body() {
                     body.unsplit(extra);
                 }
-                add_body_and_update_cl(&mut decode_struct.message, body);
+                add_body_and_update_cl(decode_struct.message, body);
                 Self::End
             }
             DecodeState::End => Self::End,
         }
     }
 
-    fn is_ended(&self) -> bool {
+    pub fn is_ended(&self) -> bool {
         matches!(self, DecodeState::End)
     }
 }
@@ -110,7 +110,7 @@ fn apply_encoding<T>(
 where
     T: DecompressTrait,
 {
-    match runner(
+    match decompression_runner(
         &decode_struct.body,
         decode_struct.extra_body.as_deref(),
         encoding_info,