PowerToys/Plugins/Wox.Plugin.Program/Main.cs

281 lines
9.0 KiB
C#
Raw Normal View History

using System;
2014-01-04 20:26:13 +08:00
using System.Collections.Generic;
using System.ComponentModel;
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;
using Wox.Infrastructure.Logger;
using Wox.Infrastructure.Storage;
2016-01-06 14:45:08 +08:00
using Wox.Plugin.Program.ProgramSources;
using Stopwatch = Wox.Infrastructure.Stopwatch;
2014-01-04 20:26:13 +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
{
private static List<Program> _programs = new List<Program>();
private static List<UWPApp> _uwpApps = new List<UWPApp>();
private PluginInitContext _context;
private static ProgramIndexCache _cache;
private static BinaryStorage<ProgramIndexCache> _cacheStorage;
private static Settings _settings;
2016-04-27 05:45:31 +08:00
private readonly PluginJsonStorage<Settings> _settingsStorage;
2016-05-07 10:55:09 +08:00
public Main()
{
2016-04-27 05:45:31 +08:00
_settingsStorage = new PluginJsonStorage<Settings>();
_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);
}
public void Save()
{
_settingsStorage.Save();
_cacheStorage.Save();
}
public List<Result> Query(Query query)
2014-01-04 20:26:13 +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;
}
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
private int Score(Program program, string query)
2014-01-05 17:56:02 +08:00
{
var score1 = StringMatcher.Score(program.Title, query);
var score2 = StringMatcher.ScoreForPinyin(program.Title, query);
var score3 = StringMatcher.Score(program.ExecutableName, query);
var score = new[] { score1, score2, score3 }.Max();
program.Score = score;
return score;
2014-01-05 17:56:02 +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;
}
public void Init(PluginInitContext context)
2014-01-04 20:26:13 +08:00
{
_context = context;
}
public static void IndexPrograms()
{
2016-07-22 03:49:01 +08:00
var sources = ProgramSources();
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-07-22 03:49:01 +08:00
_programs = programs.ToList();
_cache.Programs = _programs;
var windows10 = new Version(10, 0);
var support = Environment.OSVersion.Version.Major >= windows10.Major;
if (support)
{
_uwpApps = UWPApp.All();
}
}
2014-03-22 13:50:20 +08:00
2016-07-22 03:49:01 +08:00
private static List<ProgramSource> ProgramSources()
{
2016-07-22 03:49:01 +08:00
var sources = new List<ProgramSource>
{
new CommonStartMenuProgramSource
{
BonusPoints = 0,
Enabled = _settings.EnableStartMenuSource,
},
new UserStartMenuProgramSource
{
BonusPoints = 0,
Enabled = _settings.EnableStartMenuSource,
},
new AppPathsProgramSource
{
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
{
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" };
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
}
return p;
2014-01-05 17:56:02 +08:00
}
2016-01-07 05:34:42 +08:00
public Control CreateSettingPanel()
{
return new ProgramSetting(_context, _settings);
}
2015-02-07 20:17:49 +08:00
public string GetTranslatedPluginTitle()
{
return _context.API.GetTranslation("wox_plugin_program_plugin_name");
2015-02-07 20:17:49 +08:00
}
public string GetTranslatedPluginDescription()
{
return _context.API.GetTranslation("wox_plugin_program_plugin_description");
2015-02-07 20:17:49 +08:00
}
public List<Result> LoadContextMenus(Result selectedResult)
{
Program p = selectedResult.ContextData as Program;
if (p != null)
{
List<Result> contextMenus = new List<Result>
{
new Result
{
Title = _context.API.GetTranslation("wox_plugin_program_run_as_administrator"),
Action = _ =>
{
var info = new ProcessStartInfo
{
FileName = p.Path,
WorkingDirectory = p.Directory,
Verb = "runas"
};
var hide = StartProcess(info);
return hide;
},
IcoPath = "Images/cmd.png"
},
new Result
{
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>();
}
}
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
}
}