oneone-plz 0.0.7

use decompression_plz::DecompressTrait;
use std::fmt::Display;

use body_plz::{
    reader::{
        chunked_reader::ChunkReaderState,
        content_length_reader::read_content_length,
    },
    variants::Body,
};
use buffer_plz::Event;
use bytes::BytesMut;
use header_plz::{
    OneHeader, OneInfoLine,
    body_headers::{parse::ParseBodyHeaders, transfer_types::TransferType},
    message_head::MessageHead,
};
use http_plz::OneOne;

use crate::error::{HttpStateError, IncorrectState};

#[derive(Debug, PartialEq)]
pub enum State<T>
where
    T: OneInfoLine + std::fmt::Debug,
{
    ReadMessageHead,
    ReadBodyContentLength(OneOne<T>, usize),
    ReadBodyContentLengthExtra(OneOne<T>),
    ReadBodyChunked(OneOne<T>, ChunkReaderState),
    ReadBodyChunkedExtra(OneOne<T>),
    ReadBodyClose(OneOne<T>),
    End(OneOne<T>),
    ReadBodyContentLengthExtraEnd(OneOne<T>, BytesMut),
    ReadBodyChunkedExtraEnd(OneOne<T>, BytesMut),
}

impl<T> State<T>
where
    T: OneInfoLine + std::fmt::Debug,
    MessageHead<T, OneHeader>: ParseBodyHeaders,
{
    #[allow(clippy::new_without_default)]
    pub fn new() -> State<T> {
        State::<T>::ReadMessageHead
    }

    /* Steps:
     *      1. Build OneOne from headers
     *
     *      2. Match body_headers
     *          a. None => End
     *          b. Some, match transfer_type
     *              1. ContentLength,
     *                  a. size == 0    => End
     *                  b. size != 0    => ReadBodyContentLength(size)
     *              2. Chunked          => ReadBodyChunked
     *              3. Close            => ReadBodyClose
     *              3. Unknown          => End
     *
     *      3. Default => End
     */

    #[allow(clippy::result_large_err)]
    fn build_oneone(headers: BytesMut) -> Result<Self, HttpStateError<T>> {
        let mut one = OneOne::try_from_message_head_buf(headers)?;
        let next_state = match one.body_headers() {
            None => Self::End(one),
            Some(body_headers) => match body_headers.transfer_type {
                Some(tt) => match tt {
                    TransferType::ContentLength(size) => {
                        if size == 0 {
                            Self::End(one)
                        } else {
                            Self::ReadBodyContentLength(one, size)
                        }
                    }
                    TransferType::Chunked => {
                        one.set_body(Body::Chunked(Vec::new()));
                        Self::ReadBodyChunked(one, ChunkReaderState::ReadSize)
                    }
                    TransferType::Close => Self::ReadBodyClose(one),
                },
                None => Self::End(one),
            },
        };
        Ok(next_state)
    }
}

