This commit is contained in:
Alex Vasilev 2023-01-21 23:42:11 +05:30
parent dd9dcca7a2
commit 2a4837089e

View File

@ -34,7 +34,7 @@ pub(crate) fn add_braces(acc: &mut Assists, ctx: &AssistContext<'_>) -> Option<(
acc.add( acc.add(
AssistId("add_braces", AssistKind::RefactorRewrite), AssistId("add_braces", AssistKind::RefactorRewrite),
match expr_type { match expr_type {
ParentType::ClosureExpr => "Add braces to lambda expression", ParentType::ClosureExpr => "Add braces to closure body",
ParentType::MatchArmExpr => "Add braces to arm expression", ParentType::MatchArmExpr => "Add braces to arm expression",
}, },
expr.syntax().text_range(), expr.syntax().text_range(),
@ -46,9 +46,7 @@ pub(crate) fn add_braces(acc: &mut Assists, ctx: &AssistContext<'_>) -> Option<(
builder.replace(expr.syntax().text_range(), block_expr.syntax().text()); builder.replace(expr.syntax().text_range(), block_expr.syntax().text());
}, },
); )
Some(())
} }
enum ParentType { enum ParentType {
@ -58,7 +56,7 @@ enum ParentType {
fn get_replacement_node(ctx: &AssistContext<'_>) -> Option<(ParentType, ast::Expr)> { fn get_replacement_node(ctx: &AssistContext<'_>) -> Option<(ParentType, ast::Expr)> {
if let Some(match_arm) = ctx.find_node_at_offset::<ast::MatchArm>() { if let Some(match_arm) = ctx.find_node_at_offset::<ast::MatchArm>() {
let match_arm_expr = match_arm.syntax().children().find_map(ast::Expr::cast)?; let match_arm_expr = match_arm.expr()?;
if matches!(match_arm_expr, ast::Expr::BlockExpr(_)) { if matches!(match_arm_expr, ast::Expr::BlockExpr(_)) {
return None; return None;