Flow.Launcher/Plugins/Flow.Launcher.Plugin.PluginsManager/PluginsManager.cs

using Flow.Launcher.Infrastructure;
using Flow.Launcher.Infrastructure.Http;
using Flow.Launcher.Infrastructure.Logger;
using Flow.Launcher.Infrastructure.UserSettings;
using Flow.Launcher.Plugin.PluginsManager.Models;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Windows;

namespace Flow.Launcher.Plugin.PluginsManager
{
    internal class PluginsManager
    {
        private readonly PluginsManifest pluginsManifest;

        private PluginInitContext Context { get; set; }

        private Settings Settings { get; set; }

        private bool shouldHideWindow = true;
        private bool ShouldHideWindow 
        {
            set { shouldHideWindow = value; }
            get
            {
                var setValue = shouldHideWindow;
                // Default value for hide main window is true. Revert after get call.
                // This ensures when set by another method to false, it is only used once.
                shouldHideWindow = true;

                return setValue;
            }
        }

        private readonly string icoPath = "Images\\pluginsmanager.png";

        internal PluginsManager(PluginInitContext context, Settings settings)
        {
            pluginsManifest = new PluginsManifest();
            Context = context;
            Settings = settings;
        }
        internal void InstallOrUpdate(UserPlugin plugin)
        {
            if (PluginExists(plugin.ID))
            {
                if (Context.API.GetAllPlugins().Any(x => x.Metadata.ID == plugin.ID && x.Metadata.Version != plugin.Version))
                {
                    if (MessageBox.Show(Context.API.GetTranslation("plugin_pluginsmanager_update_exists"),
                                        Context.API.GetTranslation("plugin_pluginsmanager_update_title"),
                                        MessageBoxButton.YesNo) == MessageBoxResult.Yes)
                         Context
                                .API
                                .ChangeQuery($"{Context.CurrentPluginMetadata.ActionKeywords.FirstOrDefault()} {Settings.HotkeyUpdate} {plugin.Name}");

                    Application.Current.MainWindow.Show();
                    shouldHideWindow = false;

                    return;
                }

                Context.API.ShowMsg(Context.API.GetTranslation("plugin_pluginsmanager_update_alreadyexists"));
                return;
            }

            var message = string.Format(Context.API.GetTranslation("plugin_pluginsmanager_install_prompt"),
                                                                        plugin.Name, plugin.Author,
                                                                        Environment.NewLine, Environment.NewLine);

            if (MessageBox.Show(message, Context.API.GetTranslation("plugin_pluginsmanager_install_title"), MessageBoxButton.YesNo) == MessageBoxResult.No)
                return;

            var filePath = Path.Combine(DataLocation.PluginsDirectory, $"{plugin.Name}-{plugin.Version}.zip");

            try
            {
                Context.API.ShowMsg(Context.API.GetTranslation("plugin_pluginsmanager_downloading_plugin"),
                                    Context.API.GetTranslation("plugin_pluginsmanager_please_wait"));

                Http.Download(plugin.UrlDownload, filePath);

                Context.API.ShowMsg(Context.API.GetTranslation("plugin_pluginsmanager_downloading_plugin"),
                                    Context.API.GetTranslation("plugin_pluginsmanager_download_success"));
            }
            catch (Exception e)
            {
                Context.API.ShowMsg(Context.API.GetTranslation("plugin_pluginsmanager_downloading_plugin"),
                                Context.API.GetTranslation("plugin_pluginsmanager_download_success"));

                Log.Exception("PluginsManager", "An error occured while downloading plugin", e, "PluginDownload");
            }

            Application.Current.Dispatcher.Invoke(() => { Install(plugin, filePath); Context.API.RestartApp(); });
        }

