Skip to content

Commit d0474b3

Browse files
committed
Add type check for OnExecute.
1 parent 2b4b54d commit d0474b3

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

Rubberduck.Core/UI/CodeExplorer/Commands/AddTemplateCommand.cs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,23 +29,23 @@ public bool CanExecuteForNode(ICodeExplorerNode model)
2929
protected override bool EvaluateCanExecute(object parameter)
3030
{
3131
// TODO this cast needs to be safer.
32-
var data = ((string templateName, CodeExplorerItemViewModel model)) parameter;
32+
var data = ((string templateName, ICodeExplorerNode model)) parameter;
3333

3434
return base.EvaluateCanExecute(data.model);
3535
}
3636

3737
protected override void OnExecute(object parameter)
3838
{
3939
// TODO this cast needs to be safer.
40-
var data = ((string templateName, CodeExplorerItemViewModel model)) parameter;
40+
var data = ((string templateName, ICodeExplorerNode node)) parameter;
4141

42-
if (string.IsNullOrWhiteSpace(data.templateName))
42+
if (string.IsNullOrWhiteSpace(data.templateName) || !(data.node is CodeExplorerItemViewModel model))
4343
{
4444
return;
4545
}
4646

4747
var moduleText = GetTemplate(data.templateName);
48-
AddComponent(data.model, moduleText);
48+
AddComponent(model, moduleText);
4949
}
5050

5151
private string GetTemplate(string name)

0 commit comments

Comments
 (0)