// 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.Collections;
using System.ComponentModel;
using System.ComponentModel.Design;
using System.Diagnostics;
using System.Drawing;

namespace System.Windows.Forms.Design
{
    /// <summary>
    ///  This class is going to replace the shell contextMenu and uses the ContextMenuStrip. The ContextMenuStrip contains groups and groupOrder which it uses to add items to itself. ControlDesigners can add custom items to the contextMenu, using the new member to the  group and add the groupOrder to the ContextMenu.
    /// </summary>
    internal class BaseContextMenuStrip : GroupedContextMenuStrip
    {
        private readonly IServiceProvider serviceProvider;
        private readonly Component component;
        private ToolStripMenuItem selectionMenuItem;

        public BaseContextMenuStrip(IServiceProvider provider, Component component) : base()
        {
            serviceProvider = provider;
            this.component = component;
            // Now initialiaze the contextMenu
            InitializeContextMenu();
        }

        /// <summary>
        ///  Helper function to add the "View Code" menuItem.
        /// </summary>
        private void AddCodeMenuItem()
        {
            StandardCommandToolStripMenuItem codeMenuItem = new StandardCommandToolStripMenuItem(StandardCommands.ViewCode, SR.ContextMenuViewCode, "viewcode", serviceProvider);
            Groups[StandardGroups.Code].Items.Add(codeMenuItem);
        }

        /// <summary>
        ///  Helper function to add the "SendToBack/BringToFront" menuItem.
        /// </summary>
        private void AddZorderMenuItem()
        {
            StandardCommandToolStripMenuItem ZOrderMenuItem = new StandardCommandToolStripMenuItem(MenuCommands.BringToFront, SR.ContextMenuBringToFront, "bringToFront", serviceProvider);
            Groups[StandardGroups.ZORder].Items.Add(ZOrderMenuItem);
            ZOrderMenuItem = new StandardCommandToolStripMenuItem(MenuCommands.SendToBack, SR.ContextMenuSendToBack, "sendToBack", serviceProvider);
            Groups[StandardGroups.ZORder].Items.Add(ZOrderMenuItem);
        }

        /// <summary>
        ///  Helper function to add the "Alignment" menuItem.
        /// </summary>
        private void AddGridMenuItem()
        {
            StandardCommandToolStripMenuItem gridMenuItem = new StandardCommandToolStripMenuItem(MenuCommands.AlignToGrid, SR.ContextMenuAlignToGrid, "alignToGrid", serviceProvider);
            Groups[StandardGroups.Grid].Items.Add(gridMenuItem);
        }

        /// <summary>
        ///  Helper function to add the "Locked" menuItem.
        /// </summary>
        private void AddLockMenuItem()
        {
            StandardCommandToolStripMenuItem lockMenuItem = new StandardCommandToolStripMenuItem(MenuCommands.LockControls, SR.ContextMenuLockControls, "lockControls", serviceProvider);
            Groups[StandardGroups.Lock].Items.Add(lockMenuItem);
        }

