jellyfin/MediaBrowser.Api/UserLibrary/PersonsService.cs

234 lines
8.5 KiB
C#
Raw Normal View History

2013-04-22 04:38:03 +00:00
using MediaBrowser.Controller.Dto;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Audio;
2013-04-17 05:36:56 +00:00
using MediaBrowser.Controller.Entities.Movies;
using MediaBrowser.Controller.Entities.TV;
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Persistence;
2013-04-17 05:36:56 +00:00
using MediaBrowser.Model.Dto;
using MediaBrowser.Model.Entities;
2013-04-22 04:38:03 +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.Linq;
using System.Threading.Tasks;
namespace MediaBrowser.Api.UserLibrary
{
/// <summary>
/// Class GetPersons
/// </summary>
2013-04-22 04:38:03 +00:00
[Route("/Persons", "GET")]
2013-04-10 15:32:09 +00:00
[Api(Description = "Gets all persons from a given item, folder, or the entire library")]
2013-02-21 01:33:05 +00:00
public class GetPersons : GetItemsByName
{
/// <summary>
/// Gets or sets the person types.
/// </summary>
/// <value>The person types.</value>
public string PersonTypes { get; set; }
}
2013-04-17 05:36:56 +00:00
/// <summary>
/// Class GetPersonItemCounts
/// </summary>
2013-04-22 04:38:03 +00:00
[Route("/Persons/{Name}/Counts", "GET")]
2013-04-17 05:36:56 +00:00
[Api(Description = "Gets item counts of library items that a person appears in")]
public class GetPersonItemCounts : IReturn<ItemByNameCounts>
{
/// <summary>
/// Gets or sets the user id.
/// </summary>
/// <value>The user id.</value>
[ApiMember(Name = "UserId", Description = "User Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
2013-04-17 05:36:56 +00:00
public Guid UserId { get; set; }
/// <summary>
/// Gets or sets the name.
/// </summary>
/// <value>The name.</value>
[ApiMember(Name = "Name", Description = "The person name", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
2013-04-17 05:36:56 +00:00
public string Name { get; set; }
}
2013-04-22 04:38:03 +00:00
/// <summary>
/// Class GetPerson
/// </summary>
[Route("/Persons/{Name}", "GET")]
[Api(Description = "Gets a person, by name")]
public class GetPerson : IReturn<BaseItemDto>
{
/// <summary>
/// Gets or sets the name.
/// </summary>
/// <value>The name.</value>
[ApiMember(Name = "Name", Description = "The person name", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
public string Name { get; set; }
/// <summary>
/// Gets or sets the user id.
/// </summary>
/// <value>The user id.</value>
[ApiMember(Name = "UserId", Description = "Optional. Filter by user id, and attach user data", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")]
public Guid? UserId { get; set; }
}
2013-02-21 01:33:05 +00:00
/// <summary>
/// Class PersonsService
/// </summary>
public class PersonsService : BaseItemsByNameService<Person>
{
2013-04-17 05:36:56 +00:00
/// <summary>
/// Initializes a new instance of the <see cref="PersonsService"/> class.
/// </summary>
/// <param name="userManager">The user manager.</param>
/// <param name="libraryManager">The library manager.</param>
/// <param name="userDataRepository">The user data repository.</param>
public PersonsService(IUserManager userManager, ILibraryManager libraryManager, IUserDataRepository userDataRepository)
: base(userManager, libraryManager, userDataRepository)
{
}
2013-04-22 04:38:03 +00:00
/// <summary>
/// Gets the specified request.
/// </summary>
/// <param name="request">The request.</param>
/// <returns>System.Object.</returns>
public object Get(GetPerson request)
{
var result = GetItem(request).Result;
return ToOptimizedResult(result);
}
/// <summary>
/// Gets the item.
/// </summary>
/// <param name="request">The request.</param>
/// <returns>Task{BaseItemDto}.</returns>
private async Task<BaseItemDto> GetItem(GetPerson request)
{
var item = await GetPerson(request.Name, LibraryManager).ConfigureAwait(false);
2013-04-22 04:38:03 +00:00
// Get everything
var fields = Enum.GetNames(typeof(ItemFields)).Select(i => (ItemFields)Enum.Parse(typeof(ItemFields), i, true));
var builder = new DtoBuilder(Logger, LibraryManager, UserDataRepository);
if (request.UserId.HasValue)
{
var user = UserManager.GetUserById(request.UserId.Value);
2013-05-15 16:56:38 +00:00
return await builder.GetBaseItemDto(item, fields.ToList(), user).ConfigureAwait(false);
2013-04-22 04:38:03 +00:00
}
return await builder.GetBaseItemDto(item, fields.ToList()).ConfigureAwait(false);
}
2013-02-21 01:33:05 +00:00
/// <summary>
/// Gets the specified request.
/// </summary>
/// <param name="request">The request.</param>
/// <returns>System.Object.</returns>
public object Get(GetPersons request)
{
var result = GetResult(request).Result;
return ToOptimizedResult(result);
}
2013-04-17 05:36:56 +00:00
/// <summary>
/// Gets the specified request.
/// </summary>
/// <param name="request">The request.</param>
/// <returns>System.Object.</returns>
public object Get(GetPersonItemCounts request)
{
var name = DeSlugPersonName(request.Name, LibraryManager);
var items = GetItems(request.UserId).Where(i => i.People != null && i.People.Any(p => string.Equals(p.Name, name, StringComparison.OrdinalIgnoreCase))).ToList();
2013-04-17 05:36:56 +00:00
var counts = new ItemByNameCounts
{
TotalCount = items.Count,
TrailerCount = items.OfType<Trailer>().Count(),
2013-04-17 05:36:56 +00:00
MovieCount = items.OfType<Movie>().Count(),
SeriesCount = items.OfType<Series>().Count(),
GameCount = items.OfType<BaseGame>().Count(),
2013-04-20 00:44:35 +00:00
SongCount = items.OfType<Audio>().Count(),
AlbumCount = items.OfType<MusicAlbum>().Count(),
2013-05-04 04:38:38 +00:00
EpisodeCount = items.OfType<Episode>().Count()
2013-04-17 05:36:56 +00:00
};
return ToOptimizedResult(counts);
}
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{Tuple{System.StringFunc{System.Int32}}}.</returns>
2013-04-22 04:38:03 +00:00
protected override IEnumerable<IbnStub<Person>> GetAllItems(GetItemsByName request, IEnumerable<BaseItem> items)
2013-02-21 01:33:05 +00:00
{
2013-04-17 05:36:56 +00:00
var inputPersonTypes = ((GetPersons)request).PersonTypes;
2013-02-21 01:33:05 +00:00
var personTypes = string.IsNullOrEmpty(inputPersonTypes) ? new string[] { } : inputPersonTypes.Split(',');
var itemsList = items.Where(i => i.People != null).ToList();
// Either get all people, or all people filtered by a specific person type
var allPeople = GetAllPeople(itemsList, personTypes);
return allPeople
.Select(i => i.Name)
.Distinct(StringComparer.OrdinalIgnoreCase)
.Select(name => new IbnStub<Person>(name, () =>
2013-02-21 01:33:05 +00:00
{
if (personTypes.Length == 0)
{
return itemsList.Where(i => i.People.Any(p => p.Name.Equals(name, StringComparison.OrdinalIgnoreCase)));
2013-02-21 01:33:05 +00:00
}
return itemsList.Where(i => i.People.Any(p => p.Name.Equals(name, StringComparison.OrdinalIgnoreCase) && personTypes.Contains(p.Type ?? string.Empty, StringComparer.OrdinalIgnoreCase)));
}, GetEntity)
2013-02-21 01:33:05 +00:00
);
}
/// <summary>
/// Gets all people.
/// </summary>
/// <param name="itemsList">The items list.</param>
/// <param name="personTypes">The person types.</param>
/// <returns>IEnumerable{PersonInfo}.</returns>
private IEnumerable<PersonInfo> GetAllPeople(IEnumerable<BaseItem> itemsList, string[] personTypes)
{
var people = itemsList.SelectMany(i => i.People.OrderBy(p => p.Type));
2013-02-21 01:33:05 +00:00
return personTypes.Length == 0 ?
people :
people.Where(p => personTypes.Contains(p.Type ?? string.Empty, StringComparer.OrdinalIgnoreCase));
2013-02-21 01:33:05 +00:00
}
/// <summary>
/// Gets the entity.
/// </summary>
/// <param name="name">The name.</param>
/// <returns>Task{Genre}.</returns>
protected Task<Person> GetEntity(string name)
2013-02-21 01:33:05 +00:00
{
return LibraryManager.GetPerson(name);
2013-02-21 01:33:05 +00:00
}
}
}