TVSeriesManager.cs 9.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261
  1. #pragma warning disable CS1591
  2. using System;
  3. using System.Collections.Generic;
  4. using System.Globalization;
  5. using System.Linq;
  6. using Jellyfin.Data.Entities;
  7. using Jellyfin.Data.Enums;
  8. using MediaBrowser.Controller.Configuration;
  9. using MediaBrowser.Controller.Dto;
  10. using MediaBrowser.Controller.Entities;
  11. using MediaBrowser.Controller.Library;
  12. using MediaBrowser.Controller.TV;
  13. using MediaBrowser.Model.Entities;
  14. using MediaBrowser.Model.Querying;
  15. using Episode = MediaBrowser.Controller.Entities.TV.Episode;
  16. using Series = MediaBrowser.Controller.Entities.TV.Series;
  17. namespace Emby.Server.Implementations.TV
  18. {
  19. public class TVSeriesManager : ITVSeriesManager
  20. {
  21. private readonly IUserManager _userManager;
  22. private readonly IUserDataManager _userDataManager;
  23. private readonly ILibraryManager _libraryManager;
  24. public TVSeriesManager(IUserManager userManager, IUserDataManager userDataManager, ILibraryManager libraryManager)
  25. {
  26. _userManager = userManager;
  27. _userDataManager = userDataManager;
  28. _libraryManager = libraryManager;
  29. }
  30. public QueryResult<BaseItem> GetNextUp(NextUpQuery request, DtoOptions dtoOptions)
  31. {
  32. var user = _userManager.GetUserById(request.UserId);
  33. if (user == null)
  34. {
  35. throw new ArgumentException("User not found");
  36. }
  37. string presentationUniqueKey = null;
  38. if (!string.IsNullOrEmpty(request.SeriesId))
  39. {
  40. var series = _libraryManager.GetItemById(request.SeriesId) as Series;
  41. if (series != null)
  42. {
  43. presentationUniqueKey = GetUniqueSeriesKey(series);
  44. }
  45. }
  46. if (!string.IsNullOrEmpty(presentationUniqueKey))
  47. {
  48. return GetResult(GetNextUpEpisodes(request, user, new[] { presentationUniqueKey }, dtoOptions), request);
  49. }
  50. BaseItem[] parents;
  51. if (request.ParentId.HasValue)
  52. {
  53. var parent = _libraryManager.GetItemById(request.ParentId.Value);
  54. if (parent != null)
  55. {
  56. parents = new[] { parent };
  57. }
  58. else
  59. {
  60. parents = Array.Empty<BaseItem>();
  61. }
  62. }
  63. else
  64. {
  65. parents = _libraryManager.GetUserRootFolder().GetChildren(user, true)
  66. .Where(i => i is Folder)
  67. .Where(i => !user.GetPreference(PreferenceKind.LatestItemExcludes)
  68. .Contains(i.Id.ToString("N", CultureInfo.InvariantCulture)))
  69. .ToArray();
  70. }
  71. return GetNextUp(request, parents, dtoOptions);
  72. }
  73. public QueryResult<BaseItem> GetNextUp(NextUpQuery request, BaseItem[] parentsFolders, DtoOptions dtoOptions)
  74. {
  75. var user = _userManager.GetUserById(request.UserId);
  76. if (user == null)
  77. {
  78. throw new ArgumentException("User not found");
  79. }
  80. string presentationUniqueKey = null;
  81. int? limit = null;
  82. if (!string.IsNullOrEmpty(request.SeriesId))
  83. {
  84. var series = _libraryManager.GetItemById(request.SeriesId) as Series;
  85. if (series != null)
  86. {
  87. presentationUniqueKey = GetUniqueSeriesKey(series);
  88. limit = 1;
  89. }
  90. }
  91. if (!string.IsNullOrEmpty(presentationUniqueKey))
  92. {
  93. return GetResult(GetNextUpEpisodes(request, user, new[] { presentationUniqueKey }, dtoOptions), request);
  94. }
  95. if (limit.HasValue)
  96. {
  97. limit = limit.Value + 10;
  98. }
  99. var items = _libraryManager
  100. .GetItemList(
  101. new InternalItemsQuery(user)
  102. {
  103. IncludeItemTypes = new[] { nameof(Episode) },
  104. OrderBy = new[] { new ValueTuple<string, SortOrder>(ItemSortBy.DatePlayed, SortOrder.Descending) },
  105. SeriesPresentationUniqueKey = presentationUniqueKey,
  106. Limit = limit,
  107. DtoOptions = new DtoOptions { Fields = new[] { ItemFields.SeriesPresentationUniqueKey }, EnableImages = false },
  108. GroupBySeriesPresentationUniqueKey = true
  109. }, parentsFolders.ToList())
  110. .Cast<Episode>()
  111. .Where(episode => !string.IsNullOrEmpty(episode.SeriesPresentationUniqueKey))
  112. .Select(GetUniqueSeriesKey);
  113. // Avoid implicitly captured closure
  114. var episodes = GetNextUpEpisodes(request, user, items, dtoOptions);
  115. return GetResult(episodes, request);
  116. }
  117. public IEnumerable<Episode> GetNextUpEpisodes(NextUpQuery request, User user, IEnumerable<string> seriesKeys, DtoOptions dtoOptions)
  118. {
  119. // Avoid implicitly captured closure
  120. var currentUser = user;
  121. var allNextUp = seriesKeys
  122. .Select(i => GetNextUp(i, currentUser, dtoOptions));
  123. return allNextUp
  124. .Where(i =>
  125. {
  126. return i.Item1 != DateTime.MinValue;
  127. })
  128. .Select(i => i.Item2())
  129. .Where(i => i != null);
  130. }
  131. private static string GetUniqueSeriesKey(Episode episode)
  132. {
  133. return episode.SeriesPresentationUniqueKey;
  134. }
  135. private static string GetUniqueSeriesKey(Series series)
  136. {
  137. return series.GetPresentationUniqueKey();
  138. }
  139. /// <summary>
  140. /// Gets the next up.
  141. /// </summary>
  142. /// <returns>Task{Episode}.</returns>
  143. private Tuple<DateTime, Func<Episode>> GetNextUp(string seriesKey, User user, DtoOptions dtoOptions)
  144. {
  145. var lastWatchedEpisode = _libraryManager.GetItemList(new InternalItemsQuery(user)
  146. {
  147. AncestorWithPresentationUniqueKey = null,
  148. SeriesPresentationUniqueKey = seriesKey,
  149. IncludeItemTypes = new[] { nameof(Episode) },
  150. OrderBy = new[] { new ValueTuple<string, SortOrder>(ItemSortBy.SortName, SortOrder.Descending) },
  151. IsPlayed = true,
  152. Limit = 1,
  153. ParentIndexNumberNotEquals = 0,
  154. DtoOptions = new DtoOptions
  155. {
  156. Fields = new ItemFields[]
  157. {
  158. ItemFields.SortName
  159. },
  160. EnableImages = false
  161. }
  162. }).FirstOrDefault();
  163. Func<Episode> getEpisode = () =>
  164. {
  165. var nextEpisode = _libraryManager.GetItemList(new InternalItemsQuery(user)
  166. {
  167. AncestorWithPresentationUniqueKey = null,
  168. SeriesPresentationUniqueKey = seriesKey,
  169. IncludeItemTypes = new[] { nameof(Episode) },
  170. OrderBy = new[] { new ValueTuple<string, SortOrder>(ItemSortBy.SortName, SortOrder.Ascending) },
  171. Limit = 1,
  172. IsPlayed = false,
  173. IsVirtualItem = false,
  174. ParentIndexNumberNotEquals = 0,
  175. MinSortName = lastWatchedEpisode?.SortName,
  176. DtoOptions = dtoOptions
  177. }).Cast<Episode>().FirstOrDefault();
  178. if (nextEpisode != null)
  179. {
  180. var userData = _userDataManager.GetUserData(user, nextEpisode);
  181. if (userData.PlaybackPositionTicks > 0)
  182. {
  183. return null;
  184. }
  185. }
  186. return nextEpisode;
  187. };
  188. if (lastWatchedEpisode != null)
  189. {
  190. var userData = _userDataManager.GetUserData(user, lastWatchedEpisode);
  191. var lastWatchedDate = userData.LastPlayedDate ?? DateTime.MinValue.AddDays(1);
  192. return new Tuple<DateTime, Func<Episode>>(lastWatchedDate, getEpisode);
  193. }
  194. // Return the first episode
  195. return new Tuple<DateTime, Func<Episode>>(DateTime.MinValue, getEpisode);
  196. }
  197. private static QueryResult<BaseItem> GetResult(IEnumerable<BaseItem> items, NextUpQuery query)
  198. {
  199. int totalCount = 0;
  200. if (query.EnableTotalRecordCount)
  201. {
  202. var list = items.ToList();
  203. totalCount = list.Count;
  204. items = list;
  205. }
  206. if (query.StartIndex.HasValue)
  207. {
  208. items = items.Skip(query.StartIndex.Value);
  209. }
  210. if (query.Limit.HasValue)
  211. {
  212. items = items.Take(query.Limit.Value);
  213. }
  214. return new QueryResult<BaseItem>
  215. {
  216. TotalRecordCount = totalCount,
  217. Items = items.ToArray()
  218. };
  219. }
  220. }
  221. }