Mirrors/rust-analyzer
2
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer synced 2024-11-10 07:04:22 +00:00
Code Issues Projects Releases Packages Wiki Activity

method gen assist usable in all of expression

Browse source
This commit is contained in:
mahdi-frms 2021-08-09 19:21:15 +04:30
parent 311dc5b04f
commit 9217f6dcf7
1 changed files with 33 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
crates/ide_assists/src/handlers/generate_function.rs
View file

@ -101,8 +101,10 @@ pub(crate) fn generate_function(acc: &mut Assists, ctx: &AssistContext) -> Optio
} }
pub(crate) fn generate_method(acc: &mut Assists, ctx: &AssistContext) -> Option<()> { pub(crate) fn generate_method(acc: &mut Assists, ctx: &AssistContext) -> Option<()> {
let fn_name: ast::NameRef = ctx.find_node_at_offset()?;
let call: ast::MethodCallExpr = ctx.find_node_at_offset()?; let call: ast::MethodCallExpr = ctx.find_node_at_offset()?;
let fn_name: ast::NameRef = ast::NameRef::cast(
call.syntax().children().find(|child| child.kind() == SyntaxKind::NAME_REF)?,
)?;
let ty = ctx.sema.type_of_expr(&call.receiver()?)?.original().strip_references().as_adt()?; let ty = ctx.sema.type_of_expr(&call.receiver()?)?.original().strip_references().as_adt()?;
let current_module = let current_module =
@ -1498,6 +1500,36 @@ mod s {
impl S { impl S {
fn bar(&self) ${0:-> ()} {
todo!()
}
}
",
)
}
#[test]
fn create_method_with_cursor_anywhere_on_call_expresion() {
check_assist(
generate_method,
r"
struct S;
fn foo() {
$0S.bar();
}
",
r"
struct S;
fn foo() {
S.bar();
}
impl S {
fn bar(&self) ${0:-> ()} { fn bar(&self) ${0:-> ()} {
todo!() todo!()
} }