From 9481d04a19138599cdbec6d515514bef83cb3b78 Mon Sep 17 00:00:00 2001
From: qianlifeng <qianlf2008@163.com>
Date: Mon, 3 Nov 2014 17:30:48 +0800
Subject: [PATCH] Add custom context menu setting for findfile plugin and
 improve the search speed.

---
 .../FindFileContextMenuStorage.cs             |  46 ++++++++++++++++++
 Plugins/Wox.Plugin.FindFile/Images/list.png   | Bin 0 -> 1775 bytes
 .../MFTSearch/MFTSearcher.cs                  |   2 +-
 .../MFTSearch/MFTSearcherCache.cs             |  21 +++++---
 Plugins/Wox.Plugin.FindFile/Main.cs           |  42 ++++++++++------
 .../Wox.Plugin.FindFile.csproj                |  37 ++++++++++++++
 Plugins/Wox.Plugin.FindFile/packages.config   |   6 +++
 Plugins/Wox.Plugin.FindFile/setting.xaml      |  20 ++++++++
 Plugins/Wox.Plugin.FindFile/setting.xaml.cs   |  31 ++++++++++++
 Wox.Test/MFTSearcherTest.cs                   |   3 +-
 10 files changed, 182 insertions(+), 26 deletions(-)
 create mode 100644 Plugins/Wox.Plugin.FindFile/FindFileContextMenuStorage.cs
 create mode 100644 Plugins/Wox.Plugin.FindFile/Images/list.png
 create mode 100644 Plugins/Wox.Plugin.FindFile/packages.config
 create mode 100644 Plugins/Wox.Plugin.FindFile/setting.xaml
 create mode 100644 Plugins/Wox.Plugin.FindFile/setting.xaml.cs

