2014-10-22 22:49:34 +08:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Diagnostics;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Runtime.Serialization.Formatters.Binary;
|
|
|
|
|
using System.Text;
|
|
|
|
|
using Wox.Infrastructure;
|
2014-10-27 18:22:25 +08:00
|
|
|
|
using Wox.Plugin.FindFile.MFTSearch;
|
2014-10-22 22:49:34 +08:00
|
|
|
|
|
|
|
|
|
namespace Wox.Plugin.FindFile
|
|
|
|
|
{
|
|
|
|
|
public class Main : IPlugin
|
|
|
|
|
{
|
|
|
|
|
private PluginInitContext context;
|
|
|
|
|
private bool initial = false;
|
|
|
|
|
|
|
|
|
|
public List<Result> Query(Query query)
|
|
|
|
|
{
|
|
|
|
|
if (!initial)
|
|
|
|
|
{
|
|
|
|
|
return new List<Result>()
|
|
|
|
|
{
|
|
|
|
|
new Result("Wox is indexing your files, please try later.","Images/warning.png")
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
string q = query.GetAllRemainingParameter();
|
|
|
|
|
return MFTSearcher.Search(q).Take(100).Select(t => ConvertMFTSearch(t, q)).ToList();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void Init(PluginInitContext context)
|
|
|
|
|
{
|
|
|
|
|
this.context = context;
|
|
|
|
|
var searchtimestart = DateTime.Now;
|
|
|
|
|
MFTSearcher.IndexAllVolumes();
|
|
|
|
|
initial = true;
|
|
|
|
|
var searchtimeend = DateTime.Now;
|
2014-10-24 09:43:43 +08:00
|
|
|
|
Debug.WriteLine(string.Format("{0} file, indexed, {1}ms has spent.", MFTSearcher.IndexedFileCount, searchtimeend.Subtract(searchtimestart).TotalMilliseconds));
|
2014-10-22 22:49:34 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Result ConvertMFTSearch(MFTSearchRecord record, string query)
|
|
|
|
|
{
|
|
|
|
|
string icoPath = "Images/file.png";
|
|
|
|
|
if (record.IsFolder)
|
|
|
|
|
{
|
|
|
|
|
icoPath = "Images/folder.png";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
string name = Path.GetFileName(record.FullPath);
|
|
|
|
|
FuzzyMatcher matcher = FuzzyMatcher.Create(query);
|
|
|
|
|
return new Result()
|
|
|
|
|
{
|
|
|
|
|
Title = name,
|
|
|
|
|
Score = matcher.Evaluate(name).Score,
|
|
|
|
|
SubTitle = record.FullPath,
|
|
|
|
|
IcoPath = icoPath,
|
|
|
|
|
Action = _ =>
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
Process.Start(record.FullPath);
|
|
|
|
|
}
|
|
|
|
|
catch
|
|
|
|
|
{
|
|
|
|
|
context.API.ShowMsg("Can't open " + record.FullPath, string.Empty, string.Empty);
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
return true;
|
2014-10-23 18:39:11 +08:00
|
|
|
|
},
|
|
|
|
|
ContextMenu = GetContextMenu(record)
|
2014-10-22 22:49:34 +08:00
|
|
|
|
};
|
|
|
|
|
}
|
2014-10-23 18:39:11 +08:00
|
|
|
|
|
|
|
|
|
private List<Result> GetContextMenu(MFTSearchRecord record)
|
|
|
|
|
{
|
|
|
|
|
List<Result> contextMenus = new List<Result>();
|
|
|
|
|
|
|
|
|
|
if (!record.IsFolder)
|
|
|
|
|
{
|
|
|
|
|
contextMenus.Add(new Result()
|
|
|
|
|
{
|
|
|
|
|
Title = "Open Containing Folder",
|
|
|
|
|
Action = _ =>
|
|
|
|
|
{
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
Process.Start("explorer.exe", string.Format("/select, \"{0}\"", record.FullPath));
|
|
|
|
|
}
|
|
|
|
|
catch
|
|
|
|
|
{
|
|
|
|
|
context.API.ShowMsg("Can't open " + record.FullPath, string.Empty, string.Empty);
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
return true;
|
|
|
|
|
},
|
|
|
|
|
IcoPath = "Images/folder.png"
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return contextMenus;
|
|
|
|
|
}
|
2014-10-22 22:49:34 +08:00
|
|
|
|
}
|
|
|
|
|
}
|