mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-06 11:34:57 +02:00
214 lines
7.5 KiB
C#
214 lines
7.5 KiB
C#
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Entities.TV;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.TV;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.Querying;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
|
|
namespace MediaBrowser.Server.Implementations.TV
|
|
{
|
|
public class TVSeriesManager : ITVSeriesManager
|
|
{
|
|
private readonly IUserManager _userManager;
|
|
private readonly IUserDataManager _userDataManager;
|
|
private readonly ILibraryManager _libraryManager;
|
|
|
|
public TVSeriesManager(IUserManager userManager, IUserDataManager userDataManager, ILibraryManager libraryManager)
|
|
{
|
|
_userManager = userManager;
|
|
_userDataManager = userDataManager;
|
|
_libraryManager = libraryManager;
|
|
}
|
|
|
|
public QueryResult<BaseItem> GetNextUp(NextUpQuery request)
|
|
{
|
|
var user = _userManager.GetUserById(request.UserId);
|
|
|
|
if (user == null)
|
|
{
|
|
throw new ArgumentException("User not found");
|
|
}
|
|
|
|
var parentIds = string.IsNullOrEmpty(request.ParentId)
|
|
? new string[] { }
|
|
: new[] { request.ParentId };
|
|
|
|
string presentationUniqueKey = null;
|
|
int? limit = null;
|
|
if (!string.IsNullOrWhiteSpace(request.SeriesId))
|
|
{
|
|
var series = _libraryManager.GetItemById(request.SeriesId);
|
|
|
|
if (series != null)
|
|
{
|
|
presentationUniqueKey = series.PresentationUniqueKey;
|
|
limit = 1;
|
|
}
|
|
}
|
|
|
|
var items = _libraryManager.GetItemList(new InternalItemsQuery(user)
|
|
{
|
|
IncludeItemTypes = new[] { typeof(Series).Name },
|
|
SortOrder = SortOrder.Ascending,
|
|
PresentationUniqueKey = presentationUniqueKey,
|
|
Limit = limit
|
|
|
|
}, parentIds).Cast<Series>();
|
|
|
|
// Avoid implicitly captured closure
|
|
var episodes = GetNextUpEpisodes(request, user, items);
|
|
|
|
return GetResult(episodes, null, request);
|
|
}
|
|
|
|
public QueryResult<BaseItem> GetNextUp(NextUpQuery request, IEnumerable<Folder> parentsFolders)
|
|
{
|
|
var user = _userManager.GetUserById(request.UserId);
|
|
|
|
if (user == null)
|
|
{
|
|
throw new ArgumentException("User not found");
|
|
}
|
|
|
|
string presentationUniqueKey = null;
|
|
int? limit = null;
|
|
if (!string.IsNullOrWhiteSpace(request.SeriesId))
|
|
{
|
|
var series = _libraryManager.GetItemById(request.SeriesId);
|
|
|
|
if (series != null)
|
|
{
|
|
presentationUniqueKey = series.PresentationUniqueKey;
|
|
limit = 1;
|
|
}
|
|
}
|
|
|
|
var items = _libraryManager.GetItemList(new InternalItemsQuery(user)
|
|
{
|
|
IncludeItemTypes = new[] { typeof(Series).Name },
|
|
SortOrder = SortOrder.Ascending,
|
|
PresentationUniqueKey = presentationUniqueKey,
|
|
Limit = limit
|
|
|
|
}, parentsFolders.Select(i => i.Id.ToString("N"))).Cast<Series>();
|
|
|
|
// Avoid implicitly captured closure
|
|
var episodes = GetNextUpEpisodes(request, user, items);
|
|
|
|
return GetResult(episodes, null, request);
|
|
}
|
|
|
|
public IEnumerable<Episode> GetNextUpEpisodes(NextUpQuery request, User user, IEnumerable<Series> series)
|
|
{
|
|
// Avoid implicitly captured closure
|
|
var currentUser = user;
|
|
|
|
return series
|
|
.AsParallel()
|
|
.Select(i => GetNextUp(i, currentUser))
|
|
// Include if an episode was found, and either the series is not unwatched or the specific series was requested
|
|
.Where(i => i.Item1 != null && (!i.Item3 || !string.IsNullOrWhiteSpace(request.SeriesId)))
|
|
//.OrderByDescending(i =>
|
|
//{
|
|
// var episode = i.Item1;
|
|
|
|
// var seriesUserData = _userDataManager.GetUserData(user, episode.Series);
|
|
|
|
// if (seriesUserData.IsFavorite)
|
|
// {
|
|
// return 2;
|
|
// }
|
|
|
|
// if (seriesUserData.Likes.HasValue)
|
|
// {
|
|
// return seriesUserData.Likes.Value ? 1 : -1;
|
|
// }
|
|
|
|
// return 0;
|
|
//})
|
|
.OrderByDescending(i => i.Item2)
|
|
.ThenByDescending(i => i.Item1.PremiereDate ?? DateTime.MinValue)
|
|
.Select(i => i.Item1);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the next up.
|
|
/// </summary>
|
|
/// <param name="series">The series.</param>
|
|
/// <param name="user">The user.</param>
|
|
/// <returns>Task{Episode}.</returns>
|
|
private Tuple<Episode, DateTime, bool> GetNextUp(Series series, User user)
|
|
{
|
|
// Get them in display order, then reverse
|
|
var allEpisodes = series.GetSeasons(user, true, true)
|
|
.Where(i => !i.IndexNumber.HasValue || i.IndexNumber.Value != 0)
|
|
.SelectMany(i => i.GetEpisodes(user))
|
|
.Reverse()
|
|
.ToList();
|
|
|
|
Episode lastWatched = null;
|
|
var lastWatchedDate = DateTime.MinValue;
|
|
Episode nextUp = null;
|
|
|
|
var includeMissing = user.Configuration.DisplayMissingEpisodes;
|
|
|
|
// Go back starting with the most recent episodes
|
|
foreach (var episode in allEpisodes)
|
|
{
|
|
var userData = _userDataManager.GetUserData(user, episode);
|
|
|
|
if (userData.Played)
|
|
{
|
|
if (lastWatched != null || nextUp == null)
|
|
{
|
|
break;
|
|
}
|
|
|
|
lastWatched = episode;
|
|
lastWatchedDate = userData.LastPlayedDate ?? DateTime.MinValue;
|
|
}
|
|
else
|
|
{
|
|
if (!episode.IsVirtualUnaired && (includeMissing || !episode.IsMissingEpisode))
|
|
{
|
|
nextUp = episode;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (lastWatched != null)
|
|
{
|
|
return new Tuple<Episode, DateTime, bool>(nextUp, lastWatchedDate, false);
|
|
}
|
|
|
|
var firstEpisode = allEpisodes.LastOrDefault(i => !i.IsVirtualUnaired && (includeMissing || !i.IsMissingEpisode) && !i.IsPlayed(user));
|
|
|
|
// Return the first episode
|
|
return new Tuple<Episode, DateTime, bool>(firstEpisode, DateTime.MinValue, true);
|
|
}
|
|
|
|
private QueryResult<BaseItem> GetResult(IEnumerable<BaseItem> items, int? totalRecordLimit, NextUpQuery query)
|
|
{
|
|
var itemsArray = totalRecordLimit.HasValue ? items.Take(totalRecordLimit.Value).ToArray() : items.ToArray();
|
|
var totalCount = itemsArray.Length;
|
|
|
|
if (query.Limit.HasValue)
|
|
{
|
|
itemsArray = itemsArray.Skip(query.StartIndex ?? 0).Take(query.Limit.Value).ToArray();
|
|
}
|
|
else if (query.StartIndex.HasValue)
|
|
{
|
|
itemsArray = itemsArray.Skip(query.StartIndex.Value).ToArray();
|
|
}
|
|
|
|
return new QueryResult<BaseItem>
|
|
{
|
|
TotalRecordCount = totalCount,
|
|
Items = itemsArray
|
|
};
|
|
}
|
|
}
|
|
}
|