        /// <summary>
        ///  Helper function to add the Select Parent menuItem.
        /// </summary>
        private void RefreshSelectionMenuItem()
        {
            int index = -1;
            if (selectionMenuItem != null)
            {
                index = Items.IndexOf(selectionMenuItem);
                Groups[StandardGroups.Selection].Items.Remove(selectionMenuItem);
                Items.Remove(selectionMenuItem);
            }
            ArrayList parentControls = new ArrayList();
            int nParentControls = 0;

            // Get the currently selected Control
            if (serviceProvider.GetService(typeof(ISelectionService)) is ISelectionService selectionService && serviceProvider.GetService(typeof(IDesignerHost)) is IDesignerHost host)
            {
                IComponent root = host.RootComponent;
                Debug.Assert(root != null, "Null root component. Will be unable to build selection menu");
                if (selectionService.PrimarySelection is Control selectedControl && root != null && selectedControl != root)
                {
                    Control parentControl = selectedControl.Parent;
                    while (parentControl != null)
                    {
                        if (parentControl.Site != null)
                        {
                            parentControls.Add(parentControl);
                            nParentControls++;
                        }
                        if (parentControl == root)
                        {
                            break;
                        }
                        parentControl = parentControl.Parent;
                    }
                }
                else if (selectionService.PrimarySelection is ToolStripItem)
                {
                    ToolStripItem selectedItem = selectionService.PrimarySelection as ToolStripItem;
                    if (host.GetDesigner(selectedItem) is ToolStripItemDesigner itemDesigner)
                    {
                        parentControls = itemDesigner.AddParentTree();
                        nParentControls = parentControls.Count;
                    }
                }
            }
            if (nParentControls > 0)
            {
                selectionMenuItem = new ToolStripMenuItem();

                if (serviceProvider.GetService(typeof(IUIService)) is IUIService uis)
                {
                    selectionMenuItem.DropDown.Renderer = (ToolStripProfessionalRenderer)uis.Styles["VsRenderer"];
                    //Set the right Font
                    selectionMenuItem.DropDown.Font = (Font)uis.Styles["DialogFont"];
                    if (uis.Styles["VsColorPanelText"] is Color)
                    {
                        selectionMenuItem.DropDown.ForeColor = (Color)uis.Styles["VsColorPanelText"];
                    }
                }

                selectionMenuItem.Text = SR.ContextMenuSelect;
                foreach (Component parent in parentControls)
                {
                    ToolStripMenuItem selectListItem = new SelectToolStripMenuItem(parent, serviceProvider);
                    selectionMenuItem.DropDownItems.Add(selectListItem);
                }
                Groups[StandardGroups.Selection].Items.Add(selectionMenuItem);
                // Re add the newly refreshed item..
                if (index != -1)
                {
                    Items.Insert(index, selectionMenuItem);
                }
            }
        }

        /// <summary>
        ///  Helper function to add the Verbs.
        /// </summary>
        private void AddVerbMenuItem()
        {
            //Add Designer Verbs..
            IMenuCommandService menuCommandService = (IMenuCommandService)serviceProvider.GetService(typeof(IMenuCommandService));
            if (menuCommandService != null)
            {
                DesignerVerbCollection verbCollection = menuCommandService.Verbs;
                foreach (DesignerVerb verb in verbCollection)
                {
                    DesignerVerbToolStripMenuItem verbItem = new DesignerVerbToolStripMenuItem(verb);
                    Groups[StandardGroups.Verbs].Items.Add(verbItem);
                }
            }
        }

        /// <summary>
        ///  Helper function to add the "Cut/Copy/Paste/Delete" menuItem.
        /// </summary>
        private void AddEditMenuItem()
        {
            StandardCommandToolStripMenuItem stdMenuItem = new StandardCommandToolStripMenuItem(StandardCommands.Cut, SR.ContextMenuCut, "cut", serviceProvider);
            Groups[StandardGroups.Edit].Items.Add(stdMenuItem);
            stdMenuItem = new StandardCommandToolStripMenuItem(StandardCommands.Copy, SR.ContextMenuCopy, "copy", serviceProvider);
            Groups[StandardGroups.Edit].Items.Add(stdMenuItem);
            stdMenuItem = new StandardCommandToolStripMenuItem(StandardCommands.Paste, SR.ContextMenuPaste, "paste", serviceProvider);
            Groups[StandardGroups.Edit].Items.Add(stdMenuItem);
            stdMenuItem = new StandardCommandToolStripMenuItem(StandardCommands.Delete, SR.ContextMenuDelete, "delete", serviceProvider);
            Groups[StandardGroups.Edit].Items.Add(stdMenuItem);
        }

        /// <summary>
        ///  Helper function to add the "Properties" menuItem.
        /// </summary>
        private void AddPropertiesMenuItem()
        {
            StandardCommandToolStripMenuItem stdMenuItem = new StandardCommandToolStripMenuItem(StandardCommands.DocumentOutline, SR.ContextMenuDocumentOutline, "", serviceProvider);
            Groups[StandardGroups.Properties].Items.Add(stdMenuItem);
            stdMenuItem = new StandardCommandToolStripMenuItem(MenuCommands.DesignerProperties, SR.ContextMenuProperties, "properties", serviceProvider);
            Groups[StandardGroups.Properties].Items.Add(stdMenuItem);
        }

