aboutsummaryrefslogtreecommitdiff
path: root/MediaBrowser.Api/Music
diff options
context:
space:
mode:
Diffstat (limited to 'MediaBrowser.Api/Music')
-rw-r--r--MediaBrowser.Api/Music/AlbumsService.cs115
-rw-r--r--MediaBrowser.Api/Music/InstantMixService.cs143
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;
+ }
+
+ }
+}