        internal List<Result> RequestUpdate(string search)
        {
            var autocompletedResults = AutoCompleteReturnAllResults(search,
                                                                    Settings.HotkeyUpdate,
                                                                    "Update",
                                                                    "Select a plugin to update");

            if (autocompletedResults.Any())
                return autocompletedResults;

            var uninstallSearch = search.Replace(Settings.HotkeyUpdate, string.Empty).TrimStart();


            var resultsForUpdate = 
                    from existingPlugin in Context.API.GetAllPlugins()
                    join pluginFromManifest in pluginsManifest.UserPlugins
                        on existingPlugin.Metadata.ID equals pluginFromManifest.ID
                    where existingPlugin.Metadata.Version != pluginFromManifest.Version
                    select 
                      new 
                      {
                          pluginFromManifest.Name, 
                          pluginFromManifest.Author, 
                          CurrentVersion = existingPlugin.Metadata.Version, 
                          NewVersion = pluginFromManifest.Version,
                          existingPlugin.Metadata.IcoPath,
                          PluginExistingMetadata = existingPlugin.Metadata,
                          PluginNewUserPlugin = pluginFromManifest
                      };

            if (!resultsForUpdate.Any())
                return new List<Result> { 
                                new Result 
                                    {
                                        Title = Context.API.GetTranslation("plugin_pluginsmanager_update_noresult_title"),
                                        SubTitle = Context.API.GetTranslation("plugin_pluginsmanager_update_noresult_subtitle"),
                                        IcoPath = icoPath
                                    }};


            var results = resultsForUpdate
                            .Select(x =>
                                new Result
                                    {
                                        Title = $"{x.Name} by {x.Author}",
                                        SubTitle = $"Update from version {x.CurrentVersion} to {x.NewVersion}",
                                        IcoPath = x.IcoPath,
                                        Action = e =>
                                        {
                                            string message = string.Format(Context.API.GetTranslation("plugin_pluginsmanager_update_prompt"),
                                                                        x.Name, x.Author,
                                                                        Environment.NewLine, Environment.NewLine);

                                            if (MessageBox.Show(message, Context.API.GetTranslation("plugin_pluginsmanager_update_title"),
                                                                                                        MessageBoxButton.YesNo) == MessageBoxResult.Yes)
                                            {
                                                Uninstall(x.PluginExistingMetadata);

                                                var downloadToFilePath = Path.Combine(DataLocation.PluginsDirectory, $"{x.Name}-{x.NewVersion}.zip");
                                                Http.Download(x.PluginNewUserPlugin.UrlDownload, downloadToFilePath);
                                                Install(x.PluginNewUserPlugin, downloadToFilePath);

                                                Context.API.RestartApp();

                                                return true;
                                            }

                                            return false;
                                        }
                                    })
                            .ToList();

            return Search(results, uninstallSearch);
        }

        internal bool PluginExists(string id)
        {
            return Context.API.GetAllPlugins().Any(x => x.Metadata.ID == id);
        }

        internal List<Result> Search(List<Result> results, string searchName)
        {
            if (string.IsNullOrEmpty(searchName))
                return results;

            return results
                    .Where(x =>
                            {
                                var matchResult = StringMatcher.FuzzySearch(searchName, x.Title);
                                if (matchResult.IsSearchPrecisionScoreMet())
                                    x.Score = matchResult.Score;

                                return matchResult.IsSearchPrecisionScoreMet();
                            })
                    .ToList();
        }

        internal List<Result> RequestInstallOrUpdate(string searchName)
        {
            var results =
                pluginsManifest
                .UserPlugins
                .Select(x =>
                    new Result
                    {
                        Title = $"{x.Name} by {x.Author}",
                        SubTitle = x.Description,
                        IcoPath = icoPath,
                        Action = e =>
                        {
                            Application.Current.MainWindow.Hide();
                            InstallOrUpdate(x);

                            return ShouldHideWindow;
                        }
                    })
                .ToList();

            return Search(results, searchName);
        }

