diff options
Diffstat (limited to 'crates/ra_editor')
-rw-r--r-- | crates/ra_editor/src/extend_selection.rs | 12 | ||||
-rw-r--r-- | crates/ra_editor/src/lib.rs | 66 |
2 files changed, 8 insertions, 70 deletions
diff --git a/crates/ra_editor/src/extend_selection.rs b/crates/ra_editor/src/extend_selection.rs index 4665a336a..bf0727dde 100644 --- a/crates/ra_editor/src/extend_selection.rs +++ b/crates/ra_editor/src/extend_selection.rs | |||
@@ -1,16 +1,11 @@ | |||
1 | use ra_syntax::{ | 1 | use ra_syntax::{ |
2 | algo::{find_covering_node, find_leaf_at_offset, LeafAtOffset}, | 2 | algo::{find_covering_node, find_leaf_at_offset, LeafAtOffset}, |
3 | Direction, SourceFileNode, | 3 | Direction, |
4 | SyntaxKind::*, | 4 | SyntaxKind::*, |
5 | SyntaxNodeRef, TextRange, TextUnit, | 5 | SyntaxNodeRef, TextRange, TextUnit, |
6 | }; | 6 | }; |
7 | 7 | ||
8 | pub fn extend_selection(file: &SourceFileNode, range: TextRange) -> Option<TextRange> { | 8 | pub fn extend_selection(root: SyntaxNodeRef, range: TextRange) -> Option<TextRange> { |
9 | let syntax = file.syntax(); | ||
10 | extend(syntax.borrowed(), range) | ||
11 | } | ||
12 | |||
13 | pub(crate) fn extend(root: SyntaxNodeRef, range: TextRange) -> Option<TextRange> { | ||
14 | if range.is_empty() { | 9 | if range.is_empty() { |
15 | let offset = range.start(); | 10 | let offset = range.start(); |
16 | let mut leaves = find_leaf_at_offset(root, offset); | 11 | let mut leaves = find_leaf_at_offset(root, offset); |
@@ -126,6 +121,7 @@ fn adj_comments(node: SyntaxNodeRef, dir: Direction) -> SyntaxNodeRef { | |||
126 | #[cfg(test)] | 121 | #[cfg(test)] |
127 | mod tests { | 122 | mod tests { |
128 | use super::*; | 123 | use super::*; |
124 | use ra_syntax::SourceFileNode; | ||
129 | use test_utils::extract_offset; | 125 | use test_utils::extract_offset; |
130 | 126 | ||
131 | fn do_check(before: &str, afters: &[&str]) { | 127 | fn do_check(before: &str, afters: &[&str]) { |
@@ -133,7 +129,7 @@ mod tests { | |||
133 | let file = SourceFileNode::parse(&before); | 129 | let file = SourceFileNode::parse(&before); |
134 | let mut range = TextRange::offset_len(cursor, 0.into()); | 130 | let mut range = TextRange::offset_len(cursor, 0.into()); |
135 | for &after in afters { | 131 | for &after in afters { |
136 | range = extend_selection(&file, range).unwrap(); | 132 | range = extend_selection(file.syntax(), range).unwrap(); |
137 | let actual = &before[range]; | 133 | let actual = &before[range]; |
138 | assert_eq!(after, actual); | 134 | assert_eq!(after, actual); |
139 | } | 135 | } |
diff --git a/crates/ra_editor/src/lib.rs b/crates/ra_editor/src/lib.rs index a65637d52..b03f9ea54 100644 --- a/crates/ra_editor/src/lib.rs +++ b/crates/ra_editor/src/lib.rs | |||
@@ -22,7 +22,7 @@ pub use self::{ | |||
22 | use ra_text_edit::{TextEdit, TextEditBuilder}; | 22 | use ra_text_edit::{TextEdit, TextEditBuilder}; |
23 | use ra_syntax::{ | 23 | use ra_syntax::{ |
24 | algo::find_leaf_at_offset, | 24 | algo::find_leaf_at_offset, |
25 | ast::{self, AstNode, NameOwner}, | 25 | ast::{self, AstNode}, |
26 | SourceFileNode, | 26 | SourceFileNode, |
27 | SyntaxKind::{self, *}, | 27 | SyntaxKind::{self, *}, |
28 | SyntaxNodeRef, TextRange, TextUnit, Direction, | 28 | SyntaxNodeRef, TextRange, TextUnit, Direction, |
@@ -49,18 +49,6 @@ pub struct Diagnostic { | |||
49 | pub fix: Option<LocalEdit>, | 49 | pub fix: Option<LocalEdit>, |
50 | } | 50 | } |
51 | 51 | ||
52 | #[derive(Debug)] | ||
53 | pub struct Runnable { | ||
54 | pub range: TextRange, | ||
55 | pub kind: RunnableKind, | ||
56 | } | ||
57 | |||
58 | #[derive(Debug)] | ||
59 | pub enum RunnableKind { | ||
60 | Test { name: String }, | ||
61 | Bin, | ||
62 | } | ||
63 | |||
64 | pub fn matching_brace(file: &SourceFileNode, offset: TextUnit) -> Option<TextUnit> { | 52 | pub fn matching_brace(file: &SourceFileNode, offset: TextUnit) -> Option<TextUnit> { |
65 | const BRACES: &[SyntaxKind] = &[ | 53 | const BRACES: &[SyntaxKind] = &[ |
66 | L_CURLY, R_CURLY, L_BRACK, R_BRACK, L_PAREN, R_PAREN, L_ANGLE, R_ANGLE, | 54 | L_CURLY, R_CURLY, L_BRACK, R_BRACK, L_PAREN, R_PAREN, L_ANGLE, R_ANGLE, |
@@ -79,11 +67,11 @@ pub fn matching_brace(file: &SourceFileNode, offset: TextUnit) -> Option<TextUni | |||
79 | Some(matching_node.range().start()) | 67 | Some(matching_node.range().start()) |
80 | } | 68 | } |
81 | 69 | ||
82 | pub fn highlight(file: &SourceFileNode) -> Vec<HighlightedRange> { | 70 | pub fn highlight(root: SyntaxNodeRef) -> Vec<HighlightedRange> { |
83 | // Visited nodes to handle highlighting priorities | 71 | // Visited nodes to handle highlighting priorities |
84 | let mut highlighted = FxHashSet::default(); | 72 | let mut highlighted = FxHashSet::default(); |
85 | let mut res = Vec::new(); | 73 | let mut res = Vec::new(); |
86 | for node in file.syntax().descendants() { | 74 | for node in root.descendants() { |
87 | if highlighted.contains(&node) { | 75 | if highlighted.contains(&node) { |
88 | continue; | 76 | continue; |
89 | } | 77 | } |
@@ -133,29 +121,6 @@ pub fn syntax_tree(file: &SourceFileNode) -> String { | |||
133 | ::ra_syntax::utils::dump_tree(file.syntax()) | 121 | ::ra_syntax::utils::dump_tree(file.syntax()) |
134 | } | 122 | } |
135 | 123 | ||
136 | pub fn runnables(file: &SourceFileNode) -> Vec<Runnable> { | ||
137 | file.syntax() | ||
138 | .descendants() | ||
139 | .filter_map(ast::FnDef::cast) | ||
140 | .filter_map(|f| { | ||
141 | let name = f.name()?.text(); | ||
142 | let kind = if name == "main" { | ||
143 | RunnableKind::Bin | ||
144 | } else if f.has_atom_attr("test") { | ||
145 | RunnableKind::Test { | ||
146 | name: name.to_string(), | ||
147 | } | ||
148 | } else { | ||
149 | return None; | ||
150 | }; | ||
151 | Some(Runnable { | ||
152 | range: f.syntax().range(), | ||
153 | kind, | ||
154 | }) | ||
155 | }) | ||
156 | .collect() | ||
157 | } | ||
158 | |||
159 | pub fn find_node_at_offset<'a, N: AstNode<'a>>( | 124 | pub fn find_node_at_offset<'a, N: AstNode<'a>>( |
160 | syntax: SyntaxNodeRef<'a>, | 125 | syntax: SyntaxNodeRef<'a>, |
161 | offset: TextUnit, | 126 | offset: TextUnit, |
@@ -178,7 +143,7 @@ fn main() {} | |||
178 | println!("Hello, {}!", 92); | 143 | println!("Hello, {}!", 92); |
179 | "#, | 144 | "#, |
180 | ); | 145 | ); |
181 | let hls = highlight(&file); | 146 | let hls = highlight(file.syntax()); |
182 | assert_eq_dbg( | 147 | assert_eq_dbg( |
183 | r#"[HighlightedRange { range: [1; 11), tag: "comment" }, | 148 | r#"[HighlightedRange { range: [1; 11), tag: "comment" }, |
184 | HighlightedRange { range: [12; 14), tag: "keyword" }, | 149 | HighlightedRange { range: [12; 14), tag: "keyword" }, |
@@ -191,29 +156,6 @@ fn main() {} | |||
191 | } | 156 | } |
192 | 157 | ||
193 | #[test] | 158 | #[test] |
194 | fn test_runnables() { | ||
195 | let file = SourceFileNode::parse( | ||
196 | r#" | ||
197 | fn main() {} | ||
198 | |||
199 | #[test] | ||
200 | fn test_foo() {} | ||
201 | |||
202 | #[test] | ||
203 | #[ignore] | ||
204 | fn test_foo() {} | ||
205 | "#, | ||
206 | ); | ||
207 | let runnables = runnables(&file); | ||
208 | assert_eq_dbg( | ||
209 | r#"[Runnable { range: [1; 13), kind: Bin }, | ||
210 | Runnable { range: [15; 39), kind: Test { name: "test_foo" } }, | ||
211 | Runnable { range: [41; 75), kind: Test { name: "test_foo" } }]"#, | ||
212 | &runnables, | ||
213 | ) | ||
214 | } | ||
215 | |||
216 | #[test] | ||
217 | fn test_matching_brace() { | 159 | fn test_matching_brace() { |
218 | fn do_check(before: &str, after: &str) { | 160 | fn do_check(before: &str, after: &str) { |
219 | let (pos, before) = extract_offset(before); | 161 | let (pos, before) = extract_offset(before); |