ContextMenuManager/ContextMenuManager/Controls/ShellExItem.cs

using BulePointLilac.Controls;
using BulePointLilac.Methods;
using ContextMenuManager.Controls.Interfaces;
using Microsoft.Win32;
using System;
using System.Collections.Generic;
using System.IO;
using System.Windows.Forms;

namespace ContextMenuManager.Controls
{
    sealed class ShellExItem : MyListItem, IChkVisibleItem, IBtnShowMenuItem,
        ITsiWebSearchItem, ITsiFilePathItem, ITsiRegPathItem, ITsiDeleteItem
    {
        public static Dictionary<string, Guid> GetPathAndGuids(string shellExPath)
        {
            Dictionary<string, Guid> dic = new Dictionary<string, Guid>();
            foreach(string cmhPart in CmhParts)
            {
                using(RegistryKey cmKey = RegistryEx.GetRegistryKey($@"{shellExPath}\{cmhPart}"))
                {
                    if(cmKey == null) continue;
                    foreach(string keyName in cmKey.GetSubKeyNames())
                    {
                        using(RegistryKey key = cmKey.OpenSubKey(keyName))
                        {
                            if(!GuidInfo.TryGetGuid(key.GetValue("")?.ToString(), out Guid guid))
                                GuidInfo.TryGetGuid(keyName, out guid);
                            if(!guid.Equals(Guid.Empty))
                                dic.Add(key.Name, guid);
                        }
                    }
                }
            }
            return dic;
        }

        public static readonly string[] CmhParts = { "ContextMenuHandlers", "-ContextMenuHandlers" };
        private const string LnkOpenGuid = "00021401-0000-0000-c000-000000000046";

        public ShellExItem(Guid guid, string regPath)
        {
            InitializeComponents();
            this.Guid = guid;
            this.RegPath = regPath;
        }

        private string regPath;
        public string RegPath
        {
            get => regPath;
            set
            {
                regPath = value;
                this.Text = this.ItemText;
                this.Image = GuidInfo.GetImage(Guid);
                ChkVisible.Checked = this.ItemVisible;
            }
        }

        public Guid Guid { get; set; }
        public string SearchText => Text;
        public string ItemFilePath => GuidInfo.GetFilePath(Guid);
        private string KeyName => RegistryEx.GetKeyName(RegPath);
        private string ShellExPath => RegistryEx.GetParentPath(RegistryEx.GetParentPath(RegPath));
        private string CmhKeyName => RegistryEx.GetKeyName(RegistryEx.GetParentPath(RegPath));
        private string DefaultValue => Registry.GetValue(RegPath, "", null)?.ToString();
        private string ItemText => GuidInfo.GetText(Guid) ?? ((Guid.ToString("B") == KeyName) ? DefaultValue : KeyName);
        private string BuckupPath => $@"{ShellExPath}\{(ItemVisible ? CmhParts[1] : CmhParts[0])}\{KeyName}";
        private GuidInfo.IconLocation IconLocation => GuidInfo.GetIconLocation(this.Guid);
        private bool IsOpenLnkItem => Guid.ToString() == LnkOpenGuid;
        private bool TryProtectOpenItem => IsOpenLnkItem && MessageBoxEx.Show
            (AppString.MessageBox.PromptIsOpenItem, MessageBoxButtons.YesNo) != DialogResult.Yes;

        public bool ItemVisible
        {
            get => CmhKeyName.Equals(CmhParts[0], StringComparison.OrdinalIgnoreCase);
            set
            {
                if(!value && TryProtectOpenItem) return;
                using(RegistryKey srcKey = RegistryEx.GetRegistryKey(RegPath))
                using(RegistryKey dstKey = RegistryEx.GetRegistryKey(BuckupPath, true, true))
                    srcKey?.CopyTo(dstKey);
                RegistryEx.DeleteKeyTree(RegPath);
                RegPath = BuckupPath;
            }
        }

        public VisibleCheckBox ChkVisible { get; set; }
        public MenuButton BtnShowMenu { get; set; }
        public WebSearchMenuItem TsiSearch { get; set; }
        public FilePropertiesMenuItem TsiFileProperties { get; set; }
        public FileLocationMenuItem TsiFileLocation { get; set; }
        public RegLocationMenuItem TsiRegLocation { get; set; }
        public DeleteMeMenuItem TsiDeleteMe { get; set; }
        readonly ToolStripMenuItem TsiDetails = new ToolStripMenuItem(AppString.Menu.Details);
        readonly ToolStripMenuItem TsiHandleGuid = new ToolStripMenuItem(AppString.Menu.HandleGuid);
        readonly ToolStripMenuItem TsiCopyGuid = new ToolStripMenuItem(AppString.Menu.CopyGuid);
        readonly ToolStripMenuItem TsiBlockGuid = new ToolStripMenuItem(AppString.Menu.BlockGuid);
        readonly ToolStripMenuItem TsiAddGuidDic = new ToolStripMenuItem(AppString.Menu.AddGuidDic);

        private void InitializeComponents()
        {
            BtnShowMenu = new MenuButton(this);
            ChkVisible = new VisibleCheckBox(this);
            TsiSearch = new WebSearchMenuItem(this);
            TsiFileLocation = new FileLocationMenuItem(this);
            TsiFileProperties = new FilePropertiesMenuItem(this);
            TsiRegLocation = new RegLocationMenuItem(this);
            TsiDeleteMe = new DeleteMeMenuItem(this);

            ContextMenuStrip.Items.AddRange(new ToolStripItem[] { TsiHandleGuid, new ToolStripSeparator(),
                TsiDetails, new ToolStripSeparator(), TsiDeleteMe });

            TsiHandleGuid.DropDownItems.AddRange(new ToolStripItem[] { TsiCopyGuid, new ToolStripSeparator(),
                TsiBlockGuid, new ToolStripSeparator(), TsiAddGuidDic });

            TsiDetails.DropDownItems.AddRange(new ToolStripItem[] { TsiSearch, new ToolStripSeparator(),
                TsiFileProperties, TsiFileLocation, TsiRegLocation});

            ContextMenuStrip.Opening += (sender, e) => RefreshMenuItem();
            TsiCopyGuid.Click += (sender, e) => CopyGuid();
            TsiBlockGuid.Click += (sender, e) => BlockGuid();
            TsiAddGuidDic.Click += (sender, e) => AddGuidDic();
        }

        private void CopyGuid()
        {
            Clipboard.SetText(Guid.ToString());
            MessageBoxEx.Show($"{AppString.MessageBox.CopiedToClipboard}:\n{Guid}",
                MessageBoxButtons.OK, MessageBoxIcon.Information);
        }

        private void BlockGuid()
        {
            foreach(string path in GuidBlockedItem.BlockedPaths)
            {
                if(TsiBlockGuid.Checked)
                {
                    RegistryEx.DeleteValue(path, this.Guid.ToString("B"));
                }
                else
                {
                    Registry.SetValue(path, this.Guid.ToString("B"), string.Empty);
                }
            }
            ExplorerRestarter.NeedRestart = true;
        }

        private void AddGuidDic()
        {
            using(AddGuidDicDialog dlg = new AddGuidDicDialog())
            {
                dlg.ItemName = this.Text;
                dlg.ItemIcon = this.Image;
                dlg.ItemIconPath = this.IconLocation.IconPath;
                dlg.ItemIconIndex = this.IconLocation.IconIndex;
                IniFileHelper helper = new IniFileHelper(Program.GuidInfosDicPath);
                string section = this.Guid.ToString();
                if(dlg.ShowDialog() != DialogResult.OK)
                {
                    if(dlg.IsDelete) {
                        helper.DeleteSection(section);
                        GuidInfo.ItemTextDic.Remove(this.Guid);
                        GuidInfo.ItemImageDic.Remove(this.Guid);
                        GuidInfo.IconLocationDic.Remove(this.Guid);
                        GuidInfo.UserDic.RootDic.Remove(section);
                        this.Text = this.ItemText;
                        this.Image = GuidInfo.GetImage(Guid);
                    }
                    return;
                }
                Directory.CreateDirectory(Program.ConfigDir);
                if(!dlg.ItemName.IsNullOrWhiteSpace())
                {
                    helper.SetValue(section, "Text", dlg.ItemName);
                    this.Text = ResourceString.GetDirectString(dlg.ItemName);
                    if(GuidInfo.ItemTextDic.ContainsKey(this.Guid))
                    {
                        GuidInfo.ItemTextDic[this.Guid] = this.Text;
                    }
                    else
                    {
                        GuidInfo.ItemTextDic.Add(this.Guid, this.Text);
                    }
                }
                if(dlg.ItemIconLocation != null)
                {
                    helper.SetValue(section, "Icon", dlg.ItemIconLocation);
                    var location = new GuidInfo.IconLocation { IconPath = dlg.ItemIconPath, IconIndex = dlg.ItemIconIndex };
                    if(GuidInfo.IconLocationDic.ContainsKey(this.Guid))
                    {
                        GuidInfo.IconLocationDic[this.Guid] = location;
                    }
                    else
                    {
                        GuidInfo.IconLocationDic.Add(this.Guid, location);
                    }
                    this.Image = dlg.ItemIcon;
                    if(GuidInfo.ItemImageDic.ContainsKey(this.Guid))
                    {
                        GuidInfo.ItemImageDic[this.Guid] = this.Image;
                    }
                    else
                    {
                        GuidInfo.ItemImageDic.Add(this.Guid, this.Image);
                    }
                }
            }
        }

        private void RefreshMenuItem()
        {
            TsiDeleteMe.Enabled = !IsOpenLnkItem;
            TsiBlockGuid.Checked = false;
            foreach(string path in GuidBlockedItem.BlockedPaths)
            {
                if(Registry.GetValue(path, this.Guid.ToString("B"), null) != null)
                {
                    TsiBlockGuid.Checked = true;
                    break;
                }
            }
        }

        public void DeleteMe()
        {
            RegistryEx.DeleteKeyTree(this.RegPath);
            RegistryEx.DeleteKeyTree(this.BuckupPath);
            this.Dispose();
        }
    }
}