From 281c7d85e11e4122d9ae639daf472d616cd7faa8 Mon Sep 17 00:00:00 2001
From: azakhi <azak.halilibrahim@gmail.com>
Date: Sun, 19 Aug 2018 13:19:14 +0300
Subject: [PATCH] Fix adding comment block to collapsed section

---
 .../ViewModels/ItemFilterScriptViewModel.cs      | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/Filtration/ViewModels/ItemFilterScriptViewModel.cs b/Filtration/ViewModels/ItemFilterScriptViewModel.cs
index 06e99e3..88428c5 100644
--- a/Filtration/ViewModels/ItemFilterScriptViewModel.cs
+++ b/Filtration/ViewModels/ItemFilterScriptViewModel.cs
@@ -1141,7 +1141,21 @@ namespace Filtration.ViewModels
 
         private void OnAddCommentBlockCommand()
         {
-            AddCommentBlock(SelectedBlockViewModel);
+            var selectedBlockAsCommentBlock = SelectedBlockViewModel as IItemFilterCommentBlockViewModel;
+            if (selectedBlockAsCommentBlock == null || selectedBlockAsCommentBlock.IsExpanded)
+            {
+                AddCommentBlock(SelectedBlockViewModel);
+            }
+            else
+            {
+                var sectionStart = ItemFilterBlockViewModels.IndexOf(selectedBlockAsCommentBlock);
+                var sectionEnd = sectionStart + 1;
+                while (sectionEnd < ItemFilterBlockViewModels.Count && ItemFilterBlockViewModels[sectionEnd] as IItemFilterCommentBlockViewModel == null)
+                {
+                    sectionEnd++;
+                }
+                AddCommentBlock(ItemFilterBlockViewModels[sectionEnd - 1]);
+            }
         }
 
         private void OnExpandAllBlocksCommand()