#247 - Web client - Hide views that have no content

MediaBrowser.Api/LibraryService.cs

@@ -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>