mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Don't concat path in replace_qualified assist when they start with a keyword

Browse Source
This commit is contained in:
Lukas Wirth 2021-07-20 18:57:12 +02:00
parent ea105f9396
commit 4f99c6467b
1 changed files with 19 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
crates/ide_assists/src/handlers/replace_qualified_name_with_use.rs
View File

@ -54,12 +54,23 @@ pub(crate) fn replace_qualified_name_with_use(
_ => return None, _ => return None,
}; };
let starts_with_name_ref = !matches!(
path.first_segment().and_then(|it| it.kind()),
Some(
ast::PathSegmentKind::CrateKw
| ast::PathSegmentKind::SuperKw
| ast::PathSegmentKind::SelfKw
)
);
let path_to_qualifier = starts_with_name_ref
.then(|| {
ctx.sema.scope(path.syntax()).module().and_then(|m| {
m.find_use_path_prefixed(ctx.sema.db, module, ctx.config.insert_use.prefix_kind)
})
})
.flatten();
let scope = ImportScope::find_insert_use_container_with_macros(path.syntax(), &ctx.sema)?; let scope = ImportScope::find_insert_use_container_with_macros(path.syntax(), &ctx.sema)?;
let path_to_qualifier = ctx.sema.scope(path.syntax()).module()?.find_use_path_prefixed(
ctx.sema.db,
module,
ctx.config.insert_use.prefix_kind,
)?;
let target = path.syntax().text_range(); let target = path.syntax().text_range();
acc.add( acc.add(
AssistId("replace_qualified_name_with_use", AssistKind::RefactorRewrite), AssistId("replace_qualified_name_with_use", AssistKind::RefactorRewrite),
@ -74,7 +85,7 @@ pub(crate) fn replace_qualified_name_with_use(
ImportScope::Block(it) => ImportScope::Block(builder.make_mut(it)), ImportScope::Block(it) => ImportScope::Block(builder.make_mut(it)),
}; };
// stick the found import in front of the to be replaced path // stick the found import in front of the to be replaced path
let path = match mod_path_to_ast(&path_to_qualifier).qualifier() { let path = match path_to_qualifier.and_then(|it| mod_path_to_ast(&it).qualifier()) {
Some(qualifier) => make::path_concat(qualifier, path), Some(qualifier) => make::path_concat(qualifier, path),
None => path, None => path,
}; };
@ -323,7 +334,7 @@ fn main() {
replace_qualified_name_with_use, replace_qualified_name_with_use,
r" r"
pub mod foo { pub mod foo {
struct Foo; pub struct Foo;
} }
mod bar { mod bar {
@ -338,7 +349,7 @@ fn main() {
use foo::Foo; use foo::Foo;
pub mod foo { pub mod foo {
struct Foo; pub struct Foo;
} }
mod bar { mod bar {