Flow.Launcher/Plugins/Flow.Launcher.Plugin.Explorer/Search/SearchManager.cs

using Flow.Launcher.Plugin.Explorer.Search.DirectoryInfo;
using Flow.Launcher.Plugin.Explorer.Search.QuickFolderLinks;
using Flow.Launcher.Plugin.Explorer.Search.WindowsIndex;
using Flow.Launcher.Plugin.SharedCommands;
using System;
using System.Collections.Generic;

namespace Flow.Launcher.Plugin.Explorer.Search
{
    public class SearchManager
    {
        private Settings _settings;
        private PluginInitContext _context;

        private IndexSearch _indexSearch;

        private QuickFolderAccess quickFolderAccess = new QuickFolderAccess();

        public SearchManager(Settings settings, PluginInitContext context)
        {
            _settings = settings;
            _context = context;
            _indexSearch = new IndexSearch();
        }

        internal List<Result> Search(Query query)
        {
            var querySearch = query.Search;

            var quickFolderLinks = quickFolderAccess.FolderList(query, _settings.FolderLinks);

            if (quickFolderLinks.Count > 0)
                return quickFolderLinks;

            if (!FilesFolders.IsLocationPathString(querySearch))
                return WindowsIndexFilesAndFoldersSearch(query, querySearch);

            if (EnvironmentVariables.IsEnvironmentVariableSearch(querySearch)) 
            {
                return EnvironmentVariables.GetEnvironmentStringPathSuggestions(querySearch, query);
            }

            // Query is a location path with a full environment variable, eg. %appdata%\somefolder\
            if (querySearch.Substring(1).Contains("%"))
            {
                querySearch = EnvironmentVariables.TranslateEnvironmentVariablePath(querySearch);
            }

            var results = new List<Result>();

            var currentFolderResult = CreateOpenCurrentFolderResult(FilesFolders.LocationExists, querySearch);

            if (currentFolderResult == null)
                return new List<Result>();
            
            results.Add(currentFolderResult);

            results.AddRange(TopLevelFolderSearchBehaviour(WindowsIndexTopLevelFolderSearch,
                                                           DirectoryInfoClassSearch,
                                                           WindowsIndexExists,
                                                           query,
                                                           querySearch));

            return results;
        }

        private List<Result> DirectoryInfoClassSearch(Query query, string querySearch)
        {
            var directoryInfoSearch = new DirectoryInfoSearch(_settings);

            return directoryInfoSearch.TopLevelDirectorySearch(query, querySearch);
        }

        public List<Result> TopLevelFolderSearchBehaviour(
            Func<Query, string, List<Result>> windowsIndexSearch,
            Func<Query, string, List<Result>> directoryInfoClassSearch,
            Func<string, bool> indexExists,
            Query query,
            string querySearchString)
        {
            var results = windowsIndexSearch(query, querySearchString);

            if (results.Count == 0 && !indexExists(querySearchString))
                return directoryInfoClassSearch(query, querySearchString);

            return results;
        }

        private List<Result> WindowsIndexFilesAndFoldersSearch(Query query, string querySearchString)
        {
            var queryConstructor = new QueryConstructor(_settings);

            return _indexSearch.WindowsIndexSearch(querySearchString,
                                               queryConstructor.CreateQueryHelper().ConnectionString,
                                               queryConstructor.QueryForAllFilesAndFolders,
                                               query);
        }
        
        private List<Result> WindowsIndexTopLevelFolderSearch(Query query, string path)
        {
            var queryConstructor = new QueryConstructor(_settings);

            return _indexSearch.WindowsIndexSearch(path,
                                               queryConstructor.CreateQueryHelper().ConnectionString,
                                               queryConstructor.QueryForTopLevelDirectorySearch,
                                               query);
        }

        private bool WindowsIndexExists(string path)
        {
            path = FilesFolders.GetPreviousLevelDirectoryIfPathIncomplete(path);

            return _indexSearch.PathIsIndexed(path);
        }

        private Result CreateOpenCurrentFolderResult(Func<string, bool> locationExists, string querySearchString)
        {
            if (locationExists(querySearchString))
                return ResultManager.CreateOpenCurrentFolderResult(querySearchString, false);

            var previousDirectoryPath = FilesFolders.GetPreviousExistingDirectory(FilesFolders.LocationExists, querySearchString);

            if (string.IsNullOrEmpty(previousDirectoryPath))
                return null;

            return ResultManager.CreateOpenCurrentFolderResult(previousDirectoryPath, true);
        }
    }
}