Skip to content

Commit

Permalink
Auto merge of #15743 - Young-Flash:replace_is_some_with_if_let_some, …
Browse files Browse the repository at this point in the history
…r=HKalbasi

add replace_is_some_with_if_let_some assist

This is a basic impl, if it is ok, `if my_result.is_ok()`-> `if let Ok(_x) = my_result ​`will be add next.

close #12977
  • Loading branch information
bors committed Oct 12, 2023
2 parents 7eec17e + 3605bb3 commit 84e1d27
Show file tree
Hide file tree
Showing 3 changed files with 111 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,90 @@
use syntax::ast::{self, AstNode};

use crate::{AssistContext, AssistId, AssistKind, Assists};

// Assist: replace_is_some_with_if_let_some
//
// Replace `if x.is_some()` with `if let Some(_tmp) = x`.
//
// ```
// fn main() {
// let x = Some(1);
// if x.is_som$0e() {}
// }
// ```
// ->
// ```
// fn main() {
// let x = Some(1);
// if let Some(_tmp) = x {}
// }
// ```
pub(crate) fn replace_is_some_with_if_let_some(
acc: &mut Assists,
ctx: &AssistContext<'_>,
) -> Option<()> {
let if_expr = ctx.find_node_at_offset::<ast::IfExpr>()?;

let cond = if_expr.condition()?;
let call_expr = match cond {
ast::Expr::MethodCallExpr(call) => call,
_ => return None,
};

let name_ref = call_expr.name_ref()?;
if name_ref.text() != "is_some" {
return None;
}

let receiver = call_expr.receiver()?;
let target = call_expr.syntax().text_range();

acc.add(
AssistId("replace_is_some_with_if_let_some", AssistKind::RefactorRewrite),
"Replace `is_some` with `if let Some`",
target,
|edit| {
let replacement = format!("let Some(_tmp) = {}", receiver);
edit.replace(target, replacement);
},
)
}

#[cfg(test)]
mod tests {
use crate::tests::{check_assist, check_assist_not_applicable};

use super::replace_is_some_with_if_let_some;

#[test]
fn replace_is_some_with_if_let_some_works() {
check_assist(
replace_is_some_with_if_let_some,
r#"
fn main() {
let x = Some(1);
if x.is_som$0e() {}
}
"#,
r#"
fn main() {
let x = Some(1);
if let Some(_tmp) = x {}
}
"#,
);
}

#[test]
fn replace_is_some_with_if_let_some_not_applicable() {
check_assist_not_applicable(
replace_is_some_with_if_let_some,
r#"
fn main() {
let x = Some(1);
if x.is_non$0e() {}
}
"#,
);
}
}
2 changes: 2 additions & 0 deletions crates/ide-assists/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -213,6 +213,7 @@ mod handlers {
mod unwrap_block;
mod unwrap_result_return_type;
mod unqualify_method_call;
mod replace_is_some_with_if_let_some;
mod wrap_return_type_in_result;
mod into_to_qualified_from;

Expand Down Expand Up @@ -332,6 +333,7 @@ mod handlers {
unwrap_result_return_type::unwrap_result_return_type,
unwrap_tuple::unwrap_tuple,
unqualify_method_call::unqualify_method_call,
replace_is_some_with_if_let_some::replace_is_some_with_if_let_some,
wrap_return_type_in_result::wrap_return_type_in_result,
// These are manually sorted for better priorities. By default,
// priority is determined by the size of the target range (smaller
Expand Down
19 changes: 19 additions & 0 deletions crates/ide-assists/src/tests/generated.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2558,6 +2558,25 @@ fn handle(action: Action) {
)
}

#[test]
fn doctest_replace_is_some_with_if_let_some() {
check_doc_test(
"replace_is_some_with_if_let_some",
r#####"
fn main() {
let x = Some(1);
if x.is_som$0e() {}
}
"#####,
r#####"
fn main() {
let x = Some(1);
if let Some(_tmp) = x {}
}
"#####,
)
}

#[test]
fn doctest_replace_let_with_if_let() {
check_doc_test(
Expand Down

0 comments on commit 84e1d27

Please sign in to comment.