2015-01-05 22:41:17 +08:00
|
|
|
|
using System;
|
2014-01-04 20:26:13 +08:00
|
|
|
|
using System.Collections.Generic;
|
2016-03-27 09:49:05 +08:00
|
|
|
|
using System.ComponentModel;
|
2015-11-01 02:06:57 +08:00
|
|
|
|
using System.Diagnostics;
|
2014-01-04 20:26:13 +08:00
|
|
|
|
using System.Linq;
|
2016-01-07 05:34:42 +08:00
|
|
|
|
using System.Windows.Controls;
|
2015-10-31 07:17:34 +08:00
|
|
|
|
using Wox.Infrastructure;
|
2015-11-08 01:32:58 +08:00
|
|
|
|
using Wox.Infrastructure.Logger;
|
2016-04-21 08:53:21 +08:00
|
|
|
|
using Wox.Infrastructure.Storage;
|
2016-01-06 14:45:08 +08:00
|
|
|
|
using Wox.Plugin.Program.ProgramSources;
|
2015-11-05 05:49:36 +08:00
|
|
|
|
using Stopwatch = Wox.Infrastructure.Stopwatch;
|
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
|
|
|
|
{
|
2016-05-07 10:55:09 +08:00
|
|
|
|
public class Main : ISettingProvider, IPlugin, IPluginI18n, IContextMenu, ISavable
|
2014-01-04 20:26:13 +08:00
|
|
|
|
{
|
2016-04-23 06:03:32 +08:00
|
|
|
|
private static List<Program> _programs = new List<Program>();
|
2016-08-18 08:16:40 +08:00
|
|
|
|
private static List<UWPApp> _uwpApps = new List<UWPApp>();
|
2016-04-23 06:03:32 +08:00
|
|
|
|
|
2016-03-27 09:49:05 +08:00
|
|
|
|
private PluginInitContext _context;
|
2016-04-21 08:53:21 +08:00
|
|
|
|
|
|
|
|
|
private static ProgramIndexCache _cache;
|
|
|
|
|
private static BinaryStorage<ProgramIndexCache> _cacheStorage;
|
2016-04-23 06:03:32 +08:00
|
|
|
|
private static Settings _settings;
|
2016-04-27 05:45:31 +08:00
|
|
|
|
private readonly PluginJsonStorage<Settings> _settingsStorage;
|
2016-04-21 08:53:21 +08:00
|
|
|
|
|
2016-05-07 10:55:09 +08:00
|
|
|
|
public Main()
|
2016-04-21 08:53:21 +08:00
|
|
|
|
{
|
2016-04-27 05:45:31 +08:00
|
|
|
|
_settingsStorage = new PluginJsonStorage<Settings>();
|
2016-04-21 08:53:21 +08:00
|
|
|
|
_settings = _settingsStorage.Load();
|
2016-07-22 03:49:01 +08:00
|
|
|
|
|
|
|
|
|
Stopwatch.Debug("Preload programs", () =>
|
|
|
|
|
{
|
|
|
|
|
_cacheStorage = new BinaryStorage<ProgramIndexCache>();
|
|
|
|
|
_cache = _cacheStorage.Load();
|
|
|
|
|
_programs = _cache.Programs;
|
|
|
|
|
});
|
|
|
|
|
Log.Info($"Preload {_programs.Count} programs from cache");
|
|
|
|
|
Stopwatch.Debug("Program Index", IndexPrograms);
|
2016-04-21 08:53:21 +08:00
|
|
|
|
}
|
|
|
|
|
|
2016-05-03 05:37:01 +08:00
|
|
|
|
public void Save()
|
2016-04-21 08:53:21 +08:00
|
|
|
|
{
|
|
|
|
|
_settingsStorage.Save();
|
|
|
|
|
_cacheStorage.Save();
|
|
|
|
|
}
|
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
|
|
|
|
{
|
2016-08-18 08:16:40 +08:00
|
|
|
|
var results1 = _programs.AsParallel()
|
|
|
|
|
.Where(p => Score(p, query.Search) > 0)
|
|
|
|
|
.Select(ResultFromProgram);
|
|
|
|
|
|
|
|
|
|
var results2 = _uwpApps.AsParallel()
|
|
|
|
|
.Where(u => Score(u, query.Search) > 0)
|
|
|
|
|
.Select(ResultFromUWPApp);
|
|
|
|
|
var result = results1.Concat(results2).ToList();
|
|
|
|
|
|
|
|
|
|
return result;
|
2014-01-04 20:26:13 +08:00
|
|
|
|
}
|
|
|
|
|
|
2016-07-22 03:49:01 +08:00
|
|
|
|
public Result ResultFromProgram(Program p)
|
|
|
|
|
{
|
|
|
|
|
var result = new Result
|
|
|
|
|
{
|
|
|
|
|
Title = p.Title,
|
|
|
|
|
SubTitle = p.Path,
|
|
|
|
|
IcoPath = p.IcoPath,
|
|
|
|
|
Score = p.Score,
|
|
|
|
|
ContextData = p,
|
|
|
|
|
Action = e =>
|
|
|
|
|
{
|
|
|
|
|
var info = new ProcessStartInfo
|
|
|
|
|
{
|
|
|
|
|
FileName = p.Path,
|
|
|
|
|
WorkingDirectory = p.Directory
|
|
|
|
|
};
|
|
|
|
|
var hide = StartProcess(info);
|
|
|
|
|
return hide;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
return result;
|
|
|
|
|
}
|
2016-08-18 08:16:40 +08:00
|
|
|
|
public Result ResultFromUWPApp(UWPApp uwpApp)
|
|
|
|
|
{
|
|
|
|
|
var app = uwpApp.Apps[0];
|
|
|
|
|
var result = new Result
|
|
|
|
|
{
|
|
|
|
|
Title = app.DisplayName,
|
|
|
|
|
SubTitle = $"Windows Store app: {app.Description}",
|
|
|
|
|
Icon = app.Logo,
|
|
|
|
|
Score = uwpApp.Score,
|
|
|
|
|
ContextData = app,
|
|
|
|
|
Action = e =>
|
|
|
|
|
{
|
|
|
|
|
app.Launch();
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
return result;
|
|
|
|
|
}
|
|
|
|
|
|
2016-07-22 03:49:01 +08:00
|
|
|
|
|
2016-04-24 07:37:25 +08:00
|
|
|
|
private int Score(Program program, string query)
|
2014-01-05 17:56:02 +08:00
|
|
|
|
{
|
2016-04-24 07:37:25 +08:00
|
|
|
|
var score1 = StringMatcher.Score(program.Title, query);
|
|
|
|
|
var score2 = StringMatcher.ScoreForPinyin(program.Title, query);
|
2016-04-24 20:35:21 +08:00
|
|
|
|
var score3 = StringMatcher.Score(program.ExecutableName, query);
|
|
|
|
|
var score = new[] { score1, score2, score3 }.Max();
|
2016-04-24 07:37:25 +08:00
|
|
|
|
program.Score = score;
|
|
|
|
|
return score;
|
2014-01-05 17:56:02 +08:00
|
|
|
|
}
|
|
|
|
|
|
2016-08-18 08:16:40 +08:00
|
|
|
|
private int Score(UWPApp app, string query)
|
|
|
|
|
{
|
|
|
|
|
var score1 = StringMatcher.Score(app.Apps[0].DisplayName, query);
|
|
|
|
|
var score2 = StringMatcher.ScoreForPinyin(app.Apps[0].DisplayName, query);
|
|
|
|
|
var score3 = StringMatcher.Score(app.Apps[0].Description, query);
|
|
|
|
|
var score = new[] { score1, score2, score3 }.Max();
|
|
|
|
|
app.Score = score;
|
|
|
|
|
return score;
|
|
|
|
|
}
|
|
|
|
|
|
2015-01-03 15:20:34 +08:00
|
|
|
|
public void Init(PluginInitContext context)
|
2014-01-04 20:26:13 +08:00
|
|
|
|
{
|
2016-04-21 08:53:21 +08:00
|
|
|
|
_context = context;
|
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
|
|
|
|
{
|
2016-07-22 03:49:01 +08:00
|
|
|
|
var sources = ProgramSources();
|
2016-04-23 06:03:32 +08:00
|
|
|
|
|
2016-07-22 03:49:01 +08:00
|
|
|
|
var programs = sources.AsParallel()
|
|
|
|
|
.SelectMany(s => s.LoadPrograms())
|
|
|
|
|
// filter duplicate program
|
|
|
|
|
.GroupBy(x => new { ExecutePath = x.Path, ExecuteName = x.ExecutableName })
|
|
|
|
|
.Select(g => g.First());
|
|
|
|
|
programs = programs.Select(ScoreFilter);
|
2016-04-23 06:03:32 +08:00
|
|
|
|
|
2016-07-22 03:49:01 +08:00
|
|
|
|
_programs = programs.ToList();
|
2016-07-22 03:07:23 +08:00
|
|
|
|
_cache.Programs = _programs;
|
2016-08-18 08:16:40 +08:00
|
|
|
|
|
|
|
|
|
var windows10 = new Version(10, 0);
|
|
|
|
|
var support = Environment.OSVersion.Version.Major >= windows10.Major;
|
|
|
|
|
if (support)
|
|
|
|
|
{
|
|
|
|
|
_uwpApps = UWPApp.All();
|
|
|
|
|
}
|
2014-08-10 22:22:54 +08:00
|
|
|
|
}
|
2014-03-22 13:50:20 +08:00
|
|
|
|
|
2016-07-22 03:49:01 +08:00
|
|
|
|
private static List<ProgramSource> ProgramSources()
|
2014-08-10 22:22:54 +08:00
|
|
|
|
{
|
2016-07-22 03:49:01 +08:00
|
|
|
|
var sources = new List<ProgramSource>
|
2014-08-10 22:22:54 +08:00
|
|
|
|
{
|
2016-07-22 02:51:47 +08:00
|
|
|
|
new CommonStartMenuProgramSource
|
2016-04-23 06:03:32 +08:00
|
|
|
|
{
|
|
|
|
|
BonusPoints = 0,
|
|
|
|
|
Enabled = _settings.EnableStartMenuSource,
|
|
|
|
|
},
|
2016-07-22 02:51:47 +08:00
|
|
|
|
new UserStartMenuProgramSource
|
2016-04-23 06:03:32 +08:00
|
|
|
|
{
|
|
|
|
|
BonusPoints = 0,
|
|
|
|
|
Enabled = _settings.EnableStartMenuSource,
|
|
|
|
|
},
|
2016-07-22 02:51:47 +08:00
|
|
|
|
new AppPathsProgramSource
|
2016-04-23 06:03:32 +08:00
|
|
|
|
{
|
|
|
|
|
BonusPoints = -10,
|
|
|
|
|
Enabled = _settings.EnableRegistrySource,
|
|
|
|
|
}
|
|
|
|
|
};
|
2016-07-22 03:49:01 +08:00
|
|
|
|
|
|
|
|
|
if (_settings.ProgramSources.Count(o => o.Enabled) > 0)
|
|
|
|
|
{
|
|
|
|
|
sources.AddRange(_settings.ProgramSources);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return sources;
|
2014-01-04 20:26:13 +08:00
|
|
|
|
}
|
|
|
|
|
|
2016-07-22 03:49:01 +08:00
|
|
|
|
private static Program ScoreFilter(Program p)
|
2014-01-05 17:56:02 +08:00
|
|
|
|
{
|
2014-03-19 02:06:51 +08:00
|
|
|
|
p.Score += p.Source.BonusPoints;
|
2016-07-22 03:49:01 +08:00
|
|
|
|
var start = new[] { "启动", "start" };
|
|
|
|
|
var doc = new[] { "帮助", "help", "文档", "documentation" };
|
|
|
|
|
var uninstall = new[] { "卸载", "uninstall" };
|
2014-03-19 02:06:51 +08:00
|
|
|
|
|
2016-07-22 03:49:01 +08:00
|
|
|
|
var contained = start.Any(s => p.Title.ToLower().Contains(s));
|
|
|
|
|
if (contained)
|
|
|
|
|
{
|
2014-03-18 13:17:09 +08:00
|
|
|
|
p.Score += 10;
|
2016-07-22 03:49:01 +08:00
|
|
|
|
}
|
|
|
|
|
contained = doc.Any(d => p.Title.ToLower().Contains(d));
|
|
|
|
|
if (contained)
|
|
|
|
|
{
|
2014-03-18 13:17:09 +08:00
|
|
|
|
p.Score -= 10;
|
2016-07-22 03:49:01 +08:00
|
|
|
|
}
|
|
|
|
|
contained = uninstall.Any(u => p.Title.ToLower().Contains(u));
|
|
|
|
|
if (contained)
|
|
|
|
|
{
|
2014-03-18 13:17:09 +08:00
|
|
|
|
p.Score -= 20;
|
2016-07-22 03:49:01 +08:00
|
|
|
|
}
|
|
|
|
|
|
2015-11-11 14:20:52 +08:00
|
|
|
|
return p;
|
2014-01-05 17:56:02 +08:00
|
|
|
|
}
|
2014-03-28 22:42:28 +08:00
|
|
|
|
|
2016-01-07 05:34:42 +08:00
|
|
|
|
public Control CreateSettingPanel()
|
2014-03-29 15:52:57 +08:00
|
|
|
|
{
|
2016-03-28 10:09:57 +08:00
|
|
|
|
return new ProgramSetting(_context, _settings);
|
2014-03-29 15:52:57 +08:00
|
|
|
|
}
|
|
|
|
|
|
2015-02-07 20:17:49 +08:00
|
|
|
|
public string GetTranslatedPluginTitle()
|
|
|
|
|
{
|
2016-03-27 09:49:05 +08:00
|
|
|
|
return _context.API.GetTranslation("wox_plugin_program_plugin_name");
|
2015-02-07 20:17:49 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public string GetTranslatedPluginDescription()
|
|
|
|
|
{
|
2016-03-27 09:49:05 +08:00
|
|
|
|
return _context.API.GetTranslation("wox_plugin_program_plugin_description");
|
2015-02-07 20:17:49 +08:00
|
|
|
|
}
|
2015-02-07 23:49:46 +08:00
|
|
|
|
|
|
|
|
|
public List<Result> LoadContextMenus(Result selectedResult)
|
|
|
|
|
{
|
|
|
|
|
Program p = selectedResult.ContextData as Program;
|
2016-08-18 08:16:40 +08:00
|
|
|
|
if (p != null)
|
2015-02-07 23:49:46 +08:00
|
|
|
|
{
|
2016-08-18 08:16:40 +08:00
|
|
|
|
List<Result> contextMenus = new List<Result>
|
2015-02-07 23:49:46 +08:00
|
|
|
|
{
|
2016-08-18 08:16:40 +08:00
|
|
|
|
new Result
|
2015-02-07 23:49:46 +08:00
|
|
|
|
{
|
2016-08-18 08:16:40 +08:00
|
|
|
|
Title = _context.API.GetTranslation("wox_plugin_program_run_as_administrator"),
|
|
|
|
|
Action = _ =>
|
2016-01-04 07:18:51 +08:00
|
|
|
|
{
|
2016-08-18 08:16:40 +08:00
|
|
|
|
var info = new ProcessStartInfo
|
|
|
|
|
{
|
|
|
|
|
FileName = p.Path,
|
|
|
|
|
WorkingDirectory = p.Directory,
|
|
|
|
|
Verb = "runas"
|
|
|
|
|
};
|
|
|
|
|
var hide = StartProcess(info);
|
|
|
|
|
return hide;
|
|
|
|
|
},
|
|
|
|
|
IcoPath = "Images/cmd.png"
|
2015-02-07 23:49:46 +08:00
|
|
|
|
},
|
2016-08-18 08:16:40 +08:00
|
|
|
|
new Result
|
2015-02-07 23:49:46 +08:00
|
|
|
|
{
|
2016-08-18 08:16:40 +08:00
|
|
|
|
Title = _context.API.GetTranslation("wox_plugin_program_open_containing_folder"),
|
|
|
|
|
Action = _ =>
|
|
|
|
|
{
|
|
|
|
|
var hide = StartProcess(new ProcessStartInfo(p.Directory));
|
|
|
|
|
return hide;
|
|
|
|
|
},
|
|
|
|
|
IcoPath = "Images/folder.png"
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
return contextMenus;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
return new List<Result>();
|
|
|
|
|
}
|
2015-02-07 23:49:46 +08:00
|
|
|
|
}
|
2016-03-27 09:49:05 +08:00
|
|
|
|
|
|
|
|
|
private bool StartProcess(ProcessStartInfo info)
|
|
|
|
|
{
|
|
|
|
|
bool hide;
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
Process.Start(info);
|
|
|
|
|
hide = true;
|
|
|
|
|
}
|
|
|
|
|
catch (Win32Exception)
|
|
|
|
|
{
|
|
|
|
|
var name = $"Plugin: {_context.CurrentPluginMetadata.Name}";
|
|
|
|
|
var message = "Can't open this file";
|
|
|
|
|
_context.API.ShowMsg(name, message, string.Empty);
|
|
|
|
|
hide = false;
|
|
|
|
|
}
|
|
|
|
|
return hide;
|
|
|
|
|
}
|
2014-01-04 20:26:13 +08:00
|
|
|
|
}
|
2015-02-07 23:49:46 +08:00
|
|
|
|
}
|