diff options
Diffstat (limited to 'MediaBrowser.Api/LibraryService.cs')
| -rw-r--r-- | MediaBrowser.Api/LibraryService.cs | 209 |
1 files changed, 5 insertions, 204 deletions
diff --git a/MediaBrowser.Api/LibraryService.cs b/MediaBrowser.Api/LibraryService.cs index e3b93c340..681bbd851 100644 --- a/MediaBrowser.Api/LibraryService.cs +++ b/MediaBrowser.Api/LibraryService.cs @@ -127,32 +127,6 @@ 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<ItemsResult> - { - /// <summary> - /// Gets or sets the user id. - /// </summary> - /// <value>The user id.</value> - [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; } - - /// <summary> - /// Gets or sets the id. - /// </summary> - /// <value>The id.</value> - [ApiMember(Name = "Id", Description = "Item Id", IsRequired = true, DataType = "string", ParameterType = "path", Verb = "GET")] - public string Id { get; set; } - - /// <summary> - /// The maximum number of items to return - /// </summary> - /// <value>The limit.</value> - [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; } - } - /// <summary> /// Class LibraryService /// </summary> @@ -420,6 +394,11 @@ namespace MediaBrowser.Api Task.WaitAll(task); } + /// <summary> + /// Refreshes the item. + /// </summary> + /// <param name="request">The request.</param> + /// <returns>Task.</returns> private async Task RefreshItem(RefreshItem request) { var item = DtoBuilder.GetItemByClientId(request.Id, _userManager, _libraryManager); @@ -441,183 +420,5 @@ namespace MediaBrowser.Api Logger.ErrorException("Error refreshing library", ex); } } - - /// <summary> - /// Gets the specified request. - /// </summary> - /// <param name="request">The request.</param> - /// <returns>System.Object.</returns> - 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); - } - - /// <summary> - /// Gets the similiar items. - /// </summary> - /// <param name="item">The item.</param> - /// <param name="inputItems">The input items.</param> - /// <returns>IEnumerable{BaseItem}.</returns> - private IEnumerable<BaseItem> GetSimilaritems(BaseItem item, IEnumerable<BaseItem> 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<BaseItem, int>(i, GetSimiliarityScore(item, i))) - .Where(i => i.Item2 > 0) - .OrderByDescending(i => i.Item2) - .ThenByDescending(i => i.Item1.CriticRating ?? 0) - .Select(i => i.Item1); - } - - /// <summary> - /// Gets the similiarity score. - /// </summary> - /// <param name="item1">The item1.</param> - /// <param name="item2">The item2.</param> - /// <returns>System.Int32.</returns> - 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; - } - - /// <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(MusicAlbum item1, MusicAlbum item2) - { - var artists1 = item1.RecursiveChildren - .OfType<Audio>() - .SelectMany(i => new[] { i.AlbumArtist, i.Artist }) - .Where(i => !string.IsNullOrEmpty(i)) - .Distinct(StringComparer.OrdinalIgnoreCase) - .ToList(); - - var artists2 = item2.RecursiveChildren - .OfType<Audio>() - .SelectMany(i => new[] { i.AlbumArtist, i.Artist }) - .Where(i => !string.IsNullOrEmpty(i)) - .Distinct(StringComparer.OrdinalIgnoreCase) - .ToList(); - - return artists1.Where(i => artists2.Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 5); - } } } |