        /// <summary>
        ///  Basic Initialize method.
        /// </summary>
        private void InitializeContextMenu()
        {
            //this.Opening += new CancelEventHandler(OnContextMenuOpening);
            Name = "designerContextMenuStrip";
            if (serviceProvider.GetService(typeof(IUIService)) is IUIService uis)
            {
                Renderer = (ToolStripProfessionalRenderer)uis.Styles["VsRenderer"];
                if (uis.Styles["VsColorPanelText"] is Color)
                {
                    ForeColor = (Color)uis.Styles["VsColorPanelText"];
                }
            }
            GroupOrdering.AddRange(new string[] { StandardGroups.Code, StandardGroups.ZORder, StandardGroups.Grid, StandardGroups.Lock, StandardGroups.Verbs, StandardGroups.Custom, StandardGroups.Selection, StandardGroups.Edit, StandardGroups.Properties });
            // ADD MENUITEMS
            AddCodeMenuItem();
            AddZorderMenuItem();
            AddGridMenuItem();
            AddLockMenuItem();
            AddVerbMenuItem();
            RefreshSelectionMenuItem();
            AddEditMenuItem();
            AddPropertiesMenuItem();
        }

        /// <summary>
        ///  Public function that allows the individual MenuItems to get refreshed each time the ContextMenu is opened.
        /// </summary>
        public override void RefreshItems()
        {
            if (serviceProvider.GetService(typeof(IUIService)) is IUIService uis)
            {
                Font = (Font)uis.Styles["DialogFont"];
            }

            foreach (ToolStripItem item in Items)
            {
                if (item is StandardCommandToolStripMenuItem stdItem)
                {
                    stdItem.RefreshItem();
                }
            }
            RefreshSelectionMenuItem();
        }

        /// <summary>
        ///  A ToolStripMenuItem that gets added for the "Select" menuitem.
        /// </summary>
        private class SelectToolStripMenuItem : ToolStripMenuItem
        {
            private readonly Component _comp;
            private readonly IServiceProvider _serviceProvider;
            private readonly Type _itemType;
            private bool _cachedImage;
            private Image _image;
            private static readonly string s_systemWindowsFormsNamespace = typeof(ToolStripItem).Namespace;

            public SelectToolStripMenuItem(Component c, IServiceProvider provider)
            {
                _comp = c;
                _serviceProvider = provider;
                // Get NestedSiteName...
                string compName = null;
                if (_comp != null)
                {
                    ISite site = _comp.Site;
                    if (site != null)
                    {
                        if (site is INestedSite nestedSite && !string.IsNullOrEmpty(nestedSite.FullName))
                        {
                            compName = nestedSite.FullName;
                        }
                        else if (!string.IsNullOrEmpty(site.Name))
                        {
                            compName = site.Name;
                        }
                    }
                }
                Text = string.Format(SR.ToolStripSelectMenuItem, compName);
                _itemType = c.GetType();
            }

            public override Image Image
            {
                get
                {
                    // Defer loading the image until we're sure we need it
                    if (!_cachedImage)
                    {
                        _cachedImage = true;
                        // else attempt to get the resource from a known place in the manifest. if and only if the namespace of the type is System.Windows.Forms. else attempt to get the resource from a known place in the manifest
                        if (_itemType.Namespace == s_systemWindowsFormsNamespace)
                        {
                            _image = ToolboxBitmapAttribute.GetImageFromResource(_itemType, null, false);
                        }

                        // if all else fails, throw up a default image.
                        if (_image is null)
                        {
                            _image = ToolboxBitmapAttribute.GetImageFromResource(_comp.GetType(), null, false);
                        }
                    }
                    return _image;
                }
                set
                {
                    _image = value;
                    _cachedImage = true;
                }
            }

            /// <summary>
            ///  Items OnClick event, to select the Parent Control.
            /// </summary>
            protected override void OnClick(System.EventArgs e)
            {
                if (_serviceProvider.GetService(typeof(ISelectionService)) is ISelectionService selectionService)
                {
                    selectionService.SetSelectedComponents(new object[] { _comp }, SelectionTypes.Replace);
                }
            }
        }
    }
}
