UserViewBuilder.cs 34 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027
  1. #nullable disable
  2. #pragma warning disable CS1591
  3. using System;
  4. using System.Collections.Generic;
  5. using System.Globalization;
  6. using System.Linq;
  7. using Jellyfin.Data;
  8. using Jellyfin.Data.Enums;
  9. using Jellyfin.Database.Implementations.Entities;
  10. using Jellyfin.Database.Implementations.Enums;
  11. using Jellyfin.Extensions;
  12. using MediaBrowser.Controller.Library;
  13. using MediaBrowser.Controller.TV;
  14. using MediaBrowser.Model.Entities;
  15. using MediaBrowser.Model.Querying;
  16. using Microsoft.Extensions.Logging;
  17. using Episode = MediaBrowser.Controller.Entities.TV.Episode;
  18. using MetadataProvider = MediaBrowser.Model.Entities.MetadataProvider;
  19. using Series = MediaBrowser.Controller.Entities.TV.Series;
  20. namespace MediaBrowser.Controller.Entities
  21. {
  22. public class UserViewBuilder
  23. {
  24. private readonly IUserViewManager _userViewManager;
  25. private readonly ILibraryManager _libraryManager;
  26. private readonly ILogger<BaseItem> _logger;
  27. private readonly IUserDataManager _userDataManager;
  28. private readonly ITVSeriesManager _tvSeriesManager;
  29. public UserViewBuilder(
  30. IUserViewManager userViewManager,
  31. ILibraryManager libraryManager,
  32. ILogger<BaseItem> logger,
  33. IUserDataManager userDataManager,
  34. ITVSeriesManager tvSeriesManager)
  35. {
  36. _userViewManager = userViewManager;
  37. _libraryManager = libraryManager;
  38. _logger = logger;
  39. _userDataManager = userDataManager;
  40. _tvSeriesManager = tvSeriesManager;
  41. }
  42. public QueryResult<BaseItem> GetUserItems(Folder queryParent, Folder displayParent, CollectionType? viewType, InternalItemsQuery query)
  43. {
  44. var user = query.User;
  45. // if (query.IncludeItemTypes is not null &&
  46. // query.IncludeItemTypes.Length == 1 &&
  47. // string.Equals(query.IncludeItemTypes[0], "Playlist", StringComparison.OrdinalIgnoreCase))
  48. // {
  49. // if (!string.Equals(viewType, CollectionType.Playlists, StringComparison.OrdinalIgnoreCase))
  50. // {
  51. // return await FindPlaylists(queryParent, user, query).ConfigureAwait(false);
  52. // }
  53. // }
  54. switch (viewType)
  55. {
  56. case CollectionType.folders:
  57. return GetResult(_libraryManager.GetUserRootFolder().GetChildren(user, true), query);
  58. case CollectionType.tvshows:
  59. return GetTvView(queryParent, user, query);
  60. case CollectionType.movies:
  61. return GetMovieFolders(queryParent, user, query);
  62. case CollectionType.tvshowseries:
  63. return GetTvSeries(queryParent, user, query);
  64. case CollectionType.tvgenres:
  65. return GetTvGenres(queryParent, user, query);
  66. case CollectionType.tvgenre:
  67. return GetTvGenreItems(queryParent, displayParent, user, query);
  68. case CollectionType.tvresume:
  69. return GetTvResume(queryParent, user, query);
  70. case CollectionType.tvnextup:
  71. return GetTvNextUp(queryParent, query);
  72. case CollectionType.tvlatest:
  73. return GetTvLatest(queryParent, user, query);
  74. case CollectionType.moviefavorites:
  75. return GetFavoriteMovies(queryParent, user, query);
  76. case CollectionType.movielatest:
  77. return GetMovieLatest(queryParent, user, query);
  78. case CollectionType.moviegenres:
  79. return GetMovieGenres(queryParent, user, query);
  80. case CollectionType.moviegenre:
  81. return GetMovieGenreItems(queryParent, displayParent, user, query);
  82. case CollectionType.movieresume:
  83. return GetMovieResume(queryParent, user, query);
  84. case CollectionType.moviemovies:
  85. return GetMovieMovies(queryParent, user, query);
  86. case CollectionType.moviecollection:
  87. return GetMovieCollections(user, query);
  88. case CollectionType.tvfavoriteepisodes:
  89. return GetFavoriteEpisodes(queryParent, user, query);
  90. case CollectionType.tvfavoriteseries:
  91. return GetFavoriteSeries(queryParent, user, query);
  92. default:
  93. {
  94. if (queryParent is UserView)
  95. {
  96. return GetResult(GetMediaFolders(user).OfType<Folder>().SelectMany(i => i.GetChildren(user, true)), query);
  97. }
  98. return queryParent.GetItems(query);
  99. }
  100. }
  101. }
  102. private int GetSpecialItemsLimit()
  103. {
  104. return 50;
  105. }
  106. private QueryResult<BaseItem> GetMovieFolders(Folder parent, User user, InternalItemsQuery query)
  107. {
  108. if (query.Recursive)
  109. {
  110. query.Recursive = true;
  111. query.SetUser(user);
  112. if (query.IncludeItemTypes.Length == 0)
  113. {
  114. query.IncludeItemTypes = new[] { BaseItemKind.Movie };
  115. }
  116. return parent.QueryRecursive(query);
  117. }
  118. var list = new List<BaseItem>
  119. {
  120. GetUserView(CollectionType.movieresume, "HeaderContinueWatching", "0", parent),
  121. GetUserView(CollectionType.movielatest, "Latest", "1", parent),
  122. GetUserView(CollectionType.moviemovies, "Movies", "2", parent),
  123. GetUserView(CollectionType.moviecollection, "Collections", "3", parent),
  124. GetUserView(CollectionType.moviefavorites, "Favorites", "4", parent),
  125. GetUserView(CollectionType.moviegenres, "Genres", "5", parent)
  126. };
  127. return GetResult(list, query);
  128. }
  129. private QueryResult<BaseItem> GetFavoriteMovies(Folder parent, User user, InternalItemsQuery query)
  130. {
  131. query.Recursive = true;
  132. query.Parent = parent;
  133. query.SetUser(user);
  134. query.IsFavorite = true;
  135. query.IncludeItemTypes = new[] { BaseItemKind.Movie };
  136. return _libraryManager.GetItemsResult(query);
  137. }
  138. private QueryResult<BaseItem> GetFavoriteSeries(Folder parent, User user, InternalItemsQuery query)
  139. {
  140. query.Recursive = true;
  141. query.Parent = parent;
  142. query.SetUser(user);
  143. query.IsFavorite = true;
  144. query.IncludeItemTypes = new[] { BaseItemKind.Series };
  145. return _libraryManager.GetItemsResult(query);
  146. }
  147. private QueryResult<BaseItem> GetFavoriteEpisodes(Folder parent, User user, InternalItemsQuery query)
  148. {
  149. query.Recursive = true;
  150. query.Parent = parent;
  151. query.SetUser(user);
  152. query.IsFavorite = true;
  153. query.IncludeItemTypes = new[] { BaseItemKind.Episode };
  154. return _libraryManager.GetItemsResult(query);
  155. }
  156. private QueryResult<BaseItem> GetMovieMovies(Folder parent, User user, InternalItemsQuery query)
  157. {
  158. query.Recursive = true;
  159. query.Parent = parent;
  160. query.SetUser(user);
  161. query.IncludeItemTypes = new[] { BaseItemKind.Movie };
  162. return _libraryManager.GetItemsResult(query);
  163. }
  164. private QueryResult<BaseItem> GetMovieCollections(User user, InternalItemsQuery query)
  165. {
  166. query.Parent = null;
  167. query.IncludeItemTypes = new[] { BaseItemKind.BoxSet };
  168. query.SetUser(user);
  169. query.Recursive = true;
  170. return _libraryManager.GetItemsResult(query);
  171. }
  172. private QueryResult<BaseItem> GetMovieLatest(Folder parent, User user, InternalItemsQuery query)
  173. {
  174. query.OrderBy = new[] { (ItemSortBy.DateCreated, SortOrder.Descending), (ItemSortBy.SortName, SortOrder.Descending) };
  175. query.Recursive = true;
  176. query.Parent = parent;
  177. query.SetUser(user);
  178. query.Limit = GetSpecialItemsLimit();
  179. query.IncludeItemTypes = new[] { BaseItemKind.Movie };
  180. return ConvertToResult(_libraryManager.GetItemList(query));
  181. }
  182. private QueryResult<BaseItem> GetMovieResume(Folder parent, User user, InternalItemsQuery query)
  183. {
  184. query.OrderBy = new[] { (ItemSortBy.DatePlayed, SortOrder.Descending), (ItemSortBy.SortName, SortOrder.Descending) };
  185. query.IsResumable = true;
  186. query.Recursive = true;
  187. query.Parent = parent;
  188. query.SetUser(user);
  189. query.Limit = GetSpecialItemsLimit();
  190. query.IncludeItemTypes = new[] { BaseItemKind.Movie };
  191. return ConvertToResult(_libraryManager.GetItemList(query));
  192. }
  193. private QueryResult<BaseItem> ConvertToResult(IReadOnlyList<BaseItem> items)
  194. {
  195. return new QueryResult<BaseItem>(items);
  196. }
  197. private QueryResult<BaseItem> GetMovieGenres(Folder parent, User user, InternalItemsQuery query)
  198. {
  199. var genres = parent.QueryRecursive(new InternalItemsQuery(user)
  200. {
  201. IncludeItemTypes = new[] { BaseItemKind.Movie },
  202. Recursive = true,
  203. EnableTotalRecordCount = false
  204. }).Items
  205. .SelectMany(i => i.Genres)
  206. .DistinctNames()
  207. .Select(i =>
  208. {
  209. try
  210. {
  211. return _libraryManager.GetGenre(i);
  212. }
  213. catch (Exception ex)
  214. {
  215. _logger.LogError(ex, "Error getting genre");
  216. return null;
  217. }
  218. })
  219. .Where(i => i is not null)
  220. .Select(i => GetUserViewWithName(CollectionType.moviegenre, i.SortName, parent));
  221. return GetResult(genres, query);
  222. }
  223. private QueryResult<BaseItem> GetMovieGenreItems(Folder queryParent, Folder displayParent, User user, InternalItemsQuery query)
  224. {
  225. query.Recursive = true;
  226. query.Parent = queryParent;
  227. query.GenreIds = new[] { displayParent.Id };
  228. query.SetUser(user);
  229. query.IncludeItemTypes = new[] { BaseItemKind.Movie };
  230. return _libraryManager.GetItemsResult(query);
  231. }
  232. private QueryResult<BaseItem> GetTvView(Folder parent, User user, InternalItemsQuery query)
  233. {
  234. if (query.Recursive)
  235. {
  236. query.Recursive = true;
  237. query.SetUser(user);
  238. if (query.IncludeItemTypes.Length == 0)
  239. {
  240. query.IncludeItemTypes = new[]
  241. {
  242. BaseItemKind.Series,
  243. BaseItemKind.Season,
  244. BaseItemKind.Episode
  245. };
  246. }
  247. return parent.QueryRecursive(query);
  248. }
  249. var list = new List<BaseItem>
  250. {
  251. GetUserView(CollectionType.tvresume, "HeaderContinueWatching", "0", parent),
  252. GetUserView(CollectionType.tvnextup, "HeaderNextUp", "1", parent),
  253. GetUserView(CollectionType.tvlatest, "Latest", "2", parent),
  254. GetUserView(CollectionType.tvshowseries, "Shows", "3", parent),
  255. GetUserView(CollectionType.tvfavoriteseries, "HeaderFavoriteShows", "4", parent),
  256. GetUserView(CollectionType.tvfavoriteepisodes, "HeaderFavoriteEpisodes", "5", parent),
  257. GetUserView(CollectionType.tvgenres, "Genres", "6", parent)
  258. };
  259. return GetResult(list, query);
  260. }
  261. private QueryResult<BaseItem> GetTvLatest(Folder parent, User user, InternalItemsQuery query)
  262. {
  263. query.OrderBy = new[] { (ItemSortBy.DateCreated, SortOrder.Descending), (ItemSortBy.SortName, SortOrder.Descending) };
  264. query.Recursive = true;
  265. query.Parent = parent;
  266. query.SetUser(user);
  267. query.Limit = GetSpecialItemsLimit();
  268. query.IncludeItemTypes = new[] { BaseItemKind.Episode };
  269. query.IsVirtualItem = false;
  270. return ConvertToResult(_libraryManager.GetItemList(query));
  271. }
  272. private QueryResult<BaseItem> GetTvNextUp(Folder parent, InternalItemsQuery query)
  273. {
  274. var parentFolders = GetMediaFolders(parent, query.User, new[] { CollectionType.tvshows });
  275. var result = _tvSeriesManager.GetNextUp(
  276. new NextUpQuery
  277. {
  278. Limit = query.Limit,
  279. StartIndex = query.StartIndex,
  280. User = query.User
  281. },
  282. parentFolders,
  283. query.DtoOptions);
  284. return result;
  285. }
  286. private QueryResult<BaseItem> GetTvResume(Folder parent, User user, InternalItemsQuery query)
  287. {
  288. query.OrderBy = new[] { (ItemSortBy.DatePlayed, SortOrder.Descending), (ItemSortBy.SortName, SortOrder.Descending) };
  289. query.IsResumable = true;
  290. query.Recursive = true;
  291. query.Parent = parent;
  292. query.SetUser(user);
  293. query.Limit = GetSpecialItemsLimit();
  294. query.IncludeItemTypes = new[] { BaseItemKind.Episode };
  295. return ConvertToResult(_libraryManager.GetItemList(query));
  296. }
  297. private QueryResult<BaseItem> GetTvSeries(Folder parent, User user, InternalItemsQuery query)
  298. {
  299. query.Recursive = true;
  300. query.Parent = parent;
  301. query.SetUser(user);
  302. query.IncludeItemTypes = new[] { BaseItemKind.Series };
  303. return _libraryManager.GetItemsResult(query);
  304. }
  305. private QueryResult<BaseItem> GetTvGenres(Folder parent, User user, InternalItemsQuery query)
  306. {
  307. var genres = parent.QueryRecursive(new InternalItemsQuery(user)
  308. {
  309. IncludeItemTypes = new[] { BaseItemKind.Series },
  310. Recursive = true,
  311. EnableTotalRecordCount = false
  312. }).Items
  313. .SelectMany(i => i.Genres)
  314. .DistinctNames()
  315. .Select(i =>
  316. {
  317. try
  318. {
  319. return _libraryManager.GetGenre(i);
  320. }
  321. catch (Exception ex)
  322. {
  323. _logger.LogError(ex, "Error getting genre");
  324. return null;
  325. }
  326. })
  327. .Where(i => i is not null)
  328. .Select(i => GetUserViewWithName(CollectionType.tvgenre, i.SortName, parent));
  329. return GetResult(genres, query);
  330. }
  331. private QueryResult<BaseItem> GetTvGenreItems(Folder queryParent, Folder displayParent, User user, InternalItemsQuery query)
  332. {
  333. query.Recursive = true;
  334. query.Parent = queryParent;
  335. query.GenreIds = new[] { displayParent.Id };
  336. query.SetUser(user);
  337. query.IncludeItemTypes = new[] { BaseItemKind.Series };
  338. return _libraryManager.GetItemsResult(query);
  339. }
  340. private QueryResult<BaseItem> GetResult<T>(
  341. IEnumerable<T> items,
  342. InternalItemsQuery query)
  343. where T : BaseItem
  344. {
  345. items = items.Where(i => Filter(i, query.User, query, _userDataManager, _libraryManager));
  346. return PostFilterAndSort(items, null, query, _libraryManager);
  347. }
  348. public static bool FilterItem(BaseItem item, InternalItemsQuery query)
  349. {
  350. return Filter(item, query.User, query, BaseItem.UserDataManager, BaseItem.LibraryManager);
  351. }
  352. public static QueryResult<BaseItem> PostFilterAndSort(
  353. IEnumerable<BaseItem> items,
  354. int? totalRecordLimit,
  355. InternalItemsQuery query,
  356. ILibraryManager libraryManager)
  357. {
  358. // This must be the last filter
  359. if (!query.AdjacentTo.IsNullOrEmpty())
  360. {
  361. items = FilterForAdjacency(items.ToList(), query.AdjacentTo.Value);
  362. }
  363. return SortAndPage(items, totalRecordLimit, query, libraryManager);
  364. }
  365. public static QueryResult<BaseItem> SortAndPage(
  366. IEnumerable<BaseItem> items,
  367. int? totalRecordLimit,
  368. InternalItemsQuery query,
  369. ILibraryManager libraryManager)
  370. {
  371. if (query.OrderBy.Count > 0)
  372. {
  373. items = libraryManager.Sort(items, query.User, query.OrderBy);
  374. }
  375. var itemsArray = totalRecordLimit.HasValue ? items.Take(totalRecordLimit.Value).ToArray() : items.ToArray();
  376. var totalCount = itemsArray.Length;
  377. if (query.Limit.HasValue)
  378. {
  379. itemsArray = itemsArray.Skip(query.StartIndex ?? 0).Take(query.Limit.Value).ToArray();
  380. }
  381. else if (query.StartIndex.HasValue)
  382. {
  383. itemsArray = itemsArray.Skip(query.StartIndex.Value).ToArray();
  384. }
  385. return new QueryResult<BaseItem>(
  386. query.StartIndex,
  387. totalCount,
  388. itemsArray);
  389. }
  390. public static bool Filter(BaseItem item, User user, InternalItemsQuery query, IUserDataManager userDataManager, ILibraryManager libraryManager)
  391. {
  392. if (!string.IsNullOrEmpty(query.NameStartsWith) && !item.SortName.StartsWith(query.NameStartsWith, StringComparison.InvariantCultureIgnoreCase))
  393. {
  394. return false;
  395. }
  396. #pragma warning disable CA1309 // Use ordinal string comparison
  397. if (!string.IsNullOrEmpty(query.NameStartsWithOrGreater) && string.Compare(query.NameStartsWithOrGreater, item.SortName, StringComparison.InvariantCultureIgnoreCase) == 1)
  398. {
  399. return false;
  400. }
  401. if (!string.IsNullOrEmpty(query.NameLessThan) && string.Compare(query.NameLessThan, item.SortName, StringComparison.InvariantCultureIgnoreCase) != 1)
  402. #pragma warning restore CA1309 // Use ordinal string comparison
  403. {
  404. return false;
  405. }
  406. if (query.MediaTypes.Length > 0 && !query.MediaTypes.Contains(item.MediaType))
  407. {
  408. return false;
  409. }
  410. if (query.IncludeItemTypes.Length > 0 && !query.IncludeItemTypes.Contains(item.GetBaseItemKind()))
  411. {
  412. return false;
  413. }
  414. if (query.ExcludeItemTypes.Length > 0 && query.ExcludeItemTypes.Contains(item.GetBaseItemKind()))
  415. {
  416. return false;
  417. }
  418. if (query.IsVirtualItem.HasValue && item.IsVirtualItem != query.IsVirtualItem.Value)
  419. {
  420. return false;
  421. }
  422. if (query.IsFolder.HasValue && query.IsFolder.Value != item.IsFolder)
  423. {
  424. return false;
  425. }
  426. UserItemData userData = null;
  427. if (query.IsLiked.HasValue)
  428. {
  429. userData = userDataManager.GetUserData(user, item);
  430. if (!userData.Likes.HasValue || userData.Likes != query.IsLiked.Value)
  431. {
  432. return false;
  433. }
  434. }
  435. if (query.IsFavoriteOrLiked.HasValue)
  436. {
  437. userData ??= userDataManager.GetUserData(user, item);
  438. var isFavoriteOrLiked = userData.IsFavorite || (userData.Likes ?? false);
  439. if (isFavoriteOrLiked != query.IsFavoriteOrLiked.Value)
  440. {
  441. return false;
  442. }
  443. }
  444. if (query.IsFavorite.HasValue)
  445. {
  446. userData ??= userDataManager.GetUserData(user, item);
  447. if (userData.IsFavorite != query.IsFavorite.Value)
  448. {
  449. return false;
  450. }
  451. }
  452. if (query.IsResumable.HasValue)
  453. {
  454. userData ??= userDataManager.GetUserData(user, item);
  455. var isResumable = userData.PlaybackPositionTicks > 0;
  456. if (isResumable != query.IsResumable.Value)
  457. {
  458. return false;
  459. }
  460. }
  461. if (query.IsPlayed.HasValue)
  462. {
  463. userData ??= userDataManager.GetUserData(user, item);
  464. if (item.IsPlayed(user, userData) != query.IsPlayed.Value)
  465. {
  466. return false;
  467. }
  468. }
  469. // Filter by Video3DFormat
  470. if (query.Is3D.HasValue)
  471. {
  472. var val = query.Is3D.Value;
  473. var video = item as Video;
  474. if (video is null || val != video.Video3DFormat.HasValue)
  475. {
  476. return false;
  477. }
  478. }
  479. /*
  480. * fuck - fix this
  481. if (query.IsHD.HasValue)
  482. {
  483. if (item.IsHD != query.IsHD.Value)
  484. {
  485. return false;
  486. }
  487. }
  488. */
  489. if (query.IsLocked.HasValue)
  490. {
  491. var val = query.IsLocked.Value;
  492. if (item.IsLocked != val)
  493. {
  494. return false;
  495. }
  496. }
  497. if (query.HasOverview.HasValue)
  498. {
  499. var filterValue = query.HasOverview.Value;
  500. var hasValue = !string.IsNullOrEmpty(item.Overview);
  501. if (hasValue != filterValue)
  502. {
  503. return false;
  504. }
  505. }
  506. if (query.HasImdbId.HasValue)
  507. {
  508. var filterValue = query.HasImdbId.Value;
  509. var hasValue = !string.IsNullOrEmpty(item.GetProviderId(MetadataProvider.Imdb));
  510. if (hasValue != filterValue)
  511. {
  512. return false;
  513. }
  514. }
  515. if (query.HasTmdbId.HasValue)
  516. {
  517. var filterValue = query.HasTmdbId.Value;
  518. var hasValue = !string.IsNullOrEmpty(item.GetProviderId(MetadataProvider.Tmdb));
  519. if (hasValue != filterValue)
  520. {
  521. return false;
  522. }
  523. }
  524. if (query.HasTvdbId.HasValue)
  525. {
  526. var filterValue = query.HasTvdbId.Value;
  527. var hasValue = !string.IsNullOrEmpty(item.GetProviderId(MetadataProvider.Tvdb));
  528. if (hasValue != filterValue)
  529. {
  530. return false;
  531. }
  532. }
  533. if (query.HasOfficialRating.HasValue)
  534. {
  535. var filterValue = query.HasOfficialRating.Value;
  536. var hasValue = !string.IsNullOrEmpty(item.OfficialRating);
  537. if (hasValue != filterValue)
  538. {
  539. return false;
  540. }
  541. }
  542. if (query.IsPlaceHolder.HasValue)
  543. {
  544. var filterValue = query.IsPlaceHolder.Value;
  545. var isPlaceHolder = false;
  546. if (item is ISupportsPlaceHolders hasPlaceHolder)
  547. {
  548. isPlaceHolder = hasPlaceHolder.IsPlaceHolder;
  549. }
  550. if (isPlaceHolder != filterValue)
  551. {
  552. return false;
  553. }
  554. }
  555. if (query.HasSpecialFeature.HasValue)
  556. {
  557. var filterValue = query.HasSpecialFeature.Value;
  558. if (item is IHasSpecialFeatures movie)
  559. {
  560. var ok = filterValue
  561. ? movie.SpecialFeatureIds.Count > 0
  562. : movie.SpecialFeatureIds.Count == 0;
  563. if (!ok)
  564. {
  565. return false;
  566. }
  567. }
  568. else
  569. {
  570. return false;
  571. }
  572. }
  573. if (query.HasSubtitles.HasValue)
  574. {
  575. var val = query.HasSubtitles.Value;
  576. var video = item as Video;
  577. if (video is null || val != video.HasSubtitles)
  578. {
  579. return false;
  580. }
  581. }
  582. if (query.HasParentalRating.HasValue)
  583. {
  584. var val = query.HasParentalRating.Value;
  585. var rating = item.CustomRating;
  586. if (string.IsNullOrEmpty(rating))
  587. {
  588. rating = item.OfficialRating;
  589. }
  590. if (val)
  591. {
  592. if (string.IsNullOrEmpty(rating))
  593. {
  594. return false;
  595. }
  596. }
  597. else
  598. {
  599. if (!string.IsNullOrEmpty(rating))
  600. {
  601. return false;
  602. }
  603. }
  604. }
  605. if (query.HasTrailer.HasValue)
  606. {
  607. var val = query.HasTrailer.Value;
  608. var trailerCount = 0;
  609. if (item is IHasTrailers hasTrailers)
  610. {
  611. trailerCount = hasTrailers.GetTrailerCount();
  612. }
  613. var ok = val ? trailerCount > 0 : trailerCount == 0;
  614. if (!ok)
  615. {
  616. return false;
  617. }
  618. }
  619. if (query.HasThemeSong.HasValue)
  620. {
  621. var filterValue = query.HasThemeSong.Value;
  622. var themeCount = item.GetThemeSongs(user).Count;
  623. var ok = filterValue ? themeCount > 0 : themeCount == 0;
  624. if (!ok)
  625. {
  626. return false;
  627. }
  628. }
  629. if (query.HasThemeVideo.HasValue)
  630. {
  631. var filterValue = query.HasThemeVideo.Value;
  632. var themeCount = item.GetThemeVideos(user).Count;
  633. var ok = filterValue ? themeCount > 0 : themeCount == 0;
  634. if (!ok)
  635. {
  636. return false;
  637. }
  638. }
  639. // Apply genre filter
  640. if (query.Genres.Count > 0 && !query.Genres.Any(v => item.Genres.Contains(v, StringComparison.OrdinalIgnoreCase)))
  641. {
  642. return false;
  643. }
  644. // Filter by VideoType
  645. if (query.VideoTypes.Length > 0)
  646. {
  647. var video = item as Video;
  648. if (video is null || !query.VideoTypes.Contains(video.VideoType))
  649. {
  650. return false;
  651. }
  652. }
  653. if (query.ImageTypes.Length > 0 && !query.ImageTypes.Any(item.HasImage))
  654. {
  655. return false;
  656. }
  657. // Apply studio filter
  658. if (query.StudioIds.Length > 0 && !query.StudioIds.Any(id =>
  659. {
  660. var studioItem = libraryManager.GetItemById(id);
  661. return studioItem is not null && item.Studios.Contains(studioItem.Name, StringComparison.OrdinalIgnoreCase);
  662. }))
  663. {
  664. return false;
  665. }
  666. // Apply genre filter
  667. if (query.GenreIds.Count > 0 && !query.GenreIds.Any(id =>
  668. {
  669. var genreItem = libraryManager.GetItemById(id);
  670. return genreItem is not null && item.Genres.Contains(genreItem.Name, StringComparison.OrdinalIgnoreCase);
  671. }))
  672. {
  673. return false;
  674. }
  675. // Apply year filter
  676. if (query.Years.Length > 0)
  677. {
  678. if (!(item.ProductionYear.HasValue && query.Years.Contains(item.ProductionYear.Value)))
  679. {
  680. return false;
  681. }
  682. }
  683. // Apply official rating filter
  684. if (query.OfficialRatings.Length > 0 && !query.OfficialRatings.Contains(item.OfficialRating ?? string.Empty))
  685. {
  686. return false;
  687. }
  688. if (query.ItemIds.Length > 0)
  689. {
  690. if (!query.ItemIds.Contains(item.Id))
  691. {
  692. return false;
  693. }
  694. }
  695. // Apply tag filter
  696. var tags = query.Tags;
  697. if (tags.Length > 0)
  698. {
  699. if (!tags.Any(v => item.Tags.Contains(v, StringComparison.OrdinalIgnoreCase)))
  700. {
  701. return false;
  702. }
  703. }
  704. if (query.MinCommunityRating.HasValue)
  705. {
  706. var val = query.MinCommunityRating.Value;
  707. if (!(item.CommunityRating.HasValue && item.CommunityRating >= val))
  708. {
  709. return false;
  710. }
  711. }
  712. if (query.MinCriticRating.HasValue)
  713. {
  714. var val = query.MinCriticRating.Value;
  715. if (!(item.CriticRating.HasValue && item.CriticRating >= val))
  716. {
  717. return false;
  718. }
  719. }
  720. if (query.MinIndexNumber.HasValue)
  721. {
  722. var val = query.MinIndexNumber.Value;
  723. if (!(item.IndexNumber.HasValue && item.IndexNumber.Value >= val))
  724. {
  725. return false;
  726. }
  727. }
  728. if (query.MinPremiereDate.HasValue)
  729. {
  730. var val = query.MinPremiereDate.Value;
  731. if (!(item.PremiereDate.HasValue && item.PremiereDate.Value >= val))
  732. {
  733. return false;
  734. }
  735. }
  736. if (query.MaxPremiereDate.HasValue)
  737. {
  738. var val = query.MaxPremiereDate.Value;
  739. if (!(item.PremiereDate.HasValue && item.PremiereDate.Value <= val))
  740. {
  741. return false;
  742. }
  743. }
  744. if (query.ParentIndexNumber.HasValue)
  745. {
  746. var filterValue = query.ParentIndexNumber.Value;
  747. if (item.ParentIndexNumber.HasValue && item.ParentIndexNumber.Value != filterValue)
  748. {
  749. return false;
  750. }
  751. }
  752. if (query.SeriesStatuses.Length > 0)
  753. {
  754. var ok = new[] { item }.OfType<Series>().Any(p => p.Status.HasValue && query.SeriesStatuses.Contains(p.Status.Value));
  755. if (!ok)
  756. {
  757. return false;
  758. }
  759. }
  760. if (query.AiredDuringSeason.HasValue)
  761. {
  762. var episode = item as Episode;
  763. if (episode is null)
  764. {
  765. return false;
  766. }
  767. if (!Series.FilterEpisodesBySeason(new[] { episode }, query.AiredDuringSeason.Value, true).Any())
  768. {
  769. return false;
  770. }
  771. }
  772. if (query.ExcludeItemIds.Contains(item.Id))
  773. {
  774. return false;
  775. }
  776. return true;
  777. }
  778. private IEnumerable<BaseItem> GetMediaFolders(User user)
  779. {
  780. if (user is null)
  781. {
  782. return _libraryManager.RootFolder
  783. .Children
  784. .OfType<Folder>()
  785. .Where(UserView.IsEligibleForGrouping);
  786. }
  787. return _libraryManager.GetUserRootFolder()
  788. .GetChildren(user, true)
  789. .OfType<Folder>()
  790. .Where(i => user.IsFolderGrouped(i.Id) && UserView.IsEligibleForGrouping(i));
  791. }
  792. private BaseItem[] GetMediaFolders(User user, IEnumerable<CollectionType> viewTypes)
  793. {
  794. if (user is null)
  795. {
  796. return GetMediaFolders(null)
  797. .Where(i =>
  798. {
  799. var folder = i as ICollectionFolder;
  800. return folder?.CollectionType is not null && viewTypes.Contains(folder.CollectionType.Value);
  801. }).ToArray();
  802. }
  803. return GetMediaFolders(user)
  804. .Where(i =>
  805. {
  806. var folder = i as ICollectionFolder;
  807. return folder?.CollectionType is not null && viewTypes.Contains(folder.CollectionType.Value);
  808. }).ToArray();
  809. }
  810. private BaseItem[] GetMediaFolders(Folder parent, User user, IEnumerable<CollectionType> viewTypes)
  811. {
  812. if (parent is null || parent is UserView)
  813. {
  814. return GetMediaFolders(user, viewTypes);
  815. }
  816. return new BaseItem[] { parent };
  817. }
  818. private UserView GetUserViewWithName(CollectionType? type, string sortName, BaseItem parent)
  819. {
  820. return _userViewManager.GetUserSubView(parent.Id, type, parent.Id.ToString("N", CultureInfo.InvariantCulture), sortName);
  821. }
  822. private UserView GetUserView(CollectionType? type, string localizationKey, string sortName, BaseItem parent)
  823. {
  824. return _userViewManager.GetUserSubView(parent.Id, type, localizationKey, sortName);
  825. }
  826. public static IEnumerable<BaseItem> FilterForAdjacency(List<BaseItem> list, Guid adjacentTo)
  827. {
  828. var adjacentToItem = list.FirstOrDefault(i => i.Id.Equals(adjacentTo));
  829. var index = list.IndexOf(adjacentToItem);
  830. var previousId = Guid.Empty;
  831. var nextId = Guid.Empty;
  832. if (index > 0)
  833. {
  834. previousId = list[index - 1].Id;
  835. }
  836. if (index < list.Count - 1)
  837. {
  838. nextId = list[index + 1].Id;
  839. }
  840. return list.Where(i => i.Id.Equals(previousId) || i.Id.Equals(nextId) || i.Id.Equals(adjacentTo));
  841. }
  842. }
  843. }