Add type check for OnExecute.

rubberduck-vba · Jan 17, 2019 · d0474b3 · d0474b3
1 parent 2b4b54d
commit d0474b3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Rubberduck.Core/UI/CodeExplorer/Commands/AddTemplateCommand.cs b/Rubberduck.Core/UI/CodeExplorer/Commands/AddTemplateCommand.cs
@@ -29,23 +29,23 @@ public bool CanExecuteForNode(ICodeExplorerNode model)
         protected override bool EvaluateCanExecute(object parameter)
         {
             // TODO this cast needs to be safer.
-            var data = ((string templateName, CodeExplorerItemViewModel model)) parameter;
+            var data = ((string templateName, ICodeExplorerNode model)) parameter;
 
             return base.EvaluateCanExecute(data.model);
         }
 
         protected override void OnExecute(object parameter)
         {
             // TODO this cast needs to be safer.
-            var data = ((string templateName, CodeExplorerItemViewModel model)) parameter;
+            var data = ((string templateName, ICodeExplorerNode node)) parameter;
 
-            if (string.IsNullOrWhiteSpace(data.templateName))
+            if (string.IsNullOrWhiteSpace(data.templateName) || !(data.node is CodeExplorerItemViewModel model))
             {
                 return;
             }
 
             var moduleText = GetTemplate(data.templateName);
-            AddComponent(data.model, moduleText);
+            AddComponent(model, moduleText);
         }
 
         private string GetTemplate(string name)