PowerToys/Wox.Plugin.System/DirectoryIndicator.cs

using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Text;
using System.Windows.Forms;
using Wox.Infrastructure;

namespace Wox.Plugin.System
{
    public class DirectoryIndicator : BaseSystemPlugin
    {
        private PluginInitContext context;
        private static  List<string> driverNames = null;
        private static Dictionary<string, DirectoryInfo[]> parentDirectories = new Dictionary<string, DirectoryInfo[]>();

        protected override List<Result> QueryInternal(Query query)
        {
            List<Result> results = new List<Result>();
            if (string.IsNullOrEmpty(query.RawQuery))
            {
                // clear the cache
                if (parentDirectories.Count > 0)
                    parentDirectories.Clear();

                return results;
            }

            InitialDriverList();

            var input = query.RawQuery.ToLower();
            if (driverNames.FirstOrDefault(x => input.StartsWith(x)) == null) return results;

            if (Directory.Exists(query.RawQuery))
            {
                // show all child directory
                if (input.EndsWith("\\") || input.EndsWith("/"))
                {
                    var dirInfo = new DirectoryInfo(query.RawQuery);
                    var dirs = dirInfo.GetDirectories();

                    var parentDirKey = input.TrimEnd('\\', '/');
                    if (!parentDirectories.ContainsKey(parentDirKey))
                        parentDirectories.Add(parentDirKey, dirs);

                    foreach (var dir in dirs)
                    {
                        if ((dir.Attributes & FileAttributes.Hidden) == FileAttributes.Hidden)
                            continue;

                        var dirPath = dir.FullName;
                        Result result = new Result
                        {
                            Title = dir.Name,
                            IcoPath = "Images/folder.png",
                            Action = (c) =>
                            {
                                context.ChangeQuery(dirPath);
                                return false;
                            }
                        };
                        results.Add(result);
                    }

                    if (results.Count == 0)
                    {
                        Result result = new Result
                        {
                            Title = "Open this directory",
                            SubTitle = "No files in this directory",
                            IcoPath = "Images/folder.png",
                            Action = (c) =>
                            {
                                Process.Start(query.RawQuery);
                                return true;
                            }
                        };
                        results.Add(result);
                    }
                }
                else
                {
                    Result result = new Result
                    {
                        Title = "Open this directory",
                        SubTitle = string.Format("path: {0}", query.RawQuery),
                        Score = 50,
                        IcoPath = "Images/folder.png",
                        Action = (c) =>
                        {
                            Process.Start(query.RawQuery);
                            return true;
                        }
                    };
                    results.Add(result);
                }

            }

            // change to search in current directory
            var parentDir = Path.GetDirectoryName(input);
            if (!string.IsNullOrEmpty(parentDir) && results.Count == 0)
            {
                parentDir = parentDir.TrimEnd('\\', '/');
                if (parentDirectories.ContainsKey(parentDir))
                {
                    
                    var dirs = parentDirectories[parentDir];
                    var queryFileName = Path.GetFileName(query.RawQuery).ToLower();
                    var fuzzy = FuzzyMatcher.Create(queryFileName);
                    foreach (var dir in dirs)
                    {
                        if ((dir.Attributes & FileAttributes.Hidden) == FileAttributes.Hidden)
                            continue;

                        var matchResult = fuzzy.Evaluate(dir.Name);
                        if (!matchResult.Success)
                            continue;

                        var dirPath = dir.FullName;
                        Result result = new Result
                        {
                            Title = dir.Name,
                            IcoPath = "Images/folder.png",
                            Score = matchResult.Score,
                            Action = (c) =>
                            {
                                context.ChangeQuery(dirPath);
                                return false;
                            }
                        };
                        results.Add(result);
                    }
                }
            }
            

            return results;
        }

        private void InitialDriverList()
        {
            if (driverNames == null)
            {
                driverNames = new List<string>();
                var allDrives = DriveInfo.GetDrives();
                foreach (var driver in allDrives)
                {
                    driverNames.Add(driver.Name.ToLower().TrimEnd('\\'));
                }
            }
        }

        protected override void InitInternal(PluginInitContext context)
        {
            this.context = context;
        }

    }
}