Skip to content

Commit

Permalink
Move context menu and socre into program model
Browse files Browse the repository at this point in the history
  • Loading branch information
@bao-qian
bao-qian committed Aug 22, 2016
1 parent af9c809 commit 165c12f
Show file tree
Hide file tree
Showing 5 changed files with 215 additions and 166 deletions.
144 changes: 13 additions & 131 deletions Plugins/Wox.Plugin.Program/Main.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ public class Main : ISettingProvider, IPlugin, IPluginI18n, IContextMenu, ISavab
private static Win32[] _win32s = { };
private static UWP.Application[] _uwps = { };

private PluginInitContext _context;
private static PluginInitContext _context;

private static ProgramIndexCache _cache;
private static BinaryStorage<ProgramIndexCache> _cacheStorage;
Expand Down Expand Up @@ -49,105 +49,15 @@ public void Save()
public List<Result> Query(Query query)
{
var results1 = _win32s.AsParallel()
.Where(p => Score(p, query.Search) > 0)
.Select(ResultFromWin32);
.Select(p => p.Result(query.Search, _context.API))
.Where(r => r.Score > 0);
var results2 = _uwps.AsParallel()
.Where(app => Score(app, query.Search) > 0)
.Select(ResultFromUWP);
.Select(p => p.Result(query.Search, _context.API))
.Where(r => r.Score > 0);
var result = results1.Concat(results2).ToList();
return result;
}

public Result ResultFromWin32(Win32 program)
{
var result = new Result
{
SubTitle = program.FullPath,
IcoPath = program.IcoPath,
Score = program.Score,
ContextData = program,
Action = e =>
{
var info = new ProcessStartInfo
{
FileName = program.FullPath,
WorkingDirectory = program.ParentDirectory
};
var hide = StartProcess(info);
return hide;
}
};

if (program.Description.Length >= program.Name.Length &&
program.Description.Substring(0, program.Name.Length) == program.Name)
{
result.Title = program.Description;
}
else if (!string.IsNullOrEmpty(program.Description))
{
result.Title = $"{program.Name}: {program.Description}";
}
else
{
result.Title = program.Name;
}

return result;
}
public Result ResultFromUWP(UWP.Application app)
{
var result = new Result
{
SubTitle = $"{app.Location}",
Icon = app.Logo,
Score = app.Score,
ContextData = app,
Action = e =>
{
app.Launch();
return true;
}
};

if (app.Description.Length >= app.DisplayName.Length &&
app.Description.Substring(0, app.DisplayName.Length) == app.DisplayName)
{
result.Title = app.Description;
}
else if (!string.IsNullOrEmpty(app.Description))
{
result.Title = $"{app.DisplayName}: {app.Description}";
}
else
{
result.Title = app.DisplayName;
}
return result;
}


private int Score(Win32 program, string query)
{
var score1 = StringMatcher.Score(program.Name, query);
var score2 = StringMatcher.ScoreForPinyin(program.Name, query);
var score3 = StringMatcher.Score(program.Description, query);
var score4 = StringMatcher.ScoreForPinyin(program.Description, query);
var score5 = StringMatcher.Score(program.ExecutableName, query);
var score = new[] { score1, score2, score3, score4, score5 }.Max();
program.Score = score;
return score;
}

private int Score(UWP.Application app, string query)
{
var score1 = StringMatcher.Score(app.DisplayName, query);
var score2 = StringMatcher.ScoreForPinyin(app.DisplayName, query);
var score3 = StringMatcher.Score(app.Description, query);
var score = new[] { score1, score2, score3 }.Max();
app.Score = score;
return score;
}

public void Init(PluginInitContext context)
{
_context = context;
Expand Down Expand Up @@ -176,58 +86,30 @@ public string GetTranslatedPluginDescription()

public List<Result> LoadContextMenus(Result selectedResult)
{
Win32 p = selectedResult.ContextData as Win32;
if (p != null)
var program = selectedResult.ContextData as IProgram;
if (program != 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.FullPath,
WorkingDirectory = p.ParentDirectory,
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.ParentDirectory));
return hide;
},
IcoPath = "Images/folder.png"
}
};
return contextMenus;
var menus = program.ContextMenus(_context.API);
return menus;
}
else
{
return new List<Result>();
}
}

