Close #2048

Author: Hosch250

RetailCoder.VBE/UI/CodeExplorer/Commands/ImportCommand.cs

@@ -11,10 +11,12 @@ namespace Rubberduck.UI.CodeExplorer.Commands
     [CodeExplorerCommand]
     public class ImportCommand : CommandBase, IDisposable
     {
+        private readonly VBE _vbe;
         private readonly IOpenFileDialog _openFileDialog;
 
-        public ImportCommand(IOpenFileDialog openFileDialog) : base(LogManager.GetCurrentClassLogger())
+        public ImportCommand(VBE vbe, IOpenFileDialog openFileDialog) : base(LogManager.GetCurrentClassLogger())
         {
+            _vbe = vbe;
             _openFileDialog = openFileDialog;
 
             _openFileDialog.AddExtension = true;
@@ -26,23 +28,24 @@ public ImportCommand(IOpenFileDialog openFileDialog) : base(LogManager.GetCurren
             _openFileDialog.CheckFileExists = true;
         }
 
-        protected override void ExecuteImpl(object parameter)
+        protected override bool CanExecuteImpl(object parameter)
         {
-            VBProject project;
+            return parameter != null || _vbe.VBProjects.Count == 1 || _vbe.ActiveVBProject != null;
+        }
 
-            if (parameter is ICodeExplorerDeclarationViewModel)
-            {
-                project = ((ICodeExplorerDeclarationViewModel) parameter).Declaration.QualifiedName.QualifiedModuleName.Project;
-            }
-            else
+        protected override void ExecuteImpl(object parameter)
+        {
+            var project = GetNodeProject((CodeExplorerItemViewModel)parameter);
+            if (project == null)
             {
-                var node = ((CodeExplorerItemViewModel) parameter).Parent;
-                while (!(node is ICodeExplorerDeclarationViewModel))
+                if (_vbe.VBProjects.Count == 1)
                 {
-                    node = node.Parent;
+                    project = _vbe.VBProjects.Item(1);
+                }
+                else if (_vbe.ActiveVBProject != null)
+                {
+                    project = _vbe.ActiveVBProject;
                 }
-
-                project = ((ICodeExplorerDeclarationViewModel) node).Declaration.QualifiedName.QualifiedModuleName.Project;
             }
 
             if (_openFileDialog.ShowDialog() == DialogResult.OK)
@@ -60,6 +63,22 @@ protected override void ExecuteImpl(object parameter)
             }
         }
 
+        private VBProject GetNodeProject(CodeExplorerItemViewModel parameter)
+        {
+            if (parameter is ICodeExplorerDeclarationViewModel)
+            {
+                return parameter.GetSelectedDeclaration().QualifiedName.QualifiedModuleName.Project;
+            }
+
+            var node = parameter.Parent;
+            while (!(node is ICodeExplorerDeclarationViewModel))
+            {
+                node = node.Parent;
+            }
+
+            return ((ICodeExplorerDeclarationViewModel)node).Declaration.QualifiedName.QualifiedModuleName.Project;
+        }
+
         public void Dispose()
         {
             if (_openFileDialog != null)