Set up reference removal and modules to reresolve in the ParseCoordinator. (Not yet activated.)

Author: MDoerner

Rubberduck.Parsing/VBA/ParseCoordinator.cs

@@ -214,10 +214,8 @@ private void ExecuteCommonParseActivities(List<IVBComponent> toParse, Cancellati
 
                 token.ThrowIfCancellationRequested();
 
-            _projectDeclarations.Clear();
-            State.ClearAllReferences();
-
-            ClearModuleToModuleReferences(toParse);
+            var toResolveReferences = ModulesForWhichToResolveReferences(toParse);
+            PerformPreParseCleanup(toParse, toResolveReferences);
 
             ParseComponents(toParse, token);
 
@@ -241,7 +239,7 @@ private void ExecuteCommonParseActivities(List<IVBComponent> toParse, Cancellati
                     throw new OperationCanceledException(token);
                 }
 
-            var toResolveReferences = State.ParseTrees.Select(kvp => kvp.Key).ToHashSet();
+            toResolveReferences = State.ParseTrees.Select(kvp => kvp.Key).ToHashSet();
             ResolveAllReferences(toResolveReferences, token);
 
                 if (token.IsCancellationRequested || State.Status >= ParserState.Error)
@@ -271,6 +269,24 @@ private void SetModuleStates(List<IVBComponent> components, ParserState parserSt
                 }
         }
 
+        private ICollection<QualifiedModuleName> ModulesForWhichToResolveReferences(List<IVBComponent> toParse)
+        {
+            var toResolveReferences = new HashSet<QualifiedModuleName>();
+            foreach (var qmn in toParse.Select(component => new QualifiedModuleName(component)))
+            { 
+                toResolveReferences.UnionWith(State.ModulesReferencing(qmn));
+            }
+            return toResolveReferences;
+        }
+
+        private void PerformPreParseCleanup(List<IVBComponent> toParse, ICollection<QualifiedModuleName> toResolveReferences)
+        {
+            ClearModuleToModuleReferences(toParse);
+            State.RemoveAllReferencesBy(toResolveReferences);
+            State.ClearAllReferences(); //This has to stay here until the selective resolving actually works.
+            _projectDeclarations.Clear();
+        }
+
         private void ClearModuleToModuleReferences(List<IVBComponent> toClear)
         {
             foreach (var component in toClear)
@@ -711,10 +727,12 @@ private bool CleanUpRemovedComponents(List<IVBComponent> components)
         {
             var removedModuledecalrations = RemovedModuleDeclarations(components);
             var componentRemoved = removedModuledecalrations.Any();
-            foreach (var declaration in removedModuledecalrations)
+            var removedModules = removedModuledecalrations.Select(declaration => declaration.QualifiedName.QualifiedModuleName).ToHashSet();
+            State.RemoveAllReferencesBy(removedModules);
+            foreach (var qmn in removedModules)
             {
-                State.ClearModuleToModuleReferencesFromModule(declaration.QualifiedName.QualifiedModuleName);
-                State.ClearStateCache(declaration.QualifiedName.QualifiedModuleName);
+                State.ClearModuleToModuleReferencesFromModule(qmn);
+                State.ClearStateCache(qmn);
             }
             return componentRemoved;
         }