|
@@ -0,0 +1,361 @@
|
|
|
+using MediaBrowser.Common.Configuration;
|
|
|
+using MediaBrowser.Common.Security;
|
|
|
+using MediaBrowser.Controller.Channels;
|
|
|
+using MediaBrowser.Controller.Entities;
|
|
|
+using MediaBrowser.Controller.Entities.Movies;
|
|
|
+using MediaBrowser.Controller.Entities.TV;
|
|
|
+using MediaBrowser.Controller.Library;
|
|
|
+using MediaBrowser.Controller.Localization;
|
|
|
+using MediaBrowser.Model.Channels;
|
|
|
+using MediaBrowser.Model.Configuration;
|
|
|
+using MediaBrowser.Model.Entities;
|
|
|
+using System;
|
|
|
+using System.Collections.Generic;
|
|
|
+using System.Linq;
|
|
|
+using System.Threading;
|
|
|
+using System.Threading.Tasks;
|
|
|
+
|
|
|
+namespace MediaBrowser.Server.Implementations.Intros
|
|
|
+{
|
|
|
+ public class DefaultIntroProvider : IIntroProvider
|
|
|
+ {
|
|
|
+ private readonly ISecurityManager _security;
|
|
|
+ private readonly IChannelManager _channelManager;
|
|
|
+ private readonly ILocalizationManager _localization;
|
|
|
+ private readonly IConfigurationManager _serverConfig;
|
|
|
+
|
|
|
+ public DefaultIntroProvider(ISecurityManager security, IChannelManager channelManager, ILocalizationManager localization, IConfigurationManager serverConfig)
|
|
|
+ {
|
|
|
+ _security = security;
|
|
|
+ _channelManager = channelManager;
|
|
|
+ _localization = localization;
|
|
|
+ _serverConfig = serverConfig;
|
|
|
+ }
|
|
|
+
|
|
|
+ public async Task<IEnumerable<IntroInfo>> GetIntros(BaseItem item, User user)
|
|
|
+ {
|
|
|
+ var config = GetOptions();
|
|
|
+
|
|
|
+ if (item is Movie)
|
|
|
+ {
|
|
|
+ if (!config.EnableIntrosForMovies)
|
|
|
+ {
|
|
|
+ return new List<IntroInfo>();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ else if (item is Episode)
|
|
|
+ {
|
|
|
+ if (!config.EnableIntrosForEpisodes)
|
|
|
+ {
|
|
|
+ return new List<IntroInfo>();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ else
|
|
|
+ {
|
|
|
+ return new List<IntroInfo>();
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!IsSupporter)
|
|
|
+ {
|
|
|
+ return new List<IntroInfo>();
|
|
|
+ }
|
|
|
+
|
|
|
+ var ratingLevel = string.IsNullOrWhiteSpace(item.OfficialRating)
|
|
|
+ ? (int?)null
|
|
|
+ : _localization.GetRatingLevel(item.OfficialRating);
|
|
|
+
|
|
|
+ var libaryItems = user.RootFolder.GetRecursiveChildren(user, false)
|
|
|
+ .ToList();
|
|
|
+
|
|
|
+ var random = new Random(Environment.TickCount + Guid.NewGuid().GetHashCode());
|
|
|
+
|
|
|
+ var candidates = new List<ItemWithTrailer>();
|
|
|
+
|
|
|
+ if (config.EnableIntrosFromMoviesInLibrary)
|
|
|
+ {
|
|
|
+ var itemsWithTrailers = libaryItems
|
|
|
+ .Where(i =>
|
|
|
+ {
|
|
|
+ var hasTrailers = i as IHasTrailers;
|
|
|
+
|
|
|
+ if (hasTrailers != null && hasTrailers.LocalTrailerIds.Count > 0)
|
|
|
+ {
|
|
|
+ if (i is Movie)
|
|
|
+ {
|
|
|
+ return true;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return false;
|
|
|
+ });
|
|
|
+
|
|
|
+ candidates.AddRange(itemsWithTrailers.Select(i => new ItemWithTrailer
|
|
|
+ {
|
|
|
+ Item = i,
|
|
|
+ Type = ItemWithTrailerType.ItemWithTrailer,
|
|
|
+ User = user,
|
|
|
+ WatchingItem = item,
|
|
|
+ Random = random
|
|
|
+ }));
|
|
|
+ }
|
|
|
+
|
|
|
+ if (config.EnableIntrosFromUpcomingTrailers)
|
|
|
+ {
|
|
|
+ var channelTrailers = await _channelManager.GetAllMediaInternal(new AllChannelMediaQuery
|
|
|
+ {
|
|
|
+ ContentTypes = new[] { ChannelMediaContentType.Trailer },
|
|
|
+ UserId = user.Id.ToString("N")
|
|
|
+
|
|
|
+ }, CancellationToken.None);
|
|
|
+
|
|
|
+ candidates.AddRange(channelTrailers.Items.Select(i => new ItemWithTrailer
|
|
|
+ {
|
|
|
+ Item = i,
|
|
|
+ Type = ItemWithTrailerType.ChannelTrailer,
|
|
|
+ User = user,
|
|
|
+ WatchingItem = item,
|
|
|
+ Random = random
|
|
|
+ }));
|
|
|
+
|
|
|
+ candidates.AddRange(libaryItems.Where(i => i is Trailer).Select(i => new ItemWithTrailer
|
|
|
+ {
|
|
|
+ Item = i,
|
|
|
+ Type = ItemWithTrailerType.LibraryTrailer,
|
|
|
+ User = user,
|
|
|
+ WatchingItem = item,
|
|
|
+ Random = random
|
|
|
+ }));
|
|
|
+ }
|
|
|
+
|
|
|
+ var customIntros = config.EnableCustomIntro ?
|
|
|
+ GetCustomIntros(item) :
|
|
|
+ new List<IntroInfo>();
|
|
|
+
|
|
|
+ var trailerLimit = 2;
|
|
|
+ if (customIntros.Count > 0)
|
|
|
+ {
|
|
|
+ trailerLimit--;
|
|
|
+ }
|
|
|
+
|
|
|
+ // Avoid implicitly captured closure
|
|
|
+ var currentUser = user;
|
|
|
+ return candidates.Where(i =>
|
|
|
+ {
|
|
|
+ if (config.EnableIntrosParentalControl && !FilterByParentalRating(ratingLevel, i.Item))
|
|
|
+ {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!config.EnableIntrosForWatchedContent && i.IsPlayed)
|
|
|
+ {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+ return true;
|
|
|
+ })
|
|
|
+ .OrderByDescending(i => i.Score)
|
|
|
+ .ThenBy(i => Guid.NewGuid())
|
|
|
+ .ThenByDescending(i => (i.IsPlayed ? 0 : 1))
|
|
|
+ .Select(i => i.IntroInfo)
|
|
|
+ .Take(trailerLimit)
|
|
|
+ .Concat(customIntros.Take(1));
|
|
|
+ }
|
|
|
+
|
|
|
+ private CinemaModeConfiguration GetOptions()
|
|
|
+ {
|
|
|
+ return _serverConfig.GetConfiguration<CinemaModeConfiguration>("cinemamode");
|
|
|
+ }
|
|
|
+
|
|
|
+ private List<IntroInfo> GetCustomIntros(BaseItem item)
|
|
|
+ {
|
|
|
+ return new List<IntroInfo>();
|
|
|
+ }
|
|
|
+
|
|
|
+ private bool FilterByParentalRating(int? ratingLevel, BaseItem item)
|
|
|
+ {
|
|
|
+ // Only content rated same or lower
|
|
|
+ if (ratingLevel.HasValue)
|
|
|
+ {
|
|
|
+ var level = string.IsNullOrWhiteSpace(item.OfficialRating)
|
|
|
+ ? (int?)null
|
|
|
+ : _localization.GetRatingLevel(item.OfficialRating);
|
|
|
+
|
|
|
+ return level.HasValue && level.Value <= ratingLevel.Value;
|
|
|
+ }
|
|
|
+
|
|
|
+ return true;
|
|
|
+ }
|
|
|
+
|
|
|
+ internal static int GetSimiliarityScore(BaseItem item1, BaseItem item2, Random random)
|
|
|
+ {
|
|
|
+ var points = 0;
|
|
|
+
|
|
|
+ if (!string.IsNullOrEmpty(item1.OfficialRating) && string.Equals(item1.OfficialRating, item2.OfficialRating, StringComparison.OrdinalIgnoreCase))
|
|
|
+ {
|
|
|
+ points += 10;
|
|
|
+ }
|
|
|
+
|
|
|
+ // Find common genres
|
|
|
+ points += item1.Genres.Where(i => item2.Genres.Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 10);
|
|
|
+
|
|
|
+ // Find common tags
|
|
|
+ points += GetTags(item1).Where(i => GetTags(item2).Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 10);
|
|
|
+
|
|
|
+ // Find common keywords
|
|
|
+ points += GetKeywords(item1).Where(i => GetKeywords(item2).Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 10);
|
|
|
+
|
|
|
+ // Find common studios
|
|
|
+ points += item1.Studios.Where(i => item2.Studios.Contains(i, StringComparer.OrdinalIgnoreCase)).Sum(i => 5);
|
|
|
+
|
|
|
+ var item2PeopleNames = item2.People.Select(i => i.Name)
|
|
|
+ .Distinct(StringComparer.OrdinalIgnoreCase)
|
|
|
+ .ToDictionary(i => i, StringComparer.OrdinalIgnoreCase);
|
|
|
+
|
|
|
+ points += item1.People.Where(i => item2PeopleNames.ContainsKey(i.Name)).Sum(i =>
|
|
|
+ {
|
|
|
+ if (string.Equals(i.Type, PersonType.Director, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.Director, StringComparison.OrdinalIgnoreCase))
|
|
|
+ {
|
|
|
+ return 5;
|
|
|
+ }
|
|
|
+ if (string.Equals(i.Type, PersonType.Actor, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.Actor, StringComparison.OrdinalIgnoreCase))
|
|
|
+ {
|
|
|
+ return 3;
|
|
|
+ }
|
|
|
+ if (string.Equals(i.Type, PersonType.Composer, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.Composer, StringComparison.OrdinalIgnoreCase))
|
|
|
+ {
|
|
|
+ return 3;
|
|
|
+ }
|
|
|
+ if (string.Equals(i.Type, PersonType.GuestStar, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.GuestStar, StringComparison.OrdinalIgnoreCase))
|
|
|
+ {
|
|
|
+ return 3;
|
|
|
+ }
|
|
|
+ if (string.Equals(i.Type, PersonType.Writer, StringComparison.OrdinalIgnoreCase) || string.Equals(i.Role, PersonType.Writer, StringComparison.OrdinalIgnoreCase))
|
|
|
+ {
|
|
|
+ return 2;
|
|
|
+ }
|
|
|
+
|
|
|
+ return 1;
|
|
|
+ });
|
|
|
+
|
|
|
+ // Add some randomization so that you're not always seeing the same ones for a given movie
|
|
|
+ points += random.Next(0, 50);
|
|
|
+
|
|
|
+ return points;
|
|
|
+ }
|
|
|
+
|
|
|
+ private static IEnumerable<string> GetTags(BaseItem item)
|
|
|
+ {
|
|
|
+ var hasTags = item as IHasTags;
|
|
|
+ if (hasTags != null)
|
|
|
+ {
|
|
|
+ return hasTags.Tags;
|
|
|
+ }
|
|
|
+
|
|
|
+ return new List<string>();
|
|
|
+ }
|
|
|
+
|
|
|
+ private static IEnumerable<string> GetKeywords(BaseItem item)
|
|
|
+ {
|
|
|
+ var hasTags = item as IHasKeywords;
|
|
|
+ if (hasTags != null)
|
|
|
+ {
|
|
|
+ return hasTags.Keywords;
|
|
|
+ }
|
|
|
+
|
|
|
+ return new List<string>();
|
|
|
+ }
|
|
|
+
|
|
|
+ public IEnumerable<string> GetAllIntroFiles()
|
|
|
+ {
|
|
|
+ return new List<string>();
|
|
|
+ }
|
|
|
+
|
|
|
+ private bool IsSupporter
|
|
|
+ {
|
|
|
+ get { return _security.IsMBSupporter; }
|
|
|
+ }
|
|
|
+
|
|
|
+ public string Name
|
|
|
+ {
|
|
|
+ get { return "Default"; }
|
|
|
+ }
|
|
|
+
|
|
|
+ internal class ItemWithTrailer
|
|
|
+ {
|
|
|
+ internal BaseItem Item;
|
|
|
+ internal ItemWithTrailerType Type;
|
|
|
+ internal User User;
|
|
|
+ internal BaseItem WatchingItem;
|
|
|
+ internal Random Random;
|
|
|
+
|
|
|
+ private bool? _isPlayed;
|
|
|
+ public bool IsPlayed
|
|
|
+ {
|
|
|
+ get
|
|
|
+ {
|
|
|
+ if (!_isPlayed.HasValue)
|
|
|
+ {
|
|
|
+ _isPlayed = Item.IsPlayed(User);
|
|
|
+ }
|
|
|
+ return _isPlayed.Value;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ private int? _score;
|
|
|
+ public int Score
|
|
|
+ {
|
|
|
+ get
|
|
|
+ {
|
|
|
+ if (!_score.HasValue)
|
|
|
+ {
|
|
|
+ _score = GetSimiliarityScore(WatchingItem, Item, Random);
|
|
|
+ }
|
|
|
+ return _score.Value;
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ public IntroInfo IntroInfo
|
|
|
+ {
|
|
|
+ get
|
|
|
+ {
|
|
|
+ var id = Item.Id;
|
|
|
+
|
|
|
+ if (Type == ItemWithTrailerType.ItemWithTrailer)
|
|
|
+ {
|
|
|
+ var hasTrailers = Item as IHasTrailers;
|
|
|
+
|
|
|
+ if (hasTrailers != null)
|
|
|
+ {
|
|
|
+ id = hasTrailers.LocalTrailerIds.FirstOrDefault();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ return new IntroInfo
|
|
|
+ {
|
|
|
+ ItemId = id
|
|
|
+ };
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ internal enum ItemWithTrailerType
|
|
|
+ {
|
|
|
+ LibraryTrailer,
|
|
|
+ ChannelTrailer,
|
|
|
+ ItemWithTrailer
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ public class CinemaModeConfigurationFactory : IConfigurationFactory
|
|
|
+ {
|
|
|
+ public IEnumerable<ConfigurationStore> GetConfigurations()
|
|
|
+ {
|
|
|
+ return new[]
|
|
|
+ {
|
|
|
+ new ConfigurationStore
|
|
|
+ {
|
|
|
+ ConfigurationType = typeof(CinemaModeConfiguration),
|
|
|
+ Key = "cinemamode"
|
|
|
+ }
|
|
|
+ };
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+}
|