mod generated; use std::sync::Arc; use smol_str::SmolStr; use { SyntaxNode, SyntaxRoot, TreeRoot, SyntaxError, SyntaxKind::*, }; pub use self::generated::*; pub trait AstNode: Sized { fn cast(syntax: SyntaxNode) -> Option; fn syntax(&self) -> &SyntaxNode; } pub trait NameOwner: AstNode { fn name(&self) -> Option> { self.syntax() .children() .filter_map(Name::cast) .next() } } 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() } } impl FnDef { 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) -> SmolStr { let ident = self.syntax().first_child() .unwrap(); ident.leaf_text().unwrap() } } impl NameRef { pub fn text(&self) -> SmolStr { let ident = self.syntax().first_child() .unwrap(); ident.leaf_text().unwrap() } }