using System; using System.Collections.Generic; using System.Diagnostics; using System.IO; using System.Linq; using System.Windows.Forms; using WindowsInput; using WindowsInput.Native; using Wox.Infrastructure.Hotkey; using Wox.Infrastructure.Logger; using Wox.Infrastructure.Storage; using Control = System.Windows.Controls.Control; namespace Wox.Plugin.CMD { public class CMD : IPlugin, ISettingProvider, IPluginI18n, IInstantQuery, IContextMenu { private PluginInitContext context; private bool WinRStroked; private readonly KeyboardSimulator keyboardSimulator = new KeyboardSimulator(new InputSimulator()); private readonly CMDHistory _settings; private readonly PluginSettingsStorage _storage; public CMD() { _storage = new PluginSettingsStorage(); _settings = _storage.Load(); } ~CMD() { _storage.Save(); } public List Query(Query query) { List results = new List(); string cmd = query.Search; if (string.IsNullOrEmpty(cmd)) { return ResultsFromlHistory(); } else { var queryCmd = GetCurrentCmd(cmd); results.Add(queryCmd); var history = GetHistoryCmds(cmd, queryCmd); results.AddRange(history); 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) ?? string.Empty)) { basedir = Path.GetDirectoryName(excmd); var dirn = Path.GetDirectoryName(cmd); dir = (dirn.EndsWith("/") || dirn.EndsWith(@"\")) ? dirn : cmd.Substring(0, dirn.Length + 1); } if (basedir != null) { var 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)) && !results.Any(p => o.Equals(p.Title, StringComparison.OrdinalIgnoreCase))).ToList(); autocomplete.Sort(); results.AddRange(autocomplete.ConvertAll(m => new Result { Title = m, IcoPath = "Images/cmd.png", Action = c => { ExecuteCMD(m); return true; } })); } } catch (Exception e) { Log.Error(e); } return results; } } private List GetHistoryCmds(string cmd, Result result) { IEnumerable history = _settings.Count.Where(o => o.Key.Contains(cmd)) .OrderByDescending(o => o.Value) .Select(m => { if (m.Key == cmd) { result.SubTitle = string.Format(context.API.GetTranslation("wox_plugin_cmd_cmd_has_been_executed_times"), m.Value); return null; } var ret = new Result { Title = m.Key, SubTitle = string.Format(context.API.GetTranslation("wox_plugin_cmd_cmd_has_been_executed_times"), m.Value), IcoPath = "Images/cmd.png", Action = c => { ExecuteCMD(m.Key); return true; } }; 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 = context.API.GetTranslation("wox_plugin_cmd_execute_through_shell"), IcoPath = "Images/cmd.png", Action = c => { ExecuteCMD(cmd); return true; } }; return result; } private List ResultsFromlHistory() { IEnumerable history = _settings.Count.OrderByDescending(o => o.Value) .Select(m => new Result { Title = m.Key, SubTitle = string.Format(context.API.GetTranslation("wox_plugin_cmd_cmd_has_been_executed_times"), m.Value), IcoPath = "Images/cmd.png", Action = c => { ExecuteCMD(m.Key); return true; } }).Take(5); return history.ToList(); } private void ExecuteCMD(string cmd, bool runAsAdministrator = false) { var arguments = _settings.LeaveCmdOpen ? $"/k {cmd}" : $"/c {cmd} & pause"; var info = new ProcessStartInfo { UseShellExecute = true, FileName = "cmd.exe", WorkingDirectory = Environment.GetFolderPath(Environment.SpecialFolder.UserProfile), Arguments = arguments, Verb = runAsAdministrator ? "runas" : "" }; try { Process.Start(info); _settings.AddCmdHistory(cmd); } catch (FileNotFoundException e) { MessageBox.Show($"Command not found: {e.Message}"); } } public void Init(PluginInitContext context) { this.context = context; context.API.GlobalKeyboardEvent += API_GlobalKeyboardEvent; } bool API_GlobalKeyboardEvent(int keyevent, int vkcode, SpecialKeyState state) { if (_settings.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($"{context.CurrentPluginMetadata.ActionKeywords[0]}{Plugin.Query.TermSeperater}"); } public Control CreateSettingPanel() { return new CMDSetting(_settings); } public string GetTranslatedPluginTitle() { return context.API.GetTranslation("wox_plugin_cmd_plugin_name"); } public string GetTranslatedPluginDescription() { return context.API.GetTranslation("wox_plugin_cmd_plugin_description"); } public bool IsInstantQuery(string query) => false; public List LoadContextMenus(Result selectedResult) { return new List { new Result { Title = context.API.GetTranslation("wox_plugin_cmd_run_as_administrator"), Action = c => { context.API.HideApp(); ExecuteCMD(selectedResult.Title, true); return true; }, IcoPath = "Images/cmd.png" } }; } } }