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

214 lines
7.4 KiB
C#
Raw Normal View History

2013-03-07 05:34:00 +00:00
using MediaBrowser.Controller.Entities;
2013-02-21 01:33:05 +00:00
using MediaBrowser.Controller.Library;
using MediaBrowser.Model.Dto;
2013-03-10 05:36:39 +00:00
using MediaBrowser.Model.Querying;
2013-02-21 01:33:05 +00:00
using ServiceStack.ServiceHost;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
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
2013-02-21 01:33:05 +00:00
where TItemType : BaseItem
{
/// <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;
/// <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>
protected BaseItemsByNameService(IUserManager userManager, ILibraryManager libraryManager)
{
UserManager = userManager;
LibraryManager = libraryManager;
}
2013-02-21 01:33:05 +00:00
/// <summary>
/// Gets the specified request.
/// </summary>
/// <param name="request">The request.</param>
/// <returns>Task{ItemsResult}.</returns>
protected async Task<ItemsResult> GetResult(GetItemsByName request)
{
var user = UserManager.GetUserById(request.UserId);
2013-02-21 01:33:05 +00:00
2013-03-11 05:06:55 +00:00
var item = string.IsNullOrEmpty(request.ParentId) ? user.RootFolder : DtoBuilder.GetItemByClientId(request.ParentId, UserManager, LibraryManager, user.Id);
2013-02-21 01:33:05 +00:00
IEnumerable<BaseItem> items;
if (item.IsFolder)
{
var folder = (Folder)item;
items = request.Recursive ? folder.GetRecursiveChildren(user) : folder.GetChildren(user);
}
else
{
items = new[] { item };
}
2013-04-11 19:36:50 +00:00
items = FilterItems(request, items, user);
2013-04-12 03:50:47 +00:00
var extractedItems = GetAllItems(request, items, user);
var ibnItemsArray = SortItems(request, extractedItems).ToArray();
2013-04-11 19:36:50 +00:00
IEnumerable<Tuple<string, Func<IEnumerable<BaseItem>>>> ibnItems = ibnItemsArray;
2013-02-21 01:33:05 +00:00
var result = new ItemsResult
{
TotalRecordCount = ibnItemsArray.Length
};
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
}
}
2013-04-11 19:36:50 +00:00
var fields = request.GetItemFields().ToList();
2013-03-02 07:16:52 +00:00
var tasks = ibnItems.Select(i => GetDto(i, user, fields));
2013-02-21 01:33:05 +00:00
var resultItems = await Task.WhenAll(tasks).ConfigureAwait(false);
2013-03-05 23:11:45 +00:00
result.Items = resultItems.Where(i => i != null).ToArray();
2013-02-21 01:33:05 +00:00
return result;
}
2013-04-12 03:50:47 +00:00
/// <summary>
/// Sorts the items.
/// </summary>
/// <param name="request">The request.</param>
/// <param name="items">The items.</param>
/// <returns>IEnumerable{BaseItem}.</returns>
private IEnumerable<Tuple<string, Func<IEnumerable<BaseItem>>>> SortItems(GetItemsByName request, IEnumerable<Tuple<string, Func<IEnumerable<BaseItem>>>> items)
{
if (string.Equals(request.SortBy, "SortName", StringComparison.OrdinalIgnoreCase))
{
if (request.SortOrder.HasValue && request.SortOrder.Value == Model.Entities.SortOrder.Descending)
{
items = items.OrderByDescending(i => i.Item1);
}
else
{
items = items.OrderBy(i => i.Item1);
}
}
return items;
}
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>
2013-04-11 19:36:50 +00:00
/// <param name="items">The items.</param>
/// <param name="user">The user.</param>
/// <returns>IEnumerable{BaseItem}.</returns>
private IEnumerable<BaseItem> FilterItems(GetItemsByName request, IEnumerable<BaseItem> items, User user)
2013-03-02 07:16:52 +00:00
{
2013-04-11 19:36:50 +00:00
items = items.AsParallel();
2013-03-02 07:16:52 +00:00
2013-04-11 19:36:50 +00:00
items = ItemsService.ApplyAdditionalFilters(request, items);
// Apply filters
// Run them starting with the ones that are likely to reduce the list the most
foreach (var filter in request.GetFilters().OrderByDescending(f => (int)f))
2013-03-02 07:16:52 +00:00
{
2013-04-11 19:36:50 +00:00
items = ItemsService.ApplyFilter(items, filter, user, UserManager);
2013-03-02 07:16:52 +00:00
}
2013-04-11 19:36:50 +00:00
items = items.AsEnumerable();
return items;
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>
/// <param name="user">The user.</param>
/// <returns>IEnumerable{Tuple{System.StringFunc{System.Int32}}}.</returns>
protected abstract IEnumerable<Tuple<string, Func<IEnumerable<BaseItem>>>> GetAllItems(GetItemsByName request, IEnumerable<BaseItem> items, User user);
2013-02-21 01:33:05 +00:00
/// <summary>
/// Gets the entity.
/// </summary>
/// <param name="name">The name.</param>
/// <returns>Task{BaseItem}.</returns>
protected abstract Task<TItemType> GetEntity(string name);
/// <summary>
/// Gets the dto.
/// </summary>
/// <param name="stub">The stub.</param>
/// <param name="user">The user.</param>
/// <param name="fields">The fields.</param>
/// <returns>Task{DtoBaseItem}.</returns>
private async Task<BaseItemDto> GetDto(Tuple<string, Func<IEnumerable<BaseItem>>> stub, User user, List<ItemFields> fields)
2013-02-21 01:33:05 +00:00
{
BaseItem item;
try
{
item = await GetEntity(stub.Item1).ConfigureAwait(false);
}
catch (IOException ex)
{
Logger.ErrorException("Error getting IBN item {0}", ex, stub.Item1);
return null;
}
var dto = await new DtoBuilder(Logger, LibraryManager, UserManager).GetBaseItemDto(item, user, fields).ConfigureAwait(false);
2013-02-21 01:33:05 +00:00
if (fields.Contains(ItemFields.ItemCounts))
{
var items = stub.Item2().ToList();
dto.ChildCount = items.Count;
dto.RecentlyAddedItemCount = items.Count(i => i.IsRecentlyAdded(user));
}
2013-02-21 01:33:05 +00:00
return dto;
}
}
/// <summary>
/// Class GetItemsByName
/// </summary>
2013-03-11 05:06:55 +00:00
public class GetItemsByName : BaseItemsRequest, IReturn<ItemsResult>
2013-02-21 01:33:05 +00:00
{
2013-04-12 03:50:47 +00:00
/// <summary>
/// What to sort the results by
/// </summary>
/// <value>The sort by.</value>
[ApiMember(Name = "SortBy", Description = "Optional. Options: SortName", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET", AllowMultiple = true)]
public string SortBy { get; set; }
2013-02-21 01:33:05 +00:00
}
}