Fix body selection in while loops

2022-02-26 16:45:06 +02:00
parent a2cc1d6b7b
commit 87c4a4104d
4 changed files with 34 additions and 3 deletions
--- a/crates/syntax/src/ast/node_ext.rs
+++ b/crates/syntax/src/ast/node_ext.rs
@@ -769,6 +769,15 @@ impl ast::HasLoopBody for ast::ForExpr {
    }
 }

+impl ast::HasLoopBody for ast::WhileExpr {
+    fn loop_body(&self) -> Option<ast::BlockExpr> {
+        let mut exprs = support::children(self.syntax());
+        let first = exprs.next();
+        let second = exprs.next();
+        second.or(first)
+    }
+}
+
 impl ast::HasAttrs for ast::AnyHasDocComments {}

 impl From<ast::Adt> for ast::Item {