From c39352fa43f4ff4df835d511b4415c8e770438c4 Mon Sep 17 00:00:00 2001 From: Florian Diebold Date: Tue, 24 Dec 2019 16:44:32 +0100 Subject: Fill in type params in 'add missing impl members' assist --- crates/ra_assists/src/assist_ctx.rs | 2 +- .../src/assists/add_missing_impl_members.rs | 131 ++++++++++++++++++--- crates/ra_assists/src/doc_tests/generated.rs | 22 ++-- 3 files changed, 129 insertions(+), 26 deletions(-) (limited to 'crates/ra_assists/src') diff --git a/crates/ra_assists/src/assist_ctx.rs b/crates/ra_assists/src/assist_ctx.rs index 993aebc47..28152f724 100644 --- a/crates/ra_assists/src/assist_ctx.rs +++ b/crates/ra_assists/src/assist_ctx.rs @@ -46,7 +46,7 @@ pub(crate) enum Assist { /// /// Note, however, that we don't actually use such two-phase logic at the /// moment, because the LSP API is pretty awkward in this place, and it's much -/// easier to just compute the edit eagerly :-)#[derive(Debug, Clone)] +/// easier to just compute the edit eagerly :-) #[derive(Debug)] pub(crate) struct AssistCtx<'a, DB> { pub(crate) db: &'a DB, diff --git a/crates/ra_assists/src/assists/add_missing_impl_members.rs b/crates/ra_assists/src/assists/add_missing_impl_members.rs index cef669cb5..bc49e71fe 100644 --- a/crates/ra_assists/src/assists/add_missing_impl_members.rs +++ b/crates/ra_assists/src/assists/add_missing_impl_members.rs @@ -1,3 +1,5 @@ +use std::collections::HashMap; + use hir::{db::HirDatabase, HasSource}; use ra_syntax::{ ast::{self, edit, make, AstNode, NameOwner}, @@ -17,26 +19,26 @@ enum AddMissingImplMembersMode { // Adds scaffold for required impl members. // // ``` -// trait T { +// trait Trait { // Type X; -// fn foo(&self); +// fn foo(&self) -> T; // fn bar(&self) {} // } // -// impl T for () {<|> +// impl Trait for () {<|> // // } // ``` // -> // ``` -// trait T { +// trait Trait { // Type X; -// fn foo(&self); +// fn foo(&self) -> T; // fn bar(&self) {} // } // -// impl T for () { -// fn foo(&self) { unimplemented!() } +// impl Trait for () { +// fn foo(&self) -> u32 { unimplemented!() } // // } // ``` @@ -54,13 +56,13 @@ pub(crate) fn add_missing_impl_members(ctx: AssistCtx) -> Opti // Adds scaffold for overriding default impl members. // // ``` -// trait T { +// trait Trait { // Type X; // fn foo(&self); // fn bar(&self) {} // } // -// impl T for () { +// impl Trait for () { // Type X = (); // fn foo(&self) {}<|> // @@ -68,13 +70,13 @@ pub(crate) fn add_missing_impl_members(ctx: AssistCtx) -> Opti // ``` // -> // ``` -// trait T { +// trait Trait { // Type X; // fn foo(&self); // fn bar(&self) {} // } // -// impl T for () { +// impl Trait for () { // Type X = (); // fn foo(&self) {} // fn bar(&self) {} @@ -99,7 +101,7 @@ fn add_missing_impl_members_inner( let impl_node = ctx.find_node_at_offset::()?; let impl_item_list = impl_node.item_list()?; - let trait_def = { + let (trait_, trait_def) = { let analyzer = ctx.source_analyzer(impl_node.syntax(), None); resolve_target_trait_def(ctx.db, &analyzer, &impl_node)? @@ -132,10 +134,25 @@ fn add_missing_impl_members_inner( return None; } + let file_id = ctx.frange.file_id; + let db = ctx.db; + ctx.add_assist(AssistId(assist_id), label, |edit| { let n_existing_items = impl_item_list.impl_items().count(); + let substs = get_syntactic_substs(impl_node).unwrap_or_default(); + let generic_def: hir::GenericDef = trait_.into(); + let substs_by_param: HashMap<_, _> = generic_def + .params(db) + .into_iter() + // this is a trait impl, so we need to skip the first type parameter -- this is a bit hacky + .skip(1) + .zip(substs.into_iter()) + .collect(); let items = missing_items .into_iter() + .map(|it| { + substitute_type_params(db, hir::InFile::new(file_id.into(), it), &substs_by_param) + }) .map(|it| match it { ast::ImplItem::FnDef(def) => ast::ImplItem::FnDef(add_body(def)), _ => it, @@ -160,13 +177,63 @@ fn add_body(fn_def: ast::FnDef) -> ast::FnDef { } } +// FIXME: It would probably be nicer if we could get this via HIR (i.e. get the +// trait ref, and then go from the types in the substs back to the syntax) +// FIXME: This should be a general utility (not even just for assists) +fn get_syntactic_substs(impl_block: ast::ImplBlock) -> Option> { + let target_trait = impl_block.target_trait()?; + let path_type = match target_trait { + ast::TypeRef::PathType(path) => path, + _ => return None, + }; + let type_arg_list = path_type.path()?.segment()?.type_arg_list()?; + let mut result = Vec::new(); + for type_arg in type_arg_list.type_args() { + let type_arg: ast::TypeArg = type_arg; + result.push(type_arg.type_ref()?); + } + Some(result) +} + +// FIXME: This should be a general utility (not even just for assists) +fn substitute_type_params( + db: &impl HirDatabase, + node: hir::InFile, + substs: &HashMap, +) -> N { + let type_param_replacements = node + .value + .syntax() + .descendants() + .filter_map(ast::TypeRef::cast) + .filter_map(|n| { + let path = match &n { + ast::TypeRef::PathType(path_type) => path_type.path()?, + _ => return None, + }; + let analyzer = hir::SourceAnalyzer::new(db, node.with_value(n.syntax()), None); + let resolution = analyzer.resolve_path(db, &path)?; + match resolution { + hir::PathResolution::TypeParam(tp) => Some((n, substs.get(&tp)?.clone())), + _ => None, + } + }) + .collect::>(); + + if type_param_replacements.is_empty() { + node.value + } else { + edit::replace_descendants(&node.value, type_param_replacements.into_iter()) + } +} + /// Given an `ast::ImplBlock`, resolves the target trait (the one being /// implemented) to a `ast::TraitDef`. fn resolve_target_trait_def( db: &impl HirDatabase, analyzer: &hir::SourceAnalyzer, impl_block: &ast::ImplBlock, -) -> Option { +) -> Option<(hir::Trait, ast::TraitDef)> { let ast_path = impl_block .target_trait() .map(|it| it.syntax().clone()) @@ -174,7 +241,9 @@ fn resolve_target_trait_def( .path()?; match analyzer.resolve_path(db, &ast_path) { - Some(hir::PathResolution::Def(hir::ModuleDef::Trait(def))) => Some(def.source(db).value), + Some(hir::PathResolution::Def(hir::ModuleDef::Trait(def))) => { + Some((def, def.source(db).value)) + } _ => None, } } @@ -280,6 +349,40 @@ impl Foo for S { ); } + #[test] + fn fill_in_type_params_1() { + check_assist( + add_missing_impl_members, + " +trait Foo { fn foo(&self, t: T) -> &T; } +struct S; +impl Foo for S { <|> }", + " +trait Foo { fn foo(&self, t: T) -> &T; } +struct S; +impl Foo for S { + <|>fn foo(&self, t: u32) -> &u32 { unimplemented!() } +}", + ); + } + + #[test] + fn fill_in_type_params_2() { + check_assist( + add_missing_impl_members, + " +trait Foo { fn foo(&self, t: T) -> &T; } +struct S; +impl Foo for S { <|> }", + " +trait Foo { fn foo(&self, t: T) -> &T; } +struct S; +impl Foo for S { + <|>fn foo(&self, t: U) -> &U { unimplemented!() } +}", + ); + } + #[test] fn test_cursor_after_empty_impl_block() { check_assist( diff --git a/crates/ra_assists/src/doc_tests/generated.rs b/crates/ra_assists/src/doc_tests/generated.rs index 4586eeb59..7d84dc8fb 100644 --- a/crates/ra_assists/src/doc_tests/generated.rs +++ b/crates/ra_assists/src/doc_tests/generated.rs @@ -101,26 +101,26 @@ fn doctest_add_impl_default_members() { check( "add_impl_default_members", r#####" -trait T { +trait Trait { Type X; fn foo(&self); fn bar(&self) {} } -impl T for () { +impl Trait for () { Type X = (); fn foo(&self) {}<|> } "#####, r#####" -trait T { +trait Trait { Type X; fn foo(&self); fn bar(&self) {} } -impl T for () { +impl Trait for () { Type X = (); fn foo(&self) {} fn bar(&self) {} @@ -135,25 +135,25 @@ fn doctest_add_impl_missing_members() { check( "add_impl_missing_members", r#####" -trait T { +trait Trait { Type X; - fn foo(&self); + fn foo(&self) -> T; fn bar(&self) {} } -impl T for () {<|> +impl Trait for () {<|> } "#####, r#####" -trait T { +trait Trait { Type X; - fn foo(&self); + fn foo(&self) -> T; fn bar(&self) {} } -impl T for () { - fn foo(&self) { unimplemented!() } +impl Trait for () { + fn foo(&self) -> u32 { unimplemented!() } } "#####, -- cgit v1.2.3