using System; using System.Collections.Generic; using System.Diagnostics; using System.IO; using System.Linq; using System.Windows; using Wox.Infrastructure.Storage.UserSettings; using Control = System.Windows.Controls.Control; namespace Wox.Plugin.Folder { public class FolderPlugin : IPlugin, ISettingProvider { private static List driverNames; private PluginInitContext context; public Control CreateSettingPanel() { return new FileSystemSettings(); } public void Init(PluginInitContext context) { this.context = context; this.context.API.BackKeyDownEvent += ApiBackKeyDownEvent; InitialDriverList(); if (UserSettingStorage.Instance.FolderLinks == null) { UserSettingStorage.Instance.FolderLinks = new List(); UserSettingStorage.Instance.Save(); } } private void ApiBackKeyDownEvent(object sender, WoxKeyDownEventArgs e) { string query = e.Query; if (Directory.Exists(query)) { if (query.EndsWith("\\")) { query = query.Remove(query.Length - 1); } if (query.Contains("\\")) { int index = query.LastIndexOf("\\"); query = query.Remove(index) + "\\"; } context.API.ChangeQuery(query); } } public List Query(Query query) { if(string.IsNullOrEmpty(query.RawQuery)) return new List(); string input = query.RawQuery.ToLower(); List userFolderLinks = UserSettingStorage.Instance.FolderLinks.Where( x => x.Nickname.StartsWith(input, StringComparison.OrdinalIgnoreCase)).ToList(); List results = userFolderLinks.Select( item => new Result(item.Nickname, "Images/folder.png", "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); return false; } }).ToList(); if (!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(); if (!Directory.Exists(rawQuery)) return results; results.Add(new Result("Open current directory", "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; DirectoryInfo dirCopy = dir; var result = new Result(dir.Name, "Images/folder.png", "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; string filePath = file.FullName; var result = new Result(Path.GetFileName(filePath), "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; } } }