Keybinding on a treeview item, with CommandParameter

I'm trying to execute a command located on my ViewModel, using a TreeViewItem with a KeyBinding, and a MenuContext.

Currently, using the context menu, the command is invoked on the correct ViewModel instance. However, when I select a TreeViewItem and press the "C" key, the command is invoked on the "root" ViewModel.

I tried extending KeyBinding class as well ( Keybinding a RelayCommand ) with no luck.

Maybe I'm going to the wrong path : I just want to display the correct MessageBox, if I use the context menu or the key.

Code sample for a WPF project named WpfTest.

MainWindow.xaml

<Window x:Class="WpfTest.MainWindow"
        xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
        xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
        xmlns:vm="clr-namespace:WpfTest"
        Title="MainWindow" Height="350" Width="525">
    <Grid>
        <TreeView ItemsSource="{Binding}">
            <TreeView.ItemTemplate>
                <HierarchicalDataTemplate ItemsSource="{Binding Child}" DataType="{x:Type vm:ViewModel}">
                    <TextBlock Text="{Binding Name}" />
                </HierarchicalDataTemplate>
            </TreeView.ItemTemplate>
            <TreeView.ItemContainerStyle>
                <Style TargetType="{x:Type TreeViewItem}">
                    <Setter Property="ContextMenu">
                        <Setter.Value>
                            <ContextMenu>
                                <MenuItem Header="{Binding Name}" Command="{Binding SomeCommand}" CommandParameter="{Binding}"/>
                            </ContextMenu>
                        </Setter.Value>
                    </Setter>
                    <Setter Property="vm:MyAttached.InputBindings">
                        <Setter.Value>
                       开发者_开发知识库     <InputBindingCollection>
                                <KeyBinding Key="C" Command="{Binding SomeCommand}" CommandParameter="{Binding}"/>
                            </InputBindingCollection>
                        </Setter.Value>
                    </Setter>
                </Style>
            </TreeView.ItemContainerStyle>
        </TreeView>
    </Grid>
</Window>

MainWindow.xaml.cs:

namespace WpfTest
{
    using System;
    using System.Collections.Generic;
    using System.Collections.ObjectModel;
    using System.Windows;
    using System.Windows.Input;

    public partial class MainWindow : Window
    {
        public MainWindow()
        {
            InitializeComponent();

            this.DataContext = new List<ViewModel>
            {
                new ViewModel
                {
                    Name = "Parent",
                    Child = new ObservableCollection<ViewModel>
                    {
                        new ViewModel { Name = "Child 1" },
                        new ViewModel { Name = "Child 2" },
                        new ViewModel { Name = "Child 3" }
                    }
                }
            };
        }
    }

    public class ViewModel
    {
        public string Name { get; set; }
        public ObservableCollection<ViewModel> Child { get; set; }
        public ICommand SomeCommand { get; set; }

        public ViewModel()
        {
            this.SomeCommand = new RelayCommand<ViewModel>(OnCommandExecuted);
        }

        private void OnCommandExecuted(ViewModel parameter)
        {
            MessageBox.Show("CommandExecuted on " + Name + " with parameter " + parameter.Name);
        }
    }

    public class MyAttached
    {
        public static readonly DependencyProperty InputBindingsProperty =
            DependencyProperty.RegisterAttached("InputBindings", typeof(InputBindingCollection), typeof(MyAttached),
            new FrameworkPropertyMetadata(new InputBindingCollection(),
            (sender, e) =>
            {
                var element = sender as UIElement;
                if (element == null) return;
                element.InputBindings.Clear();
                element.InputBindings.AddRange((InputBindingCollection)e.NewValue);
            }));

        public static InputBindingCollection GetInputBindings(UIElement element)
        {
            return (InputBindingCollection)element.GetValue(InputBindingsProperty);
        }

        public static void SetInputBindings(UIElement element, InputBindingCollection inputBindings)
        {
            element.SetValue(InputBindingsProperty, inputBindings);
        }
    }

    public class RelayCommand<T> : ICommand
    {
        readonly Action<T> _execute = null;
        public RelayCommand(Action<T> execute)  { _execute = execute; }
        public bool CanExecute(object parameter) { return true; }
        public void Execute(object parameter) { _execute((T)parameter); }
        public event EventHandler CanExecuteChanged { add { CommandManager.RequerySuggested += value; } remove { CommandManager.RequerySuggested -= value; } }
    }
}

Here is the problem: The Style only creates one InputBindingCollection for all ListViewItems, you have to be very careful with Setter.Values for that reason.

And here is the fix:

    <TreeView ItemsSource="{Binding}">
        <TreeView.Resources>
            <!-- x:Shared="False" forces the new creation of that object whenever referenced -->
            <InputBindingCollection x:Shared="False" x:Key="InputBindings">
                <KeyBinding Key="C" Command="{Binding SomeCommand}" CommandParameter="{Binding}" />
            </InputBindingCollection>
        </TreeView.Resources>
        <!-- ... -->
        <TreeView.ItemContainerStyle>
            <Style TargetType="{x:Type TreeViewItem}">
                <!-- ... -->
                <Setter Property="vm:MyAttached.InputBindings" Value="{StaticResource InputBindings}"/>
            </Style>
        </TreeView.ItemContainerStyle>
    </TreeView>