mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 05:47:14 -04:00
188 lines
6.6 KiB
C#
188 lines
6.6 KiB
C#
using MediaBrowser.Controller.Dto;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Entities.Audio;
|
|
using MediaBrowser.Controller.Entities.Movies;
|
|
using MediaBrowser.Controller.Entities.TV;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.Persistence;
|
|
using MediaBrowser.Model.Dto;
|
|
using MediaBrowser.Model.Querying;
|
|
using ServiceStack.ServiceHost;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace MediaBrowser.Api.UserLibrary
|
|
{
|
|
/// <summary>
|
|
/// Class GetStudios
|
|
/// </summary>
|
|
[Route("/Studios", "GET")]
|
|
[Api(Description = "Gets all studios from a given item, folder, or the entire library")]
|
|
public class GetStudios : GetItemsByName
|
|
{
|
|
}
|
|
|
|
[Route("/Studios/{Name}/Counts", "GET")]
|
|
[Api(Description = "Gets item counts of library items that a studio appears in")]
|
|
public class GetStudioItemCounts : 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")]
|
|
public Guid UserId { get; set; }
|
|
|
|
/// <summary>
|
|
/// Gets or sets the name.
|
|
/// </summary>
|
|
/// <value>The name.</value>
|
|
[ApiMember(Name = "Name", Description = "The studio name", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")]
|
|
public string Name { get; set; }
|
|
}
|
|
|
|
/// <summary>
|
|
/// Class GetStudio
|
|
/// </summary>
|
|
[Route("/Studios/{Name}", "GET")]
|
|
[Api(Description = "Gets a studio, by name")]
|
|
public class GetStudio : IReturn<BaseItemDto>
|
|
{
|
|
/// <summary>
|
|
/// Gets or sets the name.
|
|
/// </summary>
|
|
/// <value>The name.</value>
|
|
[ApiMember(Name = "Name", Description = "The studio 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; }
|
|
}
|
|
|
|
/// <summary>
|
|
/// Class StudiosService
|
|
/// </summary>
|
|
public class StudiosService : BaseItemsByNameService<Studio>
|
|
{
|
|
public StudiosService(IUserManager userManager, ILibraryManager libraryManager, IUserDataRepository userDataRepository, IItemRepository itemRepo)
|
|
: base(userManager, libraryManager, userDataRepository, itemRepo)
|
|
{
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the specified request.
|
|
/// </summary>
|
|
/// <param name="request">The request.</param>
|
|
/// <returns>System.Object.</returns>
|
|
public object Get(GetStudio 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(GetStudio request)
|
|
{
|
|
var item = await GetStudio(request.Name, LibraryManager).ConfigureAwait(false);
|
|
|
|
// Get everything
|
|
var fields = Enum.GetNames(typeof(ItemFields)).Select(i => (ItemFields)Enum.Parse(typeof(ItemFields), i, true));
|
|
|
|
var builder = new DtoBuilder(Logger, LibraryManager, UserDataRepository, ItemRepository);
|
|
|
|
if (request.UserId.HasValue)
|
|
{
|
|
var user = UserManager.GetUserById(request.UserId.Value);
|
|
|
|
return await builder.GetBaseItemDto(item, fields.ToList(), user).ConfigureAwait(false);
|
|
}
|
|
|
|
return await builder.GetBaseItemDto(item, fields.ToList()).ConfigureAwait(false);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the specified request.
|
|
/// </summary>
|
|
/// <param name="request">The request.</param>
|
|
/// <returns>System.Object.</returns>
|
|
public object Get(GetStudioItemCounts request)
|
|
{
|
|
var name = DeSlugStudioName(request.Name, LibraryManager);
|
|
|
|
var items = GetItems(request.UserId).Where(i => i.Studios != null && i.Studios.Contains(name, StringComparer.OrdinalIgnoreCase)).ToList();
|
|
|
|
var counts = new ItemByNameCounts
|
|
{
|
|
TotalCount = items.Count,
|
|
|
|
TrailerCount = items.OfType<Trailer>().Count(),
|
|
|
|
MovieCount = items.OfType<Movie>().Count(),
|
|
|
|
SeriesCount = items.OfType<Series>().Count(),
|
|
|
|
GameCount = items.OfType<Game>().Count(),
|
|
|
|
SongCount = items.OfType<Audio>().Count(),
|
|
|
|
AlbumCount = items.OfType<MusicAlbum>().Count(),
|
|
|
|
MusicVideoCount = items.OfType<MusicVideo>().Count(),
|
|
|
|
AdultVideoCount = items.OfType<AdultVideo>().Count()
|
|
};
|
|
|
|
return ToOptimizedResult(counts);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the specified request.
|
|
/// </summary>
|
|
/// <param name="request">The request.</param>
|
|
/// <returns>System.Object.</returns>
|
|
public object Get(GetStudios request)
|
|
{
|
|
var result = GetResult(request).Result;
|
|
|
|
return ToOptimizedResult(result);
|
|
}
|
|
|
|
/// <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>
|
|
protected override IEnumerable<IbnStub<Studio>> GetAllItems(GetItemsByName request, IEnumerable<BaseItem> items)
|
|
{
|
|
var itemsList = items.Where(i => i.Studios != null).ToList();
|
|
|
|
return itemsList
|
|
.SelectMany(i => i.Studios)
|
|
.Distinct(StringComparer.OrdinalIgnoreCase)
|
|
.Select(name => new IbnStub<Studio>(name, () => itemsList.Where(i => i.Studios.Contains(name, StringComparer.OrdinalIgnoreCase)), GetEntity));
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the entity.
|
|
/// </summary>
|
|
/// <param name="name">The name.</param>
|
|
/// <returns>Task{Studio}.</returns>
|
|
protected Task<Studio> GetEntity(string name)
|
|
{
|
|
return LibraryManager.GetStudio(name);
|
|
}
|
|
}
|
|
}
|