2
0

GenresHandler.cs 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using System.Threading.Tasks;
  5. using MediaBrowser.Common.Net.Handlers;
  6. using MediaBrowser.Controller;
  7. using MediaBrowser.Model.DTO;
  8. using MediaBrowser.Model.Entities;
  9. namespace MediaBrowser.Api.HttpHandlers
  10. {
  11. public class GenresHandler : BaseSerializationHandler<IBNItem[]>
  12. {
  13. protected override Task<IBNItem[]> GetObjectToSerialize()
  14. {
  15. Folder parent = ApiService.GetItemById(QueryString["id"]) as Folder;
  16. Guid userId = Guid.Parse(QueryString["userid"]);
  17. User user = Kernel.Instance.Users.First(u => u.Id == userId);
  18. return GetAllGenres(parent, user);
  19. }
  20. /// <summary>
  21. /// Gets all genres from all recursive children of a folder
  22. /// The CategoryInfo class is used to keep track of the number of times each genres appears
  23. /// </summary>
  24. private async Task<IBNItem[]> GetAllGenres(Folder parent, User user)
  25. {
  26. Dictionary<string, int> data = new Dictionary<string, int>();
  27. // Get all the allowed recursive children
  28. IEnumerable<BaseItem> allItems = parent.GetParentalAllowedRecursiveChildren(user);
  29. foreach (var item in allItems)
  30. {
  31. // Add each genre from the item to the data dictionary
  32. // If the genre already exists, increment the count
  33. if (item.Genres == null)
  34. {
  35. continue;
  36. }
  37. foreach (string val in item.Genres)
  38. {
  39. if (!data.ContainsKey(val))
  40. {
  41. data.Add(val, 1);
  42. }
  43. else
  44. {
  45. data[val]++;
  46. }
  47. }
  48. }
  49. // Get the Genre objects
  50. Genre[] entities = await Task.WhenAll<Genre>(data.Keys.Select(key => { return Kernel.Instance.ItemController.GetGenre(key); })).ConfigureAwait(false);
  51. // Convert to an array of IBNItem
  52. IBNItem[] items = new IBNItem[entities.Length];
  53. for (int i = 0; i < entities.Length; i++)
  54. {
  55. Genre e = entities[i];
  56. items[i] = ApiService.GetIBNItem(e, data[e.Name]);
  57. }
  58. return items;
  59. }
  60. }
  61. }