using System; using System.Collections.Generic; using System.Diagnostics; using System.IO; using System.Linq; using System.Windows; using System.Windows.Controls; using Wox.Infrastructure.Storage; namespace Wox.Plugin.Folder { public class FolderPlugin : IPlugin, ISettingProvider, IPluginI18n { private static List driverNames; private PluginInitContext context; private readonly Settings _settings; private readonly PluginJsonStorage _storage; public FolderPlugin() { _storage = new PluginJsonStorage(); _settings = _storage.Load(); } ~FolderPlugin() { _storage.Save(); } public Control CreateSettingPanel() { return new FileSystemSettings(context.API, _settings); } public void Init(PluginInitContext context) { this.context = context; InitialDriverList(); if (_settings.FolderLinks == null) { _settings.FolderLinks = new List(); } } public List Query(Query query) { string input = query.Search.ToLower(); List userFolderLinks = _settings.FolderLinks.Where( x => x.Nickname.StartsWith(input, StringComparison.OrdinalIgnoreCase)).ToList(); List results = userFolderLinks.Select( item => new Result() { Title = item.Nickname, IcoPath = "Images/folder.png", SubTitle = "Ctrl + Enter to open the directory", Action = c => { if (c.SpecialKeyState.CtrlPressed) { try { Process.Start(item.Path); return true; } catch (Exception ex) { MessageBox.Show(ex.Message, "Could not start " + item.Path); return false; } } context.API.ChangeQuery(item.Path + (item.Path.EndsWith("\\")? "": "\\")); return false; }, ContextData = item }).ToList(); if (driverNames != null && !driverNames.Any(input.StartsWith)) return results; //if (!input.EndsWith("\\")) //{ // //"c:" means "the current directory on the C drive" whereas @"c:\" means "root of the C drive" // input = input + "\\"; //} results.AddRange(QueryInternal_Directory_Exists(input)); return results; } private void InitialDriverList() { if (driverNames == null) { driverNames = new List(); DriveInfo[] allDrives = DriveInfo.GetDrives(); foreach (DriveInfo driver in allDrives) { driverNames.Add(driver.Name.ToLower().TrimEnd('\\')); } } } private List QueryInternal_Directory_Exists(string rawQuery) { var results = new List(); string incompleteName = ""; if (!Directory.Exists(rawQuery + "\\")) { //if the last component of the path is incomplete, //then make auto complete for it. int index = rawQuery.LastIndexOf('\\'); if (index > 0 && index < (rawQuery.Length - 1)) { incompleteName = rawQuery.Substring(index + 1); incompleteName = incompleteName.ToLower(); rawQuery = rawQuery.Substring(0, index + 1); if (!Directory.Exists(rawQuery)) return results; } else return results; } else { if (!rawQuery.EndsWith("\\")) rawQuery += "\\"; } string firstResult = "Open current directory"; if (incompleteName.Length > 0) firstResult = "Open " + rawQuery; results.Add(new Result { Title = firstResult, IcoPath = "Images/folder.png", Score = 10000, Action = c => { Process.Start(rawQuery); return true; } }); //Add children directories DirectoryInfo[] dirs = new DirectoryInfo(rawQuery).GetDirectories(); foreach (DirectoryInfo dir in dirs) { if ((dir.Attributes & FileAttributes.Hidden) == FileAttributes.Hidden) continue; if (incompleteName.Length != 0 && !dir.Name.ToLower().StartsWith(incompleteName)) continue; DirectoryInfo dirCopy = dir; var result = new Result { Title = dir.Name, IcoPath = "Images/folder.png", SubTitle = "Ctrl + Enter to open the directory", Action = c => { if (c.SpecialKeyState.CtrlPressed) { try { Process.Start(dirCopy.FullName); return true; } catch (Exception ex) { MessageBox.Show(ex.Message, "Could not start " + dirCopy.FullName); return false; } } context.API.ChangeQuery(dirCopy.FullName + "\\"); return false; } }; results.Add(result); } //Add children files FileInfo[] files = new DirectoryInfo(rawQuery).GetFiles(); foreach (FileInfo file in files) { if ((file.Attributes & FileAttributes.Hidden) == FileAttributes.Hidden) continue; if (incompleteName.Length != 0 && !file.Name.ToLower().StartsWith(incompleteName)) continue; string filePath = file.FullName; var result = new Result { Title = Path.GetFileName(filePath), IcoPath = "Images/file.png", Action = c => { try { Process.Start(filePath); } catch (Exception ex) { MessageBox.Show(ex.Message, "Could not start " + filePath); } return true; } }; results.Add(result); } return results; } public string GetTranslatedPluginTitle() { return context.API.GetTranslation("wox_plugin_folder_plugin_name"); } public string GetTranslatedPluginDescription() { return context.API.GetTranslation("wox_plugin_folder_plugin_description"); } } }