jellyfin/Emby.Server.Implementations/Library/SearchEngine.cs

213 lines
7.4 KiB
C#
Raw Normal View History

2014-01-07 20:12:39 +00:00
using System;
using System.Collections.Generic;
using System.Linq;
2017-05-21 07:25:49 +00:00
using MediaBrowser.Controller.Dto;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Audio;
using MediaBrowser.Controller.Extensions;
using MediaBrowser.Controller.Library;
2017-09-04 19:28:22 +00:00
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Querying;
using MediaBrowser.Model.Search;
using Microsoft.Extensions.Logging;
2014-01-07 20:12:39 +00:00
namespace Emby.Server.Implementations.Library
2014-01-07 20:12:39 +00:00
{
/// <summary>
/// </summary>
public class SearchEngine : ISearchEngine
{
private readonly ILibraryManager _libraryManager;
private readonly IUserManager _userManager;
private readonly ILogger _logger;
public SearchEngine(ILoggerFactory loggerFactory, ILibraryManager libraryManager, IUserManager userManager)
2014-01-07 20:12:39 +00:00
{
_libraryManager = libraryManager;
_userManager = userManager;
_logger = loggerFactory.CreateLogger("SearchEngine");
2014-01-07 20:12:39 +00:00
}
2018-09-12 17:26:21 +00:00
public QueryResult<SearchHintInfo> GetSearchHints(SearchQuery query)
2014-01-07 20:12:39 +00:00
{
2015-07-18 19:32:59 +00:00
User user = null;
2018-09-12 17:26:21 +00:00
if (query.UserId.Equals(Guid.Empty))
2014-04-10 15:06:54 +00:00
{
}
else
{
2015-07-18 19:32:59 +00:00
user = _userManager.GetUserById(query.UserId);
2014-04-10 15:06:54 +00:00
}
2018-09-12 17:26:21 +00:00
var results = GetSearchHints(query, user);
var totalRecordCount = results.Count;
2014-01-07 20:12:39 +00:00
if (query.StartIndex.HasValue)
{
2018-09-12 17:26:21 +00:00
results = results.Skip(query.StartIndex.Value).ToList();
2014-01-07 20:12:39 +00:00
}
if (query.Limit.HasValue)
{
2018-09-12 17:26:21 +00:00
results = results.Take(query.Limit.Value).ToList();
2014-01-07 20:12:39 +00:00
}
return new QueryResult<SearchHintInfo>
{
2018-09-12 17:26:21 +00:00
TotalRecordCount = totalRecordCount,
2014-01-07 20:12:39 +00:00
Items = results.ToArray()
};
}
private static void AddIfMissing(List<string> list, string value)
2015-08-22 02:59:10 +00:00
{
if (!list.Contains(value, StringComparer.OrdinalIgnoreCase))
{
list.Add(value);
}
}
2014-01-07 20:12:39 +00:00
/// <summary>
/// Gets the search hints.
/// </summary>
/// <param name="query">The query.</param>
2015-07-18 19:32:59 +00:00
/// <param name="user">The user.</param>
2014-01-07 20:12:39 +00:00
/// <returns>IEnumerable{SearchHintResult}.</returns>
2019-01-13 20:37:13 +00:00
/// <exception cref="ArgumentNullException">searchTerm</exception>
2018-09-12 17:26:21 +00:00
private List<SearchHintInfo> GetSearchHints(SearchQuery query, User user)
2014-01-07 20:12:39 +00:00
{
var searchTerm = query.SearchTerm;
2018-09-12 17:26:21 +00:00
if (string.IsNullOrEmpty(searchTerm))
2014-01-07 20:12:39 +00:00
{
throw new ArgumentNullException(nameof(searchTerm));
2014-01-07 20:12:39 +00:00
}
2018-09-12 17:26:21 +00:00
searchTerm = searchTerm.Trim().RemoveDiacritics();
2014-01-07 20:12:39 +00:00
2017-05-06 20:21:08 +00:00
var excludeItemTypes = query.ExcludeItemTypes.ToList();
2018-09-12 17:26:21 +00:00
var includeItemTypes = (query.IncludeItemTypes ?? Array.Empty<string>()).ToList();
2014-01-07 20:12:39 +00:00
2015-08-22 02:59:10 +00:00
excludeItemTypes.Add(typeof(Year).Name);
2016-10-10 18:18:28 +00:00
excludeItemTypes.Add(typeof(Folder).Name);
2014-01-07 20:12:39 +00:00
2015-08-22 02:59:10 +00:00
if (query.IncludeGenres && (includeItemTypes.Count == 0 || includeItemTypes.Contains("Genre", StringComparer.OrdinalIgnoreCase)))
2014-01-07 20:12:39 +00:00
{
2015-08-22 02:59:10 +00:00
if (!query.IncludeMedia)
2014-01-07 20:12:39 +00:00
{
2015-08-22 02:59:10 +00:00
AddIfMissing(includeItemTypes, typeof(Genre).Name);
AddIfMissing(includeItemTypes, typeof(MusicGenre).Name);
2014-01-07 20:12:39 +00:00
}
}
2015-08-22 02:59:10 +00:00
else
{
AddIfMissing(excludeItemTypes, typeof(Genre).Name);
AddIfMissing(excludeItemTypes, typeof(MusicGenre).Name);
}
2014-01-07 20:12:39 +00:00
2016-05-29 21:02:32 +00:00
if (query.IncludePeople && (includeItemTypes.Count == 0 || includeItemTypes.Contains("People", StringComparer.OrdinalIgnoreCase) || includeItemTypes.Contains("Person", StringComparer.OrdinalIgnoreCase)))
2014-01-07 20:12:39 +00:00
{
2015-08-22 02:59:10 +00:00
if (!query.IncludeMedia)
2014-01-07 20:12:39 +00:00
{
2015-08-22 02:59:10 +00:00
AddIfMissing(includeItemTypes, typeof(Person).Name);
2014-01-07 20:12:39 +00:00
}
2015-08-22 02:59:10 +00:00
}
else
{
AddIfMissing(excludeItemTypes, typeof(Person).Name);
}
2014-01-07 20:12:39 +00:00
2015-08-22 02:59:10 +00:00
if (query.IncludeStudios && (includeItemTypes.Count == 0 || includeItemTypes.Contains("Studio", StringComparer.OrdinalIgnoreCase)))
{
if (!query.IncludeMedia)
2014-01-07 20:12:39 +00:00
{
2015-08-22 02:59:10 +00:00
AddIfMissing(includeItemTypes, typeof(Studio).Name);
2014-01-07 20:12:39 +00:00
}
2015-08-22 02:59:10 +00:00
}
else
{
AddIfMissing(excludeItemTypes, typeof(Studio).Name);
}
2014-01-07 20:12:39 +00:00
2015-08-22 02:59:10 +00:00
if (query.IncludeArtists && (includeItemTypes.Count == 0 || includeItemTypes.Contains("MusicArtist", StringComparer.OrdinalIgnoreCase)))
{
if (!query.IncludeMedia)
2014-01-07 20:12:39 +00:00
{
2015-08-22 02:59:10 +00:00
AddIfMissing(includeItemTypes, typeof(MusicArtist).Name);
2014-01-07 20:12:39 +00:00
}
}
2015-08-22 02:59:10 +00:00
else
{
AddIfMissing(excludeItemTypes, typeof(MusicArtist).Name);
}
2014-01-07 20:12:39 +00:00
2015-10-08 01:49:40 +00:00
AddIfMissing(excludeItemTypes, typeof(CollectionFolder).Name);
2017-01-09 17:05:34 +00:00
AddIfMissing(excludeItemTypes, typeof(Folder).Name);
2017-11-01 19:50:44 +00:00
var mediaTypes = query.MediaTypes.ToList();
if (includeItemTypes.Count > 0)
{
excludeItemTypes.Clear();
mediaTypes.Clear();
}
2015-10-29 13:28:05 +00:00
2017-11-01 19:50:44 +00:00
var searchQuery = new InternalItemsQuery(user)
2014-01-07 20:12:39 +00:00
{
2018-09-12 17:26:21 +00:00
SearchTerm = searchTerm,
2018-12-28 15:48:26 +00:00
ExcludeItemTypes = excludeItemTypes.ToArray(),
IncludeItemTypes = includeItemTypes.ToArray(),
2015-11-02 17:25:01 +00:00
Limit = query.Limit,
2018-09-12 17:26:21 +00:00
IncludeItemsByName = string.IsNullOrEmpty(query.ParentId),
ParentId = string.IsNullOrEmpty(query.ParentId) ? Guid.Empty : new Guid(query.ParentId),
OrderBy = new[] { new ValueTuple<string, SortOrder>(ItemSortBy.SortName, SortOrder.Ascending) },
2017-05-06 19:45:23 +00:00
Recursive = true,
IsKids = query.IsKids,
IsMovie = query.IsMovie,
IsNews = query.IsNews,
IsSeries = query.IsSeries,
2017-05-06 20:21:08 +00:00
IsSports = query.IsSports,
2017-11-01 19:50:44 +00:00
MediaTypes = mediaTypes.ToArray(),
2017-05-21 07:25:49 +00:00
DtoOptions = new DtoOptions
{
2017-08-19 19:43:35 +00:00
Fields = new ItemFields[]
2017-05-21 07:25:49 +00:00
{
ItemFields.AirTime,
ItemFields.DateCreated,
2017-11-23 15:46:16 +00:00
ItemFields.ChannelInfo,
ItemFields.ParentId
2017-05-21 07:25:49 +00:00
}
}
2017-11-01 19:50:44 +00:00
};
List<BaseItem> mediaItems;
if (searchQuery.IncludeItemTypes.Length == 1 && string.Equals(searchQuery.IncludeItemTypes[0], "MusicArtist", StringComparison.OrdinalIgnoreCase))
{
2018-09-12 17:26:21 +00:00
if (!searchQuery.ParentId.Equals(Guid.Empty))
2017-11-01 19:50:44 +00:00
{
2018-09-12 17:26:21 +00:00
searchQuery.AncestorIds = new[] { searchQuery.ParentId };
2017-11-01 19:50:44 +00:00
}
2018-09-12 17:26:21 +00:00
searchQuery.ParentId = Guid.Empty;
2017-11-01 19:50:44 +00:00
searchQuery.IncludeItemsByName = true;
2018-09-12 17:26:21 +00:00
searchQuery.IncludeItemTypes = Array.Empty<string>();
mediaItems = _libraryManager.GetAllArtists(searchQuery).Items.Select(i => i.Item1).ToList();
2017-11-01 19:50:44 +00:00
}
else
{
mediaItems = _libraryManager.GetItemList(searchQuery);
}
2014-01-07 20:12:39 +00:00
2018-09-12 17:26:21 +00:00
return mediaItems.Select(i => new SearchHintInfo
2014-01-07 20:12:39 +00:00
{
2018-09-12 17:26:21 +00:00
Item = i
2014-01-07 20:12:39 +00:00
2018-09-12 17:26:21 +00:00
}).ToList();
2014-01-07 20:12:39 +00:00
}
}
}