Fix GenerateFoldMarkers

punker76 · Nov 22, 2017 · fe2d448 · fe2d448
1 parent 50024b0
commit fe2d448
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/Kaxaml/CodeCompletion/XmlFoldingStrategy.cs b/Kaxaml/CodeCompletion/XmlFoldingStrategy.cs
@@ -120,9 +120,9 @@ public List<FoldMarker> GenerateFoldMarkers(IDocument document, string fileName,
                 {
                     using (var reader = new XmlTextReader(stringReader))
                     {
+                        var stack = new Stack();
                         while (reader.Read())
                         {
-                            var stack = new Stack();
                             switch (reader.NodeType)
                             {
                                 case XmlNodeType.Element:
@@ -134,8 +134,11 @@ public List<FoldMarker> GenerateFoldMarkers(IDocument document, string fileName,
                                     break;
 
                                 case XmlNodeType.EndElement:
-                                    XmlFoldStart foldStart = (XmlFoldStart)stack.Pop();
-                                    CreateElementFold(document, foldMarkers, reader, foldStart);
+                                    if (stack.Count > 0)
+                                    {
+                                        XmlFoldStart foldStart = (XmlFoldStart)stack.Pop();
+                                        CreateElementFold(document, foldMarkers, reader, foldStart);
+                                    }
                                     break;
 
                                 case XmlNodeType.Comment: