using System; using System.Collections.Generic; using System.Diagnostics; using System.IO; using System.Linq; using System.Reflection; using System.Threading; using System.Windows; using Wox.Infrastructure; using Wox.Plugin.Program.ProgramSources; using IWshRuntimeLibrary; using Wox.Plugin.Features; namespace Wox.Plugin.Program { public class Programs : ISettingProvider, IPlugin, IPluginI18n, IContextMenu { private static object lockObject = new object(); private static List programs = new List(); private static List sources = new List(); private static Dictionary SourceTypes = new Dictionary() { {"FileSystemProgramSource", typeof(FileSystemProgramSource)}, {"CommonStartMenuProgramSource", typeof(CommonStartMenuProgramSource)}, {"UserStartMenuProgramSource", typeof(UserStartMenuProgramSource)}, {"AppPathsProgramSource", typeof(AppPathsProgramSource)}, }; private PluginInitContext context; public List Query(Query query) { var fuzzyMather = FuzzyMatcher.Create(query.Search); List returnList = programs.Where(o => MatchProgram(o, fuzzyMather)).ToList(); returnList.ForEach(ScoreFilter); returnList = returnList.OrderByDescending(o => o.Score).ToList(); return returnList.Select(c => new Result() { Title = c.Title, SubTitle = c.ExecutePath, IcoPath = c.IcoPath, Score = c.Score, ContextData = c, Action = (e) => { context.API.HideApp(); context.API.ShellRun(c.ExecutePath); return true; } }).ToList(); } static string ResolveShortcut(string filePath) { // IWshRuntimeLibrary is in the COM library "Windows Script Host Object Model" WshShell shell = new WshShell(); IWshShortcut shortcut = (IWshShortcut)shell.CreateShortcut(filePath); return shortcut.TargetPath; } private bool MatchProgram(Program program, FuzzyMatcher matcher) { if ((program.Score = matcher.Evaluate(program.Title).Score) > 0) return true; if ((program.Score = matcher.Evaluate(program.PinyinTitle).Score) > 0) return true; if (program.AbbrTitle != null && (program.Score = matcher.Evaluate(program.AbbrTitle).Score) > 0) return true; if (program.ExecuteName != null && (program.Score = matcher.Evaluate(program.ExecuteName).Score) > 0) return true; return false; } public void Init(PluginInitContext context) { this.context = context; this.context.API.ResultItemDropEvent += API_ResultItemDropEvent; using (new Timeit("Preload programs")) { programs = ProgramCacheStorage.Instance.Programs; } DebugHelper.WriteLine(string.Format("Preload {0} programs from cache", programs.Count)); using (new Timeit("Program Index")) { IndexPrograms(); } } void API_ResultItemDropEvent(Result result, IDataObject dropObject, DragEventArgs e) { e.Handled = true; } public static void IndexPrograms() { lock (lockObject) { List programSources = new List(); programSources.AddRange(LoadDeaultProgramSources()); if (ProgramStorage.Instance.ProgramSources != null && ProgramStorage.Instance.ProgramSources.Count(o => o.Enabled) > 0) { programSources.AddRange(ProgramStorage.Instance.ProgramSources.Where(o => o.Enabled)); } sources.Clear(); programSources.ForEach(source => { Type sourceClass; if (SourceTypes.TryGetValue(source.Type, out sourceClass)) { ConstructorInfo constructorInfo = sourceClass.GetConstructor(new[] { typeof(ProgramSource) }); if (constructorInfo != null) { IProgramSource programSource = constructorInfo.Invoke(new object[] { source }) as IProgramSource; sources.Add(programSource); } } }); var tempPrograms = new List(); foreach (var source in sources) { var list = source.LoadPrograms(); list.ForEach(o => { o.Source = source; }); tempPrograms.AddRange(list); } // filter duplicate program programs = tempPrograms.GroupBy(x => new { x.ExecutePath, x.ExecuteName }) .Select(g => g.First()).ToList(); ProgramCacheStorage.Instance.Programs = programs; ProgramCacheStorage.Instance.Save(); } } /// /// Load program sources that wox always provide /// private static List LoadDeaultProgramSources() { var list = new List(); list.Add(new ProgramSource() { BonusPoints = 0, Enabled = true, Type = "CommonStartMenuProgramSource" }); list.Add(new ProgramSource() { BonusPoints = 0, Enabled = true, Type = "UserStartMenuProgramSource" }); list.Add(new ProgramSource() { BonusPoints = -10, Enabled = true, Type = "AppPathsProgramSource" }); return list; } private void ScoreFilter(Program p) { p.Score += p.Source.BonusPoints; if (p.Title.Contains("启动") || p.Title.ToLower().Contains("start")) p.Score += 10; if (p.Title.Contains("帮助") || p.Title.ToLower().Contains("help") || p.Title.Contains("文档") || p.Title.ToLower().Contains("documentation")) p.Score -= 10; if (p.Title.Contains("卸载") || p.Title.ToLower().Contains("uninstall")) p.Score -= 20; } #region ISettingProvider Members public System.Windows.Controls.Control CreateSettingPanel() { return new ProgramSetting(context); } #endregion public string GetLanguagesFolder() { return Path.Combine(Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location), "Languages"); } 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 LoadContextMenus(Result selectedResult) { Program p = selectedResult.ContextData as Program; List contextMenus = new List() { new Result() { Title = context.API.GetTranslation("wox_plugin_program_run_as_administrator"), Action = _ => { context.API.HideApp(); context.API.ShellRun(p.ExecutePath, true); return true; }, IcoPath = "Images/cmd.png" }, new Result() { Title = context.API.GetTranslation("wox_plugin_program_open_containing_folder"), Action = _ => { context.API.HideApp(); String Path = p.ExecutePath; //check if shortcut if (Path.EndsWith(".lnk")) { //get location of shortcut Path = ResolveShortcut(Path); } //get parent folder Path = Directory.GetParent(Path).FullName; //open the folder context.API.ShellRun("explorer.exe " + Path, false); return true; }, IcoPath = "Images/folder.png" } }; return contextMenus; } } }