@ -1,10 +1,16 @@
using System.Threading ;
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 ;
using System.Threading.Tasks ;
namespace MediaBrowser.Api
@ -43,7 +49,15 @@ namespace MediaBrowser.Api
public class RefreshLibrary : IReturnVoid
{
}
[Route("/Items/Counts", "GET")]
[Api(Description = "Gets counts of various item types")]
public class GetItemCounts : IReturn < ItemCounts >
{
[ApiMember(Name = "UserId", Description = "Optional. Get counts from a specific user's library.", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")]
public Guid ? UserId { get ; set ; }
}
/// <summary>
/// Class LibraryService
/// </summary>
@ -55,16 +69,19 @@ namespace MediaBrowser.Api
private readonly IItemRepository _itemRepo ;
private readonly ILibraryManager _libraryManager ;
private readonly IUserManager _userManager ;
/// <summary>
/// Initializes a new instance of the <see cref="LibraryService" /> class.
/// </summary>
/// <param name="itemRepo">The item repo.</param>
/// <param name="libraryManager">The library manager.</param>
public LibraryService ( IItemRepository itemRepo , ILibraryManager libraryManager )
/// <param name="userManager">The user manager.</param>
public LibraryService ( IItemRepository itemRepo , ILibraryManager libraryManager , IUserManager userManager )
{
_itemRepo = itemRepo ;
_libraryManager = libraryManager ;
_userManager = userManager ;
}
/// <summary>
@ -79,6 +96,41 @@ namespace MediaBrowser.Api
return ToOptimizedResult ( result ) ;
}
/// <summary>
/// Gets the specified request.
/// </summary>
/// <param name="request">The request.</param>
/// <returns>System.Object.</returns>
public object Get ( GetItemCounts request )
{
var items = GetItems ( request . UserId ) . ToList ( ) ;
var counts = new ItemCounts
{
AlbumCount = items . OfType < MusicAlbum > ( ) . Count ( ) ,
EpisodeCount = items . OfType < Episode > ( ) . Count ( ) ,
GameCount = items . OfType < BaseGame > ( ) . Count ( ) ,
MovieCount = items . OfType < Movie > ( ) . Count ( ) ,
SeriesCount = items . OfType < Series > ( ) . Count ( ) ,
SongCount = items . OfType < Audio > ( ) . Count ( ) ,
TrailerCount = items . OfType < Trailer > ( ) . Count ( )
} ;
return ToOptimizedResult ( counts ) ;
}
protected IEnumerable < BaseItem > GetItems ( Guid ? userId )
{
if ( userId . HasValue )
{
var user = _userManager . GetUserById ( userId . Value ) ;
return _userManager . GetUserById ( userId . Value ) . RootFolder . GetRecursiveChildren ( user ) ;
}
return _libraryManager . RootFolder . RecursiveChildren ;
}
/// <summary>
/// Posts the specified request.
/// </summary>