use hir::{db::HirDatabase, source_binder, ApplicationTy, Ty, TypeCtor}; use ra_db::SourceDatabase; use ra_syntax::{algo::find_node_at_offset, ast, AstNode}; use crate::{db::RootDatabase, FilePosition, NavigationTarget, RangeInfo}; pub(crate) fn goto_implementation( db: &RootDatabase, position: FilePosition, ) -> Option>> { let parse = db.parse(position.file_id); let syntax = parse.tree().syntax().clone(); let module = source_binder::module_from_position(db, position)?; if let Some(nominal_def) = find_node_at_offset::(&syntax, position.offset) { return Some(RangeInfo::new( nominal_def.syntax().text_range(), impls_for_def(db, &nominal_def, module)?, )); } else if let Some(trait_def) = find_node_at_offset::(&syntax, position.offset) { return Some(RangeInfo::new( trait_def.syntax().text_range(), impls_for_trait(db, &trait_def, module)?, )); } None } fn impls_for_def( db: &RootDatabase, node: &ast::NominalDef, module: hir::Module, ) -> Option> { let ty = match node { ast::NominalDef::StructDef(def) => { source_binder::struct_from_module(db, module, &def).ty(db) } ast::NominalDef::EnumDef(def) => source_binder::enum_from_module(db, module, &def).ty(db), }; let krate = module.krate(db)?; let impls = db.impls_in_crate(krate); Some( impls .all_impls() .filter(|impl_block| is_equal_for_find_impls(&ty, &impl_block.target_ty(db))) .map(|imp| NavigationTarget::from_impl_block(db, imp)) .collect(), ) } fn impls_for_trait( db: &RootDatabase, node: &ast::TraitDef, module: hir::Module, ) -> Option> { let tr = source_binder::trait_from_module(db, module, node); let krate = module.krate(db)?; let impls = db.impls_in_crate(krate); Some( impls .lookup_impl_blocks_for_trait(tr) .map(|imp| NavigationTarget::from_impl_block(db, imp)) .collect(), ) } fn is_equal_for_find_impls(original_ty: &Ty, impl_ty: &Ty) -> bool { match (original_ty, impl_ty) { (Ty::Apply(a_original_ty), Ty::Apply(ApplicationTy { ctor, parameters })) => match ctor { TypeCtor::Ref(..) => match parameters.as_single() { Ty::Apply(a_ty) => a_original_ty.ctor == a_ty.ctor, _ => false, }, _ => a_original_ty.ctor == *ctor, }, _ => false, } } #[cfg(test)] mod tests { use crate::mock_analysis::analysis_and_position; fn check_goto(fixture: &str, expected: &[&str]) { let (analysis, pos) = analysis_and_position(fixture); let mut navs = analysis.goto_implementation(pos).unwrap().unwrap().info; assert_eq!(navs.len(), expected.len()); navs.sort_by_key(|nav| (nav.file_id(), nav.full_range().start())); navs.into_iter().enumerate().for_each(|(i, nav)| nav.assert_match(expected[i])); } #[test] fn goto_implementation_works() { check_goto( " //- /lib.rs struct Foo<|>; impl Foo {} ", &["impl IMPL_BLOCK FileId(1) [12; 23)"], ); } #[test] fn goto_implementation_works_multiple_blocks() { check_goto( " //- /lib.rs struct Foo<|>; impl Foo {} impl Foo {} ", &["impl IMPL_BLOCK FileId(1) [12; 23)", "impl IMPL_BLOCK FileId(1) [24; 35)"], ); } #[test] fn goto_implementation_works_multiple_mods() { check_goto( " //- /lib.rs struct Foo<|>; mod a { impl super::Foo {} } mod b { impl super::Foo {} } ", &["impl IMPL_BLOCK FileId(1) [24; 42)", "impl IMPL_BLOCK FileId(1) [57; 75)"], ); } #[test] fn goto_implementation_works_multiple_files() { check_goto( " //- /lib.rs struct Foo<|>; mod a; mod b; //- /a.rs impl crate::Foo {} //- /b.rs impl crate::Foo {} ", &["impl IMPL_BLOCK FileId(2) [0; 18)", "impl IMPL_BLOCK FileId(3) [0; 18)"], ); } #[test] fn goto_implementation_for_trait() { check_goto( " //- /lib.rs trait T<|> {} struct Foo; impl T for Foo {} ", &["impl IMPL_BLOCK FileId(1) [23; 40)"], ); } #[test] fn goto_implementation_for_trait_multiple_files() { check_goto( " //- /lib.rs trait T<|> {}; struct Foo; mod a; mod b; //- /a.rs impl crate::T for crate::Foo {} //- /b.rs impl crate::T for crate::Foo {} ", &["impl IMPL_BLOCK FileId(2) [0; 31)", "impl IMPL_BLOCK FileId(3) [0; 31)"], ); } #[test] fn goto_implementation_all_impls() { check_goto( " //- /lib.rs trait T {} struct Foo<|>; impl Foo {} impl T for Foo {} impl T for &Foo {} ", &[ "impl IMPL_BLOCK FileId(1) [23; 34)", "impl IMPL_BLOCK FileId(1) [35; 52)", "impl IMPL_BLOCK FileId(1) [53; 71)", ], ); } }