EverythingPowerToys/ContextMenuLoader.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.ComponentModel;
using System.Diagnostics;
using System.IO;
using System.Reflection;
using System.Security.AccessControl;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Input;
using Community.PowerToys.Run.Plugin.Everything.Properties;
using Wox.Infrastructure;
using Wox.Plugin;
using Wox.Plugin.Logger;

namespace Community.PowerToys.Run.Plugin.Everything
{
    internal sealed class ContextMenuLoader(PluginInitContext context, string options) : IContextMenu
    {
        private readonly PluginInitContext _context = context;

        // Extensions for adding run as admin context menu item for applications
        private readonly string[] _appExtensions = [".exe", ".bat", ".appref-ms", ".lnk"];

        private bool _swapCopy;
        private string _options = options;
        private string _customProgram;
        private string _customArg;
        internal void Update(Settings s)
        {
            _swapCopy = s.Copy;
            _options = s.Context;
            _customProgram = s.CustomProgram;
            _customArg = s.CustomArg;
        }

        public List<ContextMenuResult> LoadContextMenus(Result selectedResult)
        {
            var contextMenus = new List<ContextMenuResult>();
            if (selectedResult.ContextData is SearchResult record)
            {
                bool isFile = record.File, runAs = CanFileBeRunAsAdmin(record.Path);
                foreach (char o in _options)
                {
                    switch (o)
                    {
                        case '0':
                            // Open folder
                            if (isFile)
                            {
                                contextMenus.Add(new ContextMenuResult
                                {
                                    PluginName = Assembly.GetExecutingAssembly().GetName().Name,
                                    Title = Resources.open_containing_folder,
                                    Glyph = "\xE838",
                                    FontFamily = "Segoe MDL2 Assets",
                                    AcceleratorKey = Key.E,
                                    AcceleratorModifiers = ModifierKeys.Control | ModifierKeys.Shift,
                                    Action = _ =>
                                    {
                                        if (!Helper.OpenInShell("explorer.exe", $"/select,\"{record.Path}\""))
                                        {
                                            var message = $"{Resources.folder_open_failed} {Path.GetDirectoryName(record.Path)}";
                                            _context.API.ShowMsg(message);
                                            return false;
                                        }

                                        return true;
                                    },
                                });
                            }

                            break;
                        case '1':
                            // Run as Adsmin
                            if (runAs)
                            {
                                contextMenus.Add(new ContextMenuResult
                                {
                                    PluginName = Assembly.GetExecutingAssembly().GetName().Name,
                                    Title = Resources.run_as_admin,
                                    Glyph = "\xE7EF",
                                    FontFamily = "Segoe MDL2 Assets",
                                    AcceleratorKey = Key.Enter,
                                    AcceleratorModifiers = ModifierKeys.Control | ModifierKeys.Shift,
                                    Action = _ =>
                                    {
                                        try
                                        {
                                            Task.Run(() => Helper.RunAsAdmin(record.Path));
                                            return true;
                                        }
                                        catch (Exception e)
                                        {
                                            Log.Exception($"Failed to run {record.Path} as admin, {e.Message}", e, MethodBase.GetCurrentMethod().DeclaringType);
                                            return false;
                                        }
                                    },
                                });
                            }

                            break;
                        case '2':
                            // Run as User
                            if (runAs)
                            {
                                contextMenus.Add(new ContextMenuResult
                                {
                                    PluginName = Assembly.GetExecutingAssembly().GetName().Name,
                                    Title = Resources.run_as_user,
                                    Glyph = "\xE7EE",
                                    FontFamily = "Segoe MDL2 Assets",
                                    AcceleratorKey = Key.U,
                                    AcceleratorModifiers = ModifierKeys.Control | ModifierKeys.Shift,
                                    Action = _ =>
                                    {
                                        try
                                        {
                                            Task.Run(() => Helper.RunAsUser(record.Path));
                                            return true;
                                        }
                                        catch (Exception e)
                                        {
                                            Log.Exception($"Failed to run {record.Path} as different user, {e.Message}", e, MethodBase.GetCurrentMethod().DeclaringType);
                                            return false;
                                        }
                                    },
                                });
                            }

                            break;
                        case '3':
                            // Copy File/Folder
                            contextMenus.Add(new ContextMenuResult
                            {
                                PluginName = Assembly.GetExecutingAssembly().GetName().Name,
                                Title = Resources.copy_file + (_swapCopy ? Resources.copy_shortcut : Resources.copy_shortcutAlt),
                                Glyph = "\xE8C8",
                                FontFamily = "Segoe MDL2 Assets",
                                AcceleratorKey = Key.C,
                                AcceleratorModifiers = _swapCopy ? ModifierKeys.Control : ModifierKeys.Control | ModifierKeys.Alt,

                                Action = (context) =>
                                {
                                    try
                                    {
                                        Clipboard.SetData(DataFormats.FileDrop, new string[] { record.Path });
                                        return true;
                                    }
                                    catch (Exception e)
                                    {
                                        var message = Resources.clipboard_failed;
                                        Log.Exception(message, e, GetType());

                                        _context.API.ShowMsg(message);
                                        return false;
                                    }
                                },
                            });
                            break;
                        case '4':
                            // Copy Path
                            contextMenus.Add(new ContextMenuResult
                            {
                                PluginName = Assembly.GetExecutingAssembly().GetName().Name,
                                Title = Resources.copy_path + (_swapCopy ? Resources.copy_shortcutAlt : Resources.copy_shortcut),
                                Glyph = "\xE71B",
                                FontFamily = "Segoe MDL2 Assets",
                                AcceleratorKey = Key.C,
                                AcceleratorModifiers = _swapCopy ? ModifierKeys.Control | ModifierKeys.Alt : ModifierKeys.Control,

                                Action = (context) =>
                                {
                                    try
                                    {
                                        Clipboard.SetDataObject(record.Path);
                                        return true;
                                    }
                                    catch (Exception e)
                                    {
                                        var message = Resources.clipboard_failed;
                                        Log.Exception(message, e, GetType());

                                        _context.API.ShowMsg(message);
                                        return false;
                                    }
                                },
                            });
                            break;
                        case '5':
                            // Open in Shell
                            contextMenus.Add(new ContextMenuResult
                            {
                                PluginName = Assembly.GetExecutingAssembly().GetName().Name,
                                Title = Resources.open_in_console,
                                Glyph = "\xE756",
                                FontFamily = "Segoe MDL2 Assets",
                                AcceleratorKey = Key.C,
                                AcceleratorModifiers = ModifierKeys.Control | ModifierKeys.Shift,

                                Action = (context) =>
                                {
                                    try
                                    {
                                        if (isFile)
                                        {
                                            Helper.OpenInConsole(Path.GetDirectoryName(record.Path));
                                        }
                                        else
                                        {
                                            Helper.OpenInConsole(record.Path);
                                        }

                                        return true;
                                    }
                                    catch (Exception e)
                                    {
                                        Log.Exception($"Failed to open {record.Path} in console, {e.Message}", e, GetType());
                                        return false;
                                    }
                                },
                            });
                            break;
                        case '6':
                            // Pass to custom program as parameter
                            contextMenus.Add(new ContextMenuResult
                            {
                                PluginName = Assembly.GetExecutingAssembly().GetName().Name,
                                Title = Resources.open_in_custom,
                                Glyph = "\xE8A7",
                                FontFamily = "Segoe MDL2 Assets",
                                AcceleratorKey = Key.N,
                                AcceleratorModifiers = ModifierKeys.Control,

                                Action = (context) =>
                                {
                                    using var process = new Process();
                                    process.StartInfo.FileName = _customProgram;
                                    process.StartInfo.Arguments = $"\"{_customArg.Replace("$P", record.Path)}\"";
                                    try
                                    {
                                        process.Start();
                                        return true;
                                    }
                                    catch (Exception e)
                                    {
                                        Log.Exception($"Failed to execute {_customProgram} with arguments {_customArg}", e, GetType());
                                        return false;
                                    }
                                },
                            });
                            break;
                        default:
                            break;
                    }
                }
            }

            return contextMenus;
        }

        private bool CanFileBeRunAsAdmin(string path)
        {
            string fileExtension = Path.GetExtension(path);
            foreach (string extension in _appExtensions)
            {
                // Using OrdinalIgnoreCase since this is internal
                if (extension.Equals(fileExtension, StringComparison.OrdinalIgnoreCase))
                {
                    return true;
                }
            }

            return false;
        }
    }
}