using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Runtime.Serialization.Formatters.Binary; using System.Windows.Forms; namespace Wox.Plugin.System.CMD { public class CMD : BaseSystemPlugin { private PluginInitContext context; protected override List QueryInternal(Query query) { List results = new List(); if (query.RawQuery == ">") { IEnumerable 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; } }).Take(5); results.AddRange(history); } if (query.RawQuery.StartsWith(">") && query.RawQuery.Length > 1) { string cmd = query.RawQuery.Substring(1); Result result = new Result { Title = cmd, Score = 5000, SubTitle = "execute command through command shell", IcoPath = "Images/cmd.png", Action = (c) => { ExecuteCmd(cmd); return true; } }; try { if (File.Exists(cmd) || Directory.Exists(cmd)) { result.IcoPath = cmd; } } catch (Exception) { } results.Add(result); IEnumerable 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; } }; try { if (File.Exists(m.Key) || Directory.Exists(m.Key)) { ret.IcoPath = m.Key; } } catch (Exception) { } return ret; }).Where(o => o != null).Take(4); 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))) { basedir = Path.GetDirectoryName(excmd); var dirn = Path.GetDirectoryName(cmd); dir = (dirn.EndsWith("/") || dirn.EndsWith(@"\")) ? dirn : cmd.Substring(0, dirn.Length + 1); } if (basedir != null) { List 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))).ToList(); autocomplete.Sort(); results.AddRange(autocomplete.ConvertAll(m => new Result() { Title = m, SubTitle = "", IcoPath = m, Action = (c) => { ExecuteCmd(m); return true; } })); } } catch (Exception) { } } return results; } private void ExecuteCmd(string cmd) { if (context.ShellRun(cmd)) CMDStorage.Instance.AddCmdHistory(cmd); } protected override void InitInternal(PluginInitContext context) { this.context = context; } } }