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 = while_kw.parent().and_then(ast::WhileExpr::cast)?;
43 let while_body = while_expr.loop_body()?;
44 let while_cond = while_expr.condition()?;
45 let while_cond_expr = while_cond.expr()?;
47 let target = while_expr.syntax().text_range();
49 AssistId("convert_while_to_loop", AssistKind::RefactorRewrite),
50 "Convert while to loop",
53 let while_indent_level = IndentLevel::from_node(while_expr.syntax());
56 make::block_expr(once(make::expr_stmt(make::expr_break(None)).into()), None)
57 .indent(while_indent_level);
58 let block_expr = match while_cond.pat() {
60 let if_expr = make::expr_if(while_cond, while_body, Some(break_block.into()));
61 let stmts = once(make::expr_stmt(if_expr).into());
62 make::block_expr(stmts, None)
65 let if_cond = make::condition(invert_boolean_expression(while_cond_expr), None);
66 let if_expr = make::expr_if(if_cond, break_block, None);
68 once(make::expr_stmt(if_expr).into()).chain(while_body.statements());
69 make::block_expr(stmts, while_body.tail_expr())
73 let replacement = make::expr_loop(block_expr.indent(while_indent_level));
74 edit.replace(target, replacement.syntax().text())
81 use crate::tests::{check_assist, check_assist_not_applicable};
86 fn convert_inside_fn() {
88 convert_while_to_loop,
110 fn convert_busy_wait() {
112 convert_while_to_loop,
131 fn convert_trailing_expr() {
133 convert_while_to_loop,
155 fn convert_while_let() {
157 convert_while_to_loop,
160 while$0 let Some(_) = foo() {
168 if let Some(_) = foo() {
180 fn ignore_cursor_in_body() {
181 check_assist_not_applicable(
182 convert_while_to_loop,