using MediaBrowser.Common.Implementations.HttpServer;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Library;
using MediaBrowser.Model.Dto;
using MediaBrowser.Model.Entities;
using ServiceStack.ServiceHost;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
namespace MediaBrowser.Api.UserLibrary
{
///
/// Class BaseItemsByNameService
///
/// The type of the T item type.
public abstract class BaseItemsByNameService : BaseRestService
where TItemType : BaseItem
{
///
/// The _user manager
///
protected readonly IUserManager UserManager;
protected readonly ILibraryManager LibraryManager;
///
/// Initializes a new instance of the class.
///
/// The user manager.
protected BaseItemsByNameService(IUserManager userManager, ILibraryManager libraryManager)
{
UserManager = userManager;
LibraryManager = libraryManager;
}
///
/// Gets the specified request.
///
/// The request.
/// Task{ItemsResult}.
protected async Task GetResult(GetItemsByName request)
{
var user = UserManager.GetUserById(request.UserId);
var item = string.IsNullOrEmpty(request.Id) ? user.RootFolder : DtoBuilder.GetItemByClientId(request.Id, UserManager, LibraryManager, user.Id);
IEnumerable items;
if (item.IsFolder)
{
var folder = (Folder)item;
items = request.Recursive ? folder.GetRecursiveChildren(user) : folder.GetChildren(user);
}
else
{
items = new[] { item };
}
var ibnItemsArray = GetAllItems(request, items, user).ToArray();
IEnumerable>> ibnItems = ibnItemsArray;
var result = new ItemsResult
{
TotalRecordCount = ibnItemsArray.Length
};
if (request.StartIndex.HasValue || request.PageSize.HasValue)
{
if (request.StartIndex.HasValue)
{
ibnItems = ibnItems.Skip(request.StartIndex.Value);
}
if (request.PageSize.HasValue)
{
ibnItems = ibnItems.Take(request.PageSize.Value);
}
}
var fields = GetItemFields(request).ToList();
var tasks = ibnItems.Select(i => GetDto(i, user, fields));
var resultItems = await Task.WhenAll(tasks).ConfigureAwait(false);
result.Items = resultItems.Where(i => i != null).ToArray();
return result;
}
///
/// Gets the item fields.
///
/// The request.
/// IEnumerable{ItemFields}.
private IEnumerable GetItemFields(GetItemsByName request)
{
var val = request.Fields;
if (string.IsNullOrEmpty(val))
{
return new ItemFields[] { };
}
return val.Split(',').Select(v => (ItemFields)Enum.Parse(typeof(ItemFields), v, true));
}
///
/// Gets all items.
///
/// The request.
/// The items.
/// The user.
/// IEnumerable{Tuple{System.StringFunc{System.Int32}}}.
protected abstract IEnumerable>> GetAllItems(GetItemsByName request, IEnumerable items, User user);
///
/// Gets the entity.
///
/// The name.
/// Task{BaseItem}.
protected abstract Task GetEntity(string name);
///
/// Gets the dto.
///
/// The stub.
/// The user.
/// The fields.
/// Task{DtoBaseItem}.
private async Task GetDto(Tuple> stub, User user, List fields)
{
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).GetDtoBaseItem(item, user, fields, LibraryManager).ConfigureAwait(false);
dto.ChildCount = stub.Item2();
return dto;
}
}
///
/// Class GetItemsByName
///
public class GetItemsByName : IReturn
{
///
/// Gets or sets the user id.
///
/// The user id.
public Guid UserId { get; set; }
///
/// Gets or sets the start index.
///
/// The start index.
public int? StartIndex { get; set; }
///
/// Gets or sets the size of the page.
///
/// The size of the page.
public int? PageSize { get; set; }
///
/// Gets or sets a value indicating whether this is recursive.
///
/// true if recursive; otherwise, false.
public bool Recursive { get; set; }
///
/// Gets or sets the sort order.
///
/// The sort order.
public SortOrder? SortOrder { get; set; }
///
/// If specified the search will be localized within a specific item or folder
///
/// The item id.
public string Id { get; set; }
///
/// Fields to return within the items, in addition to basic information
///
/// The fields.
public string Fields { get; set; }
}
}