language-servers/rust-analyzer
1
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer.git synced 2025-09-28 04:44:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Introduce and use get_fn_target_info

Browse source
This commit is contained in:
Dorian Scheidt 2022-08-18 18:39:42 -05:00
parent d8e7419c64
commit 48ea3825b8
1 changed files with 16 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

32
crates/ide-assists/src/handlers/generate_function.rs
View file

@ -104,13 +104,10 @@ fn fn_target_info(
call: &CallExpr,
fn_name: &str,
) -> Option<TargetInfo> {
let target_module;
let adt_name = None;
let (target, file, insert_offset) = match path.qualifier() {
match path.qualifier() {
Some(qualifier) => match ctx.sema.resolve_path(&qualifier) {
Some(hir::PathResolution::Def(hir::ModuleDef::Module(module))) => {
target_module = Some(module);
get_fn_target(ctx, &target_module, call.clone())?
get_fn_target_info(ctx, &Some(module), call.clone())
}
Some(hir::PathResolution::Def(hir::ModuleDef::Adt(adt))) => {
if let hir::Adt::Enum(_) = adt {
@ -120,22 +117,16 @@ fn fn_target_info(
}
}
return assoc_fn_target_info(ctx, call, adt, fn_name);
assoc_fn_target_info(ctx, call, adt, fn_name)
}
Some(hir::PathResolution::SelfType(impl_)) => {
let adt = impl_.self_ty(ctx.db()).as_adt()?;
return assoc_fn_target_info(ctx, call, adt, fn_name);
}
_ => {
return None;
assoc_fn_target_info(ctx, call, adt, fn_name)
}
_ => None,
},
_ => {
target_module = None;
get_fn_target(ctx, &target_module, call.clone())?
}
};
Some(TargetInfo::new(target_module, adt_name, target, file, insert_offset))
_ => get_fn_target_info(ctx, &None, call.clone()),
}
}
fn gen_method(acc: &mut Assists, ctx: &AssistContext<'_>) -> Option<()> {
@ -391,6 +382,15 @@ fn make_return_type(
(ret_type, should_focus_return_type)
}
fn get_fn_target_info(
ctx: &AssistContext<'_>,
target_module: &Option<Module>,
call: CallExpr,
) -> Option<TargetInfo> {
let (target, file, insert_offset) = get_fn_target(ctx, target_module, call)?;
Some(TargetInfo::new(*target_module, None, target, file, insert_offset))
}
fn get_fn_target(
ctx: &AssistContext<'_>,
target_module: &Option<Module>,