bendns' repos / rust-analyzer
Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/ide-assists/src/handlers/convert_for_to_while_let.rs')
-rw-r--r--crates/ide-assists/src/handlers/convert_for_to_while_let.rs38
1 files changed, 18 insertions, 20 deletions
diff --git a/crates/ide-assists/src/handlers/convert_for_to_while_let.rs b/crates/ide-assists/src/handlers/convert_for_to_while_let.rs
index 07ebe981b4..9eb4c0584b 100644
--- a/crates/ide-assists/src/handlers/convert_for_to_while_let.rs
+++ b/crates/ide-assists/src/handlers/convert_for_to_while_let.rs
@@ -48,24 +48,22 @@ pub(crate) fn convert_for_loop_to_while_let(
"Replace this for loop with `while let`",
for_loop.syntax().text_range(),
|builder| {
- let mut editor = builder.make_editor(for_loop.syntax());
+ let editor = builder.make_editor(for_loop.syntax());
+ let make = editor.make();
let (iterable, method) = if impls_core_iter(&ctx.sema, &iterable) {
(iterable, None)
} else if let Some((expr, method)) = is_ref_and_impls_iter_method(&ctx.sema, &iterable)
{
- (expr, Some(editor.make().name_ref(method.as_str())))
+ (expr, Some(make.name_ref(method.as_str())))
} else if let ast::Expr::RefExpr(_) = iterable {
- (
- editor.make().expr_paren(iterable).into(),
- Some(editor.make().name_ref("into_iter")),
- )
+ (make.expr_paren(iterable).into(), Some(make.name_ref("into_iter")))
} else {
- (iterable, Some(editor.make().name_ref("into_iter")))
+ (iterable, Some(make.name_ref("into_iter")))
};
let iterable = if let Some(method) = method {
- editor.make().expr_method_call(iterable, method, editor.make().arg_list([])).into()
+ make.expr_method_call(iterable, method, make.arg_list([])).into()
} else {
iterable
};
@@ -75,8 +73,8 @@ pub(crate) fn convert_for_loop_to_while_let(
);
let tmp_var = new_name.suggest_name("iter");
- let mut_expr = editor.make().let_stmt(
- editor.make().ident_pat(false, true, editor.make().name(&tmp_var)).into(),
+ let mut_expr = make.let_stmt(
+ make.ident_pat(false, true, make.name(&tmp_var)).into(),
None,
Some(iterable),
);
@@ -84,26 +82,26 @@ pub(crate) fn convert_for_loop_to_while_let(
if let Some(label) = for_loop.label() {
let label = label.syntax();
- editor.insert(Position::before(for_loop.syntax()), editor.make().whitespace(" "));
+ editor.insert(Position::before(for_loop.syntax()), make.whitespace(" "));
editor.insert(Position::before(for_loop.syntax()), label);
}
- crate::utils::insert_attributes(for_loop.syntax(), &mut editor, for_loop.attrs());
+ crate::utils::insert_attributes(for_loop.syntax(), &editor, for_loop.attrs());
editor.insert(
Position::before(for_loop.syntax()),
- editor.make().whitespace(format!("\n{indent}").as_str()),
+ make.whitespace(format!("\n{indent}").as_str()),
);
editor.insert(Position::before(for_loop.syntax()), mut_expr.syntax());
- let opt_pat = editor.make().tuple_struct_pat(editor.make().ident_path("Some"), [pat]);
- let iter_next_expr = editor.make().expr_method_call(
- editor.make().expr_path(editor.make().ident_path(&tmp_var)),
- editor.make().name_ref("next"),
- editor.make().arg_list([]),
+ let opt_pat = make.tuple_struct_pat(make.ident_path("Some"), [pat]);
+ let iter_next_expr = make.expr_method_call(
+ make.expr_path(make.ident_path(&tmp_var)),
+ make.name_ref("next"),
+ make.arg_list([]),
);
- let cond = editor.make().expr_let(opt_pat.into(), iter_next_expr.into());
+ let cond = make.expr_let(opt_pat.into(), iter_next_expr.into());
- let while_loop = editor.make().expr_while_loop(cond.into(), body);
+ let while_loop = make.expr_while_loop(cond.into(), body);
editor.replace(for_loop.syntax(), while_loop.syntax());