diff options
Diffstat (limited to 'crates/ra_assists/src/handlers/change_visibility.rs')
-rw-r--r-- | crates/ra_assists/src/handlers/change_visibility.rs | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/crates/ra_assists/src/handlers/change_visibility.rs b/crates/ra_assists/src/handlers/change_visibility.rs index 157c7b665..34c7e481d 100644 --- a/crates/ra_assists/src/handlers/change_visibility.rs +++ b/crates/ra_assists/src/handlers/change_visibility.rs | |||
@@ -6,7 +6,7 @@ use ra_syntax::{ | |||
6 | }; | 6 | }; |
7 | use test_utils::mark; | 7 | use test_utils::mark; |
8 | 8 | ||
9 | use crate::{utils::vis_offset, AssistContext, AssistId, Assists}; | 9 | use crate::{utils::vis_offset, AssistContext, AssistId, AssistKind, Assists}; |
10 | 10 | ||
11 | // Assist: change_visibility | 11 | // Assist: change_visibility |
12 | // | 12 | // |
@@ -62,9 +62,15 @@ fn add_vis(acc: &mut Assists, ctx: &AssistContext) -> Option<()> { | |||
62 | return None; | 62 | return None; |
63 | }; | 63 | }; |
64 | 64 | ||
65 | acc.add(AssistId("change_visibility"), "Change visibility to pub(crate)", target, |edit| { | 65 | acc.add( |
66 | edit.insert(offset, "pub(crate) "); | 66 | AssistId("change_visibility"), |
67 | }) | 67 | AssistKind::RefactorRewrite, |
68 | "Change visibility to pub(crate)", | ||
69 | target, | ||
70 | |edit| { | ||
71 | edit.insert(offset, "pub(crate) "); | ||
72 | }, | ||
73 | ) | ||
68 | } | 74 | } |
69 | 75 | ||
70 | fn change_vis(acc: &mut Assists, vis: ast::Visibility) -> Option<()> { | 76 | fn change_vis(acc: &mut Assists, vis: ast::Visibility) -> Option<()> { |
@@ -72,6 +78,7 @@ fn change_vis(acc: &mut Assists, vis: ast::Visibility) -> Option<()> { | |||
72 | let target = vis.syntax().text_range(); | 78 | let target = vis.syntax().text_range(); |
73 | return acc.add( | 79 | return acc.add( |
74 | AssistId("change_visibility"), | 80 | AssistId("change_visibility"), |
81 | AssistKind::RefactorRewrite, | ||
75 | "Change Visibility to pub(crate)", | 82 | "Change Visibility to pub(crate)", |
76 | target, | 83 | target, |
77 | |edit| { | 84 | |edit| { |
@@ -83,6 +90,7 @@ fn change_vis(acc: &mut Assists, vis: ast::Visibility) -> Option<()> { | |||
83 | let target = vis.syntax().text_range(); | 90 | let target = vis.syntax().text_range(); |
84 | return acc.add( | 91 | return acc.add( |
85 | AssistId("change_visibility"), | 92 | AssistId("change_visibility"), |
93 | AssistKind::RefactorRewrite, | ||
86 | "Change visibility to pub", | 94 | "Change visibility to pub", |
87 | target, | 95 | target, |
88 | |edit| { | 96 | |edit| { |