parsing.rs

use amq_protocol::frame::parsing::traits::*;
use std::{
    iter::{Chain, Cloned, Enumerate},
    ops::RangeFrom,
    slice::Iter,
};

#[derive(Debug, PartialEq)]
pub(crate) struct ParsingContext<'a> {
    buffers: [&'a [u8]; 2],
}

impl<'a> From<&'a [u8]> for ParsingContext<'a> {
    fn from(buffer: &'a [u8]) -> Self {
        Self {
            buffers: [buffer, &buffer[buffer.len()..]],
        }
    }
}

impl<'a> From<[&'a [u8]; 2]> for ParsingContext<'a> {
    fn from(buffers: [&'a [u8]; 2]) -> Self {
        Self { buffers }
    }
}

impl<'a> Clone for ParsingContext<'a> {
    fn clone(&self) -> Self {
        [&self.buffers[0][..], &self.buffers[1][..]].into()
    }
}

impl<'a> ParsingContext<'a> {
    pub(crate) fn as_ptr(&self) -> *const u8 {
        self.buffers[0].as_ptr()
    }

    fn iter(&self) -> Chain<Iter<'a, u8>, Iter<'a, u8>> {
        self.buffers[0].iter().chain(self.buffers[1].iter())
    }
}

impl<'a> InputIter for ParsingContext<'a> {
    type Item = u8;
    type Iter = Enumerate<Self::IterElem>;
    type IterElem = Cloned<Chain<Iter<'a, u8>, Iter<'a, u8>>>;

    #[inline]
    fn iter_indices(&self) -> Self::Iter {
        self.iter_elements().enumerate()
    }

    #[inline]
    fn iter_elements(&self) -> Self::IterElem {
        self.iter().cloned()
    }

    #[inline]
    fn position<P>(&self, predicate: P) -> Option<usize>
    where
        P: Fn(Self::Item) -> bool,
    {
        self.iter().position(|b| predicate(*b))
    }

    #[inline]
    fn slice_index(&self, count: usize) -> Result<usize, Needed> {
        if self.input_len() >= count {
            Ok(count)
        } else {
            Err(Needed::new(count - self.input_len()))
        }
    }
}

impl<'a> InputLength for ParsingContext<'a> {
    #[inline]
    fn input_len(&self) -> usize {
        self.buffers.iter().map(|buf| buf.len()).sum()
    }
}

impl<'a> InputTake for ParsingContext<'a> {
    #[inline]
    fn take(&self, count: usize) -> Self {
        if self.buffers[0].len() > count {
            self.buffers[0][..count].into()
        } else {
            let needed = count - self.buffers[0].len();
            [&self.buffers[0][..], &self.buffers[1][..needed]].into()
        }
    }

    #[inline]
    fn take_split(&self, count: usize) -> (Self, Self) {
        if self.buffers[0].len() > count {
            (
                [&self.buffers[0][count..], &self.buffers[1][..]].into(),
                self.buffers[0][..count].into(),
            )
        } else {
            let needed = count - self.buffers[0].len();
            (
                self.buffers[1][needed..].into(),
                [&self.buffers[0][..], &self.buffers[1][..needed]].into(),
            )
        }
    }
}

impl<'a> Slice<RangeFrom<usize>> for ParsingContext<'a> {
    #[inline]
    fn slice(&self, range: RangeFrom<usize>) -> Self {
        if range.start < self.buffers[0].len() {
            [&self.buffers[0][range.start..], &self.buffers[1][..]].into()
        } else {
            let needed = range.start - self.buffers[0].len();
            self.buffers[1][needed..].into()
        }
    }
}

impl<'a> UnspecializedInput for ParsingContext<'a> {}