Snippetify fix_visibility

This commit is contained in:
Aleksey Kladov 2020-05-20 14:13:17 +02:00
parent c446fd76a2
commit ba3a58d1b2
3 changed files with 36 additions and 32 deletions

View file

@ -25,7 +25,7 @@ use crate::{AssistContext, AssistId, Assists};
// -> // ->
// ``` // ```
// mod m { // mod m {
// pub(crate) fn frobnicate() {} // $0pub(crate) fn frobnicate() {}
// } // }
// fn main() { // fn main() {
// m::frobnicate() {} // m::frobnicate() {}
@ -62,10 +62,12 @@ fn add_vis_to_referenced_module_def(acc: &mut Assists, ctx: &AssistContext) -> O
Some(name) => format!("Change visibility of {} to {}", name, missing_visibility), Some(name) => format!("Change visibility of {} to {}", name, missing_visibility),
}; };
acc.add(AssistId("fix_visibility"), assist_label, target, |edit| { acc.add(AssistId("fix_visibility"), assist_label, target, |builder| {
edit.set_file(target_file); builder.set_file(target_file);
edit.insert(offset, format!("{} ", missing_visibility)); match ctx.config.snippet_cap {
edit.set_cursor(offset); Some(cap) => builder.insert_snippet(cap, offset, format!("$0{} ", missing_visibility)),
None => builder.insert(offset, format!("{} ", missing_visibility)),
}
}) })
} }
@ -103,10 +105,12 @@ fn add_vis_to_referenced_record_field(acc: &mut Assists, ctx: &AssistContext) ->
let assist_label = let assist_label =
format!("Change visibility of {}.{} to {}", parent_name, target_name, missing_visibility); format!("Change visibility of {}.{} to {}", parent_name, target_name, missing_visibility);
acc.add(AssistId("fix_visibility"), assist_label, target, |edit| { acc.add(AssistId("fix_visibility"), assist_label, target, |builder| {
edit.set_file(target_file); builder.set_file(target_file);
edit.insert(offset, format!("{} ", missing_visibility)); match ctx.config.snippet_cap {
edit.set_cursor(offset) Some(cap) => builder.insert_snippet(cap, offset, format!("$0{} ", missing_visibility)),
None => builder.insert(offset, format!("{} ", missing_visibility)),
}
}) })
} }
@ -196,7 +200,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { fn foo() {} } r"mod foo { fn foo() {} }
fn main() { foo::foo<|>() } ", fn main() { foo::foo<|>() } ",
r"mod foo { <|>pub(crate) fn foo() {} } r"mod foo { $0pub(crate) fn foo() {} }
fn main() { foo::foo() } ", fn main() { foo::foo() } ",
); );
check_assist_not_applicable( check_assist_not_applicable(
@ -212,7 +216,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { struct Foo; } r"mod foo { struct Foo; }
fn main() { foo::Foo<|> } ", fn main() { foo::Foo<|> } ",
r"mod foo { <|>pub(crate) struct Foo; } r"mod foo { $0pub(crate) struct Foo; }
fn main() { foo::Foo } ", fn main() { foo::Foo } ",
); );
check_assist_not_applicable( check_assist_not_applicable(
@ -224,7 +228,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { enum Foo; } r"mod foo { enum Foo; }
fn main() { foo::Foo<|> } ", fn main() { foo::Foo<|> } ",
r"mod foo { <|>pub(crate) enum Foo; } r"mod foo { $0pub(crate) enum Foo; }
fn main() { foo::Foo } ", fn main() { foo::Foo } ",
); );
check_assist_not_applicable( check_assist_not_applicable(
@ -236,7 +240,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { union Foo; } r"mod foo { union Foo; }
fn main() { foo::Foo<|> } ", fn main() { foo::Foo<|> } ",
r"mod foo { <|>pub(crate) union Foo; } r"mod foo { $0pub(crate) union Foo; }
fn main() { foo::Foo } ", fn main() { foo::Foo } ",
); );
check_assist_not_applicable( check_assist_not_applicable(
@ -258,7 +262,7 @@ mod tests {
//- /foo.rs //- /foo.rs
struct Foo; struct Foo;
", ",
r"<|>pub(crate) struct Foo; r"$0pub(crate) struct Foo;
", ",
); );
@ -270,7 +274,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { pub struct Foo { bar: (), } } r"mod foo { pub struct Foo { bar: (), } }
fn main() { foo::Foo { <|>bar: () }; } ", fn main() { foo::Foo { <|>bar: () }; } ",
r"mod foo { pub struct Foo { <|>pub(crate) bar: (), } } r"mod foo { pub struct Foo { $0pub(crate) bar: (), } }
fn main() { foo::Foo { bar: () }; } ", fn main() { foo::Foo { bar: () }; } ",
); );
check_assist( check_assist(
@ -281,7 +285,7 @@ mod tests {
//- /foo.rs //- /foo.rs
pub struct Foo { bar: () } pub struct Foo { bar: () }
", ",
r"pub struct Foo { <|>pub(crate) bar: () } r"pub struct Foo { $0pub(crate) bar: () }
", ",
); );
@ -307,7 +311,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { pub enum Foo { Bar { bar: () } } } r"mod foo { pub enum Foo { Bar { bar: () } } }
fn main() { foo::Foo::Bar { <|>bar: () }; } ", fn main() { foo::Foo::Bar { <|>bar: () }; } ",
r"mod foo { pub enum Foo { Bar { <|>pub(crate) bar: () } } } r"mod foo { pub enum Foo { Bar { $0pub(crate) bar: () } } }
fn main() { foo::Foo::Bar { bar: () }; } ", fn main() { foo::Foo::Bar { bar: () }; } ",
); );
check_assist( check_assist(
@ -318,7 +322,7 @@ mod tests {
//- /foo.rs //- /foo.rs
pub enum Foo { Bar { bar: () } } pub enum Foo { Bar { bar: () } }
", ",
r"pub enum Foo { Bar { <|>pub(crate) bar: () } } r"pub enum Foo { Bar { $0pub(crate) bar: () } }
", ",
); );
@ -346,7 +350,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { pub union Foo { bar: (), } } r"mod foo { pub union Foo { bar: (), } }
fn main() { foo::Foo { <|>bar: () }; } ", fn main() { foo::Foo { <|>bar: () }; } ",
r"mod foo { pub union Foo { <|>pub(crate) bar: (), } } r"mod foo { pub union Foo { $0pub(crate) bar: (), } }
fn main() { foo::Foo { bar: () }; } ", fn main() { foo::Foo { bar: () }; } ",
); );
check_assist( check_assist(
@ -357,7 +361,7 @@ mod tests {
//- /foo.rs //- /foo.rs
pub union Foo { bar: () } pub union Foo { bar: () }
", ",
r"pub union Foo { <|>pub(crate) bar: () } r"pub union Foo { $0pub(crate) bar: () }
", ",
); );
@ -383,7 +387,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { const FOO: () = (); } r"mod foo { const FOO: () = (); }
fn main() { foo::FOO<|> } ", fn main() { foo::FOO<|> } ",
r"mod foo { <|>pub(crate) const FOO: () = (); } r"mod foo { $0pub(crate) const FOO: () = (); }
fn main() { foo::FOO } ", fn main() { foo::FOO } ",
); );
check_assist_not_applicable( check_assist_not_applicable(
@ -399,7 +403,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { static FOO: () = (); } r"mod foo { static FOO: () = (); }
fn main() { foo::FOO<|> } ", fn main() { foo::FOO<|> } ",
r"mod foo { <|>pub(crate) static FOO: () = (); } r"mod foo { $0pub(crate) static FOO: () = (); }
fn main() { foo::FOO } ", fn main() { foo::FOO } ",
); );
check_assist_not_applicable( check_assist_not_applicable(
@ -415,7 +419,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { trait Foo { fn foo(&self) {} } } r"mod foo { trait Foo { fn foo(&self) {} } }
fn main() { let x: &dyn foo::<|>Foo; } ", fn main() { let x: &dyn foo::<|>Foo; } ",
r"mod foo { <|>pub(crate) trait Foo { fn foo(&self) {} } } r"mod foo { $0pub(crate) trait Foo { fn foo(&self) {} } }
fn main() { let x: &dyn foo::Foo; } ", fn main() { let x: &dyn foo::Foo; } ",
); );
check_assist_not_applicable( check_assist_not_applicable(
@ -431,7 +435,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { type Foo = (); } r"mod foo { type Foo = (); }
fn main() { let x: foo::Foo<|>; } ", fn main() { let x: foo::Foo<|>; } ",
r"mod foo { <|>pub(crate) type Foo = (); } r"mod foo { $0pub(crate) type Foo = (); }
fn main() { let x: foo::Foo; } ", fn main() { let x: foo::Foo; } ",
); );
check_assist_not_applicable( check_assist_not_applicable(
@ -447,7 +451,7 @@ mod tests {
fix_visibility, fix_visibility,
r"mod foo { mod bar { fn bar() {} } } r"mod foo { mod bar { fn bar() {} } }
fn main() { foo::bar<|>::bar(); } ", fn main() { foo::bar<|>::bar(); } ",
r"mod foo { <|>pub(crate) mod bar { fn bar() {} } } r"mod foo { $0pub(crate) mod bar { fn bar() {} } }
fn main() { foo::bar::bar(); } ", fn main() { foo::bar::bar(); } ",
); );
@ -463,7 +467,7 @@ mod tests {
pub fn baz() {} pub fn baz() {}
} }
", ",
r"<|>pub(crate) mod bar { r"$0pub(crate) mod bar {
pub fn baz() {} pub fn baz() {}
} }
@ -493,7 +497,7 @@ mod tests {
pub fn baz() {} pub fn baz() {}
} }
", ",
r"<|>pub(crate) mod bar; r"$0pub(crate) mod bar;
", ",
); );
} }
@ -510,7 +514,7 @@ mod tests {
mod bar { mod bar {
pub fn baz() {} pub fn baz() {}
}", }",
r"<|>pub(crate) mod bar { r"$0pub(crate) mod bar {
pub fn baz() {} pub fn baz() {}
} }
", ",
@ -525,7 +529,7 @@ mod tests {
foo::Bar<|> foo::Bar<|>
//- /lib.rs crate:foo //- /lib.rs crate:foo
struct Bar;", struct Bar;",
r"<|>pub struct Bar; r"$0pub struct Bar;
", ",
) )
} }
@ -545,7 +549,7 @@ mod tests {
", ",
r" r"
mod foo { mod foo {
<|>pub(crate) use bar::Baz; $0pub(crate) use bar::Baz;
mod bar { pub(super) struct Baz; } mod bar { pub(super) struct Baz; }
} }
foo::Baz foo::Baz

View file

@ -358,7 +358,7 @@ fn main() {
"#####, "#####,
r#####" r#####"
mod m { mod m {
pub(crate) fn frobnicate() {} $0pub(crate) fn frobnicate() {}
} }
fn main() { fn main() {
m::frobnicate() {} m::frobnicate() {}

View file

@ -346,7 +346,7 @@ fn main() {
// AFTER // AFTER
mod m { mod m {
pub(crate) fn frobnicate() {} $0pub(crate) fn frobnicate() {}
} }
fn main() { fn main() {
m::frobnicate() {} m::frobnicate() {}