WI #2608 Protect against null references in GetCompletionPerformParag…

…raphAndSection
TypeCobolTeam · Jan 15, 2024 · b6531e6 · b6531e6
1 parent c720cf8
commit b6531e6
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/TypeCobol.LanguageServer/Completion Factory/CompletionFactory.cs b/TypeCobol.LanguageServer/Completion Factory/CompletionFactory.cs
@@ -44,8 +44,16 @@ public static List<CompletionItem> GetCompletionPerformParagraphAndSection(FileC
                 }
             }
 
-            completionItems.AddRange(paragraphs.Select(para => new CompletionItem(para.Name) { kind = CompletionItemKind.Reference }));
-            completionItems.AddRange(sections.Select(s => new CompletionItem(s.Name) { kind = CompletionItemKind.Reference }));
+            if (paragraphs != null)
+            {
+                completionItems.AddRange(paragraphs.Select(para => new CompletionItem(para.Name) { kind = CompletionItemKind.Reference }));
+            }
+
+            if (sections != null)
+            {
+                completionItems.AddRange(sections.Select(s => new CompletionItem(s.Name) { kind = CompletionItemKind.Reference }));
+            }
+
             if (variables != null)
             {
                 foreach (var variable in variables)