Skip to content

Commit cd24349

Browse files
bors[bot]viorina
andauthored
Merge #1691
1691: Show inherent and trait impls of structs and enums r=viorina a=viorina Co-authored-by: Ekaterina Babshukova <ekaterina.babshukova@yandex.ru>
2 parents af1052b + 35a04ec commit cd24349

File tree

2 files changed

+44
-2
lines changed

2 files changed

+44
-2
lines changed

crates/ra_hir/src/ty/method_resolution.rs

+9
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,15 @@ impl CrateImplBlocks {
6868
)
6969
}
7070

71+
pub fn all_impls<'a>(&'a self) -> impl Iterator<Item = ImplBlock> + 'a {
72+
self.impls.values().chain(self.impls_by_trait.values()).flat_map(|i| i.iter()).map(
73+
move |(module_id, impl_id)| {
74+
let module = Module { krate: self.krate, module_id: *module_id };
75+
ImplBlock::from_id(module, *impl_id)
76+
},
77+
)
78+
}
79+
7180
fn collect_recursive(&mut self, db: &impl HirDatabase, module: Module) {
7281
let module_impl_blocks = db.impls_in_module(module);
7382

crates/ra_ide_api/src/impls.rs

+35-2
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
use hir::{db::HirDatabase, source_binder};
1+
use hir::{db::HirDatabase, source_binder, ApplicationTy, Ty, TypeCtor};
22
use ra_db::SourceDatabase;
33
use ra_syntax::{algo::find_node_at_offset, ast, AstNode};
44

@@ -47,7 +47,8 @@ fn impls_for_def(
4747

4848
Some(
4949
impls
50-
.lookup_impl_blocks(&ty)
50+
.all_impls()
51+
.filter(|impl_block| is_equal_for_find_impls(&ty, &impl_block.target_ty(db)))
5152
.map(|imp| NavigationTarget::from_impl_block(db, imp))
5253
.collect(),
5354
)
@@ -71,6 +72,19 @@ fn impls_for_trait(
7172
)
7273
}
7374

75+
fn is_equal_for_find_impls(original_ty: &Ty, impl_ty: &Ty) -> bool {
76+
match (original_ty, impl_ty) {
77+
(Ty::Apply(a_original_ty), Ty::Apply(ApplicationTy { ctor, parameters })) => match ctor {
78+
TypeCtor::Ref(..) => match parameters.as_single() {
79+
Ty::Apply(a_ty) => a_original_ty.ctor == a_ty.ctor,
80+
_ => false,
81+
},
82+
_ => a_original_ty.ctor == *ctor,
83+
},
84+
_ => false,
85+
}
86+
}
87+
7488
#[cfg(test)]
7589
mod tests {
7690
use crate::mock_analysis::analysis_and_position;
@@ -173,4 +187,23 @@ mod tests {
173187
&["impl IMPL_BLOCK FileId(2) [0; 31)", "impl IMPL_BLOCK FileId(3) [0; 31)"],
174188
);
175189
}
190+
191+
#[test]
192+
fn goto_implementation_all_impls() {
193+
check_goto(
194+
"
195+
//- /lib.rs
196+
trait T {}
197+
struct Foo<|>;
198+
impl Foo {}
199+
impl T for Foo {}
200+
impl T for &Foo {}
201+
",
202+
&[
203+
"impl IMPL_BLOCK FileId(1) [23; 34)",
204+
"impl IMPL_BLOCK FileId(1) [35; 52)",
205+
"impl IMPL_BLOCK FileId(1) [53; 71)",
206+
],
207+
);
208+
}
176209
}

0 commit comments

Comments
 (0)