diff --git a/Plugins/Wox.Plugin.FindFile/FindFileContextMenuStorage.cs b/Plugins/Wox.Plugin.FindFile/FindFileContextMenuStorage.cs
new file mode 100644
index 0000000000..7bb3b61a7f
--- /dev/null
+++ b/Plugins/Wox.Plugin.FindFile/FindFileContextMenuStorage.cs
@@ -0,0 +1,46 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using Newtonsoft.Json;
+using Wox.Infrastructure.Storage;
+
+namespace Wox.Plugin.FindFile
+{
+    public class FindFileContextMenuStorage : BaseStorage<FindFileContextMenuStorage>
+    {
+        [JsonProperty]
+        public List<ContextMenu> ContextMenus = new List<ContextMenu>();
+
+        protected override string ConfigName
+        {
+            get { return "FindFileContextMenu"; }
+        }
+
+        protected override FindFileContextMenuStorage LoadDefaultConfig()
+        {
+            ContextMenus = new List<ContextMenu>()
+            {
+                new ContextMenu()
+                {
+                    Name = "Open Containing Folder",
+                    Command = "explorer.exe",
+                    Argument = " /select \"{path}\""
+                }
+            };
+            return this;
+        }
+    }
+
+    public class ContextMenu
+    {
+        [JsonProperty]
+        public string Name { get; set; }
+
+        [JsonProperty]
+        public string Command { get; set; }
+
+        [JsonProperty]
+        public string Argument { get; set; }
+    }
+}
diff --git a/Plugins/Wox.Plugin.FindFile/Images/list.png b/Plugins/Wox.Plugin.FindFile/Images/list.png
new file mode 100644
index 0000000000000000000000000000000000000000..ab96e07812b12479ad11a2bd6ed66bf430df671e
GIT binary patch
literal 1775
zcmV<L1`zp)P)<h;3K|Lk000e1NJLTq002M$002M;1^@s6s%dfF0000PbVXQnQ*UN;
zcVTj606}DLVr3vnZDD6+Qe|Oed2z{QJOBU&kx4{BRCwC#TU$&cWfVTWO(_%#lm!-K
zg}A~ZO9*kZ$>M`18+|b`#y1m9G-elHjPcR9>x1~BPbQk^KAC9rMUzcXAJEn4df7y|
zCM%a^xri*caw)W(nNH{8`Tuh1l$K80X=i5fBtOGUXWE|gpZ`1GcV=o+6a|j6+2NSm
zu@8VV-#7Sitk+NnE_PkZ>w2433_wTganqQ1tc3A14x7tnwpmp`W-pxS8_hcaH%2de
zaq~WoXOf+FIm>`?`b?bM3Xgw=(ZAPw+TR(>IsoW=3vPZ*cANr-%?Td&339m=mU&E3
zq*r>{`##SK1Tiq5;%FoRQV)pFyh2a1!oW3Z|L~8Aw;$9TfbWMdb>g*;r+Po9qYN{E
zv54>c!-tdiS{7A6X-gR`Ont`bh4}Pa&6Ez!0l;W6sg5(Qid~eoUH~u)m}~-Y=$Ok<
z3U<32tje6zfqj|)B$W!A9m}2%tSlSSeRoY6;HIf>1LZ!9K*j(#9VOtjd%&vGFDoJl
zVi-h;<8umt%UMqM9S%1%*Ik54Zwv6kHq5OI0Luqd*Oiu>K%8I<aa4%y;0W#SOLBDA
zasGfI=Mn%ro<dN+weeY~t?Hu3yq-oV_cXwtQy<U|LZSg@TV4cz#Ti3qdv<vM=5RzY
z7?LdHE{7ADpRCB}paW`rkEn5hsf3EsCVGy)qTO)kiM7-}2{D{;QHsK02Ad-WAdki%
z*g;Za7%i8YWQ#n7211a`V5C}Q3)Y!4K$hv4T^dlG6A1nd+mQu&&RXy`gxGtA2Vidb
z2Jq2f%Gu8&1L9<QaqSj_xmBnrYl5f{gw@Tl1Hh34_-pbbs4Q;*592R}-Um_SRO@^z
zG=mfJk3JnCIUxWP19tZ}cV_5F{jEgU*q+9JrVFcWB@U}lkS@)FwZs&3`XB%jt`pf5
zM217hV`8N6JwZiISOU2I6@&X8vK#|8x<R*kTSg2Z00?&DSfGG2&I?-wqv4t`KsJa5
zleHdAV0T3efI0)DqKhEHJUM=Y0$!ml4F<?+3@Bm%2)akOfMzON_zX~rmjMcv(l9z-
z2(Zf?19mgO;b5Tc)YIViJq%oQ6DH>Sai-Q(*Ohr{Q9iqKM0ye5cME)FB{^Ns;&&p~
z7oZF(bab^o4~@0w=&?#~6OL2x)8M<v0;80JpYP~}y6O%?%jNLYXK;Jw>#VNPpamp!
z&;ciFx>V<Q+*MFr-b?|j&fjUc^TfJaU!dKqytROwzyu}4014Noh@iyST}d^XrOU-k
zYEVh~{&{Brp~a*UjL-F}&RJR?MO`pYDeAqz2yBHH3=hEM{CB|en>wh$1PqWRiSEz-
z`wQ%Zm!Jw2ABVJbadjx6{UpH8gYQ9&?=*NzYc&BL>DuU7T>@KED8^>>uL4p66I|Q8
zLq|%JM3l&rbR1<QSYdXJmRB=a=6(PvgF4*_B7hnq&SM6c#WFHM86?$`$cN#D*j3s*
zF_UT10{Pw(AQ=>jtkPzs*(@Ul*dTv3Na(E~@?b{O)Dxr~B3o`<5spVmD}=$Umaz=(
z>4A95Im>27j0Kk40yA62p5Pu1xLqaC+4>w(-ZqG!Rv(-C3bw;5s`Gr_dU`K0UhZNj
zme!Fn7w18$FdLvGNOug_qsHgYz68yU7wCSquNkT<n&F$9Z-OWiF=T@$&b@(RsmsvW
z{C)Hq432-HuNWeO-1}=f`Dl{#^+QeNY5MvDwcUn0Ppqfw5|l7LeQ%5q6EU4dtly5L
zJHLX!(h_G+5=lmiA(D-{-WsEl7-{avgj#1WEDu5;c$dCT{@)1B8y<kMJJ%5lHuTQ`
zf&r8BKR__F2sKp?K?FZ!Zt2$k>oFua-`>26%c2^5<qd{7zp_4#I%QfP5IhtB$_B}Y
zOwf_awj)cZ$nJvGOk)6!?hJF$9V9yuu%c-SfO14<n1ExLT?cDEJef@fq;QZBrX3VH
ziXwwmP1AFN^6o>@huBwS1Bs9TRyJ)*AjR#mSd61<-|D7$2|$uU+=;Bg{qT#kbWe+~
zsF#4cg+)Y8F>hb#MHRBa0K#&Jw5pD0zySu3C&O4iWW~wu*Gz@qx;(^TdS5wsR#4i1
zW-y-ma$34+(s3Pq(jEX}S9*s(#tr#RTzGG-lkSe$nwlEyOB^q!6;PRgtnZWVdQ@3Y
zRhBnA77Q&}spgO4`0eVeBSS|j$b69;17xU|=wN;l`u`tKA9FkQ0r*dV0RR!F$#kTn
RHR=EW002ovPDHLkV1l4JGhzS$

literal 0
HcmV?d00001

diff --git a/Plugins/Wox.Plugin.FindFile/MFTSearch/MFTSearcher.cs b/Plugins/Wox.Plugin.FindFile/MFTSearch/MFTSearcher.cs
index 870aef9cd8..6ce6f51e69 100644
--- a/Plugins/Wox.Plugin.FindFile/MFTSearch/MFTSearcher.cs
+++ b/Plugins/Wox.Plugin.FindFile/MFTSearch/MFTSearcher.cs
@@ -39,7 +39,7 @@ namespace Wox.Plugin.FindFile.MFTSearch
         {
             if (string.IsNullOrEmpty(item)) return new List<MFTSearchRecord>();
 
-            List<USNRecord> found = cache.FindByName(item);
+            List<USNRecord> found = cache.FindByName(item,100);
             found.ForEach(x => FillPath(x.VolumeName, x, cache));
             return found.ConvertAll(o => new MFTSearchRecord(o));
         }
diff --git a/Plugins/Wox.Plugin.FindFile/MFTSearch/MFTSearcherCache.cs b/Plugins/Wox.Plugin.FindFile/MFTSearch/MFTSearcherCache.cs
index 849279ff48..1e1f4e65fe 100644
--- a/Plugins/Wox.Plugin.FindFile/MFTSearch/MFTSearcherCache.cs
+++ b/Plugins/Wox.Plugin.FindFile/MFTSearch/MFTSearcherCache.cs
@@ -71,17 +71,22 @@ namespace Wox.Plugin.FindFile.MFTSearch
             }
         }
 
-        public List<USNRecord> FindByName(string filename)
+        public List<USNRecord> FindByName(string filename, long maxResult = -1)
         {
-            filename = filename.ToLower();
-            var fileQuery = from filesInVolumeDic in VolumeRecords.Values
-                            from eachFilePair in filesInVolumeDic
-                            where eachFilePair.Value.Name.ToLower().Contains(filename)
-                            select eachFilePair.Value;
-
             List<USNRecord> result = new List<USNRecord>();
 
-            result.AddRange(fileQuery);
+            foreach (Dictionary<ulong, USNRecord> dictionary in VolumeRecords.Values)
+            {
+                foreach (var usnRecord in dictionary)
+                {
+                    if (usnRecord.Value.Name.IndexOf(filename, StringComparison.OrdinalIgnoreCase) >= 0)
+                    {
+                        result.Add(usnRecord.Value);
+                        if (maxResult > 0 && result.Count() >= maxResult) break;
+                    }
+                    if (maxResult > 0 && result.Count() >= maxResult) break;
+                }
+            }
 
             return result;
         }
diff --git a/Plugins/Wox.Plugin.FindFile/Main.cs b/Plugins/Wox.Plugin.FindFile/Main.cs
index 5311191bfc..917712b033 100644
--- a/Plugins/Wox.Plugin.FindFile/Main.cs
+++ b/Plugins/Wox.Plugin.FindFile/Main.cs
@@ -5,12 +5,13 @@ using System.IO;
 using System.Linq;
 using System.Runtime.Serialization.Formatters.Binary;
 using System.Text;
+using System.Windows.Controls;
 using Wox.Infrastructure;
 using Wox.Plugin.FindFile.MFTSearch;
 
 namespace Wox.Plugin.FindFile
 {
-    public class Main : IPlugin
+    public class Main : IPlugin, ISettingProvider
     {
         private PluginInitContext context;
         private bool initial = false;
@@ -78,27 +79,36 @@ namespace Wox.Plugin.FindFile
 
             if (!record.IsFolder)
             {
-                contextMenus.Add(new Result()
+                foreach (ContextMenu contextMenu in FindFileContextMenuStorage.Instance.ContextMenus)
                 {
-                    Title = "Open Containing Folder",
-                    Action = _ =>
+                    contextMenus.Add(new Result()
                     {
-                        try
+                        Title = contextMenu.Name,
+                        Action = _ =>
                         {
-                            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"
-                });
+                            string argument = contextMenu.Argument.Replace("{path}", record.FullPath);
+                            try
+                            {
+                                Process.Start(contextMenu.Command,argument);
+                            }
+                            catch
+                            {
+                                context.API.ShowMsg("Can't start " + record.FullPath, string.Empty, string.Empty);
+                                return false;
+                            }
+                            return true;
+                        },
+                        IcoPath = "Images/list.png"
+                    });
+                }
             }
 
             return contextMenus;
         }
+
+        public Control CreateSettingPanel()
+        {
+            return new Setting();
+        }
     }
 }
diff --git a/Plugins/Wox.Plugin.FindFile/Wox.Plugin.FindFile.csproj b/Plugins/Wox.Plugin.FindFile/Wox.Plugin.FindFile.csproj
index 81ecab8d62..0324d2b966 100644
--- a/Plugins/Wox.Plugin.FindFile/Wox.Plugin.FindFile.csproj
+++ b/Plugins/Wox.Plugin.FindFile/Wox.Plugin.FindFile.csproj
@@ -12,6 +12,8 @@
     <TargetFrameworkVersion>v3.5</TargetFrameworkVersion>
     <FileAlignment>512</FileAlignment>
     <TargetFrameworkProfile />
+    <SolutionDir Condition="$(SolutionDir) == '' Or $(SolutionDir) == '*Undefined*'">..\..\</SolutionDir>
+    <RestorePackages>true</RestorePackages>
   </PropertyGroup>
   <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug|AnyCPU' ">
     <DebugSymbols>true</DebugSymbols>
@@ -33,15 +35,36 @@
     <AllowUnsafeBlocks>true</AllowUnsafeBlocks>
   </PropertyGroup>
   <ItemGroup>
+    <Reference Include="log4net, Version=1.2.13.0, Culture=neutral, PublicKeyToken=669e0ddf0bb1aa2a, processorArchitecture=MSIL">
+      <SpecificVersion>False</SpecificVersion>
+      <HintPath>..\..\packages\log4net.2.0.3\lib\net35-full\log4net.dll</HintPath>
+    </Reference>
+    <Reference Include="Newtonsoft.Json, Version=6.0.0.0, Culture=neutral, PublicKeyToken=30ad4fe6b2a6aeed, processorArchitecture=MSIL">
+      <SpecificVersion>False</SpecificVersion>
+      <HintPath>..\..\packages\Newtonsoft.Json.6.0.5\lib\net35\Newtonsoft.Json.dll</HintPath>
+    </Reference>
+    <Reference Include="PresentationCore" />
+    <Reference Include="PresentationFramework" />
     <Reference Include="System" />
     <Reference Include="System.Core" />
+    <Reference Include="System.Windows.Controls.Input.Toolkit">
+      <HintPath>..\..\packages\WPFToolkit.3.5.50211.1\lib\System.Windows.Controls.Input.Toolkit.dll</HintPath>
+    </Reference>
+    <Reference Include="System.Windows.Controls.Layout.Toolkit">
+      <HintPath>..\..\packages\WPFToolkit.3.5.50211.1\lib\System.Windows.Controls.Layout.Toolkit.dll</HintPath>
+    </Reference>
     <Reference Include="System.Xml.Linq" />
     <Reference Include="System.Data.DataSetExtensions" />
     <Reference Include="Microsoft.CSharp" />
     <Reference Include="System.Data" />
     <Reference Include="System.Xml" />
+    <Reference Include="WindowsBase" />
+    <Reference Include="WPFToolkit, Version=3.5.40128.1, Culture=neutral, PublicKeyToken=31bf3856ad364e35, processorArchitecture=MSIL">
+      <HintPath>..\..\packages\WPFToolkit.3.5.50211.1\lib\WPFToolkit.dll</HintPath>
+    </Reference>
   </ItemGroup>
   <ItemGroup>
+    <Compile Include="FindFileContextMenuStorage.cs" />
     <Compile Include="Main.cs" />
     <Compile Include="MFTSearch\MFTSearcher.cs" />
     <Compile Include="MFTSearch\MFTSearcherCache.cs" />
@@ -51,6 +74,9 @@
     <Compile Include="MFTSearch\USNRecord.cs" />
     <Compile Include="MFTSearch\VolumeMonitor.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />
+    <Compile Include="Setting.xaml.cs">
+      <DependentUpon>Setting.xaml</DependentUpon>
+    </Compile>
   </ItemGroup>
   <ItemGroup>
     <ProjectReference Include="..\..\Wox.Infrastructure\Wox.Infrastructure.csproj">
@@ -72,6 +98,9 @@
     <Content Include="Images\file.png">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </Content>
+    <Content Include="Images\list.png">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </Content>
     <Content Include="Images\open.png">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </Content>
@@ -80,11 +109,19 @@
     </Content>
   </ItemGroup>
   <ItemGroup>
+    <None Include="packages.config" />
     <None Include="plugin.json">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
     </None>
   </ItemGroup>
+  <ItemGroup>
+    <Page Include="Setting.xaml">
+      <SubType>Designer</SubType>
+      <Generator>MSBuild:Compile</Generator>
+    </Page>
+  </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
+  <Import Project="$(SolutionDir)\.nuget\NuGet.targets" Condition="Exists('$(SolutionDir)\.nuget\NuGet.targets')" />
   <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 
        Other similar extension points exist, see Microsoft.Common.targets.
   <Target Name="BeforeBuild">
diff --git a/Plugins/Wox.Plugin.FindFile/packages.config b/Plugins/Wox.Plugin.FindFile/packages.config
new file mode 100644
index 0000000000..30b7b9cf8c
--- /dev/null
+++ b/Plugins/Wox.Plugin.FindFile/packages.config
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="utf-8"?>
+<packages>
+  <package id="log4net" version="2.0.3" targetFramework="net35" />
+  <package id="Newtonsoft.Json" version="6.0.5" targetFramework="net35" />
+  <package id="WPFToolkit" version="3.5.50211.1" targetFramework="net35" />
+</packages>
\ No newline at end of file
diff --git a/Plugins/Wox.Plugin.FindFile/setting.xaml b/Plugins/Wox.Plugin.FindFile/setting.xaml
new file mode 100644
index 0000000000..a0218954d8
--- /dev/null
+++ b/Plugins/Wox.Plugin.FindFile/setting.xaml
@@ -0,0 +1,20 @@
+<UserControl x:Class="Wox.Plugin.FindFile.Setting"
+             xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
+             xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+             xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006" 
+             xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
+             xmlns:controls="http://schemas.microsoft.com/wpf/2008/toolkit"
+             mc:Ignorable="d" 
+             d:DesignHeight="300" d:DesignWidth="600">
+    <TabControl Height="auto" Margin="10" VerticalAlignment="Top">
+        <TabItem Header="Context Menu">
+            <controls:DataGrid x:Name="menuGrid" AutoGenerateColumns="False" CanUserAddRows="True" CanUserDeleteRows="True" Height="420">
+                <controls:DataGrid.Columns>
+                    <controls:DataGridTextColumn Header="Name" Width="150"  Binding="{Binding Name}" />
+                    <controls:DataGridTextColumn Header="Command" Width="250"  Binding="{Binding Command}" />
+                    <controls:DataGridTextColumn Header="Arguments" Width="*"  Binding="{Binding Argument}" />
+                </controls:DataGrid.Columns>
+            </controls:DataGrid>
+        </TabItem>
+    </TabControl>
+</UserControl>
diff --git a/Plugins/Wox.Plugin.FindFile/setting.xaml.cs b/Plugins/Wox.Plugin.FindFile/setting.xaml.cs
new file mode 100644
index 0000000000..1df74a4291
--- /dev/null
+++ b/Plugins/Wox.Plugin.FindFile/setting.xaml.cs
@@ -0,0 +1,31 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Windows;
+using System.Windows.Controls;
+using System.Windows.Data;
+using System.Windows.Documents;
+using System.Windows.Input;
+using System.Windows.Media;
+using System.Windows.Media.Imaging;
+using System.Windows.Navigation;
+using System.Windows.Shapes;
+
+namespace Wox.Plugin.FindFile
+{
+    public partial class Setting : UserControl
+    {
+        public Setting()
+        {
+            InitializeComponent();
+            menuGrid.ItemsSource = FindFileContextMenuStorage.Instance.ContextMenus;
+            menuGrid.CurrentCellChanged += menuGrid_CurrentCellChanged;
+        }
+
+        void menuGrid_CurrentCellChanged(object sender, EventArgs e)
+        {
+            FindFileContextMenuStorage.Instance.Save();
+        }
+    }
+}
diff --git a/Wox.Test/MFTSearcherTest.cs b/Wox.Test/MFTSearcherTest.cs
index dac4e23dee..3e4992926b 100644
--- a/Wox.Test/MFTSearcherTest.cs
+++ b/Wox.Test/MFTSearcherTest.cs
@@ -39,9 +39,10 @@ namespace Wox.Test
             Console.WriteLine(string.Format("Index: {0}M", (newWorkingSet - oldWorkingSet)/(1024*1024)));
 
             oldWorkingSet = System.Diagnostics.Process.GetCurrentProcess().WorkingSet64;
-            MFTSearcher.Search("q");
+            List<MFTSearchRecord> mftSearchRecords = MFTSearcher.Search("q");
             newWorkingSet = System.Diagnostics.Process.GetCurrentProcess().WorkingSet64;
             Console.WriteLine(string.Format("Search: {0}M", (newWorkingSet - oldWorkingSet) / (1024 * 1024)));
+            Console.WriteLine(string.Format("Search results: {0}",mftSearchRecords.Count));
         }
     }
 }