        private void Install(UserPlugin plugin, string downloadedFilePath)
        {
            if (!File.Exists(downloadedFilePath))
                return;

            var tempFolderPath = Path.Combine(Path.GetTempPath(), "flowlauncher");
            var tempFolderPluginPath = Path.Combine(tempFolderPath, "plugin");

            if (Directory.Exists(tempFolderPath))
                Directory.Delete(tempFolderPath, true);

            Directory.CreateDirectory(tempFolderPath);

            var zipFilePath = Path.Combine(tempFolderPath, Path.GetFileName(downloadedFilePath));

            File.Move(downloadedFilePath, zipFilePath);

            Utilities.UnZip(zipFilePath, tempFolderPluginPath, true);

            var pluginFolderPath = Utilities.GetContainingFolderPathAfterUnzip(tempFolderPluginPath);

            var metadataJsonFilePath = string.Empty;
            if (File.Exists(Path.Combine(pluginFolderPath, Constant.PluginMetadataFileName)))
                metadataJsonFilePath = Path.Combine(pluginFolderPath, Constant.PluginMetadataFileName);

            if (string.IsNullOrEmpty(metadataJsonFilePath) || string.IsNullOrEmpty(pluginFolderPath))
            {
                MessageBox.Show(Context.API.GetTranslation("plugin_pluginsmanager_install_errormetadatafile"));
                return;
            }

            string newPluginPath = Path.Combine(DataLocation.PluginsDirectory, $"{plugin.Name}-{plugin.Version}");

            Directory.Move(pluginFolderPath, newPluginPath);
        }

        internal List<Result> RequestUninstall(string search)
        {
            var autocompletedResults = AutoCompleteReturnAllResults(search, 
                                                                    Settings.HotkeyUninstall, 
                                                                    "Uninstall", 
                                                                    "Select a plugin to uninstall");

            if (autocompletedResults.Any())
                return autocompletedResults;

            var uninstallSearch = search.Replace(Settings.HotkeyUninstall, string.Empty).TrimStart();

            var results= Context.API
                                .GetAllPlugins()
                                .Select(x =>
                                    new Result
                                    {
                                        Title = $"{x.Metadata.Name} by {x.Metadata.Author}",
                                        SubTitle = x.Metadata.Description,
                                        IcoPath = x.Metadata.IcoPath,
                                        Action = e =>
                                        {
                                            string message = string.Format(Context.API.GetTranslation("plugin_pluginsmanager_uninstall_prompt"),
                                                                        x.Metadata.Name, x.Metadata.Author,
                                                                        Environment.NewLine, Environment.NewLine);

                                            if (MessageBox.Show(message, Context.API.GetTranslation("plugin_pluginsmanager_uninstall_title"),
                                                                                                        MessageBoxButton.YesNo) == MessageBoxResult.Yes)
                                            {
                                                Application.Current.MainWindow.Hide();
                                                Uninstall(x.Metadata);
                                                Context.API.RestartApp();

                                                return true;
                                            }

                                            return false;
                                        }
                                    })
                                .ToList();

            return Search(results, uninstallSearch);
        }

        private void Uninstall(PluginMetadata plugin)
        {
            // Marked for deletion. Will be deleted on next start up
            using var _ = File.CreateText(Path.Combine(plugin.PluginDirectory, "NeedDelete.txt"));
        }

        private List<Result> AutoCompleteReturnAllResults(string search, string hotkey, string title, string subtitle)
        {
            if (!string.IsNullOrEmpty(search)
                && hotkey.StartsWith(search)
                && (hotkey != search || !search.StartsWith(hotkey)))
            {
                return
                    new List<Result>
                    {
                        new Result
                        {
                            Title = title,
                            IcoPath = icoPath,
                            SubTitle = subtitle,
                            Action = e =>
                            {
                                Context
                                .API
                                .ChangeQuery($"{Context.CurrentPluginMetadata.ActionKeywords.FirstOrDefault()} {hotkey} ");

                                return false;
                            }
                        }
                    };
            }

            return new List<Result>();
        }
    }
}