From 1c059050585e8610c69a91cea33ef1e01ca88a61 Mon Sep 17 00:00:00 2001
From: Brian Fiete <bfiete@gmail.com>
Date: Wed, 2 Mar 2022 07:53:18 -0800
Subject: [PATCH] Handle `while` collapse

---
 IDEHelper/Compiler/BfCompiler.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/IDEHelper/Compiler/BfCompiler.cpp b/IDEHelper/Compiler/BfCompiler.cpp
index cc14c865..b4188ce5 100644
--- a/IDEHelper/Compiler/BfCompiler.cpp
+++ b/IDEHelper/Compiler/BfCompiler.cpp
@@ -9312,6 +9312,13 @@ BF_EXPORT const char* BF_CALLTYPE BfCompiler_GetCollapseRegions(BfCompiler* bfCo
 			BfElementVisitor::Visit(repeatStatement);
 		}
 
+		virtual void Visit(BfWhileStatement* whileStatement) override
+		{
+			Add(whileStatement->mWhileToken, whileStatement->mEmbeddedStatement);
+
+			BfElementVisitor::Visit(whileStatement);
+		}
+
 		virtual void Visit(BfDoStatement* doStatement) override
 		{
 			Add(doStatement->mDoToken, doStatement->mEmbeddedStatement);