From 5817295f7c802547e2fb3ca58afc1d63dc422902 Mon Sep 17 00:00:00 2001
From: azakhi <azak.halilibrahim@gmail.com>
Date: Sun, 12 Aug 2018 11:39:36 +0300
Subject: [PATCH] Fix adding block to a 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 9b41df8..dc333a2 100644
--- a/Filtration/ViewModels/ItemFilterScriptViewModel.cs
+++ b/Filtration/ViewModels/ItemFilterScriptViewModel.cs
@@ -1037,7 +1037,21 @@ namespace Filtration.ViewModels
 
         private void OnAddBlockCommand()
         {
-            AddBlock(SelectedBlockViewModel);
+            var selectedBlockAsCommentBlock = SelectedBlockViewModel as IItemFilterCommentBlockViewModel;
+            if(selectedBlockAsCommentBlock == null || selectedBlockAsCommentBlock.IsExpanded)
+            {
+                AddBlock(SelectedBlockViewModel);
+            }
+            else
+            {
+                var sectionStart = ItemFilterBlockViewModels.IndexOf(selectedBlockAsCommentBlock);
+                var sectionEnd = sectionStart + 1;
+                while (sectionEnd < ItemFilterBlockViewModels.Count && ItemFilterBlockViewModels[sectionEnd] as IItemFilterCommentBlockViewModel == null)
+                {
+                    sectionEnd++;
+                }
+                AddBlock(ItemFilterBlockViewModels[sectionEnd - 1]);
+            }
         }
 
         public void AddBlock(IItemFilterBlockViewModelBase targetBlockViewModelBase)