mirror of
https://github.com/rust-lang/rust-analyzer.git
synced 2025-09-29 13:25:09 +00:00
Remove "Sort methods by trait definition" assist
It was replaced by the "Sort items by trait definition" assist.
This commit is contained in:
parent
2b20a05fc6
commit
e315124798
4 changed files with 25 additions and 268 deletions
|
@ -1,217 +0,0 @@
|
||||||
use hir::{PathResolution, Semantics};
|
|
||||||
use ide_db::{FxHashMap, RootDatabase};
|
|
||||||
use itertools::Itertools;
|
|
||||||
use syntax::{
|
|
||||||
ast::{self, HasName},
|
|
||||||
ted, AstNode,
|
|
||||||
};
|
|
||||||
|
|
||||||
use crate::{utils::get_methods, AssistContext, AssistId, AssistKind, Assists};
|
|
||||||
|
|
||||||
// Assist: reorder_impl
|
|
||||||
//
|
|
||||||
// Reorder the methods of an `impl Trait`. The methods will be ordered
|
|
||||||
// in the same order as in the trait definition.
|
|
||||||
//
|
|
||||||
// ```
|
|
||||||
// trait Foo {
|
|
||||||
// fn a() {}
|
|
||||||
// fn b() {}
|
|
||||||
// fn c() {}
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// struct Bar;
|
|
||||||
// $0impl Foo for Bar {
|
|
||||||
// fn b() {}
|
|
||||||
// fn c() {}
|
|
||||||
// fn a() {}
|
|
||||||
// }
|
|
||||||
// ```
|
|
||||||
// ->
|
|
||||||
// ```
|
|
||||||
// trait Foo {
|
|
||||||
// fn a() {}
|
|
||||||
// fn b() {}
|
|
||||||
// fn c() {}
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// struct Bar;
|
|
||||||
// impl Foo for Bar {
|
|
||||||
// fn a() {}
|
|
||||||
// fn b() {}
|
|
||||||
// fn c() {}
|
|
||||||
// }
|
|
||||||
// ```
|
|
||||||
pub(crate) fn reorder_impl(acc: &mut Assists, ctx: &AssistContext) -> Option<()> {
|
|
||||||
let impl_ast = ctx.find_node_at_offset::<ast::Impl>()?;
|
|
||||||
let items = impl_ast.assoc_item_list()?;
|
|
||||||
let methods = get_methods(&items);
|
|
||||||
|
|
||||||
let path = impl_ast
|
|
||||||
.trait_()
|
|
||||||
.and_then(|t| match t {
|
|
||||||
ast::Type::PathType(path) => Some(path),
|
|
||||||
_ => None,
|
|
||||||
})?
|
|
||||||
.path()?;
|
|
||||||
|
|
||||||
let ranks = compute_method_ranks(&path, ctx)?;
|
|
||||||
let sorted: Vec<_> = methods
|
|
||||||
.iter()
|
|
||||||
.cloned()
|
|
||||||
.sorted_by_key(|f| {
|
|
||||||
f.name().and_then(|n| ranks.get(&n.to_string()).copied()).unwrap_or(usize::max_value())
|
|
||||||
})
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
// Don't edit already sorted methods:
|
|
||||||
if methods == sorted {
|
|
||||||
cov_mark::hit!(not_applicable_if_sorted);
|
|
||||||
return None;
|
|
||||||
}
|
|
||||||
|
|
||||||
let target = items.syntax().text_range();
|
|
||||||
acc.add(
|
|
||||||
AssistId("reorder_impl", AssistKind::RefactorRewrite),
|
|
||||||
"Sort methods by trait definition",
|
|
||||||
target,
|
|
||||||
|builder| {
|
|
||||||
let methods = methods.into_iter().map(|fn_| builder.make_mut(fn_)).collect::<Vec<_>>();
|
|
||||||
methods
|
|
||||||
.into_iter()
|
|
||||||
.zip(sorted)
|
|
||||||
.for_each(|(old, new)| ted::replace(old.syntax(), new.clone_for_update().syntax()));
|
|
||||||
},
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn compute_method_ranks(path: &ast::Path, ctx: &AssistContext) -> Option<FxHashMap<String, usize>> {
|
|
||||||
let td = trait_definition(path, &ctx.sema)?;
|
|
||||||
|
|
||||||
Some(
|
|
||||||
td.items(ctx.db())
|
|
||||||
.iter()
|
|
||||||
.flat_map(|i| match i {
|
|
||||||
hir::AssocItem::Function(f) => Some(f),
|
|
||||||
_ => None,
|
|
||||||
})
|
|
||||||
.enumerate()
|
|
||||||
.map(|(idx, func)| (func.name(ctx.db()).to_string(), idx))
|
|
||||||
.collect(),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn trait_definition(path: &ast::Path, sema: &Semantics<RootDatabase>) -> Option<hir::Trait> {
|
|
||||||
match sema.resolve_path(path)? {
|
|
||||||
PathResolution::Def(hir::ModuleDef::Trait(trait_)) => Some(trait_),
|
|
||||||
_ => None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
use crate::tests::{check_assist, check_assist_not_applicable};
|
|
||||||
|
|
||||||
use super::*;
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn not_applicable_if_sorted() {
|
|
||||||
cov_mark::check!(not_applicable_if_sorted);
|
|
||||||
check_assist_not_applicable(
|
|
||||||
reorder_impl,
|
|
||||||
r#"
|
|
||||||
trait Bar {
|
|
||||||
fn a() {}
|
|
||||||
fn z() {}
|
|
||||||
fn b() {}
|
|
||||||
}
|
|
||||||
struct Foo;
|
|
||||||
$0impl Bar for Foo {
|
|
||||||
fn a() {}
|
|
||||||
fn z() {}
|
|
||||||
fn b() {}
|
|
||||||
}
|
|
||||||
"#,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn not_applicable_if_empty() {
|
|
||||||
check_assist_not_applicable(
|
|
||||||
reorder_impl,
|
|
||||||
r#"
|
|
||||||
trait Bar {};
|
|
||||||
struct Foo;
|
|
||||||
$0impl Bar for Foo {}
|
|
||||||
"#,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn reorder_impl_trait_functions() {
|
|
||||||
check_assist(
|
|
||||||
reorder_impl,
|
|
||||||
r#"
|
|
||||||
trait Bar {
|
|
||||||
fn a() {}
|
|
||||||
fn c() {}
|
|
||||||
fn b() {}
|
|
||||||
fn d() {}
|
|
||||||
}
|
|
||||||
|
|
||||||
struct Foo;
|
|
||||||
$0impl Bar for Foo {
|
|
||||||
fn d() {}
|
|
||||||
fn b() {}
|
|
||||||
fn c() {}
|
|
||||||
fn a() {}
|
|
||||||
}
|
|
||||||
"#,
|
|
||||||
r#"
|
|
||||||
trait Bar {
|
|
||||||
fn a() {}
|
|
||||||
fn c() {}
|
|
||||||
fn b() {}
|
|
||||||
fn d() {}
|
|
||||||
}
|
|
||||||
|
|
||||||
struct Foo;
|
|
||||||
impl Bar for Foo {
|
|
||||||
fn a() {}
|
|
||||||
fn c() {}
|
|
||||||
fn b() {}
|
|
||||||
fn d() {}
|
|
||||||
}
|
|
||||||
"#,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn reorder_impl_trait_methods_uneven_ident_lengths() {
|
|
||||||
check_assist(
|
|
||||||
reorder_impl,
|
|
||||||
r#"
|
|
||||||
trait Bar {
|
|
||||||
fn foo(&mut self) {}
|
|
||||||
fn fooo(&mut self) {}
|
|
||||||
}
|
|
||||||
|
|
||||||
struct Foo;
|
|
||||||
impl Bar for Foo {
|
|
||||||
fn fooo(&mut self) {}
|
|
||||||
fn foo(&mut self) {$0}
|
|
||||||
}"#,
|
|
||||||
r#"
|
|
||||||
trait Bar {
|
|
||||||
fn foo(&mut self) {}
|
|
||||||
fn fooo(&mut self) {}
|
|
||||||
}
|
|
||||||
|
|
||||||
struct Foo;
|
|
||||||
impl Bar for Foo {
|
|
||||||
fn foo(&mut self) {}
|
|
||||||
fn fooo(&mut self) {}
|
|
||||||
}"#,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -47,13 +47,6 @@ pub(crate) fn reorder_impl_items(acc: &mut Assists, ctx: &AssistContext) -> Opti
|
||||||
let items = impl_ast.assoc_item_list()?;
|
let items = impl_ast.assoc_item_list()?;
|
||||||
let assoc_items = items.assoc_items().collect::<Vec<_>>();
|
let assoc_items = items.assoc_items().collect::<Vec<_>>();
|
||||||
|
|
||||||
// If all items are either function or macro calls, then reorder_impl assist can be used
|
|
||||||
if assoc_items.iter().all(|i| matches!(i, ast::AssocItem::Fn(_) | ast::AssocItem::MacroCall(_)))
|
|
||||||
{
|
|
||||||
cov_mark::hit!(not_applicable_if_all_functions);
|
|
||||||
return None;
|
|
||||||
}
|
|
||||||
|
|
||||||
let path = impl_ast
|
let path = impl_ast
|
||||||
.trait_()
|
.trait_()
|
||||||
.and_then(|t| match t {
|
.and_then(|t| match t {
|
||||||
|
@ -152,21 +145,39 @@ $0impl Bar for Foo {
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn not_applicable_if_all_functions() {
|
fn reorder_impl_trait_functions() {
|
||||||
cov_mark::check!(not_applicable_if_all_functions);
|
check_assist(
|
||||||
check_assist_not_applicable(
|
|
||||||
reorder_impl_items,
|
reorder_impl_items,
|
||||||
r#"
|
r#"
|
||||||
trait Bar {
|
trait Bar {
|
||||||
fn a() {}
|
fn a() {}
|
||||||
fn z() {}
|
fn c() {}
|
||||||
fn b() {}
|
fn b() {}
|
||||||
|
fn d() {}
|
||||||
}
|
}
|
||||||
|
|
||||||
struct Foo;
|
struct Foo;
|
||||||
$0impl Bar for Foo {
|
$0impl Bar for Foo {
|
||||||
fn a() {}
|
fn d() {}
|
||||||
fn z() {}
|
|
||||||
fn b() {}
|
fn b() {}
|
||||||
|
fn c() {}
|
||||||
|
fn a() {}
|
||||||
|
}
|
||||||
|
"#,
|
||||||
|
r#"
|
||||||
|
trait Bar {
|
||||||
|
fn a() {}
|
||||||
|
fn c() {}
|
||||||
|
fn b() {}
|
||||||
|
fn d() {}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct Foo;
|
||||||
|
impl Bar for Foo {
|
||||||
|
fn a() {}
|
||||||
|
fn c() {}
|
||||||
|
fn b() {}
|
||||||
|
fn d() {}
|
||||||
}
|
}
|
||||||
"#,
|
"#,
|
||||||
)
|
)
|
||||||
|
|
|
@ -169,7 +169,6 @@ mod handlers {
|
||||||
mod remove_mut;
|
mod remove_mut;
|
||||||
mod remove_unused_param;
|
mod remove_unused_param;
|
||||||
mod reorder_fields;
|
mod reorder_fields;
|
||||||
mod reorder_impl;
|
|
||||||
mod reorder_impl_items;
|
mod reorder_impl_items;
|
||||||
mod replace_try_expr_with_match;
|
mod replace_try_expr_with_match;
|
||||||
mod replace_derive_with_manual_impl;
|
mod replace_derive_with_manual_impl;
|
||||||
|
@ -257,7 +256,6 @@ mod handlers {
|
||||||
remove_mut::remove_mut,
|
remove_mut::remove_mut,
|
||||||
remove_unused_param::remove_unused_param,
|
remove_unused_param::remove_unused_param,
|
||||||
reorder_fields::reorder_fields,
|
reorder_fields::reorder_fields,
|
||||||
reorder_impl::reorder_impl,
|
|
||||||
reorder_impl_items::reorder_impl_items,
|
reorder_impl_items::reorder_impl_items,
|
||||||
replace_try_expr_with_match::replace_try_expr_with_match,
|
replace_try_expr_with_match::replace_try_expr_with_match,
|
||||||
replace_derive_with_manual_impl::replace_derive_with_manual_impl,
|
replace_derive_with_manual_impl::replace_derive_with_manual_impl,
|
||||||
|
|
|
@ -1736,41 +1736,6 @@ const test: Foo = Foo {foo: 1, bar: 0}
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn doctest_reorder_impl() {
|
|
||||||
check_doc_test(
|
|
||||||
"reorder_impl",
|
|
||||||
r#####"
|
|
||||||
trait Foo {
|
|
||||||
fn a() {}
|
|
||||||
fn b() {}
|
|
||||||
fn c() {}
|
|
||||||
}
|
|
||||||
|
|
||||||
struct Bar;
|
|
||||||
$0impl Foo for Bar {
|
|
||||||
fn b() {}
|
|
||||||
fn c() {}
|
|
||||||
fn a() {}
|
|
||||||
}
|
|
||||||
"#####,
|
|
||||||
r#####"
|
|
||||||
trait Foo {
|
|
||||||
fn a() {}
|
|
||||||
fn b() {}
|
|
||||||
fn c() {}
|
|
||||||
}
|
|
||||||
|
|
||||||
struct Bar;
|
|
||||||
impl Foo for Bar {
|
|
||||||
fn a() {}
|
|
||||||
fn b() {}
|
|
||||||
fn c() {}
|
|
||||||
}
|
|
||||||
"#####,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn doctest_reorder_impl_items() {
|
fn doctest_reorder_impl_items() {
|
||||||
check_doc_test(
|
check_doc_test(
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue