ppatch/
pattern_skip_result_iterator.rs

1use crate::{PatternSearchType, Skipper};
2
3pub struct PatternSkipResultIterator<I, T> {
4    iter: I,
5    skipper: Skipper<T>,
6}
7
8impl<I, T> PatternSkipResultIterator<I, T>
9where
10    I: Iterator,
11{
12    pub fn new(iter: I, count: usize) -> Self {
13        Self {
14            iter,
15            skipper: Skipper::new(count),
16        }
17    }
18}
19
20impl<I, E, T> Iterator for PatternSkipResultIterator<I, T>
21where
22    I: Iterator<Item = Result<PatternSearchType<T>, E>>,
23    E: std::error::Error,
24{
25    type Item = Result<PatternSearchType<T>, E>;
26
27    fn next(&mut self) -> Option<Self::Item> {
28        if let Some(byte) = self.skipper.handle_existing_data() {
29            return Some(Ok(PatternSearchType::NonMatch(byte)));
30        }
31
32        match self.iter.next() {
33            Some(result) => match result {
34                Ok(search_type) => Some(Ok(self.skipper.handle_next(search_type))),
35                Err(error) => Some(Err(error)),
36            },
37            None => None,
38        }
39    }
40}
41
42pub trait PatternSkipResultExt<E, T>:
43    Iterator<Item = Result<PatternSearchType<T>, E>> + Sized
44where
45    E: std::error::Error,
46{
47    fn skip_pattern(self, count: usize) -> PatternSkipResultIterator<Self, T>;
48}
49
50impl<I, E, T> PatternSkipResultExt<E, T> for I
51where
52    I: Iterator<Item = Result<PatternSearchType<T>, E>>,
53    E: std::error::Error,
54{
55    fn skip_pattern(self, count: usize) -> PatternSkipResultIterator<Self, T> {
56        PatternSkipResultIterator::new(self, count)
57    }
58}
ppatch/pattern_skip_result_iterator.rs

ppatch/
pattern_skip_result_iterator.rs
