Add 'src/modules/launcher/' from commit '28acd466b352a807910cebbc74477d3173b31511'

git-subtree-dir: src/modules/launcher
git-subtree-mainline: 852689b3df
git-subtree-split: 28acd466b3

src/modules/launcher/Plugins/Wox.Plugin.Program/Main.cs

@@ -0,0 +1,214 @@
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.Linq;
+using System.Threading.Tasks;
+using System.Windows.Controls;
+using Wox.Infrastructure.Logger;
+using Wox.Infrastructure.Storage;
+using Wox.Plugin.Program.Programs;
+using Wox.Plugin.Program.Views;
+using Stopwatch = Wox.Infrastructure.Stopwatch;
+
+namespace Wox.Plugin.Program
+{
+    public class Main : ISettingProvider, IPlugin, IPluginI18n, IContextMenu, ISavable, IReloadable
+    {
+        private static readonly object IndexLock = new object();
+        internal static Win32[] _win32s { get; set; }
+        internal static UWP.Application[] _uwps { get; set; }
+        internal static Settings _settings { get; set; }
+
+        private static bool IsStartupIndexProgramsRequired => _settings.LastIndexTime.AddDays(3) < DateTime.Today;
+
+        private static PluginInitContext _context;
+
+        private static BinaryStorage<Win32[]> _win32Storage;
+        private static BinaryStorage<UWP.Application[]> _uwpStorage;
+        private readonly PluginJsonStorage<Settings> _settingsStorage;
+
+        public Main()
+        {
+            _settingsStorage = new PluginJsonStorage<Settings>();
+            _settings = _settingsStorage.Load();
+
+            Stopwatch.Normal("|Wox.Plugin.Program.Main|Preload programs cost", () =>
+            {
+                _win32Storage = new BinaryStorage<Win32[]>("Win32");
+                _win32s = _win32Storage.TryLoad(new Win32[] { });
+                _uwpStorage = new BinaryStorage<UWP.Application[]>("UWP");
+                _uwps = _uwpStorage.TryLoad(new UWP.Application[] { });
+            });
+            Log.Info($"|Wox.Plugin.Program.Main|Number of preload win32 programs <{_win32s.Length}>");
+            Log.Info($"|Wox.Plugin.Program.Main|Number of preload uwps <{_uwps.Length}>");
+
+            var a = Task.Run(() =>
+            {
+                if (IsStartupIndexProgramsRequired || !_win32s.Any())
+                    Stopwatch.Normal("|Wox.Plugin.Program.Main|Win32Program index cost", IndexWin32Programs);
+            });
+
+            var b = Task.Run(() =>
+            {
+                if (IsStartupIndexProgramsRequired || !_uwps.Any())
+                    Stopwatch.Normal("|Wox.Plugin.Program.Main|Win32Program index cost", IndexUWPPrograms);
+            });
+
+            Task.WaitAll(a, b);
+
+            _settings.LastIndexTime = DateTime.Today;
+        }
+
+        public void Save()
+        {
+            _settingsStorage.Save();
+            _win32Storage.Save(_win32s);
+            _uwpStorage.Save(_uwps);
+        }
+
+        public List<Result> Query(Query query)
+        {
+            Win32[] win32;
+            UWP.Application[] uwps;
+
+            lock (IndexLock)
+            { // just take the reference inside the lock to eliminate query time issues.
+                win32 = _win32s;
+                uwps = _uwps;
+            }
+
+            var results1 = win32.AsParallel()
+                .Where(p => p.Enabled)
+                .Select(p => p.Result(query.Search, _context.API));
+
+            var results2 = uwps.AsParallel()
+                .Where(p => p.Enabled)
+                .Select(p => p.Result(query.Search, _context.API));
+
+            var result = results1.Concat(results2).Where(r => r != null && r.Score > 0).ToList();
+            return result;
+        }
+
+        public void Init(PluginInitContext context)
+        {
+            _context = context;
+        }
+
+        public static void IndexWin32Programs()
+        {
+            var win32S = Win32.All(_settings);
+            lock (IndexLock)
+            {
+                _win32s = win32S;
+            }
+        }
+
+        public static void IndexUWPPrograms()
+        {
+            var windows10 = new Version(10, 0);
+            var support = Environment.OSVersion.Version.Major >= windows10.Major;
+
+            var applications = support ? UWP.All() : new UWP.Application[] { };
+            lock (IndexLock)
+            {
+                _uwps = applications;
+            }
+        }
+
+        public static void IndexPrograms()
+        {
+            var t1 = Task.Run(() => IndexWin32Programs());
+
+            var t2 = Task.Run(() => IndexUWPPrograms());
+
+            Task.WaitAll(t1, t2);
+
+            _settings.LastIndexTime = DateTime.Today;
+        }
+
+        public Control CreateSettingPanel()
+        {
+            return new ProgramSetting(_context, _settings, _win32s, _uwps);
+        }
+
+        public string GetTranslatedPluginTitle()
+        {
+            return _context.API.GetTranslation("wox_plugin_program_plugin_name");
+        }
+
+        public string GetTranslatedPluginDescription()
+        {
+            return _context.API.GetTranslation("wox_plugin_program_plugin_description");
+        }
+
+        public List<Result> LoadContextMenus(Result selectedResult)
+        {
+            var menuOptions = new List<Result>();
+            var program = selectedResult.ContextData as IProgram;
+            if (program != null)
+            {
+                menuOptions = program.ContextMenus(_context.API);
+            }
+
+            menuOptions.Add(
+                                new Result
+                                {
+                                    Title = _context.API.GetTranslation("wox_plugin_program_disable_program"),
+                                    Action = c =>
+                                    {
+                                        DisableProgram(program);
+                                        _context.API.ShowMsg(_context.API.GetTranslation("wox_plugin_program_disable_dlgtitle_success"),
+                                                                _context.API.GetTranslation("wox_plugin_program_disable_dlgtitle_success_message"));
+                                        return false;
+                                    },
+                                    IcoPath = "Images/disable.png"
+                                }
+                           );
+
+            return menuOptions;
+        }
+
+        private void DisableProgram(IProgram programToDelete)
+        {
+            if (_settings.DisabledProgramSources.Any(x => x.UniqueIdentifier == programToDelete.UniqueIdentifier))
+                return;
+
+            if (_uwps.Any(x => x.UniqueIdentifier == programToDelete.UniqueIdentifier))
+                _uwps.Where(x => x.UniqueIdentifier == programToDelete.UniqueIdentifier).FirstOrDefault().Enabled = false;
+
+            if (_win32s.Any(x => x.UniqueIdentifier == programToDelete.UniqueIdentifier))
+                _win32s.Where(x => x.UniqueIdentifier == programToDelete.UniqueIdentifier).FirstOrDefault().Enabled = false;
+
+            _settings.DisabledProgramSources
+                     .Add(
+                             new Settings.DisabledProgramSource
+                             {
+                                 Name = programToDelete.Name,
+                                 Location = programToDelete.Location,
+                                 UniqueIdentifier = programToDelete.UniqueIdentifier,
+                                 Enabled = false
+                             }
+                         );
+        }
+
+        public static void StartProcess(Func<ProcessStartInfo, Process> runProcess, ProcessStartInfo info)
+        {
+            bool hide;
+            try
+            {
+                runProcess(info);
+            }
+            catch (Exception)
+            {
+                var name = "Plugin: Program";
+                var message = $"Unable to start: {info.FileName}";
+                _context.API.ShowMsg(name, message, string.Empty);
+            }
+        }
+
+        public void ReloadData()
+        {
+            IndexPrograms();
+        }
+    }
+}