2014-01-03 18:16:05 +08:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
2014-01-03 23:52:36 +08:00
|
|
|
|
using System.Diagnostics;
|
2014-01-26 11:01:13 +08:00
|
|
|
|
using System.IO;
|
2014-01-03 18:16:05 +08:00
|
|
|
|
using System.Linq;
|
2014-01-26 11:01:13 +08:00
|
|
|
|
using System.Runtime.Serialization.Formatters.Binary;
|
2014-01-03 18:16:05 +08:00
|
|
|
|
using System.Text;
|
2014-01-03 23:52:36 +08:00
|
|
|
|
using System.Windows.Forms;
|
2014-01-03 18:16:05 +08:00
|
|
|
|
|
2014-01-29 18:33:24 +08:00
|
|
|
|
namespace Wox.Plugin.System
|
2014-01-03 18:16:05 +08:00
|
|
|
|
{
|
2014-01-15 22:45:02 +08:00
|
|
|
|
public class CMD : BaseSystemPlugin
|
2014-01-03 18:16:05 +08:00
|
|
|
|
{
|
2014-01-26 11:01:13 +08:00
|
|
|
|
private Dictionary<string, int> cmdHistory = new Dictionary<string, int>();
|
2014-03-22 16:50:47 +08:00
|
|
|
|
private string filePath = Path.GetDirectoryName(Application.ExecutablePath) + "\\CMDHistory.dat";
|
2014-01-28 21:25:13 +08:00
|
|
|
|
private PluginInitContext context;
|
2014-01-26 11:01:13 +08:00
|
|
|
|
|
2014-01-15 22:45:02 +08:00
|
|
|
|
protected override List<Result> QueryInternal(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-01-26 18:06:38 +08:00
|
|
|
|
if (query.RawQuery == ">")
|
|
|
|
|
{
|
|
|
|
|
IEnumerable<Result> history = 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",
|
2014-02-09 20:55:18 +08:00
|
|
|
|
Action = (c) =>
|
2014-01-26 18:06:38 +08:00
|
|
|
|
{
|
|
|
|
|
ExecuteCmd(m.Key);
|
2014-02-28 23:21:01 +08:00
|
|
|
|
return true;
|
2014-01-26 18:06:38 +08:00
|
|
|
|
}
|
|
|
|
|
}).Take(5);
|
|
|
|
|
|
|
|
|
|
results.AddRange(history);
|
|
|
|
|
}
|
|
|
|
|
|
2014-01-03 23:52:36 +08:00
|
|
|
|
if (query.RawQuery.StartsWith(">") && query.RawQuery.Length > 1)
|
|
|
|
|
{
|
|
|
|
|
string cmd = query.RawQuery.Substring(1);
|
|
|
|
|
Result result = new Result
|
|
|
|
|
{
|
|
|
|
|
Title = cmd,
|
2014-01-26 11:01:13 +08:00
|
|
|
|
Score = 5000,
|
|
|
|
|
SubTitle = "execute command through command shell",
|
2014-01-03 23:52:36 +08:00
|
|
|
|
IcoPath = "Images/cmd.png",
|
2014-02-09 20:55:18 +08:00
|
|
|
|
Action = (c) =>
|
2014-01-03 23:52:36 +08:00
|
|
|
|
{
|
2014-01-26 11:01:13 +08:00
|
|
|
|
ExecuteCmd(cmd);
|
2014-02-28 23:21:01 +08:00
|
|
|
|
return true;
|
2014-01-03 23:52:36 +08:00
|
|
|
|
}
|
|
|
|
|
};
|
2014-03-20 04:12:50 +08:00
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
if (File.Exists(cmd) || Directory.Exists(cmd))
|
|
|
|
|
{
|
|
|
|
|
result.IcoPath = cmd;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
catch (Exception) { }
|
|
|
|
|
|
2014-01-03 23:52:36 +08:00
|
|
|
|
results.Add(result);
|
2014-01-26 11:01:13 +08:00
|
|
|
|
|
|
|
|
|
IEnumerable<Result> history = cmdHistory.Where(o => o.Key.Contains(cmd))
|
|
|
|
|
.OrderByDescending(o => o.Value)
|
2014-03-20 04:12:50 +08:00
|
|
|
|
.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
|
2014-01-26 11:01:13 +08:00
|
|
|
|
{
|
2014-03-20 04:12:50 +08:00
|
|
|
|
if (File.Exists(m.Key) || Directory.Exists(m.Key))
|
|
|
|
|
{
|
|
|
|
|
ret.IcoPath = m.Key;
|
|
|
|
|
}
|
2014-01-26 11:01:13 +08:00
|
|
|
|
}
|
2014-03-20 04:12:50 +08:00
|
|
|
|
catch (Exception) { }
|
|
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
|
}).Where(o => o != null).Take(4);
|
2014-01-26 11:01:13 +08:00
|
|
|
|
|
|
|
|
|
results.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)
|
|
|
|
|
{
|
|
|
|
|
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))).ToList();
|
|
|
|
|
autocomplete.Sort();
|
|
|
|
|
results.AddRange(autocomplete.ConvertAll(m => new Result() {
|
|
|
|
|
Title = m,
|
|
|
|
|
SubTitle = "",
|
|
|
|
|
IcoPath = m,
|
|
|
|
|
Action = (c) =>
|
|
|
|
|
{
|
|
|
|
|
ExecuteCmd(m);
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
}));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
catch (Exception) { }
|
2014-01-03 23:52:36 +08:00
|
|
|
|
}
|
|
|
|
|
return results;
|
2014-01-03 18:16:05 +08:00
|
|
|
|
}
|
|
|
|
|
|
2014-01-28 21:25:13 +08:00
|
|
|
|
|
|
|
|
|
private void ExecuteCmd(string cmd)
|
2014-01-26 11:01:13 +08:00
|
|
|
|
{
|
2014-03-21 03:53:18 +08:00
|
|
|
|
if (context.ShellRun(cmd))
|
2014-03-02 11:38:13 +08:00
|
|
|
|
AddCmdHistory(cmd);
|
2014-01-26 11:01:13 +08:00
|
|
|
|
}
|
|
|
|
|
|
2014-01-15 22:45:02 +08:00
|
|
|
|
protected override void InitInternal(PluginInitContext context)
|
2014-01-03 18:16:05 +08:00
|
|
|
|
{
|
2014-01-28 21:25:13 +08:00
|
|
|
|
this.context = context;
|
2014-01-26 11:01:13 +08:00
|
|
|
|
LoadCmdHistory();
|
2014-01-03 23:52:36 +08:00
|
|
|
|
}
|
|
|
|
|
|
2014-01-26 11:01:13 +08:00
|
|
|
|
//todo:we need provide a common data persist interface for user?
|
|
|
|
|
private void AddCmdHistory(string cmdName)
|
|
|
|
|
{
|
|
|
|
|
if (cmdHistory.ContainsKey(cmdName))
|
|
|
|
|
{
|
|
|
|
|
cmdHistory[cmdName] += 1;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
cmdHistory.Add(cmdName, 1);
|
|
|
|
|
}
|
|
|
|
|
PersistCmdHistory();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void LoadCmdHistory()
|
|
|
|
|
{
|
|
|
|
|
if (File.Exists(filePath))
|
|
|
|
|
{
|
|
|
|
|
FileStream fileStream = new FileStream(filePath, FileMode.Open, FileAccess.Read, FileShare.Read);
|
|
|
|
|
BinaryFormatter b = new BinaryFormatter();
|
|
|
|
|
cmdHistory = (Dictionary<string, int>)b.Deserialize(fileStream);
|
|
|
|
|
fileStream.Close();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (cmdHistory.Count > 1000)
|
|
|
|
|
{
|
|
|
|
|
List<string> onlyOnceKeys = (from c in cmdHistory where c.Value == 1 select c.Key).ToList();
|
|
|
|
|
foreach (string onlyOnceKey in onlyOnceKeys)
|
|
|
|
|
{
|
|
|
|
|
cmdHistory.Remove(onlyOnceKey);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private void PersistCmdHistory()
|
|
|
|
|
{
|
|
|
|
|
FileStream fileStream = new FileStream(filePath, FileMode.Create);
|
|
|
|
|
BinaryFormatter b = new BinaryFormatter();
|
|
|
|
|
b.Serialize(fileStream, cmdHistory);
|
|
|
|
|
fileStream.Close();
|
|
|
|
|
}
|
2014-01-03 18:16:05 +08:00
|
|
|
|
}
|
|
|
|
|
}
|