jellyfin-server/MediaBrowser.Api/UserLibrary/BaseItemsByNameService.cs

390 lines
14 KiB
C#
Raw Normal View History

using MediaBrowser.Controller.Dto;
using MediaBrowser.Controller.Entities;
2013-02-21 01:33:05 +00:00
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Persistence;
2014-10-06 23:58:46 +00:00
using MediaBrowser.Model.Entities;
2013-03-10 05:36:39 +00:00
using MediaBrowser.Model.Querying;
2013-02-21 01:33:05 +00:00
using System;
using System.Collections.Generic;
using System.Linq;
2016-11-10 14:41:24 +00:00
using MediaBrowser.Controller.Net;
2016-06-17 13:06:13 +00:00
using MediaBrowser.Model.Dto;
2016-10-25 19:02:04 +00:00
using MediaBrowser.Model.Services;
using MediaBrowser.Model.Extensions;
2013-02-21 01:33:05 +00:00
namespace MediaBrowser.Api.UserLibrary
{
/// <summary>
/// Class BaseItemsByNameService
/// </summary>
/// <typeparam name="TItemType">The type of the T item type.</typeparam>
2013-03-16 05:52:33 +00:00
public abstract class BaseItemsByNameService<TItemType> : BaseApiService
where TItemType : BaseItem, IItemByName
2013-02-21 01:33:05 +00:00
{
/// <summary>
/// The _user manager
/// </summary>
protected readonly IUserManager UserManager;
2013-03-11 05:06:55 +00:00
/// <summary>
/// The library manager
/// </summary>
protected readonly ILibraryManager LibraryManager;
2013-10-02 16:08:58 +00:00
protected readonly IUserDataManager UserDataRepository;
2013-06-18 19:16:27 +00:00
protected readonly IItemRepository ItemRepository;
2013-09-04 17:02:19 +00:00
protected IDtoService DtoService { get; private set; }
2016-11-10 14:41:24 +00:00
protected IAuthorizationContext AuthorizationContext { get; private set; }
/// <summary>
/// Initializes a new instance of the <see cref="BaseItemsByNameService{TItemType}" /> class.
/// </summary>
/// <param name="userManager">The user manager.</param>
2013-03-11 05:06:55 +00:00
/// <param name="libraryManager">The library manager.</param>
/// <param name="userDataRepository">The user data repository.</param>
/// <param name="itemRepository">The item repository.</param>
/// <param name="dtoService">The dto service.</param>
2016-11-10 14:41:24 +00:00
protected BaseItemsByNameService(IUserManager userManager, ILibraryManager libraryManager, IUserDataManager userDataRepository, IItemRepository itemRepository, IDtoService dtoService, IAuthorizationContext authorizationContext)
{
UserManager = userManager;
LibraryManager = libraryManager;
UserDataRepository = userDataRepository;
2013-06-18 19:16:27 +00:00
ItemRepository = itemRepository;
2013-09-04 17:02:19 +00:00
DtoService = dtoService;
2016-11-10 14:41:24 +00:00
AuthorizationContext = authorizationContext;
}
2015-08-15 03:15:25 +00:00
protected BaseItem GetParentItem(GetItemsByName request)
{
BaseItem parentItem;
2018-09-12 17:26:21 +00:00
if (!request.UserId.Equals(Guid.Empty))
2015-08-15 03:15:25 +00:00
{
var user = UserManager.GetUserById(request.UserId);
2018-09-12 17:26:21 +00:00
parentItem = string.IsNullOrEmpty(request.ParentId) ? LibraryManager.GetUserRootFolder() : LibraryManager.GetItemById(request.ParentId);
2015-08-15 03:15:25 +00:00
}
else
{
parentItem = string.IsNullOrEmpty(request.ParentId) ? LibraryManager.RootFolder : LibraryManager.GetItemById(request.ParentId);
}
return parentItem;
}
protected string GetParentItemViewType(GetItemsByName request)
{
var parent = GetParentItem(request);
2018-09-12 17:26:21 +00:00
var collectionFolder = parent as IHasCollectionType;
2015-08-15 03:15:25 +00:00
if (collectionFolder != null)
{
return collectionFolder.CollectionType;
}
return null;
}
2017-08-19 19:43:35 +00:00
protected QueryResult<BaseItemDto> GetResultSlim(GetItemsByName request)
2016-06-17 13:06:13 +00:00
{
2016-11-10 14:41:24 +00:00
var dtoOptions = GetDtoOptions(AuthorizationContext, request);
2016-06-17 13:06:13 +00:00
User user = null;
BaseItem parentItem;
2018-09-12 17:26:21 +00:00
if (!request.UserId.Equals(Guid.Empty))
2016-06-17 13:06:13 +00:00
{
user = UserManager.GetUserById(request.UserId);
2018-09-12 17:26:21 +00:00
parentItem = string.IsNullOrEmpty(request.ParentId) ? LibraryManager.GetUserRootFolder() : LibraryManager.GetItemById(request.ParentId);
2016-06-17 13:06:13 +00:00
}
else
{
parentItem = string.IsNullOrEmpty(request.ParentId) ? LibraryManager.RootFolder : LibraryManager.GetItemById(request.ParentId);
}
var excludeItemTypes = request.GetExcludeItemTypes();
var includeItemTypes = request.GetIncludeItemTypes();
var mediaTypes = request.GetMediaTypes();
var query = new InternalItemsQuery(user)
{
ExcludeItemTypes = excludeItemTypes,
IncludeItemTypes = includeItemTypes,
MediaTypes = mediaTypes,
StartIndex = request.StartIndex,
Limit = request.Limit,
IsFavorite = request.IsFavorite,
NameLessThan = request.NameLessThan,
NameStartsWith = request.NameStartsWith,
NameStartsWithOrGreater = request.NameStartsWithOrGreater,
Tags = request.GetTags(),
OfficialRatings = request.GetOfficialRatings(),
Genres = request.GetGenres(),
2018-09-12 17:26:21 +00:00
GenreIds = GetGuids(request.GenreIds),
StudioIds = GetGuids(request.StudioIds),
2016-06-17 13:06:13 +00:00
Person = request.Person,
2018-09-12 17:26:21 +00:00
PersonIds = GetGuids(request.PersonIds),
2016-06-17 13:06:13 +00:00
PersonTypes = request.GetPersonTypes(),
Years = request.GetYears(),
2017-05-21 07:25:49 +00:00
MinCommunityRating = request.MinCommunityRating,
2018-09-12 17:26:21 +00:00
DtoOptions = dtoOptions,
SearchTerm = request.SearchTerm,
EnableTotalRecordCount = request.EnableTotalRecordCount
2016-06-17 13:06:13 +00:00
};
if (!string.IsNullOrWhiteSpace(request.ParentId))
{
if (parentItem is Folder)
{
2018-09-12 17:26:21 +00:00
query.AncestorIds = new[] { new Guid(request.ParentId) };
2016-06-17 13:06:13 +00:00
}
else
{
2018-09-12 17:26:21 +00:00
query.ItemIds = new[] { new Guid(request.ParentId) };
2016-06-17 13:06:13 +00:00
}
}
2016-12-20 19:59:25 +00:00
// Studios
if (!string.IsNullOrEmpty(request.Studios))
{
query.StudioIds = request.Studios.Split('|').Select(i =>
{
try
{
return LibraryManager.GetStudio(i);
}
catch
{
return null;
}
2018-09-12 17:26:21 +00:00
}).Where(i => i != null).Select(i => i.Id).ToArray();
2016-12-20 19:59:25 +00:00
}
2016-06-17 13:06:13 +00:00
foreach (var filter in request.GetFilters())
{
switch (filter)
{
case ItemFilter.Dislikes:
query.IsLiked = false;
break;
case ItemFilter.IsFavorite:
query.IsFavorite = true;
break;
case ItemFilter.IsFavoriteOrLikes:
query.IsFavoriteOrLiked = true;
break;
case ItemFilter.IsFolder:
query.IsFolder = true;
break;
case ItemFilter.IsNotFolder:
query.IsFolder = false;
break;
case ItemFilter.IsPlayed:
query.IsPlayed = true;
break;
case ItemFilter.IsResumable:
query.IsResumable = true;
break;
case ItemFilter.IsUnplayed:
query.IsPlayed = false;
break;
case ItemFilter.Likes:
query.IsLiked = true;
break;
}
}
var result = GetItems(request, query);
var dtos = result.Items.Select(i =>
{
2018-09-12 17:26:21 +00:00
var dto = DtoService.GetItemByNameDto(i.Item1, dtoOptions, null, user);
2016-06-17 13:06:13 +00:00
if (!string.IsNullOrWhiteSpace(request.IncludeItemTypes))
{
SetItemCounts(dto, i.Item2);
}
return dto;
});
2017-08-19 19:43:35 +00:00
return new QueryResult<BaseItemDto>
2016-06-17 13:06:13 +00:00
{
Items = dtos.ToArray(result.Items.Length),
2016-06-17 13:06:13 +00:00
TotalRecordCount = result.TotalRecordCount
};
}
protected virtual QueryResult<Tuple<BaseItem, ItemCounts>> GetItems(GetItemsByName request, InternalItemsQuery query)
{
return new QueryResult<Tuple<BaseItem, ItemCounts>>();
}
private void SetItemCounts(BaseItemDto dto, ItemCounts counts)
{
dto.ChildCount = counts.ItemCount;
dto.ProgramCount = counts.ProgramCount;
2016-06-17 13:06:13 +00:00
dto.SeriesCount = counts.SeriesCount;
dto.EpisodeCount = counts.EpisodeCount;
dto.MovieCount = counts.MovieCount;
dto.TrailerCount = counts.TrailerCount;
dto.AlbumCount = counts.AlbumCount;
dto.SongCount = counts.SongCount;
dto.GameCount = counts.GameCount;
2016-08-13 20:54:29 +00:00
dto.ArtistCount = counts.ArtistCount;
2016-06-17 13:06:13 +00:00
}
2013-02-21 01:33:05 +00:00
/// <summary>
/// Gets the specified request.
/// </summary>
/// <param name="request">The request.</param>
/// <returns>Task{ItemsResult}.</returns>
2017-08-19 19:43:35 +00:00
protected QueryResult<BaseItemDto> GetResult(GetItemsByName request)
2013-02-21 01:33:05 +00:00
{
2016-11-10 14:41:24 +00:00
var dtoOptions = GetDtoOptions(AuthorizationContext, request);
2015-03-26 04:44:24 +00:00
2013-04-22 04:38:03 +00:00
User user = null;
2015-01-25 06:34:50 +00:00
BaseItem parentItem;
2013-02-21 01:33:05 +00:00
2018-09-12 17:26:21 +00:00
if (!request.UserId.Equals(Guid.Empty))
2013-04-22 04:38:03 +00:00
{
2015-05-29 23:51:33 +00:00
user = UserManager.GetUserById(request.UserId);
2018-09-12 17:26:21 +00:00
parentItem = string.IsNullOrEmpty(request.ParentId) ? LibraryManager.GetUserRootFolder() : LibraryManager.GetItemById(request.ParentId);
2013-04-22 04:38:03 +00:00
}
else
{
2015-01-25 06:34:50 +00:00
parentItem = string.IsNullOrEmpty(request.ParentId) ? LibraryManager.RootFolder : LibraryManager.GetItemById(request.ParentId);
2013-04-22 04:38:03 +00:00
}
2013-02-21 01:33:05 +00:00
2017-08-24 19:52:19 +00:00
IList<BaseItem> items;
2013-02-21 01:33:05 +00:00
2015-01-25 06:34:50 +00:00
var excludeItemTypes = request.GetExcludeItemTypes();
var includeItemTypes = request.GetIncludeItemTypes();
var mediaTypes = request.GetMediaTypes();
2016-05-18 05:34:10 +00:00
var query = new InternalItemsQuery(user)
{
ExcludeItemTypes = excludeItemTypes,
IncludeItemTypes = includeItemTypes,
2017-05-22 04:54:02 +00:00
MediaTypes = mediaTypes,
DtoOptions = dtoOptions
2016-05-18 05:34:10 +00:00
};
2015-01-25 06:34:50 +00:00
Func<BaseItem, bool> filter = i => FilterItem(request, i, excludeItemTypes, includeItemTypes, mediaTypes);
if (parentItem.IsFolder)
2013-02-21 01:33:05 +00:00
{
2015-01-25 06:34:50 +00:00
var folder = (Folder)parentItem;
2013-02-21 01:33:05 +00:00
2018-09-12 17:26:21 +00:00
if (!request.UserId.Equals(Guid.Empty))
2013-04-22 04:38:03 +00:00
{
2015-01-25 06:34:50 +00:00
items = request.Recursive ?
2017-08-24 19:52:19 +00:00
folder.GetRecursiveChildren(user, query).ToList() :
folder.GetChildren(user, true).Where(filter).ToList();
2013-04-22 04:38:03 +00:00
}
else
{
2015-01-25 06:34:50 +00:00
items = request.Recursive ?
folder.GetRecursiveChildren(filter) :
2017-08-24 19:52:19 +00:00
folder.Children.Where(filter).ToList();
2013-04-22 04:38:03 +00:00
}
2013-02-21 01:33:05 +00:00
}
else
{
2017-08-24 19:52:19 +00:00
items = new[] { parentItem }.Where(filter).ToList();
2013-02-21 01:33:05 +00:00
}
var extractedItems = GetAllItems(request, items);
2018-09-12 17:26:21 +00:00
var filteredItems = LibraryManager.Sort(extractedItems, user, request.GetOrderBy());
var ibnItemsArray = filteredItems.ToList();
2015-08-15 03:15:25 +00:00
IEnumerable<BaseItem> ibnItems = ibnItemsArray;
2013-02-21 01:33:05 +00:00
2017-08-19 19:43:35 +00:00
var result = new QueryResult<BaseItemDto>
2013-02-21 01:33:05 +00:00
{
TotalRecordCount = ibnItemsArray.Count
2013-02-21 01:33:05 +00:00
};
if (request.StartIndex.HasValue || request.Limit.HasValue)
2013-02-21 01:33:05 +00:00
{
if (request.StartIndex.HasValue)
{
ibnItems = ibnItems.Skip(request.StartIndex.Value);
}
if (request.Limit.HasValue)
2013-02-21 01:33:05 +00:00
{
ibnItems = ibnItems.Take(request.Limit.Value);
2013-02-21 01:33:05 +00:00
}
}
var tuples = ibnItems.Select(i => new Tuple<BaseItem, List<BaseItem>>(i, new List<BaseItem>()));
2014-11-30 19:01:33 +00:00
2018-09-12 17:26:21 +00:00
var dtos = tuples.Select(i => DtoService.GetItemByNameDto(i.Item1, dtoOptions, i.Item2, user));
2013-02-21 01:33:05 +00:00
2013-09-17 02:44:06 +00:00
result.Items = dtos.Where(i => i != null).ToArray();
2013-02-21 01:33:05 +00:00
return result;
}
2013-03-02 07:16:52 +00:00
/// <summary>
2013-04-11 19:36:50 +00:00
/// Filters the items.
2013-03-02 07:16:52 +00:00
/// </summary>
/// <param name="request">The request.</param>
2015-01-25 06:34:50 +00:00
/// <param name="f">The f.</param>
/// <param name="excludeItemTypes">The exclude item types.</param>
/// <param name="includeItemTypes">The include item types.</param>
/// <param name="mediaTypes">The media types.</param>
2013-04-11 19:36:50 +00:00
/// <returns>IEnumerable{BaseItem}.</returns>
2016-06-17 13:06:13 +00:00
private bool FilterItem(GetItemsByName request, BaseItem f, string[] excludeItemTypes, string[] includeItemTypes, string[] mediaTypes)
2013-03-02 07:16:52 +00:00
{
// Exclude item types
2015-01-25 06:34:50 +00:00
if (excludeItemTypes.Length > 0)
2013-03-02 07:16:52 +00:00
{
2015-01-25 06:34:50 +00:00
if (excludeItemTypes.Contains(f.GetType().Name, StringComparer.OrdinalIgnoreCase))
{
return false;
}
2013-03-02 07:16:52 +00:00
}
// Include item types
2015-01-25 06:34:50 +00:00
if (includeItemTypes.Length > 0)
{
2015-01-25 06:34:50 +00:00
if (!includeItemTypes.Contains(f.GetType().Name, StringComparer.OrdinalIgnoreCase))
{
return false;
}
}
2013-04-11 19:36:50 +00:00
2013-04-24 20:28:42 +00:00
// Include MediaTypes
2015-01-25 06:34:50 +00:00
if (mediaTypes.Length > 0)
2013-04-24 20:28:42 +00:00
{
2015-01-25 06:34:50 +00:00
if (!mediaTypes.Contains(f.MediaType ?? string.Empty, StringComparer.OrdinalIgnoreCase))
{
return false;
}
2013-04-24 20:28:42 +00:00
}
2013-09-10 19:30:56 +00:00
2015-01-25 06:34:50 +00:00
return true;
2013-03-02 07:16:52 +00:00
}
2013-02-21 01:33:05 +00:00
/// <summary>
/// Gets all items.
/// </summary>
/// <param name="request">The request.</param>
/// <param name="items">The items.</param>
/// <returns>IEnumerable{Task{`0}}.</returns>
2017-08-24 19:52:19 +00:00
protected abstract IEnumerable<BaseItem> GetAllItems(GetItemsByName request, IList<BaseItem> items);
2013-02-21 01:33:05 +00:00
}
/// <summary>
/// Class GetItemsByName
/// </summary>
2017-08-19 19:43:35 +00:00
public class GetItemsByName : BaseItemsRequest, IReturn<QueryResult<BaseItemDto>>
2013-02-21 01:33:05 +00:00
{
2013-04-22 04:38:03 +00:00
public GetItemsByName()
{
Recursive = true;
}
2013-02-21 01:33:05 +00:00
}
}