diff --git a/MediaBrowser.Api/LibraryService.cs b/MediaBrowser.Api/LibraryService.cs index b83d71a39..87bf08e63 100644 --- a/MediaBrowser.Api/LibraryService.cs +++ b/MediaBrowser.Api/LibraryService.cs @@ -6,6 +6,7 @@ using MediaBrowser.Controller.Entities.TV; using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Persistence; using MediaBrowser.Model.Dto; +using MediaBrowser.Model.Entities; using MediaBrowser.Model.Querying; using ServiceStack.ServiceHost; using System; @@ -103,6 +104,32 @@ namespace MediaBrowser.Api public Guid? UserId { get; set; } } + [Route("/Items/{Id}/Similar", "GET")] + [Api(Description = "Gets items similar to a given input item.")] + public class GetSimilarItems : IReturn + { + /// + /// Gets or sets the user id. + /// + /// The user id. + [ApiMember(Name = "UserId", Description = "Optional. Filter by user id, and attach user data", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")] + public Guid? UserId { get; set; } + + /// + /// Gets or sets the id. + /// + /// The id. + [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")] + public string Id { get; set; } + + /// + /// The maximum number of items to return + /// + /// The limit. + [ApiMember(Name = "Limit", Description = "Optional. The maximum number of records to return", IsRequired = false, DataType = "int", ParameterType = "query", Verb = "GET")] + public int? Limit { get; set; } + } + /// /// Class LibraryService /// @@ -227,7 +254,7 @@ namespace MediaBrowser.Api var user = request.UserId.HasValue ? _userManager.GetUserById(request.UserId.Value) : null; var item = string.IsNullOrEmpty(request.Id) ? - (request.UserId.HasValue ? user.RootFolder : + (request.UserId.HasValue ? user.RootFolder : (Folder)_libraryManager.RootFolder) : DtoBuilder.GetItemByClientId(request.Id, _userManager, _libraryManager, request.UserId); // Get everything @@ -276,5 +303,183 @@ namespace MediaBrowser.Api return ToOptimizedResult(result); } + + /// + /// Gets the specified request. + /// + /// The request. + /// System.Object. + public object Get(GetSimilarItems request) + { + var user = request.UserId.HasValue ? _userManager.GetUserById(request.UserId.Value) : null; + + var item = string.IsNullOrEmpty(request.Id) ? + (request.UserId.HasValue ? user.RootFolder : + (Folder)_libraryManager.RootFolder) : DtoBuilder.GetItemByClientId(request.Id, _userManager, _libraryManager, request.UserId); + + // Get everything + var fields = Enum.GetNames(typeof(ItemFields)).Select(i => (ItemFields)Enum.Parse(typeof(ItemFields), i, true)).ToList(); + + var dtoBuilder = new DtoBuilder(Logger, _libraryManager, _userDataRepository); + + var inputItems = user == null + ? _libraryManager.RootFolder.RecursiveChildren + : user.RootFolder.GetRecursiveChildren(user); + + var items = GetSimilaritems(item, inputItems).ToArray(); + + var result = new ItemsResult + { + Items = items.Take(request.Limit ?? items.Length).Select(i => dtoBuilder.GetBaseItemDto(i, fields, user)).Select(t => t.Result).ToArray(), + + TotalRecordCount = items.Length + }; + + return ToOptimizedResult(result); + } + + /// + /// Gets the similiar items. + /// + /// The item. + /// The input items. + /// IEnumerable{BaseItem}. + private IEnumerable GetSimilaritems(BaseItem item, IEnumerable inputItems) + { + if (item is Movie || item is Trailer) + { + inputItems = inputItems.Where(i => i is Movie || i is Trailer); + } + else if (item is Series) + { + inputItems = inputItems.Where(i => i is Series); + } + else if (item is BaseGame) + { + inputItems = inputItems.Where(i => i is BaseGame); + } + else if (item is MusicAlbum) + { + inputItems = inputItems.Where(i => i is MusicAlbum); + } + else if (item is Audio) + { + inputItems = inputItems.Where(i => i is Audio); + } + + // Avoid implicitly captured closure + var currentItem = item; + + return inputItems.Where(i => i.Id != currentItem.Id) + .Select(i => new Tuple(i, GetSimiliarityScore(item, i))) + .Where(i => i.Item2 > 0) + .OrderByDescending(i => i.Item2) + .ThenByDescending(i => i.Item1.CriticRating ?? 0) + .Select(i => i.Item1); + } + + /// + /// Gets the similiarity score. + /// + /// The item1. + /// The item2. + /// System.Int32. + private int GetSimiliarityScore(BaseItem item1, BaseItem item2) + { + var points = 0; + + if (!string.IsNullOrEmpty(item1.OfficialRating) && string.Equals(item1.OfficialRating, item2.OfficialRating, StringComparison.OrdinalIgnoreCase)) + { + points += 1; + } + + // Find common genres + points += item1.Genres.Where(i => item2.Genres.Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 5); + + // Find common tags + points += item1.Tags.Where(i => item2.Tags.Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 5); + + // Find common studios + points += item1.Studios.Where(i => item2.Studios.Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 3); + + var item2PeopleNames = item2.People.Select(i => i.Name).ToList(); + + points += item1.People.Where(i => item2PeopleNames.Contains(i.Name, StringComparer.OrdinalIgnoreCase)).Sum(i => + { + if (string.Equals(i.Name, PersonType.Director, StringComparison.OrdinalIgnoreCase)) + { + return 5; + } + if (string.Equals(i.Name, PersonType.Actor, StringComparison.OrdinalIgnoreCase)) + { + return 3; + } + if (string.Equals(i.Name, PersonType.Composer, StringComparison.OrdinalIgnoreCase)) + { + return 3; + } + if (string.Equals(i.Name, PersonType.GuestStar, StringComparison.OrdinalIgnoreCase)) + { + return 3; + } + if (string.Equals(i.Name, PersonType.Writer, StringComparison.OrdinalIgnoreCase)) + { + return 2; + } + + return 1; + }); + + if (item1.ProductionYear.HasValue && item2.ProductionYear.HasValue) + { + var diff = Math.Abs(item1.ProductionYear.Value - item2.ProductionYear.Value); + + // Add if they came out within the same decade + if (diff < 10) + { + points += 3; + } + + // And more if within five years + if (diff < 5) + { + points += 3; + } + } + + var album = item1 as MusicAlbum; + + if (album != null) + { + points += GetAlbumSimilarityScore(album, (MusicAlbum)item2); + } + + return points; + } + + /// + /// Gets the album similarity score. + /// + /// The item1. + /// The item2. + /// System.Int32. + private int GetAlbumSimilarityScore(MusicAlbum item1, MusicAlbum item2) + { + var artists1 = item1.RecursiveChildren + .OfType