// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
 
using System.Windows;
using System.Windows.Controls;
using Roslyn.Utilities;
 
namespace Microsoft.VisualStudio.LanguageServices.Implementation.InheritanceMargin.MarginGlyph
{
    internal class MenuItemContainerTemplateSelector : ItemContainerTemplateSelector
    {
        // By default, ContextMenu would create same MenuItem for each ViewModel from ItemSource,
        // this would override the default behavior, and let contextMenu create different MenuItem
        // based on the ViewModel's type
        public override DataTemplate SelectTemplate(object item, ItemsControl parentItemsControl)
        {
            if (item is HeaderMenuItemViewModel)
            {
                // Template for Header
                return (DataTemplate)parentItemsControl.FindResource("HeaderMenuItemTemplate");
            }
 
            if (item is DisambiguousTargetMenuItemViewModel)
            {
                // Template for DisambiguatingTargetMenuItemTemplate, which contains one more icon than the common TargetMenuItemTemplate.
                return (DataTemplate)parentItemsControl.FindResource("DisambiguatingTargetMenuItemTemplate");
            }
 
            if (item is TargetMenuItemViewModel)
            {
                // Template for a common target
                return (DataTemplate)parentItemsControl.FindResource("TargetMenuItemTemplate");
            }
 
            if (item is MemberMenuItemViewModel)
            {
                // Template for member
                return (DataTemplate)parentItemsControl.FindResource("MemberMenuItemTemplate");
            }
 
            throw ExceptionUtilities.Unreachable();
        }
    }
}