use std::{fmt, sync::Arc}; use smol_str::SmolStr; use { yellow::{RedNode, TreeRoot, SyntaxRoot, RedPtr}, SyntaxKind::{self, *}, TextRange, TextUnit, }; #[derive(Clone, Copy)] pub struct SyntaxNode> { pub(crate) root: R, // Guaranteed to not dangle, because `root` holds a // strong reference to red's ancestor red: RedPtr, } unsafe impl Send for SyntaxNode {} unsafe impl Sync for SyntaxNode {} impl PartialEq> for SyntaxNode { fn eq(&self, other: &SyntaxNode) -> bool { self.red == other.red } } impl Eq for SyntaxNode {} pub type SyntaxNodeRef<'a> = SyntaxNode<&'a SyntaxRoot>; #[test] fn syntax_node_ref_is_copy() { fn assert_copy(){} assert_copy::() } #[derive(Debug, Clone, PartialEq, Eq, Hash, Ord, PartialOrd)] pub struct SyntaxError { pub msg: String, pub offset: TextUnit, } impl SyntaxNode> { pub(crate) fn new_owned(root: SyntaxRoot) -> Self { let root = Arc::new(root); let red = RedPtr::new(&root.red); SyntaxNode { root, red } } } impl SyntaxNode { pub fn as_ref<'a>(&'a self) -> SyntaxNode<&'a SyntaxRoot> { SyntaxNode { root: &*self.root, red: self.red, } } pub fn kind(&self) -> SyntaxKind { self.red().green().kind() } pub fn range(&self) -> TextRange { let red = self.red(); TextRange::offset_len(red.start_offset(), red.green().text_len()) } pub fn text(&self) -> String { self.red().green().text() } pub fn children<'a>(&'a self) -> impl Iterator> + 'a { let red = self.red(); let n_children = red.n_children(); (0..n_children).map(move |i| SyntaxNode { root: self.root.clone(), red: red.get_child(i).unwrap(), }) } pub fn parent(&self) -> Option> { let parent = self.red().parent()?; Some(SyntaxNode { root: self.root.clone(), red: parent, }) } pub fn first_child(&self) -> Option> { self.children().next() } pub fn next_sibling(&self) -> Option> { let red = self.red(); let parent = self.parent()?; let next_sibling_idx = red.index_in_parent()? + 1; let sibling_red = parent.red().get_child(next_sibling_idx)?; Some(SyntaxNode { root: self.root.clone(), red: sibling_red, }) } pub fn prev_sibling(&self) -> Option> { let red = self.red(); let parent = self.parent()?; let prev_sibling_idx = red.index_in_parent()?.checked_sub(1)?; let sibling_red = parent.red().get_child(prev_sibling_idx)?; Some(SyntaxNode { root: self.root.clone(), red: sibling_red, }) } pub fn is_leaf(&self) -> bool { self.first_child().is_none() } pub fn leaf_text(&self) -> Option { self.red().green().leaf_text() } fn red(&self) -> &RedNode { unsafe { self.red.get(&self.root) } } } impl fmt::Debug for SyntaxNode { fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result { write!(fmt, "{:?}@{:?}", self.kind(), self.range())?; if has_short_text(self.kind()) { write!(fmt, " \"{}\"", self.text())?; } Ok(()) } } fn has_short_text(kind: SyntaxKind) -> bool { match kind { IDENT | LIFETIME | INT_NUMBER | FLOAT_NUMBER => true, _ => false, } }