From dc169fffda13760e46c9c6eb1a9042b420a89007 Mon Sep 17 00:00:00 2001
From: Brian Fiete <bfiete@gmail.com>
Date: Sat, 12 Sep 2020 08:11:29 -0700
Subject: [PATCH] Parser fixes

---
 IDEHelper/Compiler/BfAst.h | 34 +++++++++++++++++++++++++++++++---
 1 file changed, 31 insertions(+), 3 deletions(-)

diff --git a/IDEHelper/Compiler/BfAst.h b/IDEHelper/Compiler/BfAst.h
index acaca210..3e3da0fe 100644
--- a/IDEHelper/Compiler/BfAst.h
+++ b/IDEHelper/Compiler/BfAst.h
@@ -951,20 +951,48 @@ static void BfSizedArrayInitIndirect(BfSizedArray<T>& sizedArray, const SizedArr
 template <typename T>
 class BfDeferredAstSizedArray : public SizedArray<T, 8>
 {
-public:
+public:	
 	BfSizedArray<ASTREF(T)>* mSizedArray;
 	BfAstAllocator* mAlloc;
 
 public:
 	BfDeferredAstSizedArray(BfSizedArray<ASTREF(T)>& arr, BfAstAllocator* alloc)
-	{
+	{		
 		mSizedArray = &arr;
-		mAlloc = alloc;
+		mAlloc = alloc;		
 	}
 
 	~BfDeferredAstSizedArray()
 	{		
+		BfSizedArrayInitIndirect(*mSizedArray, *this, mAlloc);		
+	}
+};
+
+template <typename T>
+class BfDeferredAstNodeSizedArray : public SizedArray<T, 8>
+{
+public:
+	BfAstNode* mParentNode;
+	BfSizedArray<ASTREF(T)>* mSizedArray;
+	BfAstAllocator* mAlloc;
+
+public:
+	BfDeferredAstNodeSizedArray(BfAstNode* parentNode, BfSizedArray<ASTREF(T)>& arr, BfAstAllocator* alloc)
+	{
+		mParentNode = parentNode;
+		mSizedArray = &arr;
+		mAlloc = alloc;
+	}
+
+	~BfDeferredAstNodeSizedArray()
+	{
 		BfSizedArrayInitIndirect(*mSizedArray, *this, mAlloc);
+		if (!mSizedArray->IsEmpty())
+		{
+			int endPos = mSizedArray->back()->mSrcEnd;
+			if (endPos > mParentNode->mSrcEnd)
+				mParentNode->mSrcEnd = endPos;
+		}
 	}
 };