bluejay_parser/ast/executable/
selection_set.rs

1use crate::ast::executable::Selection;
2use crate::ast::{DepthLimiter, FromTokens, IsMatch, ParseError, Tokens};
3use crate::lexical_token::PunctuatorType;
4use crate::{HasSpan, Span};
5use bluejay_core::AsIter;
6
7#[derive(Debug)]
8pub struct SelectionSet<'a> {
9    selections: Vec<Selection<'a>>,
10    span: Span,
11}
12
13impl<'a> FromTokens<'a> for SelectionSet<'a> {
14    #[inline]
15    fn from_tokens(
16        tokens: &mut impl Tokens<'a>,
17        depth_limiter: DepthLimiter,
18    ) -> Result<Self, ParseError> {
19        let open_span = tokens.expect_punctuator(PunctuatorType::OpenBrace)?;
20        let mut selections: Vec<Selection> = Vec::new();
21        let close_span = loop {
22            selections.push(Selection::from_tokens(tokens, depth_limiter.bump()?)?);
23            if let Some(close_span) = tokens.next_if_punctuator(PunctuatorType::CloseBrace) {
24                break close_span;
25            }
26        };
27        let span = open_span.merge(&close_span);
28        Ok(Self { selections, span })
29    }
30}
31
32impl<'a> IsMatch<'a> for SelectionSet<'a> {
33    #[inline]
34    fn is_match(tokens: &mut impl Tokens<'a>) -> bool {
35        tokens.peek_punctuator_matches(0, PunctuatorType::OpenBrace)
36    }
37}
38
39impl bluejay_core::Indexable for SelectionSet<'_> {
40    type Id = Span;
41
42    fn id(&self) -> &Self::Id {
43        &self.span
44    }
45}
46
47impl<'a> bluejay_core::executable::SelectionSet for SelectionSet<'a> {
48    type Selection = Selection<'a>;
49}
50
51impl<'a> AsIter for SelectionSet<'a> {
52    type Item = Selection<'a>;
53    type Iterator<'b>
54        = std::slice::Iter<'b, Self::Item>
55    where
56        'a: 'b;
57
58    fn iter(&self) -> Self::Iterator<'_> {
59        self.selections.iter()
60    }
61
62    fn len(&self) -> usize {
63        self.selections.len()
64    }
65}
66
67impl HasSpan for SelectionSet<'_> {
68    fn span(&self) -> &Span {
69        &self.span
70    }
71}
bluejay_parser/ast/executable/selection_set.rs

bluejay_parser/ast/executable/
selection_set.rs
