diff options
Diffstat (limited to 'crates/ra_ide_api_light')
-rw-r--r-- | crates/ra_ide_api_light/src/formatting.rs | 19 | ||||
-rw-r--r-- | crates/ra_ide_api_light/src/typing.rs | 40 |
2 files changed, 56 insertions, 3 deletions
diff --git a/crates/ra_ide_api_light/src/formatting.rs b/crates/ra_ide_api_light/src/formatting.rs index ca0fdb928..1f34b85d6 100644 --- a/crates/ra_ide_api_light/src/formatting.rs +++ b/crates/ra_ide_api_light/src/formatting.rs | |||
@@ -7,9 +7,22 @@ use ra_syntax::{ | |||
7 | 7 | ||
8 | /// If the node is on the beginning of the line, calculate indent. | 8 | /// If the node is on the beginning of the line, calculate indent. |
9 | pub(crate) fn leading_indent(node: &SyntaxNode) -> Option<&str> { | 9 | pub(crate) fn leading_indent(node: &SyntaxNode) -> Option<&str> { |
10 | let prev = prev_leaf(node)?; | 10 | for leaf in prev_leaves(node) { |
11 | let ws_text = ast::Whitespace::cast(prev)?.text(); | 11 | if let Some(ws) = ast::Whitespace::cast(leaf) { |
12 | ws_text.rfind('\n').map(|pos| &ws_text[pos + 1..]) | 12 | let ws_text = ws.text(); |
13 | if let Some(pos) = ws_text.rfind('\n') { | ||
14 | return Some(&ws_text[pos + 1..]); | ||
15 | } | ||
16 | } | ||
17 | if leaf.leaf_text().unwrap().contains('\n') { | ||
18 | break; | ||
19 | } | ||
20 | } | ||
21 | None | ||
22 | } | ||
23 | |||
24 | fn prev_leaves(node: &SyntaxNode) -> impl Iterator<Item = &SyntaxNode> { | ||
25 | generate(prev_leaf(node), |&node| prev_leaf(node)) | ||
13 | } | 26 | } |
14 | 27 | ||
15 | fn prev_leaf(node: &SyntaxNode) -> Option<&SyntaxNode> { | 28 | fn prev_leaf(node: &SyntaxNode) -> Option<&SyntaxNode> { |
diff --git a/crates/ra_ide_api_light/src/typing.rs b/crates/ra_ide_api_light/src/typing.rs index 5ff2b7c1f..861027b9f 100644 --- a/crates/ra_ide_api_light/src/typing.rs +++ b/crates/ra_ide_api_light/src/typing.rs | |||
@@ -296,6 +296,46 @@ fn foo() { | |||
296 | } | 296 | } |
297 | 297 | ||
298 | #[test] | 298 | #[test] |
299 | fn indents_middle_of_chain_call() { | ||
300 | type_dot( | ||
301 | r" | ||
302 | fn source_impl() { | ||
303 | let var = enum_defvariant_list().unwrap() | ||
304 | <|> | ||
305 | .nth(92) | ||
306 | .unwrap(); | ||
307 | } | ||
308 | ", | ||
309 | r" | ||
310 | fn source_impl() { | ||
311 | let var = enum_defvariant_list().unwrap() | ||
312 | . | ||
313 | .nth(92) | ||
314 | .unwrap(); | ||
315 | } | ||
316 | ", | ||
317 | ); | ||
318 | type_dot( | ||
319 | r" | ||
320 | fn source_impl() { | ||
321 | let var = enum_defvariant_list().unwrap() | ||
322 | <|> | ||
323 | .nth(92) | ||
324 | .unwrap(); | ||
325 | } | ||
326 | ", | ||
327 | r" | ||
328 | fn source_impl() { | ||
329 | let var = enum_defvariant_list().unwrap() | ||
330 | . | ||
331 | .nth(92) | ||
332 | .unwrap(); | ||
333 | } | ||
334 | ", | ||
335 | ); | ||
336 | } | ||
337 | |||
338 | #[test] | ||
299 | fn dont_indent_freestanding_dot() { | 339 | fn dont_indent_freestanding_dot() { |
300 | type_dot( | 340 | type_dot( |
301 | r" | 341 | r" |