2019-01-13 21:01:16 +01:00
using System ;
2019-01-13 20:24:58 +01:00
using System.Collections.Generic ;
2019-02-28 23:22:57 +01:00
using System.Globalization ;
2019-01-13 20:24:58 +01:00
using System.Linq ;
2020-05-20 19:07:53 +02:00
using Jellyfin.Data.Entities ;
2019-01-13 20:24:58 +01:00
using MediaBrowser.Common.Extensions ;
using MediaBrowser.Controller.Configuration ;
2014-03-12 03:11:01 +01:00
using MediaBrowser.Controller.Dto ;
2013-09-04 19:02:19 +02:00
using MediaBrowser.Controller.Entities ;
2013-05-26 01:52:41 +02:00
using MediaBrowser.Controller.Library ;
2019-01-13 20:24:58 +01:00
using MediaBrowser.Controller.LiveTv ;
2014-07-02 20:34:08 +02:00
using MediaBrowser.Controller.Net ;
2014-03-12 03:11:01 +01:00
using MediaBrowser.Model.Dto ;
using MediaBrowser.Model.Entities ;
2019-01-13 20:24:58 +01:00
using MediaBrowser.Model.Querying ;
2016-10-25 21:02:04 +02:00
using MediaBrowser.Model.Services ;
2019-11-17 23:05:39 +01:00
using Microsoft.Extensions.Logging ;
2020-05-20 19:07:53 +02:00
using Movie = MediaBrowser . Controller . Entities . Movies . Movie ;
2013-05-26 01:52:41 +02:00
2014-03-07 16:53:23 +01:00
namespace MediaBrowser.Api.Movies
2013-05-26 01:52:41 +02:00
{
2014-03-25 22:13:55 +01:00
[Route("/Movies/Recommendations", "GET", Summary = "Gets movie recommendations")]
2017-05-21 09:25:49 +02:00
public class GetMovieRecommendations : IReturn < RecommendationDto [ ] > , IHasDtoOptions
2014-03-12 03:11:01 +01:00
{
[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")]
2018-09-12 19:26:21 +02:00
public Guid UserId { get ; set ; }
2014-03-12 03:11:01 +01:00
2014-05-02 04:54:33 +02:00
/// <summary>
/// Specify this to localize the search to a specific item or folder. Omit to use the root.
/// </summary>
/// <value>The parent id.</value>
[ApiMember(Name = "ParentId", Description = "Specify this to localize the search to a specific item or folder. Omit to use the root", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")]
public string ParentId { get ; set ; }
2014-09-30 06:47:30 +02:00
2017-05-21 09:25:49 +02:00
[ApiMember(Name = "EnableImages", Description = "Optional, include image information in output", IsRequired = false, DataType = "boolean", ParameterType = "query", Verb = "GET")]
public bool? EnableImages { get ; set ; }
[ApiMember(Name = "EnableUserData", Description = "Optional, include user data", IsRequired = false, DataType = "boolean", ParameterType = "query", Verb = "GET")]
public bool? EnableUserData { get ; set ; }
[ApiMember(Name = "ImageTypeLimit", Description = "Optional, the max number of images to return, per image type", IsRequired = false, DataType = "int", ParameterType = "query", Verb = "GET")]
public int? ImageTypeLimit { get ; set ; }
[ApiMember(Name = "EnableImageTypes", Description = "Optional. The image types to include in the output.", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET")]
public string EnableImageTypes { get ; set ; }
2014-03-12 03:11:01 +01:00
public GetMovieRecommendations ( )
{
CategoryLimit = 5 ;
ItemLimit = 8 ;
}
public string Fields { get ; set ; }
}
2013-05-26 01:52:41 +02:00
/// <summary>
/// Class MoviesService
/// </summary>
2014-07-02 20:34:08 +02:00
[Authenticated]
2013-05-26 01:52:41 +02:00
public class MoviesService : BaseApiService
{
/// <summary>
/// The _user manager
/// </summary>
private readonly IUserManager _userManager ;
private readonly ILibraryManager _libraryManager ;
2013-09-04 19:02:19 +02:00
private readonly IDtoService _dtoService ;
2016-11-10 15:41:24 +01:00
private readonly IAuthorizationContext _authContext ;
2013-09-04 19:02:19 +02:00
2013-05-26 01:52:41 +02:00
/// <summary>
2016-03-19 16:38:05 +01:00
/// Initializes a new instance of the <see cref="MoviesService" /> class.
2013-05-26 01:52:41 +02:00
/// </summary>
2019-11-17 23:05:39 +01:00
public MoviesService (
2020-05-15 02:11:34 +02:00
ILogger < MoviesService > logger ,
2019-11-17 23:05:39 +01:00
IServerConfigurationManager serverConfigurationManager ,
IHttpResultFactory httpResultFactory ,
IUserManager userManager ,
ILibraryManager libraryManager ,
IDtoService dtoService ,
IAuthorizationContext authContext )
: base ( logger , serverConfigurationManager , httpResultFactory )
2013-05-26 01:52:41 +02:00
{
_userManager = userManager ;
_libraryManager = libraryManager ;
2013-09-04 19:02:19 +02:00
_dtoService = dtoService ;
2016-11-10 15:41:24 +01:00
_authContext = authContext ;
2013-05-26 01:52:41 +02:00
}
2017-05-21 09:25:49 +02:00
public object Get ( GetMovieRecommendations request )
2014-03-12 03:11:01 +01:00
{
2015-05-30 01:51:33 +02:00
var user = _userManager . GetUserById ( request . UserId ) ;
2014-03-12 03:11:01 +01:00
2016-11-10 15:41:24 +01:00
var dtoOptions = GetDtoOptions ( _authContext , request ) ;
2014-12-27 06:08:39 +01:00
2016-06-01 07:50:00 +02:00
var result = GetRecommendationCategories ( user , request . ParentId , request . CategoryLimit , request . ItemLimit , dtoOptions ) ;
2014-03-12 03:11:01 +01:00
return ToOptimizedResult ( result ) ;
}
2018-09-12 19:26:21 +02:00
public QueryResult < BaseItemDto > GetSimilarItemsResult ( BaseGetSimilarItemsFromItem request )
2014-09-30 06:47:30 +02:00
{
2018-09-12 19:26:21 +02:00
var user = ! request . UserId . Equals ( Guid . Empty ) ? _userManager . GetUserById ( request . UserId ) : null ;
2014-09-30 06:47:30 +02:00
var item = string . IsNullOrEmpty ( request . Id ) ?
2018-09-12 19:26:21 +02:00
( ! request . UserId . Equals ( Guid . Empty ) ? _libraryManager . GetUserRootFolder ( ) :
2014-09-30 06:47:30 +02:00
_libraryManager . RootFolder ) : _libraryManager . GetItemById ( request . Id ) ;
2016-05-30 20:16:44 +02:00
2016-08-31 23:07:02 +02:00
var itemTypes = new List < string > { typeof ( Movie ) . Name } ;
2019-11-17 23:05:39 +01:00
if ( ServerConfigurationManager . Configuration . EnableExternalContentInSuggestions )
2016-08-31 23:07:02 +02:00
{
itemTypes . Add ( typeof ( Trailer ) . Name ) ;
itemTypes . Add ( typeof ( LiveTvProgram ) . Name ) ;
}
2016-11-10 15:41:24 +01:00
var dtoOptions = GetDtoOptions ( _authContext , request ) ;
2016-10-08 07:57:38 +02:00
2016-06-01 07:50:00 +02:00
var itemsResult = _libraryManager . GetItemList ( new InternalItemsQuery ( user )
2015-11-10 19:58:05 +01:00
{
2016-06-01 07:50:00 +02:00
Limit = request . Limit ,
2018-12-28 16:48:26 +01:00
IncludeItemTypes = itemTypes . ToArray ( ) ,
2016-06-01 07:50:00 +02:00
IsMovie = true ,
2016-06-29 05:17:27 +02:00
SimilarTo = item ,
2016-10-08 07:57:38 +02:00
EnableGroupByMetadataKey = true ,
2016-10-10 20:18:28 +02:00
DtoOptions = dtoOptions
2016-06-29 05:17:27 +02:00
2017-08-09 21:56:38 +02:00
} ) ;
2017-08-28 02:33:05 +02:00
var returnList = _dtoService . GetBaseItemDtos ( itemsResult , dtoOptions , user ) ;
2015-01-24 20:03:55 +01:00
2016-06-01 07:50:00 +02:00
var result = new QueryResult < BaseItemDto >
2014-09-30 06:47:30 +02:00
{
2017-08-19 21:43:35 +02:00
Items = returnList ,
2014-09-30 06:47:30 +02:00
2016-06-01 07:50:00 +02:00
TotalRecordCount = itemsResult . Count
2014-09-30 06:47:30 +02:00
} ;
return result ;
}
2020-05-20 19:07:53 +02:00
private IEnumerable < RecommendationDto > GetRecommendationCategories ( User user , string parentId , int categoryLimit , int itemLimit , DtoOptions dtoOptions )
2014-03-12 03:11:01 +01:00
{
var categories = new List < RecommendationDto > ( ) ;
2018-09-12 19:26:21 +02:00
var parentIdGuid = string . IsNullOrWhiteSpace ( parentId ) ? Guid . Empty : new Guid ( parentId ) ;
2016-06-30 21:01:48 +02:00
2016-05-31 20:42:32 +02:00
var query = new InternalItemsQuery ( user )
{
IncludeItemTypes = new [ ]
2014-03-12 03:11:01 +01:00
{
2016-05-31 20:42:32 +02:00
typeof ( Movie ) . Name ,
//typeof(Trailer).Name,
//typeof(LiveTvProgram).Name
} ,
// IsMovie = true
2018-09-12 19:26:21 +02:00
OrderBy = new [ ] { ItemSortBy . DatePlayed , ItemSortBy . Random } . Select ( i = > new ValueTuple < string , SortOrder > ( i , SortOrder . Descending ) ) . ToArray ( ) ,
2016-06-30 21:01:48 +02:00
Limit = 7 ,
ParentId = parentIdGuid ,
2016-08-25 05:12:25 +02:00
Recursive = true ,
2016-10-08 07:57:38 +02:00
IsPlayed = true ,
2016-10-10 20:18:28 +02:00
DtoOptions = dtoOptions
2016-05-31 20:42:32 +02:00
} ;
2014-03-12 03:11:01 +01:00
2017-08-09 21:56:38 +02:00
var recentlyPlayedMovies = _libraryManager . GetItemList ( query ) ;
2014-03-12 03:11:01 +01:00
2016-08-31 23:07:02 +02:00
var itemTypes = new List < string > { typeof ( Movie ) . Name } ;
2019-11-17 23:05:39 +01:00
if ( ServerConfigurationManager . Configuration . EnableExternalContentInSuggestions )
2016-08-31 23:07:02 +02:00
{
itemTypes . Add ( typeof ( Trailer ) . Name ) ;
itemTypes . Add ( typeof ( LiveTvProgram ) . Name ) ;
}
2016-05-31 20:42:32 +02:00
var likedMovies = _libraryManager . GetItemList ( new InternalItemsQuery ( user )
{
2018-12-28 16:48:26 +01:00
IncludeItemTypes = itemTypes . ToArray ( ) ,
2016-05-31 20:42:32 +02:00
IsMovie = true ,
2018-09-12 19:26:21 +02:00
OrderBy = new [ ] { ItemSortBy . Random } . Select ( i = > new ValueTuple < string , SortOrder > ( i , SortOrder . Descending ) ) . ToArray ( ) ,
2016-05-31 20:42:32 +02:00
Limit = 10 ,
IsFavoriteOrLiked = true ,
2018-12-28 16:48:26 +01:00
ExcludeItemIds = recentlyPlayedMovies . Select ( i = > i . Id ) . ToArray ( ) ,
2016-06-30 21:01:48 +02:00
EnableGroupByMetadataKey = true ,
ParentId = parentIdGuid ,
2016-10-08 07:57:38 +02:00
Recursive = true ,
2016-10-10 20:18:28 +02:00
DtoOptions = dtoOptions
2014-03-12 03:11:01 +01:00
2017-08-09 21:56:38 +02:00
} ) ;
2014-03-12 03:11:01 +01:00
var mostRecentMovies = recentlyPlayedMovies . Take ( 6 ) . ToList ( ) ;
// Get recently played directors
var recentDirectors = GetDirectors ( mostRecentMovies )
. ToList ( ) ;
// Get recently played actors
var recentActors = GetActors ( mostRecentMovies )
. ToList ( ) ;
2016-06-01 07:50:00 +02:00
var similarToRecentlyPlayed = GetSimilarTo ( user , recentlyPlayedMovies , itemLimit , dtoOptions , RecommendationType . SimilarToRecentlyPlayed ) . GetEnumerator ( ) ;
var similarToLiked = GetSimilarTo ( user , likedMovies , itemLimit , dtoOptions , RecommendationType . SimilarToLikedItem ) . GetEnumerator ( ) ;
2014-03-12 03:11:01 +01:00
2016-05-31 20:42:32 +02:00
var hasDirectorFromRecentlyPlayed = GetWithDirector ( user , recentDirectors , itemLimit , dtoOptions , RecommendationType . HasDirectorFromRecentlyPlayed ) . GetEnumerator ( ) ;
var hasActorFromRecentlyPlayed = GetWithActor ( user , recentActors , itemLimit , dtoOptions , RecommendationType . HasActorFromRecentlyPlayed ) . GetEnumerator ( ) ;
2014-03-12 03:11:01 +01:00
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 ;
}
}
2019-09-02 08:19:29 +02:00
return categories . OrderBy ( i = > i . RecommendationType ) ;
2014-03-12 03:11:01 +01:00
}
2020-05-13 04:10:35 +02:00
private IEnumerable < RecommendationDto > GetWithDirector (
2020-05-20 19:07:53 +02:00
User user ,
2020-05-13 04:10:35 +02:00
IEnumerable < string > names ,
int itemLimit ,
DtoOptions dtoOptions ,
RecommendationType type )
2014-03-12 03:11:01 +01:00
{
2016-08-31 23:07:02 +02:00
var itemTypes = new List < string > { typeof ( Movie ) . Name } ;
2019-11-17 23:05:39 +01:00
if ( ServerConfigurationManager . Configuration . EnableExternalContentInSuggestions )
2016-08-31 23:07:02 +02:00
{
itemTypes . Add ( typeof ( Trailer ) . Name ) ;
itemTypes . Add ( typeof ( LiveTvProgram ) . Name ) ;
}
2016-05-31 20:42:32 +02:00
foreach ( var name in names )
2014-03-12 03:11:01 +01:00
{
2016-05-31 20:42:32 +02:00
var items = _libraryManager . GetItemList ( new InternalItemsQuery ( user )
{
Person = name ,
// Account for duplicates by imdb id, since the database doesn't support this yet
Limit = itemLimit + 2 ,
PersonTypes = new [ ] { PersonType . Director } ,
2018-12-28 16:48:26 +01:00
IncludeItemTypes = itemTypes . ToArray ( ) ,
2016-06-29 05:17:27 +02:00
IsMovie = true ,
2016-10-08 07:57:38 +02:00
EnableGroupByMetadataKey = true ,
2016-10-10 20:18:28 +02:00
DtoOptions = dtoOptions
2016-05-31 20:42:32 +02:00
2020-06-06 21:17:49 +02:00
} ) . GroupBy ( i = > i . GetProviderId ( MetadataProvider . Imdb ) ? ? Guid . NewGuid ( ) . ToString ( "N" , CultureInfo . InvariantCulture ) )
2019-02-02 12:27:06 +01:00
. Select ( x = > x . First ( ) )
2016-05-31 20:42:32 +02:00
. Take ( itemLimit )
. ToList ( ) ;
2014-03-12 03:11:01 +01:00
if ( items . Count > 0 )
{
2017-08-28 02:33:05 +02:00
var returnItems = _dtoService . GetBaseItemDtos ( items , dtoOptions , user ) ;
2017-08-09 21:56:38 +02:00
2014-03-12 03:11:01 +01:00
yield return new RecommendationDto
{
2016-05-31 20:42:32 +02:00
BaselineItemName = name ,
2018-09-12 19:26:21 +02:00
CategoryId = name . GetMD5 ( ) ,
2014-03-12 03:11:01 +01:00
RecommendationType = type ,
2017-08-19 21:43:35 +02:00
Items = returnItems
2014-03-12 03:11:01 +01:00
} ;
}
}
}
2020-05-20 19:07:53 +02:00
private IEnumerable < RecommendationDto > GetWithActor ( User user , IEnumerable < string > names , int itemLimit , DtoOptions dtoOptions , RecommendationType type )
2014-03-12 03:11:01 +01:00
{
2016-08-31 23:07:02 +02:00
var itemTypes = new List < string > { typeof ( Movie ) . Name } ;
2019-11-17 23:05:39 +01:00
if ( ServerConfigurationManager . Configuration . EnableExternalContentInSuggestions )
2016-08-31 23:07:02 +02:00
{
itemTypes . Add ( typeof ( Trailer ) . Name ) ;
itemTypes . Add ( typeof ( LiveTvProgram ) . Name ) ;
}
2014-03-12 03:11:01 +01:00
foreach ( var name in names )
{
2016-05-31 20:42:32 +02:00
var items = _libraryManager . GetItemList ( new InternalItemsQuery ( user )
2015-07-08 18:10:34 +02:00
{
2016-05-31 20:42:32 +02:00
Person = name ,
// Account for duplicates by imdb id, since the database doesn't support this yet
Limit = itemLimit + 2 ,
2018-12-28 16:48:26 +01:00
IncludeItemTypes = itemTypes . ToArray ( ) ,
2016-06-29 05:17:27 +02:00
IsMovie = true ,
2016-10-08 07:57:38 +02:00
EnableGroupByMetadataKey = true ,
2016-10-10 20:18:28 +02:00
DtoOptions = dtoOptions
2016-05-31 20:42:32 +02:00
2020-06-06 21:17:49 +02:00
} ) . GroupBy ( i = > i . GetProviderId ( MetadataProvider . Imdb ) ? ? Guid . NewGuid ( ) . ToString ( "N" , CultureInfo . InvariantCulture ) )
2019-02-02 12:27:06 +01:00
. Select ( x = > x . First ( ) )
2016-05-31 20:42:32 +02:00
. Take ( itemLimit )
. ToList ( ) ;
2014-03-12 03:11:01 +01:00
if ( items . Count > 0 )
{
2017-08-28 02:33:05 +02:00
var returnItems = _dtoService . GetBaseItemDtos ( items , dtoOptions , user ) ;
2017-08-09 21:56:38 +02:00
2014-03-12 03:11:01 +01:00
yield return new RecommendationDto
{
BaselineItemName = name ,
2018-09-12 19:26:21 +02:00
CategoryId = name . GetMD5 ( ) ,
2014-03-12 03:11:01 +01:00
RecommendationType = type ,
2017-08-19 21:43:35 +02:00
Items = returnItems
2014-03-12 03:11:01 +01:00
} ;
}
}
}
2020-05-20 19:07:53 +02:00
private IEnumerable < RecommendationDto > GetSimilarTo ( User user , List < BaseItem > baselineItems , int itemLimit , DtoOptions dtoOptions , RecommendationType type )
2014-03-12 03:11:01 +01:00
{
2016-08-31 23:07:02 +02:00
var itemTypes = new List < string > { typeof ( Movie ) . Name } ;
2019-11-17 23:05:39 +01:00
if ( ServerConfigurationManager . Configuration . EnableExternalContentInSuggestions )
2016-08-31 23:07:02 +02:00
{
itemTypes . Add ( typeof ( Trailer ) . Name ) ;
itemTypes . Add ( typeof ( LiveTvProgram ) . Name ) ;
}
2014-03-12 03:11:01 +01:00
foreach ( var item in baselineItems )
{
2016-06-01 07:50:00 +02:00
var similar = _libraryManager . GetItemList ( new InternalItemsQuery ( user )
{
Limit = itemLimit ,
2018-12-28 16:48:26 +01:00
IncludeItemTypes = itemTypes . ToArray ( ) ,
2016-06-01 07:50:00 +02:00
IsMovie = true ,
2016-06-29 05:17:27 +02:00
SimilarTo = item ,
2016-10-08 07:57:38 +02:00
EnableGroupByMetadataKey = true ,
2016-10-10 20:18:28 +02:00
DtoOptions = dtoOptions
2016-06-01 07:50:00 +02:00
2017-08-09 21:56:38 +02:00
} ) ;
2014-03-12 03:11:01 +01:00
if ( similar . Count > 0 )
{
2017-08-28 02:33:05 +02:00
var returnItems = _dtoService . GetBaseItemDtos ( similar , dtoOptions , user ) ;
2017-08-09 21:56:38 +02:00
2014-03-12 03:11:01 +01:00
yield return new RecommendationDto
{
BaselineItemName = item . Name ,
2018-09-12 19:26:21 +02:00
CategoryId = item . Id ,
2014-03-12 03:11:01 +01:00
RecommendationType = type ,
2017-08-19 21:43:35 +02:00
Items = returnItems
2014-03-12 03:11:01 +01:00
} ;
}
}
}
2017-08-09 21:56:38 +02:00
private IEnumerable < string > GetActors ( List < BaseItem > items )
2014-03-12 03:11:01 +01:00
{
2015-07-08 18:10:34 +02:00
var people = _libraryManager . GetPeople ( new InternalPeopleQuery
{
2019-01-13 20:24:58 +01:00
ExcludePersonTypes = new [ ]
2015-07-08 18:10:34 +02:00
{
PersonType . Director
} ,
MaxListOrder = 3
} ) ;
2017-08-09 23:08:01 +02:00
var itemIds = items . Select ( i = > i . Id ) . ToList ( ) ;
2015-07-08 18:10:34 +02:00
return people
. Where ( i = > itemIds . Contains ( i . ItemId ) )
2014-03-12 03:11:01 +01:00
. Select ( i = > i . Name )
2015-04-09 23:11:57 +02:00
. DistinctNames ( ) ;
2014-03-12 03:11:01 +01:00
}
2017-08-09 21:56:38 +02:00
private IEnumerable < string > GetDirectors ( List < BaseItem > items )
2014-03-12 03:11:01 +01:00
{
2015-07-08 18:10:34 +02:00
var people = _libraryManager . GetPeople ( new InternalPeopleQuery
{
2020-04-05 19:00:35 +02:00
PersonTypes = new [ ]
2015-07-08 18:10:34 +02:00
{
PersonType . Director
}
} ) ;
2017-08-09 23:08:01 +02:00
var itemIds = items . Select ( i = > i . Id ) . ToList ( ) ;
2015-07-08 18:10:34 +02:00
return people
. Where ( i = > itemIds . Contains ( i . ItemId ) )
2014-03-12 03:11:01 +01:00
. Select ( i = > i . Name )
2015-04-09 23:11:57 +02:00
. DistinctNames ( ) ;
2014-03-12 03:11:01 +01:00
}
2013-05-26 01:52:41 +02:00
}
}