PowerToys/Plugins/Wox.Plugin.CMD/CMD.cs

237 lines
8.3 KiB
C#
Raw Normal View History

2014-01-03 18:16:05 +08:00
using System;
using System.Collections.Generic;
2015-02-04 23:16:41 +08:00
using System.Diagnostics;
using System.IO;
2014-01-03 18:16:05 +08:00
using System.Linq;
2015-01-07 18:51:11 +08:00
using System.Reflection;
using System.Windows.Forms;
using WindowsInput;
using WindowsInput.Native;
using Wox.Infrastructure;
using Wox.Infrastructure.Hotkey;
2015-02-05 18:43:05 +08:00
using Wox.Plugin.Features;
using Control = System.Windows.Controls.Control;
2014-01-03 18:16:05 +08:00
namespace Wox.Plugin.CMD
2014-01-03 18:16:05 +08:00
{
2015-02-05 22:20:42 +08:00
public class CMD : IPlugin, ISettingProvider, IPluginI18n, IInstantQuery, IExclusiveQuery
2014-01-03 18:16:05 +08:00
{
private PluginInitContext context;
private bool WinRStroked;
private readonly KeyboardSimulator keyboardSimulator = new KeyboardSimulator(new InputSimulator());
public List<Result> Query(Query query)
2014-01-03 18:16:05 +08:00
{
2014-01-03 23:52:36 +08:00
List<Result> results = new List<Result>();
2014-03-27 17:25:06 +08:00
List<Result> pushedResults = new List<Result>();
if (query.Search == ">")
{
2015-01-20 20:05:38 +08:00
return GetAllHistoryCmds();
}
2014-12-15 18:50:28 +08:00
if (query.Search.StartsWith(">") && query.Search.Length > 1)
2014-01-03 23:52:36 +08:00
{
string cmd = query.Search.Substring(1);
2015-01-20 20:05:38 +08:00
var queryCmd = GetCurrentCmd(cmd);
context.API.PushResults(query, context.CurrentPluginMetadata, new List<Result>() { queryCmd });
pushedResults.Add(queryCmd);
2015-01-20 20:05:38 +08:00
var history = GetHistoryCmds(cmd, queryCmd);
context.API.PushResults(query, context.CurrentPluginMetadata, history);
2014-03-27 17:25:06 +08:00
pushedResults.AddRange(history);
2014-03-20 04:12:50 +08:00
try
{
string basedir = null;
string dir = null;
string excmd = Environment.ExpandEnvironmentVariables(cmd);
if (Directory.Exists(excmd) && (cmd.EndsWith("/") || cmd.EndsWith(@"\")))
{
basedir = excmd;
dir = cmd;
}
else if (Directory.Exists(Path.GetDirectoryName(excmd)))
{
basedir = Path.GetDirectoryName(excmd);
var dirn = Path.GetDirectoryName(cmd);
dir = (dirn.EndsWith("/") || dirn.EndsWith(@"\")) ? dirn : cmd.Substring(0, dirn.Length + 1);
}
if (basedir != null)
{
2014-03-27 17:25:06 +08:00
List<string> autocomplete = Directory.GetFileSystemEntries(basedir).Select(o => dir + Path.GetFileName(o)).Where(o => o.StartsWith(cmd, StringComparison.OrdinalIgnoreCase) && !results.Any(p => o.Equals(p.Title, StringComparison.OrdinalIgnoreCase)) && !pushedResults.Any(p => o.Equals(p.Title, StringComparison.OrdinalIgnoreCase))).ToList();
2014-03-20 04:12:50 +08:00
autocomplete.Sort();
2014-03-23 16:17:41 +08:00
results.AddRange(autocomplete.ConvertAll(m => new Result()
{
2014-03-20 04:12:50 +08:00
Title = m,
2015-01-20 20:05:38 +08:00
IcoPath = "Images/cmd.png",
2014-03-20 04:12:50 +08:00
Action = (c) =>
{
ExecuteCmd(m);
return true;
2014-12-15 18:50:28 +08:00
},
ContextMenu = GetContextMenus(m)
2014-03-20 04:12:50 +08:00
}));
}
}
catch (Exception) { }
2014-01-03 23:52:36 +08:00
}
return results;
2014-01-03 18:16:05 +08:00
}
2015-01-20 20:05:38 +08:00
private List<Result> GetHistoryCmds(string cmd, Result result)
{
IEnumerable<Result> history = CMDStorage.Instance.CMDHistory.Where(o => o.Key.Contains(cmd))
.OrderByDescending(o => o.Value)
.Select(m =>
{
if (m.Key == cmd)
{
result.SubTitle = "this command has been executed " + m.Value + " times";
return null;
}
var ret = new Result
{
Title = m.Key,
SubTitle = "this command has been executed " + m.Value + " times",
IcoPath = "Images/cmd.png",
Action = (c) =>
{
ExecuteCmd(m.Key);
return true;
},
ContextMenu = GetContextMenus(m.Key)
};
return ret;
}).Where(o => o != null).Take(4);
return history.ToList();
}
private Result GetCurrentCmd(string cmd)
{
Result result = new Result
{
Title = cmd,
Score = 5000,
SubTitle = "execute command through command shell",
IcoPath = "Images/cmd.png",
Action = (c) =>
{
ExecuteCmd(cmd);
return true;
},
ContextMenu = GetContextMenus(cmd)
};
return result;
}
private List<Result> GetAllHistoryCmds()
{
IEnumerable<Result> history = CMDStorage.Instance.CMDHistory.OrderByDescending(o => o.Value)
.Select(m => new Result
{
Title = m.Key,
SubTitle = "this command has been executed " + m.Value + " times",
IcoPath = "Images/cmd.png",
Action = (c) =>
{
ExecuteCmd(m.Key);
return true;
},
ContextMenu = GetContextMenus(m.Key)
}).Take(5);
return history.ToList();
}
2014-12-15 18:50:28 +08:00
private List<Result> GetContextMenus(string cmd)
{
return new List<Result>()
{
new Result()
{
Title = "Run As Administrator",
Action = c =>
{
context.API.HideApp();
ExecuteCmd(cmd, true);
return true;
},
IcoPath = "Images/cmd.png"
}
};
}
2014-12-15 18:50:28 +08:00
private void ExecuteCmd(string cmd, bool runAsAdministrator = false)
{
2014-12-15 18:50:28 +08:00
if (context.API.ShellRun(cmd, runAsAdministrator))
2014-03-23 16:17:41 +08:00
CMDStorage.Instance.AddCmdHistory(cmd);
}
public void Init(PluginInitContext context)
2014-01-03 18:16:05 +08:00
{
this.context = context;
context.API.GlobalKeyboardEvent += API_GlobalKeyboardEvent;
}
bool API_GlobalKeyboardEvent(int keyevent, int vkcode, SpecialKeyState state)
{
if (CMDStorage.Instance.ReplaceWinR)
{
if (keyevent == (int)KeyEvent.WM_KEYDOWN && vkcode == (int)Keys.R && state.WinPressed)
{
WinRStroked = true;
OnWinRPressed();
return false;
}
if (keyevent == (int)KeyEvent.WM_KEYUP && WinRStroked && vkcode == (int)Keys.LWin)
{
WinRStroked = false;
keyboardSimulator.ModifiedKeyStroke(VirtualKeyCode.LWIN, VirtualKeyCode.CONTROL);
return false;
}
}
return true;
}
private void OnWinRPressed()
{
context.API.ShowApp();
context.API.ChangeQuery(">");
2014-01-03 23:52:36 +08:00
}
public Control CreateSettingPanel()
{
2014-12-15 18:50:28 +08:00
return new CMDSetting();
}
2015-01-07 18:51:11 +08:00
public string GetLanguagesFolder()
{
return Path.Combine(Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location), "Languages");
}
2015-02-07 20:17:49 +08:00
public string GetTranslatedPluginTitle()
{
return context.API.GetTranslation("wox_plugin_cmd_plugin_name");
}
public string GetTranslatedPluginDescription()
{
return context.API.GetTranslation("wox_plugin_cmd_plugin_description");
}
2015-02-05 22:20:42 +08:00
public bool IsInstantQuery(string query)
{
if (query.StartsWith(">")) return true;
return false;
}
2015-02-05 18:43:05 +08:00
2015-02-05 22:20:42 +08:00
public bool IsExclusiveQuery(Query query)
2015-02-05 18:43:05 +08:00
{
return query.Search.StartsWith(">");
}
2014-01-03 18:16:05 +08:00
}
}