MusicManager.cs 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156
  1. using MediaBrowser.Controller.Entities;
  2. using MediaBrowser.Controller.Entities.Audio;
  3. using MediaBrowser.Controller.Library;
  4. using MediaBrowser.Controller.Playlists;
  5. using System;
  6. using System.Collections.Generic;
  7. using System.Linq;
  8. namespace Emby.Server.Implementations.Library
  9. {
  10. public class MusicManager : IMusicManager
  11. {
  12. private readonly ILibraryManager _libraryManager;
  13. public MusicManager(ILibraryManager libraryManager)
  14. {
  15. _libraryManager = libraryManager;
  16. }
  17. public IEnumerable<Audio> GetInstantMixFromSong(Audio item, User user)
  18. {
  19. var list = new List<Audio>
  20. {
  21. item
  22. };
  23. return list.Concat(GetInstantMixFromGenres(item.Genres, user));
  24. }
  25. public IEnumerable<Audio> GetInstantMixFromArtist(MusicArtist artist, User user)
  26. {
  27. var genres = user.RootFolder
  28. .GetRecursiveChildren(user, new InternalItemsQuery(user)
  29. {
  30. IncludeItemTypes = new[] { typeof(Audio).Name }
  31. })
  32. .Cast<Audio>()
  33. .Where(i => i.HasAnyArtist(artist.Name))
  34. .SelectMany(i => i.Genres)
  35. .Concat(artist.Genres)
  36. .Distinct(StringComparer.OrdinalIgnoreCase);
  37. return GetInstantMixFromGenres(genres, user);
  38. }
  39. public IEnumerable<Audio> GetInstantMixFromAlbum(MusicAlbum item, User user)
  40. {
  41. var genres = item
  42. .GetRecursiveChildren(user, new InternalItemsQuery(user)
  43. {
  44. IncludeItemTypes = new[] { typeof(Audio).Name }
  45. })
  46. .Cast<Audio>()
  47. .SelectMany(i => i.Genres)
  48. .Concat(item.Genres)
  49. .DistinctNames();
  50. return GetInstantMixFromGenres(genres, user);
  51. }
  52. public IEnumerable<Audio> GetInstantMixFromFolder(Folder item, User user)
  53. {
  54. var genres = item
  55. .GetRecursiveChildren(user, new InternalItemsQuery(user)
  56. {
  57. IncludeItemTypes = new[] {typeof(Audio).Name}
  58. })
  59. .Cast<Audio>()
  60. .SelectMany(i => i.Genres)
  61. .Concat(item.Genres)
  62. .DistinctNames();
  63. return GetInstantMixFromGenres(genres, user);
  64. }
  65. public IEnumerable<Audio> GetInstantMixFromPlaylist(Playlist item, User user)
  66. {
  67. var genres = item
  68. .GetRecursiveChildren(user, new InternalItemsQuery(user)
  69. {
  70. IncludeItemTypes = new[] { typeof(Audio).Name }
  71. })
  72. .Cast<Audio>()
  73. .SelectMany(i => i.Genres)
  74. .Concat(item.Genres)
  75. .DistinctNames();
  76. return GetInstantMixFromGenres(genres, user);
  77. }
  78. public IEnumerable<Audio> GetInstantMixFromGenres(IEnumerable<string> genres, User user)
  79. {
  80. var genreList = genres.ToList();
  81. var inputItems = _libraryManager.GetItemList(new InternalItemsQuery(user)
  82. {
  83. IncludeItemTypes = new[] { typeof(Audio).Name },
  84. Genres = genreList.ToArray()
  85. });
  86. var genresDictionary = genreList.ToDictionary(i => i, StringComparer.OrdinalIgnoreCase);
  87. return inputItems
  88. .Cast<Audio>()
  89. .Select(i => new Tuple<Audio, int>(i, i.Genres.Count(genresDictionary.ContainsKey)))
  90. .OrderByDescending(i => i.Item2)
  91. .ThenBy(i => Guid.NewGuid())
  92. .Select(i => i.Item1)
  93. .Take(200)
  94. .OrderBy(i => Guid.NewGuid());
  95. }
  96. public IEnumerable<Audio> GetInstantMixFromItem(BaseItem item, User user)
  97. {
  98. var genre = item as MusicGenre;
  99. if (genre != null)
  100. {
  101. return GetInstantMixFromGenres(new[] { item.Name }, user);
  102. }
  103. var playlist = item as Playlist;
  104. if (playlist != null)
  105. {
  106. return GetInstantMixFromPlaylist(playlist, user);
  107. }
  108. var album = item as MusicAlbum;
  109. if (album != null)
  110. {
  111. return GetInstantMixFromAlbum(album, user);
  112. }
  113. var artist = item as MusicArtist;
  114. if (artist != null)
  115. {
  116. return GetInstantMixFromArtist(artist, user);
  117. }
  118. var song = item as Audio;
  119. if (song != null)
  120. {
  121. return GetInstantMixFromSong(song, user);
  122. }
  123. var folder = item as Folder;
  124. if (folder != null)
  125. {
  126. return GetInstantMixFromFolder(folder, user);
  127. }
  128. return new Audio[] { };
  129. }
  130. }
  131. }