2014-02-08 14:37:40 +08:00
|
|
|
using System;
|
2014-01-04 20:26:13 +08:00
|
|
|
using System.Collections.Generic;
|
|
|
|
using System.Linq;
|
2014-08-10 22:22:54 +08:00
|
|
|
using System.Reflection;
|
2014-02-06 22:22:02 +08:00
|
|
|
using Wox.Infrastructure;
|
2014-03-23 16:17:41 +08:00
|
|
|
using Wox.Infrastructure.Storage.UserSettings;
|
2015-01-03 15:20:34 +08:00
|
|
|
using Wox.Plugin.Program.ProgramSources;
|
2014-01-04 20:26:13 +08:00
|
|
|
|
2015-01-03 15:20:34 +08:00
|
|
|
namespace Wox.Plugin.Program
|
2014-01-04 20:26:13 +08:00
|
|
|
{
|
2015-01-03 15:20:34 +08:00
|
|
|
public class Programs : ISettingProvider,IPlugin
|
2014-01-04 20:26:13 +08:00
|
|
|
{
|
2014-08-10 22:22:54 +08:00
|
|
|
private static object lockObject = new object();
|
2014-12-21 20:44:31 +08:00
|
|
|
private static List<Program> programs = new List<Program>();
|
2014-08-10 22:22:54 +08:00
|
|
|
private static List<IProgramSource> sources = new List<IProgramSource>();
|
|
|
|
private static Dictionary<string, Type> SourceTypes = new Dictionary<string, Type>() {
|
2014-03-21 04:31:42 +08:00
|
|
|
{"FileSystemProgramSource", typeof(FileSystemProgramSource)},
|
2014-03-19 04:05:27 +08:00
|
|
|
{"CommonStartMenuProgramSource", typeof(CommonStartMenuProgramSource)},
|
|
|
|
{"UserStartMenuProgramSource", typeof(UserStartMenuProgramSource)},
|
|
|
|
{"AppPathsProgramSource", typeof(AppPathsProgramSource)},
|
|
|
|
};
|
2014-03-21 03:53:18 +08:00
|
|
|
private PluginInitContext context;
|
2014-01-06 21:25:24 +08:00
|
|
|
|
2015-01-03 15:20:34 +08:00
|
|
|
public List<Result> Query(Query query)
|
2014-01-04 20:26:13 +08:00
|
|
|
{
|
2014-07-21 23:18:39 +08:00
|
|
|
if (query.RawQuery.Trim().Length <= 1) return new List<Result>();
|
2014-01-04 20:26:13 +08:00
|
|
|
|
2014-02-09 00:33:10 +08:00
|
|
|
var fuzzyMather = FuzzyMatcher.Create(query.RawQuery);
|
2014-12-21 20:44:31 +08:00
|
|
|
List<Program> returnList = programs.Where(o => MatchProgram(o, fuzzyMather)).ToList();
|
2014-01-05 17:56:02 +08:00
|
|
|
returnList.ForEach(ScoreFilter);
|
2014-03-02 11:29:14 +08:00
|
|
|
returnList = returnList.OrderByDescending(o => o.Score).ToList();
|
2014-01-05 17:56:02 +08:00
|
|
|
|
|
|
|
return returnList.Select(c => new Result()
|
2014-01-04 20:26:13 +08:00
|
|
|
{
|
|
|
|
Title = c.Title,
|
2014-03-19 02:06:51 +08:00
|
|
|
SubTitle = c.ExecutePath,
|
2014-01-04 20:26:13 +08:00
|
|
|
IcoPath = c.IcoPath,
|
2014-08-10 16:50:35 +08:00
|
|
|
Score = c.Score,
|
2014-03-21 03:53:18 +08:00
|
|
|
Action = (e) =>
|
2014-01-04 20:26:13 +08:00
|
|
|
{
|
2014-07-21 19:48:17 +08:00
|
|
|
context.API.HideApp();
|
|
|
|
context.API.ShellRun(c.ExecutePath);
|
2014-02-28 23:21:01 +08:00
|
|
|
return true;
|
2014-10-24 13:09:51 +08:00
|
|
|
},
|
|
|
|
ContextMenu = new List<Result>()
|
|
|
|
{
|
|
|
|
new Result()
|
|
|
|
{
|
2014-10-24 16:48:14 +08:00
|
|
|
Title = "Run As Administrator",
|
2014-10-24 13:09:51 +08:00
|
|
|
Action = _ =>
|
|
|
|
{
|
|
|
|
context.API.HideApp();
|
|
|
|
context.API.ShellRun(c.ExecutePath,true);
|
|
|
|
return true;
|
|
|
|
},
|
|
|
|
IcoPath = "Images/cmd.png"
|
|
|
|
}
|
2014-01-04 20:26:13 +08:00
|
|
|
}
|
|
|
|
}).ToList();
|
|
|
|
}
|
|
|
|
|
2014-02-09 00:33:10 +08:00
|
|
|
private bool MatchProgram(Program program, FuzzyMatcher matcher)
|
2014-01-05 17:56:02 +08:00
|
|
|
{
|
2014-03-24 21:14:10 +08:00
|
|
|
if ((program.Score = matcher.Evaluate(program.Title).Score) > 0) return true;
|
|
|
|
if ((program.Score = matcher.Evaluate(program.PinyinTitle).Score) > 0) return true;
|
2014-08-10 16:50:35 +08:00
|
|
|
if (program.AbbrTitle != null && (program.Score = matcher.Evaluate(program.AbbrTitle).Score) > 0) return true;
|
2014-03-24 21:14:10 +08:00
|
|
|
if (program.ExecuteName != null && (program.Score = matcher.Evaluate(program.ExecuteName).Score) > 0) return true;
|
2014-01-05 17:56:02 +08:00
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2015-01-03 15:20:34 +08:00
|
|
|
public void Init(PluginInitContext context)
|
2014-01-04 20:26:13 +08:00
|
|
|
{
|
2014-03-21 03:53:18 +08:00
|
|
|
this.context = context;
|
2014-12-21 20:44:31 +08:00
|
|
|
programs = ProgramCacheStorage.Instance.Programs;
|
2014-12-16 21:36:48 +08:00
|
|
|
using (new Timeit("Program Index"))
|
2014-12-15 22:58:49 +08:00
|
|
|
{
|
2014-12-16 21:36:48 +08:00
|
|
|
IndexPrograms();
|
2014-12-15 22:58:49 +08:00
|
|
|
}
|
2014-08-10 22:22:54 +08:00
|
|
|
}
|
2014-03-21 03:53:18 +08:00
|
|
|
|
2014-08-14 22:21:07 +08:00
|
|
|
public static void IndexPrograms()
|
2014-08-10 22:22:54 +08:00
|
|
|
{
|
2014-12-16 21:36:48 +08:00
|
|
|
lock (lockObject)
|
2014-03-19 04:05:27 +08:00
|
|
|
{
|
2014-12-16 21:36:48 +08:00
|
|
|
List<ProgramSource> programSources = new List<ProgramSource>();
|
|
|
|
programSources.AddRange(LoadDeaultProgramSources());
|
|
|
|
if (UserSettingStorage.Instance.ProgramSources != null &&
|
|
|
|
UserSettingStorage.Instance.ProgramSources.Count(o => o.Enabled) > 0)
|
2014-08-10 22:22:54 +08:00
|
|
|
{
|
2014-12-16 21:36:48 +08:00
|
|
|
programSources.AddRange(UserSettingStorage.Instance.ProgramSources.Where(o => o.Enabled));
|
|
|
|
}
|
2014-08-10 22:22:54 +08:00
|
|
|
|
2014-12-16 21:36:48 +08:00
|
|
|
sources.Clear();
|
|
|
|
programSources.ForEach(source =>
|
|
|
|
{
|
|
|
|
Type sourceClass;
|
|
|
|
if (SourceTypes.TryGetValue(source.Type, out sourceClass))
|
2014-03-19 04:05:27 +08:00
|
|
|
{
|
2014-12-16 21:36:48 +08:00
|
|
|
ConstructorInfo constructorInfo = sourceClass.GetConstructor(new[] { typeof(ProgramSource) });
|
|
|
|
if (constructorInfo != null)
|
2014-08-14 22:21:07 +08:00
|
|
|
{
|
2014-12-16 21:36:48 +08:00
|
|
|
IProgramSource programSource =
|
|
|
|
constructorInfo.Invoke(new object[] { source }) as IProgramSource;
|
|
|
|
sources.Add(programSource);
|
2014-08-14 22:21:07 +08:00
|
|
|
}
|
2014-12-16 21:36:48 +08:00
|
|
|
}
|
|
|
|
});
|
2014-01-04 20:26:13 +08:00
|
|
|
|
2014-12-16 21:36:48 +08:00
|
|
|
var tempPrograms = new List<Program>();
|
|
|
|
foreach (var source in sources)
|
|
|
|
{
|
|
|
|
var list = source.LoadPrograms();
|
|
|
|
list.ForEach(o =>
|
2014-08-14 22:21:07 +08:00
|
|
|
{
|
2014-12-16 21:36:48 +08:00
|
|
|
o.Source = source;
|
|
|
|
});
|
|
|
|
tempPrograms.AddRange(list);
|
|
|
|
}
|
2014-08-14 22:21:07 +08:00
|
|
|
|
2014-12-16 21:36:48 +08:00
|
|
|
// filter duplicate program
|
2014-12-21 20:44:31 +08:00
|
|
|
programs = tempPrograms.GroupBy(x => new { x.ExecutePath, x.ExecuteName })
|
2014-12-16 21:36:48 +08:00
|
|
|
.Select(g => g.First()).ToList();
|
2014-08-10 22:22:54 +08:00
|
|
|
|
2014-12-21 20:44:31 +08:00
|
|
|
ProgramCacheStorage.Instance.Programs = programs;
|
2014-12-16 21:36:48 +08:00
|
|
|
ProgramCacheStorage.Instance.Save();
|
2014-01-04 20:26:13 +08:00
|
|
|
}
|
2014-08-10 22:22:54 +08:00
|
|
|
}
|
2014-03-22 13:50:20 +08:00
|
|
|
|
2014-08-10 22:22:54 +08:00
|
|
|
/// <summary>
|
|
|
|
/// Load program sources that wox always provide
|
|
|
|
/// </summary>
|
|
|
|
private static List<ProgramSource> LoadDeaultProgramSources()
|
|
|
|
{
|
|
|
|
var list = new List<ProgramSource>();
|
|
|
|
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;
|
2014-01-04 20:26:13 +08:00
|
|
|
}
|
|
|
|
|
2014-01-05 17:56:02 +08:00
|
|
|
private void ScoreFilter(Program p)
|
|
|
|
{
|
2014-03-19 02:06:51 +08:00
|
|
|
p.Score += p.Source.BonusPoints;
|
|
|
|
|
2014-01-05 17:56:02 +08:00
|
|
|
if (p.Title.Contains("启动") || p.Title.ToLower().Contains("start"))
|
2014-03-18 13:17:09 +08:00
|
|
|
p.Score += 10;
|
|
|
|
|
|
|
|
if (p.Title.Contains("帮助") || p.Title.ToLower().Contains("help") || p.Title.Contains("文档") || p.Title.ToLower().Contains("documentation"))
|
|
|
|
p.Score -= 10;
|
|
|
|
|
2014-01-05 17:56:02 +08:00
|
|
|
if (p.Title.Contains("卸载") || p.Title.ToLower().Contains("uninstall"))
|
2014-03-18 13:17:09 +08:00
|
|
|
p.Score -= 20;
|
2014-01-05 17:56:02 +08:00
|
|
|
}
|
2014-03-28 22:42:28 +08:00
|
|
|
|
2014-03-29 15:52:57 +08:00
|
|
|
#region ISettingProvider Members
|
|
|
|
|
|
|
|
public System.Windows.Controls.Control CreateSettingPanel()
|
|
|
|
{
|
|
|
|
return new ProgramSetting();
|
|
|
|
}
|
|
|
|
|
|
|
|
#endregion
|
2014-01-04 20:26:13 +08:00
|
|
|
}
|
2014-02-08 14:37:40 +08:00
|
|
|
}
|