vala: Ensure that the CodeContext has no reference to the node before removing it

diff --git a/vala/valasourcefile.vala b/vala/valasourcefile.vala
index 35eaf5f266d977c4042c27dffd0e5e3cc85a5208..acaea0d7c444fa899681c10ba4ccad4e2df5481d 100644 (file)
--- a/vala/valasourcefile.vala
+++ b/vala/valasourcefile.vala
@@ -197,6 +197,13 @@ public class Vala.SourceFile {
        }
 
        public void remove_node (CodeNode node) {
+               // Also update the entry point of the current code context
+               var code_context = CodeContext.get ();
+               unowned Vala.Method? entry_point = code_context.entry_point;
+               if (entry_point != null && entry_point.source_reference == node.source_reference) {
+                       code_context.entry_point = null;
+               }
+
                nodes.remove (node);
        }