Closes #3260

Author: mansellan

RetailCoder.VBE/Root/RubberduckModule.cs

@@ -535,7 +535,8 @@ private IEnumerable<IMenuItem> GetFormDesignerContextMenuItems()
         {
             return new IMenuItem[]
             {
-                KernelInstance.Get<FormDesignerRefactorRenameCommandMenuItem>()
+                KernelInstance.Get<FormDesignerRefactorRenameCommandMenuItem>(),
+                KernelInstance.Get<FormDesignerFindAllReferencesCommandMenuItem>()
             };
         }
 

RetailCoder.VBE/Rubberduck.csproj

@@ -472,7 +472,9 @@
     <Compile Include="UI\CodeExplorer\Commands\AddClassModuleCommand.cs" />
     <Compile Include="UI\CodeExplorer\Commands\AddStdModuleCommand.cs" />
     <Compile Include="UI\CodeExplorer\Commands\AddTestModuleCommand.cs" />
+    <Compile Include="UI\Command\FormDesignerFindAllReferencesCommand.cs" />
     <Compile Include="UI\Command\IndentCurrentProjectCommand.cs" />
+    <Compile Include="UI\Command\MenuItems\FormDesignerFindAllReferencesCommandMenuItem.cs" />
     <Compile Include="UI\Command\MenuItems\IndentCurrentProjectCommandMenuItem.cs" />
     <Compile Include="UI\EnvironmentProvider.cs" />
     <Compile Include="UI\ModernFolderBrowser.cs" />

RetailCoder.VBE/UI/Command/FindAllReferencesCommand.cs

@@ -87,11 +87,13 @@ private void UpdateTab()
 
         protected override bool CanExecuteImpl(object parameter)
         {
-            if (_vbe.ActiveCodePane == null || _state.Status != ParserState.Ready)
+            if (_state.Status != ParserState.Ready ||
+                (_vbe.ActiveCodePane == null 
+                    && (_vbe.SelectedVBComponent == null || !_vbe.SelectedVBComponent.HasDesigner)))
             {
                 return false;
             }
-            
+
             var target = FindTarget(parameter);
             var canExecute = target != null;
 
@@ -153,6 +155,14 @@ private SearchResultsViewModel CreateViewModel(Declaration declaration)
         }
 
         private Declaration FindTarget(object parameter)
+        {
+            var component = _vbe.SelectedVBComponent;
+            return (component != null && component.HasDesigner)
+                ? FindFormTarget()
+                : FindCodeTarget(parameter);
+        }
+
+        private Declaration FindCodeTarget(object parameter)
         {
             var declaration = parameter as Declaration;
             if (declaration != null)
@@ -163,6 +173,30 @@ private Declaration FindTarget(object parameter)
             return _state.FindSelectedDeclaration(_vbe.ActiveCodePane);
         }
 
