mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-05 21:37:10 -04:00
249 lines
9.1 KiB
C#
249 lines
9.1 KiB
C#
using MediaBrowser.Controller.Dto;
|
|
using MediaBrowser.Controller.Entities;
|
|
using MediaBrowser.Controller.Entities.TV;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.Persistence;
|
|
using MediaBrowser.Model.Dto;
|
|
using MediaBrowser.Model.Querying;
|
|
using ServiceStack.ServiceHost;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace MediaBrowser.Api
|
|
{
|
|
/// <summary>
|
|
/// Class GetNextUpEpisodes
|
|
/// </summary>
|
|
[Route("/Shows/NextUp", "GET")]
|
|
[Api(("Gets a list of currently installed plugins"))]
|
|
public class GetNextUpEpisodes : IReturn<ItemsResult>
|
|
{
|
|
/// <summary>
|
|
/// Gets or sets the user id.
|
|
/// </summary>
|
|
/// <value>The user id.</value>
|
|
[ApiMember(Name = "UserId", Description = "User Id", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "GET")]
|
|
public Guid UserId { get; set; }
|
|
|
|
/// <summary>
|
|
/// Skips over a given number of items within the results. Use for paging.
|
|
/// </summary>
|
|
/// <value>The start index.</value>
|
|
[ApiMember(Name = "StartIndex", Description = "Optional. The record index to start at. All items with a lower index will be dropped from the results.", IsRequired = false, DataType = "int", ParameterType = "query", Verb = "GET")]
|
|
public int? StartIndex { 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>
|
|
/// Fields to return within the items, in addition to basic information
|
|
/// </summary>
|
|
/// <value>The fields.</value>
|
|
[ApiMember(Name = "Fields", Description = "Optional. Specify additional fields of information to return in the output. This allows multiple, comma delimeted. Options: AudioInfo, Budget, Chapters, DateCreated, DisplayMediaType, DisplayPreferences, EndDate, Genres, HomePageUrl, ItemCounts, IndexOptions, Locations, MediaStreams, Overview, OverviewHtml, ParentId, Path, People, ProviderIds, PrimaryImageAspectRatio, Revenue, SeriesInfo, SortName, Studios, Taglines, TrailerUrls, UserData", IsRequired = false, DataType = "string", ParameterType = "query", Verb = "GET", AllowMultiple = true)]
|
|
public string Fields { get; set; }
|
|
|
|
/// <summary>
|
|
/// Gets the item fields.
|
|
/// </summary>
|
|
/// <returns>IEnumerable{ItemFields}.</returns>
|
|
public IEnumerable<ItemFields> GetItemFields()
|
|
{
|
|
var val = Fields;
|
|
|
|
if (string.IsNullOrEmpty(val))
|
|
{
|
|
return new ItemFields[] { };
|
|
}
|
|
|
|
return val.Split(',').Select(v => (ItemFields)Enum.Parse(typeof(ItemFields), v, true));
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Class TvShowsService
|
|
/// </summary>
|
|
public class TvShowsService : BaseApiService
|
|
{
|
|
/// <summary>
|
|
/// The _user manager
|
|
/// </summary>
|
|
private readonly IUserManager _userManager;
|
|
|
|
/// <summary>
|
|
/// The _user data repository
|
|
/// </summary>
|
|
private readonly IUserDataRepository _userDataRepository;
|
|
/// <summary>
|
|
/// The _library manager
|
|
/// </summary>
|
|
private readonly ILibraryManager _libraryManager;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="TvShowsService" /> class.
|
|
/// </summary>
|
|
/// <param name="userManager">The user manager.</param>
|
|
/// <param name="userDataRepository">The user data repository.</param>
|
|
/// <param name="libraryManager">The library manager.</param>
|
|
public TvShowsService(IUserManager userManager, IUserDataRepository userDataRepository, ILibraryManager libraryManager)
|
|
{
|
|
_userManager = userManager;
|
|
_userDataRepository = userDataRepository;
|
|
_libraryManager = libraryManager;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the specified request.
|
|
/// </summary>
|
|
/// <param name="request">The request.</param>
|
|
/// <returns>System.Object.</returns>
|
|
public object Get(GetNextUpEpisodes request)
|
|
{
|
|
var result = GetNextUpEpisodes(request).Result;
|
|
|
|
return ToOptimizedResult(result);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the next up episodes.
|
|
/// </summary>
|
|
/// <param name="request">The request.</param>
|
|
/// <returns>Task{ItemsResult}.</returns>
|
|
private async Task<ItemsResult> GetNextUpEpisodes(GetNextUpEpisodes request)
|
|
{
|
|
var user = _userManager.GetUserById(request.UserId);
|
|
|
|
var tasks = user.RootFolder
|
|
.GetRecursiveChildren(user)
|
|
.OfType<Series>()
|
|
.AsParallel()
|
|
.Select(i => GetNextUp(i, user));
|
|
|
|
var itemsArray = await Task.WhenAll(tasks).ConfigureAwait(false);
|
|
|
|
itemsArray = itemsArray
|
|
.Where(i => i.Item1 != null)
|
|
.OrderByDescending(i =>
|
|
{
|
|
var seriesUserData =
|
|
_userDataRepository.GetUserData(user.Id, i.Item1.Series.GetUserDataKey()).Result;
|
|
|
|
if (seriesUserData.IsFavorite)
|
|
{
|
|
return 2;
|
|
}
|
|
|
|
if (seriesUserData.Likes.HasValue)
|
|
{
|
|
return seriesUserData.Likes.Value ? 1 : -1;
|
|
}
|
|
|
|
return 0;
|
|
})
|
|
.ThenByDescending(i => i.Item1.PremiereDate ?? DateTime.MinValue)
|
|
.ToArray();
|
|
|
|
var pagedItems = ApplyPaging(request, itemsArray.Select(i => i.Item1));
|
|
|
|
var fields = request.GetItemFields().ToList();
|
|
|
|
var returnItems = await GetItemDtos(pagedItems, user, fields).ConfigureAwait(false);
|
|
|
|
return new ItemsResult
|
|
{
|
|
TotalRecordCount = itemsArray.Length,
|
|
Items = returnItems
|
|
};
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the next up.
|
|
/// </summary>
|
|
/// <param name="series">The series.</param>
|
|
/// <param name="user">The user.</param>
|
|
/// <returns>Task{Episode}.</returns>
|
|
private async Task<Tuple<Episode,DateTime>> GetNextUp(Series series, User user)
|
|
{
|
|
var allEpisodes = series.GetRecursiveChildren(user)
|
|
.OfType<Episode>()
|
|
.OrderByDescending(i => i.PremiereDate ?? DateTime.MinValue)
|
|
.ThenByDescending(i => i.IndexNumber ?? 0)
|
|
.ToList();
|
|
|
|
Episode lastWatched = null;
|
|
var lastWatchedDate = DateTime.MinValue;
|
|
Episode nextUp = null;
|
|
|
|
// Go back starting with the most recent episodes
|
|
foreach (var episode in allEpisodes)
|
|
{
|
|
var userData = await _userDataRepository.GetUserData(user.Id, episode.GetUserDataKey()).ConfigureAwait(false);
|
|
|
|
if (userData.Played)
|
|
{
|
|
if (lastWatched != null || nextUp == null)
|
|
{
|
|
break;
|
|
}
|
|
|
|
lastWatched = episode;
|
|
lastWatchedDate = userData.LastPlayedDate ?? DateTime.MinValue;
|
|
}
|
|
else
|
|
{
|
|
nextUp = episode;
|
|
}
|
|
}
|
|
|
|
if (lastWatched != null)
|
|
{
|
|
return new Tuple<Episode, DateTime>(nextUp, lastWatchedDate);
|
|
}
|
|
|
|
return new Tuple<Episode, DateTime>(null, lastWatchedDate);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the item dtos.
|
|
/// </summary>
|
|
/// <param name="pagedItems">The paged items.</param>
|
|
/// <param name="user">The user.</param>
|
|
/// <param name="fields">The fields.</param>
|
|
/// <returns>Task.</returns>
|
|
private Task<BaseItemDto[]> GetItemDtos(IEnumerable<BaseItem> pagedItems, User user, List<ItemFields> fields)
|
|
{
|
|
var dtoBuilder = new DtoBuilder(Logger, _libraryManager, _userDataRepository);
|
|
|
|
return Task.WhenAll(pagedItems.Select(i => dtoBuilder.GetBaseItemDto(i, user, fields)));
|
|
}
|
|
|
|
/// <summary>
|
|
/// Applies the paging.
|
|
/// </summary>
|
|
/// <param name="request">The request.</param>
|
|
/// <param name="items">The items.</param>
|
|
/// <returns>IEnumerable{BaseItem}.</returns>
|
|
private IEnumerable<BaseItem> ApplyPaging(GetNextUpEpisodes request, IEnumerable<BaseItem> items)
|
|
{
|
|
// Start at
|
|
if (request.StartIndex.HasValue)
|
|
{
|
|
items = items.Skip(request.StartIndex.Value);
|
|
}
|
|
|
|
// Return limit
|
|
if (request.Limit.HasValue)
|
|
{
|
|
items = items.Take(request.Limit.Value);
|
|
}
|
|
|
|
return items;
|
|
}
|
|
}
|
|
}
|