diff options
Diffstat (limited to 'crates/ra_assists')
-rw-r--r-- | crates/ra_assists/src/assists/raw_string.rs | 62 | ||||
-rw-r--r-- | crates/ra_assists/src/assists/remove_dbg.rs | 20 | ||||
-rw-r--r-- | crates/ra_assists/src/assists/replace_if_let_with_match.rs | 28 | ||||
-rw-r--r-- | crates/ra_assists/src/assists/split_import.rs | 13 | ||||
-rw-r--r-- | crates/ra_assists/src/doc_tests/generated.rs | 126 |
5 files changed, 240 insertions, 9 deletions
diff --git a/crates/ra_assists/src/assists/raw_string.rs b/crates/ra_assists/src/assists/raw_string.rs index ea756d1ca..2df48a838 100644 --- a/crates/ra_assists/src/assists/raw_string.rs +++ b/crates/ra_assists/src/assists/raw_string.rs | |||
@@ -1,5 +1,3 @@ | |||
1 | //! FIXME: write short doc here | ||
2 | |||
3 | use hir::db::HirDatabase; | 1 | use hir::db::HirDatabase; |
4 | use ra_syntax::{ | 2 | use ra_syntax::{ |
5 | SyntaxKind::{RAW_STRING, STRING}, | 3 | SyntaxKind::{RAW_STRING, STRING}, |
@@ -9,6 +7,21 @@ use rustc_lexer; | |||
9 | 7 | ||
10 | use crate::{Assist, AssistCtx, AssistId}; | 8 | use crate::{Assist, AssistCtx, AssistId}; |
11 | 9 | ||
10 | // Assist: make_raw_string | ||
11 | // | ||
12 | // Adds `r#` to a plain string literal. | ||
13 | // | ||
14 | // ``` | ||
15 | // fn main() { | ||
16 | // "Hello,<|> World!"; | ||
17 | // } | ||
18 | // ``` | ||
19 | // -> | ||
20 | // ``` | ||
21 | // fn main() { | ||
22 | // r#"Hello, World!"#; | ||
23 | // } | ||
24 | // ``` | ||
12 | pub(crate) fn make_raw_string(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { | 25 | pub(crate) fn make_raw_string(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { |
13 | let token = ctx.find_token_at_offset(STRING)?; | 26 | let token = ctx.find_token_at_offset(STRING)?; |
14 | let text = token.text().as_str(); | 27 | let text = token.text().as_str(); |
@@ -40,6 +53,21 @@ pub(crate) fn make_raw_string(mut ctx: AssistCtx<impl HirDatabase>) -> Option<As | |||
40 | ctx.build() | 53 | ctx.build() |
41 | } | 54 | } |
42 | 55 | ||
56 | // Assist: make_usual_string | ||
57 | // | ||
58 | // Turns a raw string into a plain string. | ||
59 | // | ||
60 | // ``` | ||
61 | // fn main() { | ||
62 | // r#"Hello,<|> "World!""#; | ||
63 | // } | ||
64 | // ``` | ||
65 | // -> | ||
66 | // ``` | ||
67 | // fn main() { | ||
68 | // "Hello, \"World!\""; | ||
69 | // } | ||
70 | // ``` | ||
43 | pub(crate) fn make_usual_string(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { | 71 | pub(crate) fn make_usual_string(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { |
44 | let token = ctx.find_token_at_offset(RAW_STRING)?; | 72 | let token = ctx.find_token_at_offset(RAW_STRING)?; |
45 | let text = token.text().as_str(); | 73 | let text = token.text().as_str(); |
@@ -56,6 +84,21 @@ pub(crate) fn make_usual_string(mut ctx: AssistCtx<impl HirDatabase>) -> Option< | |||
56 | ctx.build() | 84 | ctx.build() |
57 | } | 85 | } |
58 | 86 | ||
87 | // Assist: add_hash | ||
88 | // | ||
89 | // Adds a hash to a raw string literal. | ||
90 | // | ||
91 | // ``` | ||
92 | // fn main() { | ||
93 | // r#"Hello,<|> World!"#; | ||
94 | // } | ||
95 | // ``` | ||
96 | // -> | ||
97 | // ``` | ||
98 | // fn main() { | ||
99 | // r##"Hello, World!"##; | ||
100 | // } | ||
101 | // ``` | ||
59 | pub(crate) fn add_hash(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { | 102 | pub(crate) fn add_hash(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { |
60 | let token = ctx.find_token_at_offset(RAW_STRING)?; | 103 | let token = ctx.find_token_at_offset(RAW_STRING)?; |
61 | ctx.add_action(AssistId("add_hash"), "add hash to raw string", |edit| { | 104 | ctx.add_action(AssistId("add_hash"), "add hash to raw string", |edit| { |
@@ -66,6 +109,21 @@ pub(crate) fn add_hash(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { | |||
66 | ctx.build() | 109 | ctx.build() |
67 | } | 110 | } |
68 | 111 | ||
112 | // Assist: remove_hash | ||
113 | // | ||
114 | // Removes a hash from a raw string literal. | ||
115 | // | ||
116 | // ``` | ||
117 | // fn main() { | ||
118 | // r#"Hello,<|> World!"#; | ||
119 | // } | ||
120 | // ``` | ||
121 | // -> | ||
122 | // ``` | ||
123 | // fn main() { | ||
124 | // r"Hello, World!"; | ||
125 | // } | ||
126 | // ``` | ||
69 | pub(crate) fn remove_hash(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { | 127 | pub(crate) fn remove_hash(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { |
70 | let token = ctx.find_token_at_offset(RAW_STRING)?; | 128 | let token = ctx.find_token_at_offset(RAW_STRING)?; |
71 | let text = token.text().as_str(); | 129 | let text = token.text().as_str(); |
diff --git a/crates/ra_assists/src/assists/remove_dbg.rs b/crates/ra_assists/src/assists/remove_dbg.rs index ac2c43e1a..44b8de814 100644 --- a/crates/ra_assists/src/assists/remove_dbg.rs +++ b/crates/ra_assists/src/assists/remove_dbg.rs | |||
@@ -1,12 +1,26 @@ | |||
1 | //! FIXME: write short doc here | ||
2 | |||
3 | use crate::{Assist, AssistCtx, AssistId}; | ||
4 | use hir::db::HirDatabase; | 1 | use hir::db::HirDatabase; |
5 | use ra_syntax::{ | 2 | use ra_syntax::{ |
6 | ast::{self, AstNode}, | 3 | ast::{self, AstNode}, |
7 | TextUnit, T, | 4 | TextUnit, T, |
8 | }; | 5 | }; |
9 | 6 | ||
7 | use crate::{Assist, AssistCtx, AssistId}; | ||
8 | |||
9 | // Assist: remove_dbg | ||
10 | // | ||
11 | // Removes `dbg!()` macro call. | ||
12 | // | ||
13 | // ``` | ||
14 | // fn main() { | ||
15 | // <|>dbg!(92); | ||
16 | // } | ||
17 | // ``` | ||
18 | // -> | ||
19 | // ``` | ||
20 | // fn main() { | ||
21 | // 92; | ||
22 | // } | ||
23 | // ``` | ||
10 | pub(crate) fn remove_dbg(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { | 24 | pub(crate) fn remove_dbg(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { |
11 | let macro_call = ctx.find_node_at_offset::<ast::MacroCall>()?; | 25 | let macro_call = ctx.find_node_at_offset::<ast::MacroCall>()?; |
12 | 26 | ||
diff --git a/crates/ra_assists/src/assists/replace_if_let_with_match.rs b/crates/ra_assists/src/assists/replace_if_let_with_match.rs index da276e47b..58ef2ff20 100644 --- a/crates/ra_assists/src/assists/replace_if_let_with_match.rs +++ b/crates/ra_assists/src/assists/replace_if_let_with_match.rs | |||
@@ -1,5 +1,3 @@ | |||
1 | //! FIXME: write short doc here | ||
2 | |||
3 | use format_buf::format; | 1 | use format_buf::format; |
4 | use hir::db::HirDatabase; | 2 | use hir::db::HirDatabase; |
5 | use ra_fmt::extract_trivial_expression; | 3 | use ra_fmt::extract_trivial_expression; |
@@ -7,6 +5,32 @@ use ra_syntax::{ast, AstNode}; | |||
7 | 5 | ||
8 | use crate::{Assist, AssistCtx, AssistId}; | 6 | use crate::{Assist, AssistCtx, AssistId}; |
9 | 7 | ||
8 | // Assist: replace_if_let_with_match | ||
9 | // | ||
10 | // Replaces `if let` with an else branch with a `match` expression. | ||
11 | // | ||
12 | // ``` | ||
13 | // enum Action { Move { distance: u32 }, Stop } | ||
14 | // | ||
15 | // fn handle(action: Action) { | ||
16 | // <|>if let Action::Move { distance } = action { | ||
17 | // foo(distance) | ||
18 | // } else { | ||
19 | // bar() | ||
20 | // } | ||
21 | // } | ||
22 | // ``` | ||
23 | // -> | ||
24 | // ``` | ||
25 | // enum Action { Move { distance: u32 }, Stop } | ||
26 | // | ||
27 | // fn handle(action: Action) { | ||
28 | // match action { | ||
29 | // Action::Move { distance } => foo(distance), | ||
30 | // _ => bar(), | ||
31 | // } | ||
32 | // } | ||
33 | // ``` | ||
10 | pub(crate) fn replace_if_let_with_match(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { | 34 | pub(crate) fn replace_if_let_with_match(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { |
11 | let if_expr: ast::IfExpr = ctx.find_node_at_offset()?; | 35 | let if_expr: ast::IfExpr = ctx.find_node_at_offset()?; |
12 | let cond = if_expr.condition()?; | 36 | let cond = if_expr.condition()?; |
diff --git a/crates/ra_assists/src/assists/split_import.rs b/crates/ra_assists/src/assists/split_import.rs index 09bde1b72..8d8a28987 100644 --- a/crates/ra_assists/src/assists/split_import.rs +++ b/crates/ra_assists/src/assists/split_import.rs | |||
@@ -1,5 +1,3 @@ | |||
1 | //! FIXME: write short doc here | ||
2 | |||
3 | use std::iter::successors; | 1 | use std::iter::successors; |
4 | 2 | ||
5 | use hir::db::HirDatabase; | 3 | use hir::db::HirDatabase; |
@@ -7,6 +5,17 @@ use ra_syntax::{ast, AstNode, TextUnit, T}; | |||
7 | 5 | ||
8 | use crate::{Assist, AssistCtx, AssistId}; | 6 | use crate::{Assist, AssistCtx, AssistId}; |
9 | 7 | ||
8 | // Assist: split_import | ||
9 | // | ||
10 | // Wraps the tail of import into braces. | ||
11 | // | ||
12 | // ``` | ||
13 | // use std::<|>collections::HashMap; | ||
14 | // ``` | ||
15 | // -> | ||
16 | // ``` | ||
17 | // use std::{collections::HashMap}; | ||
18 | // ``` | ||
10 | pub(crate) fn split_import(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { | 19 | pub(crate) fn split_import(mut ctx: AssistCtx<impl HirDatabase>) -> Option<Assist> { |
11 | let colon_colon = ctx.find_token_at_offset(T![::])?; | 20 | let colon_colon = ctx.find_token_at_offset(T![::])?; |
12 | let path = ast::Path::cast(colon_colon.parent())?; | 21 | let path = ast::Path::cast(colon_colon.parent())?; |
diff --git a/crates/ra_assists/src/doc_tests/generated.rs b/crates/ra_assists/src/doc_tests/generated.rs index 09677af68..b8d335911 100644 --- a/crates/ra_assists/src/doc_tests/generated.rs +++ b/crates/ra_assists/src/doc_tests/generated.rs | |||
@@ -40,6 +40,23 @@ fn main() { | |||
40 | } | 40 | } |
41 | 41 | ||
42 | #[test] | 42 | #[test] |
43 | fn doctest_add_hash() { | ||
44 | check( | ||
45 | "add_hash", | ||
46 | r#####" | ||
47 | fn main() { | ||
48 | r#"Hello,<|> World!"#; | ||
49 | } | ||
50 | "#####, | ||
51 | r#####" | ||
52 | fn main() { | ||
53 | r##"Hello, World!"##; | ||
54 | } | ||
55 | "#####, | ||
56 | ) | ||
57 | } | ||
58 | |||
59 | #[test] | ||
43 | fn doctest_add_impl() { | 60 | fn doctest_add_impl() { |
44 | check( | 61 | check( |
45 | "add_impl", | 62 | "add_impl", |
@@ -275,6 +292,40 @@ fn main() { | |||
275 | } | 292 | } |
276 | 293 | ||
277 | #[test] | 294 | #[test] |
295 | fn doctest_make_raw_string() { | ||
296 | check( | ||
297 | "make_raw_string", | ||
298 | r#####" | ||
299 | fn main() { | ||
300 | "Hello,<|> World!"; | ||
301 | } | ||
302 | "#####, | ||
303 | r#####" | ||
304 | fn main() { | ||
305 | r#"Hello, World!"#; | ||
306 | } | ||
307 | "#####, | ||
308 | ) | ||
309 | } | ||
310 | |||
311 | #[test] | ||
312 | fn doctest_make_usual_string() { | ||
313 | check( | ||
314 | "make_usual_string", | ||
315 | r#####" | ||
316 | fn main() { | ||
317 | r#"Hello,<|> "World!""#; | ||
318 | } | ||
319 | "#####, | ||
320 | r#####" | ||
321 | fn main() { | ||
322 | "Hello, \"World!\""; | ||
323 | } | ||
324 | "#####, | ||
325 | ) | ||
326 | } | ||
327 | |||
328 | #[test] | ||
278 | fn doctest_merge_match_arms() { | 329 | fn doctest_merge_match_arms() { |
279 | check( | 330 | check( |
280 | "merge_match_arms", | 331 | "merge_match_arms", |
@@ -370,3 +421,78 @@ fn handle(action: Action) { | |||
370 | "#####, | 421 | "#####, |
371 | ) | 422 | ) |
372 | } | 423 | } |
424 | |||
425 | #[test] | ||
426 | fn doctest_remove_dbg() { | ||
427 | check( | ||
428 | "remove_dbg", | ||
429 | r#####" | ||
430 | fn main() { | ||
431 | <|>dbg!(92); | ||
432 | } | ||
433 | "#####, | ||
434 | r#####" | ||
435 | fn main() { | ||
436 | 92; | ||
437 | } | ||
438 | "#####, | ||
439 | ) | ||
440 | } | ||
441 | |||
442 | #[test] | ||
443 | fn doctest_remove_hash() { | ||
444 | check( | ||
445 | "remove_hash", | ||
446 | r#####" | ||
447 | fn main() { | ||
448 | r#"Hello,<|> World!"#; | ||
449 | } | ||
450 | "#####, | ||
451 | r#####" | ||
452 | fn main() { | ||
453 | r"Hello, World!"; | ||
454 | } | ||
455 | "#####, | ||
456 | ) | ||
457 | } | ||
458 | |||
459 | #[test] | ||
460 | fn doctest_replace_if_let_with_match() { | ||
461 | check( | ||
462 | "replace_if_let_with_match", | ||
463 | r#####" | ||
464 | enum Action { Move { distance: u32 }, Stop } | ||
465 | |||
466 | fn handle(action: Action) { | ||
467 | <|>if let Action::Move { distance } = action { | ||
468 | foo(distance) | ||
469 | } else { | ||
470 | bar() | ||
471 | } | ||
472 | } | ||
473 | "#####, | ||
474 | r#####" | ||
475 | enum Action { Move { distance: u32 }, Stop } | ||
476 | |||
477 | fn handle(action: Action) { | ||
478 | match action { | ||
479 | Action::Move { distance } => foo(distance), | ||
480 | _ => bar(), | ||
481 | } | ||
482 | } | ||
483 | "#####, | ||
484 | ) | ||
485 | } | ||
486 | |||
487 | #[test] | ||
488 | fn doctest_split_import() { | ||
489 | check( | ||
490 | "split_import", | ||
491 | r#####" | ||
492 | use std::<|>collections::HashMap; | ||
493 | "#####, | ||
494 | r#####" | ||
495 | use std::{collections::HashMap}; | ||
496 | "#####, | ||
497 | ) | ||
498 | } | ||