aboutsummaryrefslogtreecommitdiff
path: root/crates/ra_syntax
diff options
context:
space:
mode:
Diffstat (limited to 'crates/ra_syntax')
-rw-r--r--crates/ra_syntax/src/algo.rs10
-rw-r--r--crates/ra_syntax/src/ptr.rs9
-rw-r--r--crates/ra_syntax/src/syntax_node.rs9
3 files changed, 10 insertions, 18 deletions
diff --git a/crates/ra_syntax/src/algo.rs b/crates/ra_syntax/src/algo.rs
index 06b45135c..1f68fe467 100644
--- a/crates/ra_syntax/src/algo.rs
+++ b/crates/ra_syntax/src/algo.rs
@@ -46,13 +46,3 @@ pub fn non_trivia_sibling(element: SyntaxElement, direction: Direction) -> Optio
46pub fn find_covering_element(root: &SyntaxNode, range: TextRange) -> SyntaxElement { 46pub fn find_covering_element(root: &SyntaxNode, range: TextRange) -> SyntaxElement {
47 root.0.covering_node(range).into() 47 root.0.covering_node(range).into()
48} 48}
49
50// Replace with `std::iter::successors` in `1.34.0`
51pub fn generate<T>(seed: Option<T>, step: impl Fn(&T) -> Option<T>) -> impl Iterator<Item = T> {
52 ::itertools::unfold(seed, move |slot| {
53 slot.take().map(|curr| {
54 *slot = step(&curr);
55 curr
56 })
57 })
58}
diff --git a/crates/ra_syntax/src/ptr.rs b/crates/ra_syntax/src/ptr.rs
index d8de1c4c1..15a8b94cd 100644
--- a/crates/ra_syntax/src/ptr.rs
+++ b/crates/ra_syntax/src/ptr.rs
@@ -1,8 +1,9 @@
1use std::marker::PhantomData; 1use std::{
2 2 marker::PhantomData,
3 iter::successors,
4};
3use crate::{ 5use crate::{
4 AstNode, SourceFile, SyntaxKind, SyntaxNode, TextRange, 6 AstNode, SourceFile, SyntaxKind, SyntaxNode, TextRange,
5 algo::generate,
6}; 7};
7 8
8/// A pointer to a syntax node inside a file. It can be used to remember a 9/// A pointer to a syntax node inside a file. It can be used to remember a
@@ -19,7 +20,7 @@ impl SyntaxNodePtr {
19 } 20 }
20 21
21 pub fn to_node(self, source_file: &SourceFile) -> &SyntaxNode { 22 pub fn to_node(self, source_file: &SourceFile) -> &SyntaxNode {
22 generate(Some(source_file.syntax()), |&node| { 23 successors(Some(source_file.syntax()), |&node| {
23 node.children().find(|it| self.range.is_subrange(&it.range())) 24 node.children().find(|it| self.range.is_subrange(&it.range()))
24 }) 25 })
25 .find(|it| it.range() == self.range && it.kind() == self.kind) 26 .find(|it| it.range() == self.range && it.kind() == self.kind)
diff --git a/crates/ra_syntax/src/syntax_node.rs b/crates/ra_syntax/src/syntax_node.rs
index 64d884287..dc2352c76 100644
--- a/crates/ra_syntax/src/syntax_node.rs
+++ b/crates/ra_syntax/src/syntax_node.rs
@@ -10,6 +10,7 @@ use std::{
10 fmt::{self, Write}, 10 fmt::{self, Write},
11 any::Any, 11 any::Any,
12 borrow::Borrow, 12 borrow::Borrow,
13 iter::successors,
13}; 14};
14 15
15use ra_parser::ParseError; 16use ra_parser::ParseError;
@@ -195,7 +196,7 @@ impl SyntaxNode {
195 } 196 }
196 197
197 pub fn ancestors(&self) -> impl Iterator<Item = &SyntaxNode> { 198 pub fn ancestors(&self) -> impl Iterator<Item = &SyntaxNode> {
198 crate::algo::generate(Some(self), |&node| node.parent()) 199 successors(Some(self), |&node| node.parent())
199 } 200 }
200 201
201 pub fn descendants(&self) -> impl Iterator<Item = &SyntaxNode> { 202 pub fn descendants(&self) -> impl Iterator<Item = &SyntaxNode> {
@@ -213,7 +214,7 @@ impl SyntaxNode {
213 } 214 }
214 215
215 pub fn siblings(&self, direction: Direction) -> impl Iterator<Item = &SyntaxNode> { 216 pub fn siblings(&self, direction: Direction) -> impl Iterator<Item = &SyntaxNode> {
216 crate::algo::generate(Some(self), move |&node| match direction { 217 successors(Some(self), move |&node| match direction {
217 Direction::Next => node.next_sibling(), 218 Direction::Next => node.next_sibling(),
218 Direction::Prev => node.prev_sibling(), 219 Direction::Prev => node.prev_sibling(),
219 }) 220 })
@@ -224,7 +225,7 @@ impl SyntaxNode {
224 direction: Direction, 225 direction: Direction,
225 ) -> impl Iterator<Item = SyntaxElement> { 226 ) -> impl Iterator<Item = SyntaxElement> {
226 let me: SyntaxElement = self.into(); 227 let me: SyntaxElement = self.into();
227 crate::algo::generate(Some(me), move |el| match direction { 228 successors(Some(me), move |el| match direction {
228 Direction::Next => el.next_sibling_or_token(), 229 Direction::Next => el.next_sibling_or_token(),
229 Direction::Prev => el.prev_sibling_or_token(), 230 Direction::Prev => el.prev_sibling_or_token(),
230 }) 231 })
@@ -373,7 +374,7 @@ impl<'a> SyntaxToken<'a> {
373 direction: Direction, 374 direction: Direction,
374 ) -> impl Iterator<Item = SyntaxElement<'a>> { 375 ) -> impl Iterator<Item = SyntaxElement<'a>> {
375 let me: SyntaxElement = (*self).into(); 376 let me: SyntaxElement = (*self).into();
376 crate::algo::generate(Some(me), move |el| match direction { 377 successors(Some(me), move |el| match direction {
377 Direction::Next => el.next_sibling_or_token(), 378 Direction::Next => el.next_sibling_or_token(),
378 Direction::Prev => el.prev_sibling_or_token(), 379 Direction::Prev => el.prev_sibling_or_token(),
379 }) 380 })