|
@@ -1,9 +1,16 @@
|
|
|
-using MediaBrowser.Controller.Dto;
|
|
|
+using MediaBrowser.Common.Extensions;
|
|
|
+using MediaBrowser.Controller.Dto;
|
|
|
using MediaBrowser.Controller.Entities;
|
|
|
using MediaBrowser.Controller.Entities.Movies;
|
|
|
using MediaBrowser.Controller.Library;
|
|
|
using MediaBrowser.Controller.Persistence;
|
|
|
+using MediaBrowser.Model.Dto;
|
|
|
+using MediaBrowser.Model.Entities;
|
|
|
+using MediaBrowser.Model.Querying;
|
|
|
using ServiceStack;
|
|
|
+using System;
|
|
|
+using System.Collections.Generic;
|
|
|
+using System.Linq;
|
|
|
|
|
|
namespace MediaBrowser.Api.Movies
|
|
|
{
|
|
@@ -23,6 +30,32 @@ namespace MediaBrowser.Api.Movies
|
|
|
}
|
|
|
}
|
|
|
|
|
|
+ [Route("/Movies/Recommendations", "GET")]
|
|
|
+ [Api(Description = "Gets movie recommendations")]
|
|
|
+ public class GetMovieRecommendations : IReturn<RecommendationDto[]>, IHasItemFields
|
|
|
+ {
|
|
|
+ [ApiMember(Name = "CategoryLimit", Description = "The max number of categories to return", IsRequired = false, DataType = "int", ParameterType = "query", Verb = "GET")]
|
|
|
+ public int CategoryLimit { get; set; }
|
|
|
+
|
|
|
+ [ApiMember(Name = "ItemLimit", Description = "The max number of items to return per category", IsRequired = false, DataType = "int", ParameterType = "query", Verb = "GET")]
|
|
|
+ public int ItemLimit { get; set; }
|
|
|
+
|
|
|
+ /// <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; }
|
|
|
+
|
|
|
+ public GetMovieRecommendations()
|
|
|
+ {
|
|
|
+ CategoryLimit = 5;
|
|
|
+ ItemLimit = 8;
|
|
|
+ }
|
|
|
+
|
|
|
+ public string Fields { get; set; }
|
|
|
+ }
|
|
|
+
|
|
|
/// <summary>
|
|
|
/// Class MoviesService
|
|
|
/// </summary>
|
|
@@ -78,5 +111,214 @@ namespace MediaBrowser.Api.Movies
|
|
|
|
|
|
return ToOptimizedSerializedResultUsingCache(result);
|
|
|
}
|
|
|
+
|
|
|
+ public object Get(GetMovieRecommendations request)
|
|
|
+ {
|
|
|
+ var user = request.UserId.HasValue ? _userManager.GetUserById(request.UserId.Value) : null;
|
|
|
+
|
|
|
+ var folder = user.RootFolder;
|
|
|
+ var movies = folder.RecursiveChildren.OfType<Movie>().ToList();
|
|
|
+
|
|
|
+ var result = GetRecommendationCategories(user, movies, request.CategoryLimit, request.ItemLimit, request.GetItemFields().ToList());
|
|
|
+
|
|
|
+ return ToOptimizedResult(result);
|
|
|
+ }
|
|
|
+
|
|
|
+ private IEnumerable<RecommendationDto> GetRecommendationCategories(User user, List<Movie> allMovies, int categoryLimit, int itemLimit, List<ItemFields> fields)
|
|
|
+ {
|
|
|
+ var categories = new List<RecommendationDto>();
|
|
|
+
|
|
|
+ var recentlyPlayedMovies = allMovies
|
|
|
+ .Select(i =>
|
|
|
+ {
|
|
|
+ var userdata = _userDataRepository.GetUserData(user.Id, i.GetUserDataKey());
|
|
|
+ return new Tuple<Movie, bool, DateTime>(i, userdata.Played, userdata.LastPlayedDate ?? DateTime.MinValue);
|
|
|
+ })
|
|
|
+ .Where(i => i.Item2)
|
|
|
+ .OrderByDescending(i => i.Item3)
|
|
|
+ .Select(i => i.Item1)
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ var excludeFromLiked = recentlyPlayedMovies.Take(10);
|
|
|
+ var likedMovies = allMovies
|
|
|
+ .Select(i =>
|
|
|
+ {
|
|
|
+ var score = 0;
|
|
|
+ var userData = _userDataRepository.GetUserData(user.Id, i.GetUserDataKey());
|
|
|
+
|
|
|
+ if (userData.IsFavorite)
|
|
|
+ {
|
|
|
+ score = 2;
|
|
|
+ }
|
|
|
+ else
|
|
|
+ {
|
|
|
+ score = userData.Likes.HasValue ? userData.Likes.Value ? 1 : -1 : 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ return new Tuple<Movie, int>(i, score);
|
|
|
+ })
|
|
|
+ .OrderByDescending(i => i.Item2)
|
|
|
+ .ThenBy(i => Guid.NewGuid())
|
|
|
+ .Where(i => i.Item2 > 0)
|
|
|
+ .Select(i => i.Item1)
|
|
|
+ .Where(i => !excludeFromLiked.Contains(i));
|
|
|
+
|
|
|
+ var mostRecentMovies = recentlyPlayedMovies.Take(6).ToList();
|
|
|
+ // Get recently played directors
|
|
|
+ var recentDirectors = GetDirectors(mostRecentMovies)
|
|
|
+ .OrderBy(i => Guid.NewGuid())
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ // Get recently played actors
|
|
|
+ var recentActors = GetActors(mostRecentMovies)
|
|
|
+ .OrderBy(i => Guid.NewGuid())
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ var similarToRecentlyPlayed = GetSimilarTo(user, allMovies, recentlyPlayedMovies.Take(10).OrderBy(i => Guid.NewGuid()), itemLimit, fields, RecommendationType.SimilarToRecentlyPlayed).GetEnumerator();
|
|
|
+ var similarToLiked = GetSimilarTo(user, allMovies, likedMovies, itemLimit, fields, RecommendationType.SimilarToLikedItem).GetEnumerator();
|
|
|
+
|
|
|
+ var hasDirectorFromRecentlyPlayed = GetWithDirector(user, allMovies, recentDirectors, itemLimit, fields, RecommendationType.HasDirectorFromRecentlyPlayed).GetEnumerator();
|
|
|
+ var hasActorFromRecentlyPlayed = GetWithActor(user, allMovies, recentActors, itemLimit, fields, RecommendationType.HasActorFromRecentlyPlayed).GetEnumerator();
|
|
|
+
|
|
|
+ var categoryTypes = new List<IEnumerator<RecommendationDto>>
|
|
|
+ {
|
|
|
+ // Give this extra weight
|
|
|
+ similarToRecentlyPlayed,
|
|
|
+ similarToRecentlyPlayed,
|
|
|
+
|
|
|
+ // Give this extra weight
|
|
|
+ similarToLiked,
|
|
|
+ similarToLiked,
|
|
|
+
|
|
|
+ hasDirectorFromRecentlyPlayed,
|
|
|
+ hasActorFromRecentlyPlayed
|
|
|
+ };
|
|
|
+
|
|
|
+ while (categories.Count < categoryLimit)
|
|
|
+ {
|
|
|
+ var allEmpty = true;
|
|
|
+
|
|
|
+ foreach (var category in categoryTypes)
|
|
|
+ {
|
|
|
+ if (category.MoveNext())
|
|
|
+ {
|
|
|
+ categories.Add(category.Current);
|
|
|
+ allEmpty = false;
|
|
|
+
|
|
|
+ if (categories.Count >= categoryLimit)
|
|
|
+ {
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ if (allEmpty)
|
|
|
+ {
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ //// Get the lead actor for all movies
|
|
|
+ //var allActors = GetActors(allMovies)
|
|
|
+ // .ToList();
|
|
|
+
|
|
|
+ //foreach (var actor in recentActors)
|
|
|
+ //{
|
|
|
+
|
|
|
+ //}
|
|
|
+
|
|
|
+ return categories.OrderBy(i => i.RecommendationType).ThenBy(i => Guid.NewGuid());
|
|
|
+ }
|
|
|
+
|
|
|
+ private IEnumerable<RecommendationDto> GetWithDirector(User user, List<Movie> allMovies, IEnumerable<string> directors, int itemLimit, List<ItemFields> fields, RecommendationType type)
|
|
|
+ {
|
|
|
+ var userId = user.Id;
|
|
|
+
|
|
|
+ foreach (var director in directors)
|
|
|
+ {
|
|
|
+ var items = allMovies
|
|
|
+ .Where(i => !_userDataRepository.GetUserData(userId, i.GetUserDataKey()).Played && i.People.Any(p => string.Equals(p.Type, PersonType.Director, StringComparison.OrdinalIgnoreCase) && string.Equals(p.Name, director, StringComparison.OrdinalIgnoreCase)))
|
|
|
+ .Take(itemLimit)
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ if (items.Count > 0)
|
|
|
+ {
|
|
|
+ yield return new RecommendationDto
|
|
|
+ {
|
|
|
+ BaselineItemName = director,
|
|
|
+ CategoryId = director.GetMD5().ToString("N"),
|
|
|
+ RecommendationType = type,
|
|
|
+ Items = items.Select(i => _dtoService.GetBaseItemDto(i, fields, user)).ToArray()
|
|
|
+ };
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ private IEnumerable<RecommendationDto> GetWithActor(User user, List<Movie> allMovies, IEnumerable<string> names, int itemLimit, List<ItemFields> fields, RecommendationType type)
|
|
|
+ {
|
|
|
+ var userId = user.Id;
|
|
|
+
|
|
|
+ foreach (var name in names)
|
|
|
+ {
|
|
|
+ var items = allMovies
|
|
|
+ .Where(i => !_userDataRepository.GetUserData(userId, i.GetUserDataKey()).Played && i.People.Any(p => string.Equals(p.Name, name, StringComparison.OrdinalIgnoreCase)))
|
|
|
+ .Take(itemLimit)
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ if (items.Count > 0)
|
|
|
+ {
|
|
|
+ yield return new RecommendationDto
|
|
|
+ {
|
|
|
+ BaselineItemName = name,
|
|
|
+ CategoryId = name.GetMD5().ToString("N"),
|
|
|
+ RecommendationType = type,
|
|
|
+ Items = items.Select(i => _dtoService.GetBaseItemDto(i, fields, user)).ToArray()
|
|
|
+ };
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ private IEnumerable<RecommendationDto> GetSimilarTo(User user, List<Movie> allMovies, IEnumerable<Movie> baselineItems, int itemLimit, List<ItemFields> fields, RecommendationType type)
|
|
|
+ {
|
|
|
+ var userId = user.Id;
|
|
|
+
|
|
|
+ foreach (var item in baselineItems)
|
|
|
+ {
|
|
|
+ var similar = SimilarItemsHelper
|
|
|
+ .GetSimilaritems(item, allMovies, SimilarItemsHelper.GetSimiliarityScore)
|
|
|
+ .Where(i => !_userDataRepository.GetUserData(userId, i.GetUserDataKey()).Played)
|
|
|
+ .Take(itemLimit)
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ if (similar.Count > 0)
|
|
|
+ {
|
|
|
+ yield return new RecommendationDto
|
|
|
+ {
|
|
|
+ BaselineItemName = item.Name,
|
|
|
+ CategoryId = item.Id.ToString("N"),
|
|
|
+ RecommendationType = type,
|
|
|
+ Items = similar.Select(i => _dtoService.GetBaseItemDto(i, fields, user)).ToArray()
|
|
|
+ };
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ private IEnumerable<string> GetActors(IEnumerable<BaseItem> items)
|
|
|
+ {
|
|
|
+ // Get the two leading actors for all movies
|
|
|
+ return items
|
|
|
+ .SelectMany(i => i.People.Where(p => !string.Equals(PersonType.Director, p.Type, StringComparison.OrdinalIgnoreCase)).Take(2))
|
|
|
+ .Select(i => i.Name)
|
|
|
+ .Distinct(StringComparer.OrdinalIgnoreCase);
|
|
|
+ }
|
|
|
+
|
|
|
+ private IEnumerable<string> GetDirectors(IEnumerable<BaseItem> items)
|
|
|
+ {
|
|
|
+ return items
|
|
|
+ .Select(i => i.People.FirstOrDefault(p => string.Equals(PersonType.Director, p.Type, StringComparison.OrdinalIgnoreCase)))
|
|
|
+ .Where(i => i != null)
|
|
|
+ .Select(i => i.Name)
|
|
|
+ .Distinct(StringComparer.OrdinalIgnoreCase);
|
|
|
+ }
|
|
|
}
|
|
|
}
|