AlbumsService.cs 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118
  1. using MediaBrowser.Controller.Dto;
  2. using MediaBrowser.Controller.Entities;
  3. using MediaBrowser.Controller.Entities.Audio;
  4. using MediaBrowser.Controller.Library;
  5. using MediaBrowser.Controller.Net;
  6. using MediaBrowser.Controller.Persistence;
  7. using ServiceStack;
  8. using System;
  9. using System.Collections.Generic;
  10. using System.Linq;
  11. using System.Threading.Tasks;
  12. namespace MediaBrowser.Api.Music
  13. {
  14. [Route("/Albums/{Id}/Similar", "GET", Summary = "Finds albums similar to a given album.")]
  15. public class GetSimilarAlbums : BaseGetSimilarItemsFromItem
  16. {
  17. }
  18. [Route("/Artists/{Id}/Similar", "GET", Summary = "Finds albums similar to a given album.")]
  19. public class GetSimilarArtists : BaseGetSimilarItemsFromItem
  20. {
  21. }
  22. [Authenticated]
  23. public class AlbumsService : BaseApiService
  24. {
  25. /// <summary>
  26. /// The _user manager
  27. /// </summary>
  28. private readonly IUserManager _userManager;
  29. /// <summary>
  30. /// The _user data repository
  31. /// </summary>
  32. private readonly IUserDataManager _userDataRepository;
  33. /// <summary>
  34. /// The _library manager
  35. /// </summary>
  36. private readonly ILibraryManager _libraryManager;
  37. private readonly IItemRepository _itemRepo;
  38. private readonly IDtoService _dtoService;
  39. public AlbumsService(IUserManager userManager, IUserDataManager userDataRepository, ILibraryManager libraryManager, IItemRepository itemRepo, IDtoService dtoService)
  40. {
  41. _userManager = userManager;
  42. _userDataRepository = userDataRepository;
  43. _libraryManager = libraryManager;
  44. _itemRepo = itemRepo;
  45. _dtoService = dtoService;
  46. }
  47. public async Task<object> Get(GetSimilarArtists request)
  48. {
  49. var dtoOptions = GetDtoOptions(request);
  50. var result = await SimilarItemsHelper.GetSimilarItemsResult(dtoOptions, _userManager,
  51. _itemRepo,
  52. _libraryManager,
  53. _userDataRepository,
  54. _dtoService,
  55. Logger,
  56. request, new[] { typeof(MusicArtist) },
  57. SimilarItemsHelper.GetSimiliarityScore).ConfigureAwait(false);
  58. return ToOptimizedSerializedResultUsingCache(result);
  59. }
  60. /// <summary>
  61. /// Gets the specified request.
  62. /// </summary>
  63. /// <param name="request">The request.</param>
  64. /// <returns>System.Object.</returns>
  65. public async Task<object> Get(GetSimilarAlbums request)
  66. {
  67. var dtoOptions = GetDtoOptions(request);
  68. var result = await SimilarItemsHelper.GetSimilarItemsResult(dtoOptions, _userManager,
  69. _itemRepo,
  70. _libraryManager,
  71. _userDataRepository,
  72. _dtoService,
  73. Logger,
  74. request, new[] { typeof(MusicAlbum) },
  75. GetAlbumSimilarityScore).ConfigureAwait(false);
  76. return ToOptimizedSerializedResultUsingCache(result);
  77. }
  78. /// <summary>
  79. /// Gets the album similarity score.
  80. /// </summary>
  81. /// <param name="item1">The item1.</param>
  82. /// <param name="item1People">The item1 people.</param>
  83. /// <param name="allPeople">All people.</param>
  84. /// <param name="item2">The item2.</param>
  85. /// <returns>System.Int32.</returns>
  86. private int GetAlbumSimilarityScore(BaseItem item1, List<PersonInfo> item1People, List<PersonInfo> allPeople, BaseItem item2)
  87. {
  88. var points = SimilarItemsHelper.GetSimiliarityScore(item1, item1People, allPeople, item2);
  89. var album1 = (MusicAlbum)item1;
  90. var album2 = (MusicAlbum)item2;
  91. var artists1 = album1
  92. .AllArtists
  93. .DistinctNames()
  94. .ToList();
  95. var artists2 = album2
  96. .AllArtists
  97. .DistinctNames()
  98. .ToDictionary(i => i, StringComparer.OrdinalIgnoreCase);
  99. return points + artists1.Where(artists2.ContainsKey).Sum(i => 5);
  100. }
  101. }
  102. }