private bool StartProcess(ProcessStartInfo info)
public static bool StartProcess(ProcessStartInfo info)
{
bool hide;
try
{
Process.Start(info);
hide = true;
}
catch (Win32Exception)
catch (Exception)
{
var name = $"Plugin: {_context.CurrentPluginMetadata.Name}";
var message = "Can't open this file";
var name = "Plugin: Program";
var message = $"Can't start: {info.FileName}";
_context.API.ShowMsg(name, message, string.Empty);
hide = false;
}
Expand Down
14 changes: 14 additions & 0 deletions Plugins/Wox.Plugin.Program/Programs/IProgram.cs
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace Wox.Plugin.Program.Programs
{
public interface IProgram
{
List<Result> ContextMenus(IPublicAPI api);
Result Result(string query, IPublicAPI api);
}
}
84 changes: 74 additions & 10 deletions Plugins/Wox.Plugin.Program/Programs/UWP.cs
View file
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Runtime.InteropServices;
using System.Security.Principal;
using System.Threading.Tasks;
using System.Windows.Media;
using System.Windows.Media.Imaging;
using Windows.ApplicationModel;
Expand Down Expand Up @@ -190,7 +190,7 @@ public override int GetHashCode()
}


public class Application
public class Application : IProgram
{
public string DisplayName { get; set; }
public string Description { get; set; }
Expand All @@ -201,18 +201,82 @@ public class Application
public string BackgroundColor { get; set; }
public string LogoPath { get; set; }

public int Score { get; set; }
public string Location { get; set; }

private int Score(string query)
{
var score1 = StringMatcher.Score(DisplayName, query);
var score2 = StringMatcher.ScoreForPinyin(DisplayName, query);
var score3 = StringMatcher.Score(Description, query);
var score = new[] { score1, score2, score3 }.Max();
return score;
}

// todo: wrap with try exception
public void Launch()
public Result Result(string query, IPublicAPI api)
{
var appManager = new ApplicationActivationManager();
uint unusedPid;
const string noArgs = "";
const ACTIVATEOPTIONS noFlags = ACTIVATEOPTIONS.AO_NONE;
appManager.ActivateApplication(UserModelId, noArgs, noFlags, out unusedPid);
var result = new Result
{
SubTitle = Location,
Icon = Logo,
Score = Score(query),
ContextData = this,
Action = e =>
{
Launch(api);
return true;
}
};

if (Description.Length >= DisplayName.Length &&
Description.Substring(0, DisplayName.Length) == DisplayName)
{
result.Title = Description;
}
else if (!string.IsNullOrEmpty(Description))
{
result.Title = $"{DisplayName}: {Description}";
}
else
{
result.Title = DisplayName;
}
return result;
}

public List<Result> ContextMenus(IPublicAPI api)
{
var contextMenus = new List<Result>
{
new Result
{
Title = api.GetTranslation("wox_plugin_program_open_containing_folder"),
Action = _ =>
{
var hide = Main.StartProcess(new ProcessStartInfo(Location));
return hide;
},
IcoPath = "Images/folder.png"
}
};
return contextMenus;
}

private void Launch(IPublicAPI api)
{
try
{
var appManager = new ApplicationActivationManager();
uint unusedPid;
const string noArgs = "";
const ACTIVATEOPTIONS noFlags = ACTIVATEOPTIONS.AO_NONE;
appManager.ActivateApplication(UserModelId, noArgs, noFlags, out unusedPid);
}
catch (Exception)
{
var name = "Plugin: Program";
var message = $"Can't start UWP: {DisplayName}";
api.ShowMsg(name, message, string.Empty);
}
}


Expand Down
Loading

0 comments on commit 165c12f

Please sign in to comment.