diff options
Diffstat (limited to 'MediaBrowser.Api/Music')
| -rw-r--r-- | MediaBrowser.Api/Music/AlbumsService.cs | 115 | ||||
| -rw-r--r-- | MediaBrowser.Api/Music/InstantMixService.cs | 143 |
2 files changed, 258 insertions, 0 deletions
diff --git a/MediaBrowser.Api/Music/AlbumsService.cs b/MediaBrowser.Api/Music/AlbumsService.cs new file mode 100644 index 000000000..a80dd796a --- /dev/null +++ b/MediaBrowser.Api/Music/AlbumsService.cs @@ -0,0 +1,115 @@ +using MediaBrowser.Controller.Dto; +using MediaBrowser.Controller.Entities; +using MediaBrowser.Controller.Entities.Audio; +using MediaBrowser.Controller.Library; +using MediaBrowser.Controller.Persistence; +using ServiceStack; +using System; +using System.Collections.Generic; +using System.Linq; + +namespace MediaBrowser.Api.Music +{ + [Route("/Albums/{Id}/Similar", "GET")] + [Api(Description = "Finds albums similar to a given album.")] + public class GetSimilarAlbums : BaseGetSimilarItemsFromItem + { + } + + public class AlbumsService : BaseApiService + { + /// <summary> + /// The _user manager + /// </summary> + private readonly IUserManager _userManager; + + /// <summary> + /// The _user data repository + /// </summary> + private readonly IUserDataManager _userDataRepository; + /// <summary> + /// The _library manager + /// </summary> + private readonly ILibraryManager _libraryManager; + private readonly IItemRepository _itemRepo; + private readonly IDtoService _dtoService; + + public AlbumsService(IUserManager userManager, IUserDataManager userDataRepository, ILibraryManager libraryManager, IItemRepository itemRepo, IDtoService dtoService) + { + _userManager = userManager; + _userDataRepository = userDataRepository; + _libraryManager = libraryManager; + _itemRepo = itemRepo; + _dtoService = dtoService; + } + + /// <summary> + /// Gets the specified request. + /// </summary> + /// <param name="request">The request.</param> + /// <returns>System.Object.</returns> + public object Get(GetSimilarAlbums request) + { + var result = SimilarItemsHelper.GetSimilarItemsResult(_userManager, + _itemRepo, + _libraryManager, + _userDataRepository, + _dtoService, + Logger, + request, item => item is MusicAlbum, + GetAlbumSimilarityScore); + + return ToOptimizedSerializedResultUsingCache(result); + } + + /// <summary> + /// Gets the album similarity score. + /// </summary> + /// <param name="item1">The item1.</param> + /// <param name="item2">The item2.</param> + /// <returns>System.Int32.</returns> + private int GetAlbumSimilarityScore(BaseItem item1, BaseItem item2) + { + var points = SimilarItemsHelper.GetSimiliarityScore(item1, item2); + + var album1 = (MusicAlbum)item1; + var album2 = (MusicAlbum)item2; + + var artists1 = album1.GetRecursiveChildren() + .OfType<Audio>() + .SelectMany(i => + { + var list = new List<string>(); + + if (!string.IsNullOrEmpty(i.AlbumArtist)) + { + list.Add(i.AlbumArtist); + } + list.AddRange(i.Artists); + + return list; + }) + .Distinct(StringComparer.OrdinalIgnoreCase) + .ToList(); + + var artists2 = album2.GetRecursiveChildren() + .OfType<Audio>() + .SelectMany(i => + { + var list = new List<string>(); + + if (!string.IsNullOrEmpty(i.AlbumArtist)) + { + list.Add(i.AlbumArtist); + } + list.AddRange(i.Artists); + + return list; + }) + .Distinct(StringComparer.OrdinalIgnoreCase) + .ToDictionary(i => i, StringComparer.OrdinalIgnoreCase); + + return points + artists1.Where(artists2.ContainsKey).Sum(i => 5); + } + } +} diff --git a/MediaBrowser.Api/Music/InstantMixService.cs b/MediaBrowser.Api/Music/InstantMixService.cs new file mode 100644 index 000000000..a8446a7ef --- /dev/null +++ b/MediaBrowser.Api/Music/InstantMixService.cs @@ -0,0 +1,143 @@ +using MediaBrowser.Controller.Dto; +using MediaBrowser.Controller.Entities.Audio; +using MediaBrowser.Controller.Library; +using MediaBrowser.Model.Querying; +using ServiceStack; +using System; +using System.Collections.Generic; +using System.Linq; + +namespace MediaBrowser.Api.Music +{ + [Route("/Songs/{Id}/InstantMix", "GET")] + [Api(Description = "Creates an instant playlist based on a given song")] + public class GetInstantMixFromSong : BaseGetSimilarItemsFromItem + { + } + + [Route("/Albums/{Id}/InstantMix", "GET")] + [Api(Description = "Creates an instant playlist based on a given album")] + public class GetInstantMixFromAlbum : BaseGetSimilarItemsFromItem + { + } + + [Route("/Artists/{Name}/InstantMix", "GET")] + [Api(Description = "Creates an instant playlist based on a given artist")] + public class GetInstantMixFromArtist : BaseGetSimilarItems + { + [ApiMember(Name = "Name", Description = "The artist name", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")] + public string Name { get; set; } + } + + [Route("/MusicGenres/{Name}/InstantMix", "GET")] + [Api(Description = "Creates an instant playlist based on a music genre")] + public class GetInstantMixFromMusicGenre : BaseGetSimilarItems + { + [ApiMember(Name = "Name", Description = "The genre name", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")] + public string Name { get; set; } + } + + public class InstantMixService : BaseApiService + { + private readonly IUserManager _userManager; + private readonly ILibraryManager _libraryManager; + + private readonly IDtoService _dtoService; + + public InstantMixService(IUserManager userManager, ILibraryManager libraryManager, IDtoService dtoService) + { + _userManager = userManager; + _libraryManager = libraryManager; + _dtoService = dtoService; + } + + public object Get(GetInstantMixFromSong request) + { + var item = _dtoService.GetItemByDtoId(request.Id); + + var result = GetInstantMixResult(request, item.Genres); + + return ToOptimizedSerializedResultUsingCache(result); + } + + public object Get(GetInstantMixFromAlbum request) + { + var album = (MusicAlbum)_dtoService.GetItemByDtoId(request.Id); + + var genres = album + .RecursiveChildren + .OfType<Audio>() + .SelectMany(i => i.Genres) + .Concat(album.Genres) + .Distinct(StringComparer.OrdinalIgnoreCase); + + var result = GetInstantMixResult(request, genres); + + return ToOptimizedSerializedResultUsingCache(result); + } + + public object Get(GetInstantMixFromMusicGenre request) + { + var genre = GetMusicGenre(request.Name, _libraryManager); + + var result = GetInstantMixResult(request, new[] { genre.Name }); + + return ToOptimizedSerializedResultUsingCache(result); + } + + public object Get(GetInstantMixFromArtist request) + { + var artist = GetArtist(request.Name, _libraryManager); + + var genres = _libraryManager.RootFolder + .RecursiveChildren + .OfType<Audio>() + .Where(i => i.HasArtist(artist.Name)) + .SelectMany(i => i.Genres) + .Concat(artist.Genres) + .Distinct(StringComparer.OrdinalIgnoreCase); + + var result = GetInstantMixResult(request, genres); + + return ToOptimizedSerializedResultUsingCache(result); + } + + private ItemsResult GetInstantMixResult(BaseGetSimilarItems request, IEnumerable<string> genres) + { + var user = request.UserId.HasValue ? _userManager.GetUserById(request.UserId.Value) : null; + + var fields = request.GetItemFields().ToList(); + + var inputItems = user == null + ? _libraryManager.RootFolder.RecursiveChildren + : user.RootFolder.GetRecursiveChildren(user); + + var genresDictionary = genres.ToDictionary(i => i, StringComparer.OrdinalIgnoreCase); + + var limit = request.Limit.HasValue ? request.Limit.Value * 2 : 100; + + var items = inputItems + .OfType<Audio>() + .Select(i => new Tuple<Audio, int>(i, i.Genres.Count(genresDictionary.ContainsKey))) + .OrderByDescending(i => i.Item2) + .ThenBy(i => Guid.NewGuid()) + .Select(i => i.Item1) + .Take(limit) + .OrderBy(i => Guid.NewGuid()) + .ToList(); + + var result = new ItemsResult + { + TotalRecordCount = items.Count + }; + + var dtos = items.Take(request.Limit ?? items.Count) + .Select(i => _dtoService.GetBaseItemDto(i, fields, user)); + + result.Items = dtos.ToArray(); + + return result; + } + + } +} |
