Flow.Launcher/Plugins/Flow.Launcher.Plugin.ProcessKiller/Main.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Windows.Controls;
using Flow.Launcher.Plugin.ProcessKiller.ViewModels;
using Flow.Launcher.Plugin.ProcessKiller.Views;

namespace Flow.Launcher.Plugin.ProcessKiller
{
    public class Main : IPlugin, IPluginI18n, IContextMenu, ISettingProvider
    {
        private readonly ProcessHelper processHelper = new();

        private static PluginInitContext _context;

        internal Settings Settings;

        private SettingsViewModel _viewModel;

        public void Init(PluginInitContext context)
        {
            _context = context;
            Settings = context.API.LoadSettingJsonStorage<Settings>();
            _viewModel = new SettingsViewModel(Settings);
        }

        public List<Result> Query(Query query)
        {
            return CreateResultsFromQuery(query);
        }

        public string GetTranslatedPluginTitle()
        {
            return _context.API.GetTranslation("flowlauncher_plugin_processkiller_plugin_name");
        }

        public string GetTranslatedPluginDescription()
        {
            return _context.API.GetTranslation("flowlauncher_plugin_processkiller_plugin_description");
        }

        public List<Result> LoadContextMenus(Result result)
        {
            var menuOptions = new List<Result>();
            var processPath = result.SubTitle;

            // get all non-system processes whose file path matches that of the given result (processPath)
            var similarProcesses = processHelper.GetSimilarProcesses(processPath);

            if (similarProcesses.Any())
            {
                menuOptions.Add(new Result
                {
                    Title = _context.API.GetTranslation("flowlauncher_plugin_processkiller_kill_instances"),
                    SubTitle = processPath,
                    Action = _ =>
                    {
                        foreach (var p in similarProcesses)
                        {
                            processHelper.TryKill(_context, p);
                        }

                        return true;
                    },
                    IcoPath = processPath
                });
            }

            return menuOptions;
        }

        private List<Result> CreateResultsFromQuery(Query query)
        {
            // Get all non-system processes
            var allPocessList = processHelper.GetMatchingProcesses();
            if (!allPocessList.Any())
            {
                return null;
            }

            // Filter processes based on search term
            var searchTerm = query.Search;
            var processlist = new List<ProcessResult>();
            var processWindowTitle =
                Settings.ShowWindowTitle || Settings.PutVisibleWindowProcessesTop ?
                ProcessHelper.GetProcessesWithNonEmptyWindowTitle() :
                new Dictionary<int, string>();
            if (string.IsNullOrWhiteSpace(searchTerm))
            {
                foreach (var p in allPocessList)
                {
                    var progressNameIdTitle = ProcessHelper.GetProcessNameIdTitle(p);

                    if (processWindowTitle.TryGetValue(p.Id, out var windowTitle))
                    {
                        // Add score to prioritize processes with visible windows
                        // Use window title for those processes if enabled
                        processlist.Add(new ProcessResult(
                            p,
                            Settings.PutVisibleWindowProcessesTop ? 200 : 0,
                            Settings.ShowWindowTitle ? windowTitle : progressNameIdTitle,
                            null,
                            progressNameIdTitle));
                    }
                    else
                    {
                        processlist.Add(new ProcessResult(
                            p,
                            0,
                            progressNameIdTitle,
                            null,
                            progressNameIdTitle));
                    }
                }
            }
            else
            {
                foreach (var p in allPocessList)
                {
                    var progressNameIdTitle = ProcessHelper.GetProcessNameIdTitle(p);

                    if (processWindowTitle.TryGetValue(p.Id, out var windowTitle))
                    {
                        // Get max score from searching process name, window title and process id
                        var windowTitleMatch = _context.API.FuzzySearch(searchTerm, windowTitle);
                        var processNameIdMatch = _context.API.FuzzySearch(searchTerm, progressNameIdTitle);
                        var score = Math.Max(windowTitleMatch.Score, processNameIdMatch.Score);
                        if (score > 0)
                        {
                            // Add score to prioritize processes with visible windows
                            // Use window title for those processes
                            if (Settings.PutVisibleWindowProcessesTop)
                            {
                                score += 200;
                            }
                            processlist.Add(new ProcessResult(
                                p,
                                score,
                                Settings.ShowWindowTitle ? windowTitle : progressNameIdTitle,
                                score == windowTitleMatch.Score ? windowTitleMatch : null,
                                progressNameIdTitle));
                        }
                    }
                    else
                    {
                        var processNameIdMatch = _context.API.FuzzySearch(searchTerm, progressNameIdTitle);
                        var score = processNameIdMatch.Score;
                        if (score > 0)
                        {
                            processlist.Add(new ProcessResult(
                                p,
                                score,
                                progressNameIdTitle,
                                processNameIdMatch,
                                progressNameIdTitle));
                        }
                    }
                }
            }

            var results = new List<Result>();
            foreach (var pr in processlist)
            {
                var p = pr.Process;
                var path = processHelper.TryGetProcessFilename(p);
                results.Add(new Result()
                {
                    IcoPath = path,
                    Title = pr.Title,
                    TitleToolTip = pr.Tooltip,
                    SubTitle = path,
                    TitleHighlightData = pr.TitleMatch?.MatchData,
                    Score = pr.Score,
                    ContextData = p.ProcessName,
                    AutoCompleteText = $"{_context.CurrentPluginMetadata.ActionKeyword}{Plugin.Query.TermSeparator}{p.ProcessName}",
                    Action = (c) =>
                    {
                        processHelper.TryKill(_context, p);
                        // Re-query to refresh process list
                        _context.API.ReQuery();
                        return true;
                    }
                });
            }

            // Order results by process name for processes without visible windows
            var sortedResults = results.OrderBy(x => x.Title).ToList();

            // When there are multiple results AND all of them are instances of the same executable
            // add a quick option to kill them all at the top of the results.
            var firstResult = sortedResults.FirstOrDefault(x => !string.IsNullOrEmpty(x.SubTitle));
            if (processlist.Count > 1 && !string.IsNullOrEmpty(searchTerm) && sortedResults.All(r => r.SubTitle == firstResult?.SubTitle))
            {
                sortedResults.Insert(1, new Result()
                {
                    IcoPath = firstResult?.IcoPath,
                    Title = string.Format(_context.API.GetTranslation("flowlauncher_plugin_processkiller_kill_all"), firstResult?.ContextData),
                    SubTitle = string.Format(_context.API.GetTranslation("flowlauncher_plugin_processkiller_kill_all_count"), processlist.Count),
                    Score = 200,
                    Action = (c) =>
                    {
                        foreach (var p in processlist)
                        {
                            processHelper.TryKill(_context, p.Process);
                        }
                        // Re-query to refresh process list
                        _context.API.ReQuery();
                        return true;
                    }
                });
            }

            return sortedResults;
        }

        public Control CreateSettingPanel()
        {
            return new SettingsControl(_viewModel);
        }
    }
}