closes #436

Author: retailcoder

RetailCoder.VBE/Extensions/VbeExtensions.cs

@@ -9,16 +9,15 @@ public static class VbeExtensions
     {
         public static void SetSelection(this VBE vbe, QualifiedSelection selection)
         {
-            //not a very robust method. Breaks if there are multiple projects with the same name.
             var project = vbe.VBProjects.Cast<VBProject>()
-                            .FirstOrDefault(p => p.Protection != vbext_ProjectProtection.vbext_pp_locked 
-                                && p.Equals(selection.QualifiedName.Project));
+                             .FirstOrDefault(p => p.Protection != vbext_ProjectProtection.vbext_pp_locked 
+                                               && p.Equals(selection.QualifiedName.Project));
 
             VBComponent component = null;
             if (project != null)
             {
                 component = project.VBComponents.Cast<VBComponent>()
-                                .FirstOrDefault(c => c.Equals(selection.QualifiedName.Component));
+                                   .FirstOrDefault(c => c.Equals(selection.QualifiedName.Component));
             }
 
             if (component == null)

RetailCoder.VBE/Inspections/VariableTypeNotDeclaredInspectionResult.cs

@@ -26,13 +26,8 @@ public override IDictionary<string, Action<VBE>> GetQuickFixes()
 
         private void DeclareAsExplicitVariant(VBE vbe)
         {
-            var component = FindComponent(vbe);
-            if (component == null)
-            {
-                throw new InvalidOperationException("'" + QualifiedName + "' not found.");
-            }
-
-            var codeLine = component.CodeModule.get_Lines(QualifiedSelection.Selection.StartLine, QualifiedSelection.Selection.LineCount);
+            var codeModule = QualifiedSelection.QualifiedName.Component.CodeModule;
+            var codeLine = codeModule.get_Lines(QualifiedSelection.Selection.StartLine, QualifiedSelection.Selection.LineCount);
 
             // methods return empty string if soft-cast context is null - just concat results:
             string originalInstruction;
@@ -49,7 +44,7 @@ private void DeclareAsExplicitVariant(VBE vbe)
             }
 
             var fixedCodeLine = codeLine.Replace(originalInstruction, fix);
-            component.CodeModule.ReplaceLine(QualifiedSelection.Selection.StartLine, fixedCodeLine);
+            codeModule.ReplaceLine(QualifiedSelection.Selection.StartLine, fixedCodeLine);
         }
 
         private string DeclareExplicitVariant(VBAParser.VariableSubStmtContext context, out string instruction)