在WPF的TreeView中實現右鍵選定

在WPF中，TreeView預設情況是不支援右鍵選定的，也就是說，當右鍵點擊某節點時，是無法選中該節點的。當我們想在TreeViewItem中實現右鍵菜單時，往往希望在快顯功能表的同時選中該節點，以使得菜單針對選中的節點生效。

    

圖1：雖然是在GNU節點上彈出的右鍵菜單，但選中的節點仍然是上次左鍵單擊的DOC節點。
圖2：彈出的右鍵菜單同時選中GNU節點才是我們所需要的效果

實現這個功能並不是很難，我最開始的做法就是目前網上流行的版本：

1. 在TreeViewItem中響應PreviewMouseRightButtonDown事件
2. 在響應按右鍵事件前選中sender(TreeViewItem)節點。

private void TreeViewItem_PreviewMouseRightButtonDown(object sender, MouseButtonEventArgs e)
{
    var treeViewItem = sender as TreeViewItem;
    if (treeViewItem != null)
    {
        treeViewItem.Focus();
        e.Handled = true;
    }
}

但這樣做還存在一個問題：當選擇子節點時，sender並不是子節點的TreeViewItem，而是其父節點。導致無法選擇選擇子節點。如所示：

  

要解決這個問題也不難，那就是根據MouseButtonEventArgs的OriginalSource來擷取節點。但是OriginalSource也不是TreeViewItem，而是產生滑鼠事件的子控制項，因此還得往上尋找，從而找到所屬的TreeViewItem。

最終的解決方案如下：

1. 在TreeViewItem中響應PreviewMouseRightButtonDown事件

<TreeView.ItemContainerStyle>
    <Style TargetType="{x:Type TreeViewItem}">
        <EventSetter Event="TreeViewItem.PreviewMouseRightButtonDown" Handler="TreeViewItem_PreviewMouseRightButtonDown"/>
    </Style>
</TreeView.ItemContainerStyle>

2. 在響應按右鍵事件前選中TreeViewItem節點。

private void TreeViewItem_PreviewMouseRightButtonDown(object sender, MouseButtonEventArgs e)
{
    var treeViewItem = VisualUpwardSearch<TreeViewItem>(e.OriginalSource as DependencyObject) as TreeViewItem;
    if (treeViewItem != null)
    {
        treeViewItem.Focus();
        e.Handled = true;
    }
}

static DependencyObject VisualUpwardSearch<T>(DependencyObject source)
{
    while (source != null && source.GetType() != typeof(T))
        source = VisualTreeHelper.GetParent(source);

    return source;
}

現在才是一個比較完美的解決方案。