impl<T> State<T>
where
    T: OneInfoLine + std::fmt::Debug,
    MessageHead<T, OneHeader>: ParseBodyHeaders,
{
    pub fn try_next(
        mut self,
        event: Event,
    ) -> Result<Self, HttpStateError<T>> {
        use State::*;
        match (self, event) {
            (ReadMessageHead, Event::Read(buf)) => {
                match MessageHead::is_complete(buf) {
                    true => {
                        let raw_headers = buf.split_at_current_pos();
                        self = State::build_oneone(raw_headers)?;
                        if buf.len() > 0 {
                            self = self.try_next(Event::Read(buf))?;
                        }
                        Ok(self)
                    }
                    false => Ok(Self::ReadMessageHead),
                }
            }
            (ReadMessageHead, Event::End(buf)) => {
                Err(HttpStateError::Unparsed(buf.into_inner()))?
            }
            (ReadBodyContentLength(mut oneone, mut size), mut event) => {
                match event {
                    Event::Read(ref mut buf) | Event::End(ref mut buf) => {
                        match read_content_length(buf, &mut size) {
                            true => {
                                oneone.set_body(Body::Raw(
                                    buf.split_at_current_pos(),
                                ));
                                if buf.len() > 0 {
                                    let next_state =
                                        ReadBodyContentLengthExtra(oneone);
                                    match event {
                                        Event::Read(_) => Ok(next_state),
                                        Event::End(_) => {
                                            next_state.try_next(event)
                                        }
                                    }
                                } else {
                                    Ok(End(oneone))
                                }
                            }
                            false => match event {
                                Event::Read(_) => {
                                    Ok(ReadBodyContentLength(oneone, size))
                                }
                                Event::End(buf) => {
                                    Err(HttpStateError::ContentLengthPartial(
                                        (oneone, buf.split_at_current_pos())
                                            .into(),
                                    ))?
                                }
                            },
                        }
                    }
                }
            }
            (ReadBodyContentLengthExtra(oneone), Event::Read(_)) => {
                Ok(ReadBodyContentLengthExtra(oneone))
            }
            (ReadBodyContentLengthExtra(oneone), Event::End(buf)) => {
                let extra_body = buf.into_inner();
                Ok(ReadBodyContentLengthExtraEnd(oneone, extra_body))
            }
            (
                ReadBodyChunked(mut oneone, mut chunk_state),
                Event::Read(buf),
            ) => loop {
                match chunk_state.next(buf) {
                    Some(chunk_to_add) => {
                        oneone
                            .body_as_mut()
                            .expect("no chunked body")
                            .push_chunk(chunk_to_add);
                        match chunk_state {
                            ChunkReaderState::LastChunk => {
                                chunk_state = if oneone.has_trailers() {
                                    ChunkReaderState::ReadTrailers
                                } else {
                                    ChunkReaderState::EndCRLF
                                };
                                continue;
                            }
                            ChunkReaderState::End => {
                                return if buf.len() > 0 {
                                    Ok(State::ReadBodyChunkedExtra(oneone))
                                } else {
                                    Ok(End(oneone))
                                };
                            }
                            ChunkReaderState::Failed(e) => {
                                return Err(e.into());
                            }
                            _ => continue,
                        }
                    }
                    None => {
                        return Ok(ReadBodyChunked(oneone, chunk_state));
                    }
                }
            },
            (ReadBodyChunked(oneone, _), Event::End(buf)) => {
                Err(HttpStateError::ChunkReaderPartial(
                    (oneone, buf.into_inner()).into(),
                ))
            }
            (ReadBodyChunkedExtra(oneone), Event::Read(_)) => {
                Ok(ReadBodyChunkedExtra(oneone))
            }
            (ReadBodyChunkedExtra(oneone), Event::End(buf)) => {
                let extra_body = buf.into_inner();
                Ok(ReadBodyChunkedExtraEnd(oneone, extra_body))
            }
            (ReadBodyClose(oneone), Event::Read(_)) => {
                Ok(Self::ReadBodyClose(oneone))
            }
            (ReadBodyClose(mut oneone), Event::End(buf)) => {
                oneone.set_body(Body::Raw(buf.into_inner()));
                Ok(End(oneone))
            }
            (End(mut oneone), event) => {
                self = match oneone.body() {
                    None => {
                        oneone.set_transfer_type_close();
                        ReadBodyClose(oneone)
                    }
                    Some(Body::Raw(_)) => ReadBodyContentLengthExtra(oneone),
                    Some(Body::Chunked(_)) => ReadBodyChunkedExtra(oneone),
                };
                match event {
                    Event::Read(_) => Ok(self),
                    Event::End(_) => self.try_next(event),
                }
            }
            (ReadBodyContentLengthExtraEnd(..), _)
            | (ReadBodyChunkedExtraEnd(..), _) => {
                unreachable!("not possible")
            }
        }
    }

    pub fn is_ended(&self) -> bool {
        matches!(self, Self::End(_))
            | matches!(self, State::ReadBodyContentLengthExtraEnd(..))
            | matches!(self, State::ReadBodyChunkedExtraEnd(..))
    }

    pub fn try_into_frame(self) -> Result<OneOne<T>, IncorrectState> {
        let one = match self {
            State::End(one) => one,
            State::ReadBodyContentLengthExtraEnd(mut one, extra)
            | State::ReadBodyChunkedExtraEnd(mut one, extra) => {
                one.set_extra_body(extra);
                one
            }
            _ => {
                return Err(IncorrectState(self.to_string()));
            }
        };
        Ok(one)
    }
}

impl<T> Display for State<T>
where
    T: OneInfoLine + std::fmt::Debug,
{
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            State::ReadMessageHead => write!(f, "ReadMessageHead"),
            State::ReadBodyContentLength(_, _) => {
                write!(f, "ReadBodyContentLength")
            }
            State::ReadBodyContentLengthExtra(_) => {
                write!(f, "ReadBodyContentLengthExtra")
            }
            State::ReadBodyChunked(_, _) => write!(f, "ReadBodyChunked"),
            State::ReadBodyChunkedExtra(_) => {
                write!(f, "ReadBodyChunkedExtra")
            }
            State::ReadBodyClose(_) => write!(f, "ReadBodyClose"),
            State::End(_) => write!(f, "End"),
            State::ReadBodyContentLengthExtraEnd(_, _) => {
                write!(f, "ReadBodyContentLengthExtraEnd")
            }
            State::ReadBodyChunkedExtraEnd(_, _) => {
                write!(f, "ReadBodyChunkedExtraEnd")
            }
        }
    }
}