6 edit::{AstNodeEdit, IndentLevel},
13 assist_context::{AssistContext, Assists},
14 utils::invert_boolean_expression,
18 // Assist: convert_while_to_loop
20 // Replace a while with a loop.
40 pub(crate) fn convert_while_to_loop(acc: &mut Assists, ctx: &AssistContext) -> Option<()> {
41 let while_kw = ctx.find_token_syntax_at_offset(T![while])?;
42 let while_expr: ast::WhileExpr = while_kw.parent().and_then(ast::WhileExpr::cast)?;
43 let while_body = while_expr.loop_body()?;
44 let cond = while_expr.condition()?;
46 // Don't handle while let
47 if let Some(_) = cond.pat() {
51 let cond_expr = cond.expr()?;
53 let target = while_expr.syntax().text_range();
55 AssistId("convert_while_to_loop", AssistKind::RefactorRewrite),
56 "Convert while to loop",
59 let while_indent_level = IndentLevel::from_node(while_expr.syntax());
63 let cond = invert_boolean_expression(cond_expr);
64 let then_branch = make::block_expr(
65 once(make::expr_stmt(make::expr_break(None)).into()),
69 make::expr_if(make::condition(cond, None), then_branch, None)
72 let if_expr = if_expr.indent(while_indent_level);
73 let stmts = once(make::expr_stmt(if_expr).into()).chain(while_body.statements());
75 let block_expr = make::block_expr(stmts, while_body.tail_expr());
77 let block_expr = block_expr.indent(while_indent_level);
79 make::expr_loop(block_expr)
82 edit.replace(target, replacement.syntax().text())
89 use crate::tests::{check_assist, check_assist_not_applicable};
94 fn convert_inside_fn() {
96 convert_while_to_loop,
118 fn convert_busy_wait() {
120 convert_while_to_loop,
139 fn convert_trailing_expr() {
141 convert_while_to_loop,
163 fn ignore_while_let() {
164 check_assist_not_applicable(
165 convert_while_to_loop,
168 while$0 let Some(_) = foo() {
177 fn ignore_cursor_in_body() {
178 check_assist_not_applicable(
179 convert_while_to_loop,