From 0bfb755a3895d26144a69eb7f0ea02bba655a15f Mon Sep 17 00:00:00 2001 From: Luke Pulverenti Date: Sat, 27 Apr 2013 09:05:33 -0400 Subject: search hints progress --- .../Library/LuceneSearchEngine.cs | 156 +++++++++++++++------ 1 file changed, 110 insertions(+), 46 deletions(-) (limited to 'MediaBrowser.Server.Implementations/Library/LuceneSearchEngine.cs') diff --git a/MediaBrowser.Server.Implementations/Library/LuceneSearchEngine.cs b/MediaBrowser.Server.Implementations/Library/LuceneSearchEngine.cs index d5675578d..5c309a896 100644 --- a/MediaBrowser.Server.Implementations/Library/LuceneSearchEngine.cs +++ b/MediaBrowser.Server.Implementations/Library/LuceneSearchEngine.cs @@ -97,24 +97,26 @@ namespace MediaBrowser.Server.Implementations.Library /// The search term. /// IEnumerable{SearchHintResult}. /// searchTerm - public async Task> GetSearchHints(IEnumerable inputItems, string searchTerm) + public async Task> GetSearchHints(IEnumerable inputItems, string searchTerm) { if (string.IsNullOrEmpty(searchTerm)) { throw new ArgumentNullException("searchTerm"); } - var hints = new List>(); + var terms = GetWords(searchTerm); + + var hints = new List>(); var items = inputItems.Where(i => !(i is MusicArtist)).ToList(); foreach (var item in items) { - var index = IndexOf(item.Name, searchTerm); + var index = GetIndex(item.Name, searchTerm, terms); - if (index != -1) + if (index.Item2 != -1) { - hints.Add(new Tuple(item, index)); + hints.Add(new Tuple(item, index.Item1, index.Item2)); } } @@ -127,16 +129,23 @@ namespace MediaBrowser.Server.Implementations.Library foreach (var item in artists) { - var index = IndexOf(item, searchTerm); + var index = GetIndex(item, searchTerm, terms); - if (index != -1) + if (index.Item2 != -1) { - var artist = await _libraryManager.GetArtist(item).ConfigureAwait(false); + try + { + var artist = await _libraryManager.GetArtist(item).ConfigureAwait(false); - hints.Add(new Tuple(artist, index)); + hints.Add(new Tuple(artist, index.Item1, index.Item2)); + } + catch (Exception ex) + { + _logger.ErrorException("Error getting {0}", ex, item); + } } } - + // Find genres var genres = items.SelectMany(i => i.Genres) .Where(i => !string.IsNullOrEmpty(i)) @@ -145,13 +154,20 @@ namespace MediaBrowser.Server.Implementations.Library foreach (var item in genres) { - var index = IndexOf(item, searchTerm); + var index = GetIndex(item, searchTerm, terms); - if (index != -1) + if (index.Item2 != -1) { - var genre = await _libraryManager.GetGenre(item).ConfigureAwait(false); + try + { + var genre = await _libraryManager.GetGenre(item).ConfigureAwait(false); - hints.Add(new Tuple(genre, index)); + hints.Add(new Tuple(genre, index.Item1, index.Item2)); + } + catch (Exception ex) + { + _logger.ErrorException("Error getting {0}", ex, item); + } } } @@ -163,13 +179,20 @@ namespace MediaBrowser.Server.Implementations.Library foreach (var item in studios) { - var index = IndexOf(item, searchTerm); + var index = GetIndex(item, searchTerm, terms); - if (index != -1) + if (index.Item2 != -1) { - var studio = await _libraryManager.GetStudio(item).ConfigureAwait(false); + try + { + var studio = await _libraryManager.GetStudio(item).ConfigureAwait(false); - hints.Add(new Tuple(studio, index)); + hints.Add(new Tuple(studio, index.Item1, index.Item2)); + } + catch (Exception ex) + { + _logger.ErrorException("Error getting {0}", ex, item); + } } } @@ -182,52 +205,93 @@ namespace MediaBrowser.Server.Implementations.Library foreach (var item in persons) { - var index = IndexOf(item, searchTerm); + var index = GetIndex(item, searchTerm, terms); - if (index != -1) + if (index.Item2 != -1) { - var person = await _libraryManager.GetPerson(item).ConfigureAwait(false); + try + { + var person = await _libraryManager.GetPerson(item).ConfigureAwait(false); - hints.Add(new Tuple(person, index)); + hints.Add(new Tuple(person, index.Item1, index.Item2)); + } + catch (Exception ex) + { + _logger.ErrorException("Error getting {0}", ex, item); + } } } - return hints.OrderBy(i => i.Item2).Select(i => i.Item1); - } - - /// - /// Gets the words. - /// - /// The term. - /// System.String[][]. - private string[] GetWords(string term) - { - // TODO: Improve this to be more accurate and respect culture - var words = term.Split(' '); - - return words; + return hints.OrderBy(i => i.Item3).Select(i => new SearchHintInfo + { + Item = i.Item1, + MatchedTerm = i.Item2 + }); } /// - /// Indexes the of. + /// Gets the index. /// /// The input. - /// The term. + /// The search input. + /// The search input. /// System.Int32. - private int IndexOf(string input, string term) + private Tuple GetIndex(string input, string searchInput, string[] searchWords) { - var index = 0; + if (string.Equals(input, searchInput, StringComparison.OrdinalIgnoreCase)) + { + return new Tuple(searchInput, 0); + } + + var index = input.IndexOf(searchInput, StringComparison.OrdinalIgnoreCase); - foreach (var word in GetWords(input)) + if (index == 0) { - if (word.IndexOf(term, StringComparison.OrdinalIgnoreCase) != -1) + return new Tuple(searchInput, 1); + } + if (index > 0) + { + return new Tuple(searchInput, 2); + } + + var items = GetWords(input); + + for (var i = 0; i < searchWords.Length; i++) + { + var searchTerm = searchWords[i]; + + for (var j = 0; j < items.Length; j++) { - return index; - } + var item = items[j]; + + if (string.Equals(item, searchTerm, StringComparison.OrdinalIgnoreCase)) + { + return new Tuple(searchTerm, 3 + (i + 1) * (j + 1)); + } + + index = item.IndexOf(searchTerm, StringComparison.OrdinalIgnoreCase); - index++; + if (index == 0) + { + return new Tuple(searchTerm, 4 + (i + 1) * (j + 1)); + } + if (index > 0) + { + return new Tuple(searchTerm, 5 + (i + 1) * (j + 1)); + } + } } - return -1; + return new Tuple(null, -1); + } + + /// + /// Gets the words. + /// + /// The term. + /// System.String[][]. + private string[] GetWords(string term) + { + return term.Split().Where(i => !string.IsNullOrWhiteSpace(i)).ToArray(); } } -- cgit v1.2.3