|
@@ -0,0 +1,339 @@
|
|
|
+using MediaBrowser.Controller.Entities;
|
|
|
+using MediaBrowser.Controller.Entities.Audio;
|
|
|
+using MediaBrowser.Controller.Library;
|
|
|
+using MediaBrowser.Model.Logging;
|
|
|
+using MediaBrowser.Model.Querying;
|
|
|
+using MediaBrowser.Model.Search;
|
|
|
+using System;
|
|
|
+using System.Collections.Generic;
|
|
|
+using System.Linq;
|
|
|
+using System.Threading.Tasks;
|
|
|
+
|
|
|
+namespace MediaBrowser.Server.Implementations.Library
|
|
|
+{
|
|
|
+ /// <summary>
|
|
|
+ /// Class LuceneSearchEngine
|
|
|
+ /// http://www.codeproject.com/Articles/320219/Lucene-Net-ultra-fast-search-for-MVC-or-WebForms
|
|
|
+ /// </summary>
|
|
|
+ public class SearchEngine : ISearchEngine
|
|
|
+ {
|
|
|
+ private readonly ILibraryManager _libraryManager;
|
|
|
+ private readonly IUserManager _userManager;
|
|
|
+ private readonly ILogger _logger;
|
|
|
+
|
|
|
+ public SearchEngine(ILogManager logManager, ILibraryManager libraryManager, IUserManager userManager)
|
|
|
+ {
|
|
|
+ _libraryManager = libraryManager;
|
|
|
+ _userManager = userManager;
|
|
|
+
|
|
|
+ _logger = logManager.GetLogger("Lucene");
|
|
|
+ }
|
|
|
+
|
|
|
+ public async Task<QueryResult<SearchHintInfo>> GetSearchHints(SearchQuery query)
|
|
|
+ {
|
|
|
+ var user = _userManager.GetUserById(new Guid(query.UserId));
|
|
|
+
|
|
|
+ var inputItems = user.RootFolder.GetRecursiveChildren(user, null);
|
|
|
+
|
|
|
+ var results = await GetSearchHints(inputItems, query).ConfigureAwait(false);
|
|
|
+
|
|
|
+ var searchResultArray = results.ToArray();
|
|
|
+ results = searchResultArray;
|
|
|
+
|
|
|
+ var count = searchResultArray.Length;
|
|
|
+
|
|
|
+ if (query.StartIndex.HasValue)
|
|
|
+ {
|
|
|
+ results = results.Skip(query.StartIndex.Value);
|
|
|
+ }
|
|
|
+
|
|
|
+ if (query.Limit.HasValue)
|
|
|
+ {
|
|
|
+ results = results.Take(query.Limit.Value);
|
|
|
+ }
|
|
|
+
|
|
|
+ return new QueryResult<SearchHintInfo>
|
|
|
+ {
|
|
|
+ TotalRecordCount = count,
|
|
|
+
|
|
|
+ Items = results.ToArray()
|
|
|
+ };
|
|
|
+ }
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// Gets the search hints.
|
|
|
+ /// </summary>
|
|
|
+ /// <param name="inputItems">The input items.</param>
|
|
|
+ /// <param name="query">The query.</param>
|
|
|
+ /// <returns>IEnumerable{SearchHintResult}.</returns>
|
|
|
+ /// <exception cref="System.ArgumentNullException">searchTerm</exception>
|
|
|
+ private Task<IEnumerable<SearchHintInfo>> GetSearchHints(IEnumerable<BaseItem> inputItems, SearchQuery query)
|
|
|
+ {
|
|
|
+ var searchTerm = query.SearchTerm;
|
|
|
+
|
|
|
+ if (string.IsNullOrEmpty(searchTerm))
|
|
|
+ {
|
|
|
+ throw new ArgumentNullException("searchTerm");
|
|
|
+ }
|
|
|
+
|
|
|
+ var terms = GetWords(searchTerm);
|
|
|
+
|
|
|
+ var hints = new List<Tuple<BaseItem, string, int>>();
|
|
|
+
|
|
|
+ var items = inputItems.Where(i => !(i is MusicArtist)).ToList();
|
|
|
+
|
|
|
+ if (query.IncludeMedia)
|
|
|
+ {
|
|
|
+ // Add search hints based on item name
|
|
|
+ hints.AddRange(items.Where(i => !string.IsNullOrEmpty(i.Name)).Select(item =>
|
|
|
+ {
|
|
|
+ var index = GetIndex(item.Name, searchTerm, terms);
|
|
|
+
|
|
|
+ return new Tuple<BaseItem, string, int>(item, index.Item1, index.Item2);
|
|
|
+ }));
|
|
|
+ }
|
|
|
+
|
|
|
+ if (query.IncludeArtists)
|
|
|
+ {
|
|
|
+ // Find artists
|
|
|
+ var artists = _libraryManager.GetAllArtists(items)
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ foreach (var item in artists)
|
|
|
+ {
|
|
|
+ var index = GetIndex(item, searchTerm, terms);
|
|
|
+
|
|
|
+ if (index.Item2 != -1)
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ var artist = _libraryManager.GetArtist(item);
|
|
|
+
|
|
|
+ hints.Add(new Tuple<BaseItem, string, int>(artist, index.Item1, index.Item2));
|
|
|
+ }
|
|
|
+ catch (Exception ex)
|
|
|
+ {
|
|
|
+ _logger.ErrorException("Error getting {0}", ex, item);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if (query.IncludeGenres)
|
|
|
+ {
|
|
|
+ // Find genres, from non-audio items
|
|
|
+ var genres = items.Where(i => !(i is IHasMusicGenres) && !(i is Game))
|
|
|
+ .SelectMany(i => i.Genres)
|
|
|
+ .Where(i => !string.IsNullOrEmpty(i))
|
|
|
+ .Distinct(StringComparer.OrdinalIgnoreCase)
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ foreach (var item in genres)
|
|
|
+ {
|
|
|
+ var index = GetIndex(item, searchTerm, terms);
|
|
|
+
|
|
|
+ if (index.Item2 != -1)
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ var genre = _libraryManager.GetGenre(item);
|
|
|
+
|
|
|
+ hints.Add(new Tuple<BaseItem, string, int>(genre, index.Item1, index.Item2));
|
|
|
+ }
|
|
|
+ catch (Exception ex)
|
|
|
+ {
|
|
|
+ _logger.ErrorException("Error getting {0}", ex, item);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // Find music genres
|
|
|
+ var musicGenres = items.Where(i => i is IHasMusicGenres)
|
|
|
+ .SelectMany(i => i.Genres)
|
|
|
+ .Where(i => !string.IsNullOrEmpty(i))
|
|
|
+ .Distinct(StringComparer.OrdinalIgnoreCase)
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ foreach (var item in musicGenres)
|
|
|
+ {
|
|
|
+ var index = GetIndex(item, searchTerm, terms);
|
|
|
+
|
|
|
+ if (index.Item2 != -1)
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ var genre = _libraryManager.GetMusicGenre(item);
|
|
|
+
|
|
|
+ hints.Add(new Tuple<BaseItem, string, int>(genre, index.Item1, index.Item2));
|
|
|
+ }
|
|
|
+ catch (Exception ex)
|
|
|
+ {
|
|
|
+ _logger.ErrorException("Error getting {0}", ex, item);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // Find music genres
|
|
|
+ var gameGenres = items.OfType<Game>()
|
|
|
+ .SelectMany(i => i.Genres)
|
|
|
+ .Where(i => !string.IsNullOrEmpty(i))
|
|
|
+ .Distinct(StringComparer.OrdinalIgnoreCase)
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ foreach (var item in gameGenres)
|
|
|
+ {
|
|
|
+ var index = GetIndex(item, searchTerm, terms);
|
|
|
+
|
|
|
+ if (index.Item2 != -1)
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ var genre = _libraryManager.GetGameGenre(item);
|
|
|
+
|
|
|
+ hints.Add(new Tuple<BaseItem, string, int>(genre, index.Item1, index.Item2));
|
|
|
+ }
|
|
|
+ catch (Exception ex)
|
|
|
+ {
|
|
|
+ _logger.ErrorException("Error getting {0}", ex, item);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if (query.IncludeStudios)
|
|
|
+ {
|
|
|
+ // Find studios
|
|
|
+ var studios = items.SelectMany(i => i.Studios)
|
|
|
+ .Where(i => !string.IsNullOrEmpty(i))
|
|
|
+ .Distinct(StringComparer.OrdinalIgnoreCase)
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ foreach (var item in studios)
|
|
|
+ {
|
|
|
+ var index = GetIndex(item, searchTerm, terms);
|
|
|
+
|
|
|
+ if (index.Item2 != -1)
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ var studio = _libraryManager.GetStudio(item);
|
|
|
+
|
|
|
+ hints.Add(new Tuple<BaseItem, string, int>(studio, index.Item1, index.Item2));
|
|
|
+ }
|
|
|
+ catch (Exception ex)
|
|
|
+ {
|
|
|
+ _logger.ErrorException("Error getting {0}", ex, item);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if (query.IncludePeople)
|
|
|
+ {
|
|
|
+ // Find persons
|
|
|
+ var persons = items.SelectMany(i => i.People)
|
|
|
+ .Select(i => i.Name)
|
|
|
+ .Where(i => !string.IsNullOrEmpty(i))
|
|
|
+ .Distinct(StringComparer.OrdinalIgnoreCase)
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ foreach (var item in persons)
|
|
|
+ {
|
|
|
+ var index = GetIndex(item, searchTerm, terms);
|
|
|
+
|
|
|
+ if (index.Item2 != -1)
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ var person = _libraryManager.GetPerson(item);
|
|
|
+
|
|
|
+ hints.Add(new Tuple<BaseItem, string, int>(person, index.Item1, index.Item2));
|
|
|
+ }
|
|
|
+ catch (Exception ex)
|
|
|
+ {
|
|
|
+ _logger.ErrorException("Error getting {0}", ex, item);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ var returnValue = hints.Where(i => i.Item3 >= 0).OrderBy(i => i.Item3).Select(i => new SearchHintInfo
|
|
|
+ {
|
|
|
+ Item = i.Item1,
|
|
|
+ MatchedTerm = i.Item2
|
|
|
+ });
|
|
|
+
|
|
|
+ return Task.FromResult(returnValue);
|
|
|
+ }
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// Gets the index.
|
|
|
+ /// </summary>
|
|
|
+ /// <param name="input">The input.</param>
|
|
|
+ /// <param name="searchInput">The search input.</param>
|
|
|
+ /// <param name="searchWords">The search input.</param>
|
|
|
+ /// <returns>System.Int32.</returns>
|
|
|
+ private Tuple<string, int> GetIndex(string input, string searchInput, List<string> searchWords)
|
|
|
+ {
|
|
|
+ if (string.IsNullOrEmpty(input))
|
|
|
+ {
|
|
|
+ throw new ArgumentNullException("input");
|
|
|
+ }
|
|
|
+
|
|
|
+ if (string.Equals(input, searchInput, StringComparison.OrdinalIgnoreCase))
|
|
|
+ {
|
|
|
+ return new Tuple<string, int>(searchInput, 0);
|
|
|
+ }
|
|
|
+
|
|
|
+ var index = input.IndexOf(searchInput, StringComparison.OrdinalIgnoreCase);
|
|
|
+
|
|
|
+ if (index == 0)
|
|
|
+ {
|
|
|
+ return new Tuple<string, int>(searchInput, 1);
|
|
|
+ }
|
|
|
+ if (index > 0)
|
|
|
+ {
|
|
|
+ return new Tuple<string, int>(searchInput, 2);
|
|
|
+ }
|
|
|
+
|
|
|
+ var items = GetWords(input);
|
|
|
+
|
|
|
+ for (var i = 0; i < searchWords.Count; i++)
|
|
|
+ {
|
|
|
+ var searchTerm = searchWords[i];
|
|
|
+
|
|
|
+ for (var j = 0; j < items.Count; j++)
|
|
|
+ {
|
|
|
+ var item = items[j];
|
|
|
+
|
|
|
+ if (string.Equals(item, searchTerm, StringComparison.OrdinalIgnoreCase))
|
|
|
+ {
|
|
|
+ return new Tuple<string, int>(searchTerm, 3 + (i + 1) * (j + 1));
|
|
|
+ }
|
|
|
+
|
|
|
+ index = item.IndexOf(searchTerm, StringComparison.OrdinalIgnoreCase);
|
|
|
+
|
|
|
+ if (index == 0)
|
|
|
+ {
|
|
|
+ return new Tuple<string, int>(searchTerm, 4 + (i + 1) * (j + 1));
|
|
|
+ }
|
|
|
+ if (index > 0)
|
|
|
+ {
|
|
|
+ return new Tuple<string, int>(searchTerm, 5 + (i + 1) * (j + 1));
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return new Tuple<string, int>(null, -1);
|
|
|
+ }
|
|
|
+
|
|
|
+ /// <summary>
|
|
|
+ /// Gets the words.
|
|
|
+ /// </summary>
|
|
|
+ /// <param name="term">The term.</param>
|
|
|
+ /// <returns>System.String[][].</returns>
|
|
|
+ private List<string> GetWords(string term)
|
|
|
+ {
|
|
|
+ return term.Split().Where(i => !string.IsNullOrWhiteSpace(i)).ToList();
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|