From af6ac720c7bd40fda5a77027261377c49e2206a0 Mon Sep 17 00:00:00 2001
From: Brian Fiete <bfiete@gmail.com>
Date: Fri, 18 Mar 2022 18:34:49 -0700
Subject: [PATCH] Fixed formatting of enums with comments

---
 IDEHelper/Compiler/BfPrinter.cpp | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/IDEHelper/Compiler/BfPrinter.cpp b/IDEHelper/Compiler/BfPrinter.cpp
index 0eef2e09..839eb634 100644
--- a/IDEHelper/Compiler/BfPrinter.cpp
+++ b/IDEHelper/Compiler/BfPrinter.cpp
@@ -2354,9 +2354,13 @@ void BfPrinter::Visit(BfConstructorDeclaration* ctorDeclaration)
 {
 	//Visit((BfAstNode*)ctorDeclaration);
 
-	QueueVisitChild(ctorDeclaration->mAttributes);
 	ExpectNewLine();
-	ExpectSpace();
+	if (ctorDeclaration->mAttributes != NULL)
+	{
+		QueueVisitChild(ctorDeclaration->mAttributes);
+		ExpectNewLine();
+	}
+
 	QueueVisitChild(ctorDeclaration->mProtectionSpecifier);	
 	ExpectSpace();
 	QueueVisitChild(ctorDeclaration->mNewSpecifier);
@@ -2950,6 +2954,11 @@ void BfPrinter::Visit(BfTypeDeclaration* typeDeclaration)
 				SetAndRestoreValue<BfAstNode*> prevBlockMember(mCurBlockMember, member);
 				if (auto fieldDecl = BfNodeDynCast<BfFieldDeclaration>(member))
 					ExpectNewLine();
+				else if (auto tokenNode = BfNodeDynCast<BfTokenNode>(member))
+				{
+					mVirtualNewLineIdx = mNextStateModify.mWantNewLineIdx;
+					mNextStateModify.mExpectingSpace = false;
+				}
 				VisitChild(member);
 			}
 			ExpectUnindent();