PowerToys/Plugins/Wox.Plugin.FindFile/Main.cs

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 System.Windows.Controls;
using Wox.Infrastructure;
using Wox.Plugin.FindFile.MFTSearch;

namespace Wox.Plugin.FindFile
{
    public class Main : IPlugin, ISettingProvider
    {
        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;
            Debug.WriteLine(string.Format("{0} file, indexed, {1}ms has spent.", MFTSearcher.IndexedFileCount, searchtimeend.Subtract(searchtimestart).TotalMilliseconds));
        }

        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;
                },
                ContextMenu = GetContextMenu(record)
            };
        }

        private List<Result> GetContextMenu(MFTSearchRecord record)
        {
            List<Result> contextMenus = new List<Result>();

            if (!record.IsFolder)
            {
                foreach (ContextMenu contextMenu in FindFileContextMenuStorage.Instance.ContextMenus)
                {
                    contextMenus.Add(new Result()
                    {
                        Title = contextMenu.Name,
                        Action = _ =>
                        {
                            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();
        }
    }
}