From 3eb4091808735858b01855d298226d239be464af Mon Sep 17 00:00:00 2001 From: Luke Pulverenti Date: Thu, 3 Nov 2016 02:37:52 -0400 Subject: move additional classes to new server lib --- .../Library/SearchEngine.cs | 275 +++++++++++++++++++++ 1 file changed, 275 insertions(+) create mode 100644 Emby.Server.Implementations/Library/SearchEngine.cs (limited to 'Emby.Server.Implementations/Library/SearchEngine.cs') diff --git a/Emby.Server.Implementations/Library/SearchEngine.cs b/Emby.Server.Implementations/Library/SearchEngine.cs new file mode 100644 index 000000000..afdf65c06 --- /dev/null +++ b/Emby.Server.Implementations/Library/SearchEngine.cs @@ -0,0 +1,275 @@ +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; +using MediaBrowser.Controller.Extensions; + +namespace Emby.Server.Implementations.Library +{ + /// + /// + 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> GetSearchHints(SearchQuery query) + { + User user = null; + + if (string.IsNullOrWhiteSpace(query.UserId)) + { + } + else + { + user = _userManager.GetUserById(query.UserId); + } + + var results = await GetSearchHints(query, user).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 + { + TotalRecordCount = count, + + Items = results.ToArray() + }; + } + + private void AddIfMissing(List list, string value) + { + if (!list.Contains(value, StringComparer.OrdinalIgnoreCase)) + { + list.Add(value); + } + } + + /// + /// Gets the search hints. + /// + /// The query. + /// The user. + /// IEnumerable{SearchHintResult}. + /// searchTerm + private Task> GetSearchHints(SearchQuery query, User user) + { + var searchTerm = query.SearchTerm; + + if (searchTerm != null) + { + searchTerm = searchTerm.Trim().RemoveDiacritics(); + } + + if (string.IsNullOrWhiteSpace(searchTerm)) + { + throw new ArgumentNullException("searchTerm"); + } + + var terms = GetWords(searchTerm); + + var hints = new List>(); + + var excludeItemTypes = new List(); + var includeItemTypes = (query.IncludeItemTypes ?? new string[] { }).ToList(); + + excludeItemTypes.Add(typeof(Year).Name); + excludeItemTypes.Add(typeof(Folder).Name); + + if (query.IncludeGenres && (includeItemTypes.Count == 0 || includeItemTypes.Contains("Genre", StringComparer.OrdinalIgnoreCase))) + { + if (!query.IncludeMedia) + { + AddIfMissing(includeItemTypes, typeof(Genre).Name); + AddIfMissing(includeItemTypes, typeof(GameGenre).Name); + AddIfMissing(includeItemTypes, typeof(MusicGenre).Name); + } + } + else + { + AddIfMissing(excludeItemTypes, typeof(Genre).Name); + AddIfMissing(excludeItemTypes, typeof(GameGenre).Name); + AddIfMissing(excludeItemTypes, typeof(MusicGenre).Name); + } + + if (query.IncludePeople && (includeItemTypes.Count == 0 || includeItemTypes.Contains("People", StringComparer.OrdinalIgnoreCase) || includeItemTypes.Contains("Person", StringComparer.OrdinalIgnoreCase))) + { + if (!query.IncludeMedia) + { + AddIfMissing(includeItemTypes, typeof(Person).Name); + } + } + else + { + AddIfMissing(excludeItemTypes, typeof(Person).Name); + } + + if (query.IncludeStudios && (includeItemTypes.Count == 0 || includeItemTypes.Contains("Studio", StringComparer.OrdinalIgnoreCase))) + { + if (!query.IncludeMedia) + { + AddIfMissing(includeItemTypes, typeof(Studio).Name); + } + } + else + { + AddIfMissing(excludeItemTypes, typeof(Studio).Name); + } + + if (query.IncludeArtists && (includeItemTypes.Count == 0 || includeItemTypes.Contains("MusicArtist", StringComparer.OrdinalIgnoreCase))) + { + if (!query.IncludeMedia) + { + AddIfMissing(includeItemTypes, typeof(MusicArtist).Name); + } + } + else + { + AddIfMissing(excludeItemTypes, typeof(MusicArtist).Name); + } + + AddIfMissing(excludeItemTypes, typeof(CollectionFolder).Name); + + var mediaItems = _libraryManager.GetItemList(new InternalItemsQuery(user) + { + NameContains = searchTerm, + ExcludeItemTypes = excludeItemTypes.ToArray(), + IncludeItemTypes = includeItemTypes.ToArray(), + Limit = query.Limit, + IncludeItemsByName = true, + IsVirtualItem = false + }); + + // Add search hints based on item name + hints.AddRange(mediaItems.Select(item => + { + var index = GetIndex(item.Name, searchTerm, terms); + + return new Tuple(item, index.Item1, index.Item2); + })); + + 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); + } + + /// + /// Gets the index. + /// + /// The input. + /// The search input. + /// The search input. + /// System.Int32. + private Tuple GetIndex(string input, string searchInput, List searchWords) + { + if (string.IsNullOrWhiteSpace(input)) + { + throw new ArgumentNullException("input"); + } + + input = input.RemoveDiacritics(); + + if (string.Equals(input, searchInput, StringComparison.OrdinalIgnoreCase)) + { + return new Tuple(searchInput, 0); + } + + var index = input.IndexOf(searchInput, StringComparison.OrdinalIgnoreCase); + + if (index == 0) + { + return new Tuple(searchInput, 1); + } + if (index > 0) + { + return new Tuple(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(searchTerm, 3 + (i + 1) * (j + 1)); + } + + index = item.IndexOf(searchTerm, StringComparison.OrdinalIgnoreCase); + + 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 new Tuple(null, -1); + } + + /// + /// Gets the words. + /// + /// The term. + /// System.String[][]. + private List GetWords(string term) + { + var stoplist = GetStopList().ToList(); + + return term.Split() + .Where(i => !string.IsNullOrWhiteSpace(i) && !stoplist.Contains(i, StringComparer.OrdinalIgnoreCase)) + .ToList(); + } + + private IEnumerable GetStopList() + { + return new[] + { + "the", + "a", + "of", + "an" + }; + } + } +} -- cgit v1.2.3