2
0

UserViewBuilder.cs 35 KB

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