+        private Declaration FindFormTarget()
+        {            
+            var project = _vbe.ActiveVBProject;
+            var component = _vbe.SelectedVBComponent;
+            
+            if (component != null && _vbe.SelectedVBComponent.HasDesigner)
+            {
+                var designer = ((dynamic)component.Target).Designer;
+                var selectedCount = (int)designer.Selected.Count;
+
+                if (selectedCount > 1) { return null; }
+
+                var selectedType = selectedCount == 0 ? DeclarationType.ClassModule : DeclarationType.Control;
+                string selectedName = selectedCount == 0 ? component.Name : designer.Selected[0].Name;
+
+                return _state.DeclarationFinder.MatchName(selectedName)
+                                               .First(m => m.ProjectId == project.HelpFile
+                                                        && m.DeclarationType.HasFlag(selectedType)
+                                                        && m.ComponentName == component.Name);                
+            }
+            return null;
+        }
+
+
         public void Dispose()
         {
             if (_state != null)

RetailCoder.VBE/UI/Command/FormDesignerFindAllReferencesCommand.cs

@@ -0,0 +1,32 @@
+using System.Runtime.InteropServices;
+using NLog;
+using Rubberduck.Parsing.VBA;
+using Rubberduck.VBEditor.SafeComWrappers.Abstract;
+
+namespace Rubberduck.UI.Command
+{
+    /// <summary>
+    /// A command that locates all references to the active form designer component.
+    /// </summary>
+    [ComVisible(false)]
+    public class FormDesignerFindAllReferencesCommand : CommandBase
+    {
+        private readonly FindAllReferencesCommand _findAllReferences;
+
+        public FormDesignerFindAllReferencesCommand(FindAllReferencesCommand findAllReferences)
+             : base(LogManager.GetCurrentClassLogger())
+        {
+            _findAllReferences = findAllReferences;
+        }
+
+        protected override bool CanExecuteImpl(object parameter)
+        {
+            return _findAllReferences.CanExecute(parameter);
+        }
+
+        protected override void ExecuteImpl(object parameter)
+        {
+            _findAllReferences.Execute(parameter);
+        }
+    }
+}

RetailCoder.VBE/UI/Command/MenuItems/FormDesignerFindAllReferencesCommandMenuItem.cs

@@ -0,0 +1,22 @@
+using Rubberduck.Parsing.VBA;
+using Rubberduck.UI.Command.MenuItems.ParentMenus;
+
+namespace Rubberduck.UI.Command.MenuItems
+{
+    public class FormDesignerFindAllReferencesCommandMenuItem : CommandMenuItemBase
+    {
+        public FormDesignerFindAllReferencesCommandMenuItem(CommandBase command)
+            : base(command)
+        {
+        }
+
+        public override bool BeginGroup { get { return true; } }
+        public override string Key { get { return "ContextMenu_FindAllReferences"; } }
+        public override int DisplayOrder { get { return (int)NavigationMenuItemDisplayOrder.FindAllReferences; } }
+
+        public override bool EvaluateCanExecute(RubberduckParserState state)
+        {            
+            return state != null && Command.CanExecute(null);
+        }
+    }
+}

RetailCoder.VBE/UI/SelectionChangeService.cs

@@ -123,7 +123,7 @@ private void DispatchSelectedDesignerDeclaration(IVBComponent component)
                 return;
             }
             var form =
-                _parser.State.DeclarationFinder.UserDeclarations(DeclarationType.UserForm).SingleOrDefault(decl =>
+                _parser.State.DeclarationFinder.UserDeclarations(DeclarationType.ClassModule).SingleOrDefault(decl =>
                     decl.IdentifierName.Equals(component.Name) &&
                     decl.ProjectId.Equals(component.ParentProject.ProjectId));
 
@@ -166,7 +166,7 @@ private void DispatchSelectedProjectNodeDeclaration(IVBComponent component)
         private bool DeclarationChanged(Declaration current)
         {
             if ((_lastSelectedDeclaration == null && current == null) ||
-                ((_lastSelectedDeclaration != null && current != null) && !_lastSelectedDeclaration.Equals(current)))
+                ((_lastSelectedDeclaration != null && current != null) && _lastSelectedDeclaration.Equals(current)))
             {
                 return false;
             }

Rubberduck.VBEEditor/Rubberduck.VBEditor.csproj

@@ -128,6 +128,7 @@
     <Compile Include="Events\VBENativeServices.cs" />
     <Compile Include="Events\WindowChangedEventArgs.cs" />
     <Compile Include="Extensions\MSAccessComponentTypeExtensions.cs" />
+    <Compile Include="SafeComWrappers\Abstract\IDesigner.cs" />
     <Compile Include="SafeComWrappers\Abstract\ISafeComWrapper.cs" />
     <Compile Include="SafeComWrappers\DispatcherEventArgs.cs" />
     <Compile Include="SafeComWrappers\MSAccessComponentType.cs" />

Rubberduck.VBEEditor/SafeComWrappers/Abstract/IDesigner.cs

@@ -0,0 +1,10 @@
+using System;
+using System.Collections;
+
+namespace Rubberduck.VBEditor.SafeComWrappers.Abstract
+{
+    public interface IDesigner : ISafeComWrapper, IEquatable<IVBComponent>
+    { 
+        IList Selected { get; }
+    }
+}

Rubberduck.VBEEditor/SafeComWrappers/Abstract/IVBComponent.cs

@@ -13,6 +13,7 @@ public interface IVBComponent : ISafeComWrapper, IEquatable<IVBComponent>
         IControls Controls { get; }
         IControls SelectedControls { get; }
         bool IsSaved { get; }
+        object Designer { get; }
         bool HasDesigner { get; }
         bool HasOpenDesigner { get; }
         string DesignerId { get; }

Rubberduck.VBEEditor/SafeComWrappers/VB6/VBComponent.cs

@@ -86,6 +86,11 @@ public IControls SelectedControls
             }
         }
 
+        public object Designer
+        {
+            get { return (IsWrappingNullReference ? null : Target.Designer); }
+        }
+
         public bool HasDesigner
         {
             get