|
|
@@ -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<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>
|
|
|
@@ -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);
|
|
|
}
|
|
|
+
|
|
|
+ /// <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);
|
|
|
+ }
|
|
|
}
|
|
|
}
|
Eric Reed