spinoso_regexp/regexp/regex/utf8/
iter.rs

1use core::iter::{Enumerate, FusedIterator};
2use core::ops::Range;
3
4use regex::CaptureNames;
5
6#[derive(Debug)]
7#[must_use = "this `Captures` is an `Iterator`, which should be consumed if constructed"]
8pub struct Captures<'a> {
9    captures: regex::Captures<'a>,
10    iter: Range<usize>,
11}
12
13impl<'a> From<regex::Captures<'a>> for Captures<'a> {
14    fn from(captures: regex::Captures<'a>) -> Self {
15        let iter = 0..captures.len();
16        Self { captures, iter }
17    }
18}
19
20impl<'a> Iterator for Captures<'a> {
21    type Item = Option<&'a [u8]>;
22
23    fn next(&mut self) -> Option<Self::Item> {
24        let idx = self.iter.next()?;
25        match self.captures.get(idx) {
26            Some(capture) => Some(Some(capture.as_str().as_bytes())),
27            None => Some(None),
28        }
29    }
30
31    fn nth(&mut self, n: usize) -> Option<Self::Item> {
32        let idx = self.iter.nth(n)?;
33        match self.captures.get(idx) {
34            Some(capture) => Some(Some(capture.as_str().as_bytes())),
35            None => Some(None),
36        }
37    }
38
39    fn count(self) -> usize {
40        self.iter.count()
41    }
42}
43
44impl FusedIterator for Captures<'_> {}
45
46#[derive(Debug)]
47#[must_use = "this `CaptureIndices` is an `Iterator`, which should be consumed if constructed"]
48pub struct CaptureIndices<'a, 'b> {
49    name: &'b [u8],
50    capture_names: Enumerate<CaptureNames<'a>>,
51}
52
53impl<'a, 'b> CaptureIndices<'a, 'b> {
54    pub(crate) fn with_name_and_iter(name: &'b [u8], iter: CaptureNames<'a>) -> Self {
55        Self {
56            name,
57            capture_names: iter.enumerate(),
58        }
59    }
60
61    /// The name of the capture group this iterator targets.
62    pub const fn name(&self) -> &'b [u8] {
63        self.name
64    }
65}
66
67impl Iterator for CaptureIndices<'_, '_> {
68    type Item = usize;
69
70    fn next(&mut self) -> Option<Self::Item> {
71        for (index, group) in self.capture_names.by_ref() {
72            let group = group.map(str::as_bytes);
73            if matches!(group, Some(group) if group == self.name) {
74                return Some(index);
75            }
76        }
77        None
78    }
79}
80
81impl FusedIterator for CaptureIndices<'_, '_> {}
spinoso_regexp/regexp/regex/utf8/iter.rs

spinoso_regexp/regexp/regex/utf8/
iter.rs
