mod generated; use std::sync::Arc; use { SyntaxNode, SyntaxRoot, TreeRoot, SyntaxError, SyntaxKind::*, }; pub use self::generated::*; pub trait AstNode: Sized { fn cast(syntax: SyntaxNode) -> Option; fn syntax(&self) -> &SyntaxNode; } impl File> { pub fn parse(text: &str) -> Self { File::cast(::parse(text)).unwrap() } } impl File { pub fn errors(&self) -> Vec { self.syntax().root.errors.clone() } pub fn functions<'a>(&'a self) -> impl Iterator> + 'a { self.syntax() .children() .filter_map(Function::cast) } } impl Function { pub fn name(&self) -> Option> { self.syntax() .children() .filter_map(Name::cast) .next() } pub fn has_atom_attr(&self, atom: &str) -> bool { self.syntax() .children() .filter(|node| node.kind() == ATTR) .any(|attr| { let mut metas = attr.children().filter(|node| node.kind() == META_ITEM); let meta = match metas.next() { None => return false, Some(meta) => { if metas.next().is_some() { return false; } meta } }; let mut children = meta.children(); match children.next() { None => false, Some(child) => { if children.next().is_some() { return false; } child.kind() == IDENT && child.text() == atom } } }) } } impl Name { pub fn text(&self) -> String { self.syntax().text() } }