language-servers/rust-analyzer
1
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer.git synced 2025-08-30 23:27:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Borrow text from nodes of immutable syntax trees

Browse source
This commit is contained in:
Dawer 2021-05-06 10:07:06 +05:00
parent dc4fa504ea
commit d7e169fe55
2 changed files with 49 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

36
crates/syntax/src/ast/node_ext.rs
View file

@ -1,10 +1,11 @@
//! Various extension methods to ast Nodes, which are hard to code-generate.
//! Extensions for various expressions live in a sibling `expr_extensions` module.
use std::{fmt, iter::successors};
use std::{borrow::Cow, fmt, iter::successors};
use itertools::Itertools;
use parser::SyntaxKind;
use rowan::{GreenNodeData, GreenTokenData, NodeOrToken};
use crate::{
ast::{self, support, AstNode, AstToken, AttrsOwner, NameOwner, SyntaxNode},
@ -12,19 +13,19 @@ use crate::{
};
impl ast::Lifetime {
pub fn text(&self) -> TokenText {
pub fn text(&self) -> TokenText<'_> {
text_of_first_token(self.syntax())
}
}
impl ast::Name {
pub fn text(&self) -> TokenText {
pub fn text(&self) -> TokenText<'_> {
text_of_first_token(self.syntax())
}
}
impl ast::NameRef {
pub fn text(&self) -> TokenText {
pub fn text(&self) -> TokenText<'_> {
text_of_first_token(self.syntax())
}
@ -33,11 +34,28 @@ impl ast::NameRef {
}
}
fn text_of_first_token(node: &SyntaxNode) -> TokenText {
let first_token =
node.green().children().next().and_then(|it| it.into_token()).unwrap().to_owned();
fn _text_of_first_token(node: &SyntaxNode) -> Cow<'_, str> {
fn cow_map<F: FnOnce(&GreenNodeData) -> &str>(green: Cow<GreenNodeData>, f: F) -> Cow<str> {
match green {
Cow::Borrowed(green_ref) => Cow::Borrowed(f(green_ref)),
Cow::Owned(green) => Cow::Owned(f(&green).to_owned()),
}
}
TokenText(first_token)
cow_map(node.green(), |green_ref| {
green_ref.children().next().and_then(NodeOrToken::into_token).unwrap().text()
})
}
fn text_of_first_token(node: &SyntaxNode) -> TokenText<'_> {
fn first_token(green_ref: &GreenNodeData) -> &GreenTokenData {
green_ref.children().next().and_then(NodeOrToken::into_token).unwrap()
}
match node.green() {
Cow::Borrowed(green_ref) => TokenText::Borrowed(first_token(green_ref).text()),
Cow::Owned(green) => TokenText::Owned(first_token(&green).to_owned()),
}
}
#[derive(Debug, PartialEq, Eq, Clone)]
@ -412,7 +430,7 @@ impl fmt::Display for NameOrNameRef {
}
impl NameOrNameRef {
pub fn text(&self) -> TokenText {
pub fn text(&self) -> TokenText<'_> {
match self {
NameOrNameRef::Name(name) => name.text(),
NameOrNameRef::NameRef(name_ref) => name_ref.text(),