cairo_lang_syntax/node/
element_list.rs

1use std::marker::PhantomData;
2
3use super::SyntaxGroup;
4use crate::node::{SyntaxNode, TypedSyntaxNode};
5
6// A typed view of an element list node.
7// STEP=1 means a sequence of elements (e.g. sequence of trivia elements).
8// STEP=2 means a separated sequence (e.g. argument list separated by `,`).
9#[derive(Clone, Debug, Eq, Hash, PartialEq)]
10pub struct ElementList<T: TypedSyntaxNode, const STEP: usize> {
11    pub node: SyntaxNode,
12    phantom: PhantomData<T>,
13}
14impl<T: TypedSyntaxNode, const STEP: usize> ElementList<T, STEP> {
15    pub fn new(node: SyntaxNode) -> Self {
16        Self { node, phantom: PhantomData {} }
17    }
18    pub fn elements(&self, db: &dyn SyntaxGroup) -> Vec<T> {
19        db.get_children(self.node.clone())
20            .iter()
21            .step_by(STEP)
22            .map(|x| T::from_syntax_node(db, x.clone()))
23            .collect()
24    }
25    pub fn has_tail(&self, db: &dyn SyntaxGroup) -> bool {
26        db.get_children(self.node.clone()).iter().len() % STEP != 0
27    }
28}