BaseItemRepository.cs 77 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125
  1. using System;
  2. using System.Collections.Concurrent;
  3. using System.Collections.Generic;
  4. using System.Collections.Immutable;
  5. using System.Globalization;
  6. using System.IO;
  7. using System.Linq;
  8. using System.Linq.Expressions;
  9. using System.Reflection;
  10. using System.Text;
  11. using System.Text.Json;
  12. using System.Threading;
  13. using Jellyfin.Data.Entities;
  14. using Jellyfin.Data.Enums;
  15. using Jellyfin.Extensions;
  16. using Jellyfin.Extensions.Json;
  17. using MediaBrowser.Common;
  18. using MediaBrowser.Controller;
  19. using MediaBrowser.Controller.Channels;
  20. using MediaBrowser.Controller.Configuration;
  21. using MediaBrowser.Controller.Entities;
  22. using MediaBrowser.Controller.Entities.Audio;
  23. using MediaBrowser.Controller.Entities.TV;
  24. using MediaBrowser.Controller.LiveTv;
  25. using MediaBrowser.Controller.Persistence;
  26. using MediaBrowser.Model.Dto;
  27. using MediaBrowser.Model.Entities;
  28. using MediaBrowser.Model.LiveTv;
  29. using MediaBrowser.Model.Querying;
  30. using Microsoft.EntityFrameworkCore;
  31. using Microsoft.Extensions.Logging;
  32. using BaseItemDto = MediaBrowser.Controller.Entities.BaseItem;
  33. using BaseItemEntity = Jellyfin.Data.Entities.BaseItemEntity;
  34. #pragma warning disable RS0030 // Do not use banned APIs
  35. namespace Jellyfin.Server.Implementations.Item;
  36. /// <summary>
  37. /// Handles all storage logic for BaseItems.
  38. /// </summary>
  39. /// <remarks>
  40. /// Initializes a new instance of the <see cref="BaseItemRepository"/> class.
  41. /// </remarks>
  42. /// <param name="dbProvider">The db factory.</param>
  43. /// <param name="appHost">The Application host.</param>
  44. /// <param name="itemTypeLookup">The static type lookup.</param>
  45. /// <param name="serverConfigurationManager">The server Configuration manager.</param>
  46. /// <param name="logger">System logger.</param>
  47. public sealed class BaseItemRepository(
  48. IDbContextFactory<JellyfinDbContext> dbProvider,
  49. IServerApplicationHost appHost,
  50. IItemTypeLookup itemTypeLookup,
  51. IServerConfigurationManager serverConfigurationManager,
  52. ILogger<BaseItemRepository> logger)
  53. : IItemRepository, IDisposable
  54. {
  55. /// <summary>
  56. /// This holds all the types in the running assemblies
  57. /// so that we can de-serialize properly when we don't have strong types.
  58. /// </summary>
  59. private static readonly ConcurrentDictionary<string, Type?> _typeMap = new ConcurrentDictionary<string, Type?>();
  60. private bool _disposed;
  61. /// <inheritdoc/>
  62. public void Dispose()
  63. {
  64. if (_disposed)
  65. {
  66. return;
  67. }
  68. _disposed = true;
  69. }
  70. /// <inheritdoc />
  71. public void DeleteItem(Guid id)
  72. {
  73. ArgumentNullException.ThrowIfNull(id.IsEmpty() ? null : id);
  74. using var context = dbProvider.CreateDbContext();
  75. using var transaction = context.Database.BeginTransaction();
  76. context.PeopleBaseItemMap.Where(e => e.ItemId == id).ExecuteDelete();
  77. context.Peoples.Where(e => e.BaseItems!.Count == 0).ExecuteDelete();
  78. context.Chapters.Where(e => e.ItemId == id).ExecuteDelete();
  79. context.MediaStreamInfos.Where(e => e.ItemId == id).ExecuteDelete();
  80. context.AncestorIds.Where(e => e.ItemId == id).ExecuteDelete();
  81. context.ItemValuesMap.Where(e => e.ItemId == id).ExecuteDelete();
  82. context.ItemValues.Where(e => e.BaseItemsMap!.Count == 0).ExecuteDelete();
  83. context.BaseItemImageInfos.Where(e => e.ItemId == id).ExecuteDelete();
  84. context.BaseItemProviders.Where(e => e.ItemId == id).ExecuteDelete();
  85. context.BaseItems.Where(e => e.Id == id).ExecuteDelete();
  86. context.SaveChanges();
  87. transaction.Commit();
  88. }
  89. /// <inheritdoc />
  90. public void UpdateInheritedValues()
  91. {
  92. using var context = dbProvider.CreateDbContext();
  93. using var transaction = context.Database.BeginTransaction();
  94. context.ItemValuesMap.Where(e => e.ItemValue.Type == ItemValueType.InheritedTags).ExecuteDelete();
  95. // ItemValue Inheritence is now correctly mapped via AncestorId on demand
  96. context.SaveChanges();
  97. transaction.Commit();
  98. }
  99. /// <inheritdoc cref="IItemRepository"/>
  100. public IReadOnlyList<Guid> GetItemIdsList(InternalItemsQuery filter)
  101. {
  102. ArgumentNullException.ThrowIfNull(filter);
  103. PrepareFilterQuery(filter);
  104. using var context = dbProvider.CreateDbContext();
  105. return ApplyQueryFilter(context.BaseItems.AsNoTracking(), context, filter).Select(e => e.Id).ToImmutableArray();
  106. }
  107. /// <inheritdoc />
  108. public QueryResult<(BaseItemDto Item, ItemCounts ItemCounts)> GetAllArtists(InternalItemsQuery filter)
  109. {
  110. return GetItemValues(filter, [ItemValueType.Artist, ItemValueType.AlbumArtist], itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicArtist]!);
  111. }
  112. /// <inheritdoc />
  113. public QueryResult<(BaseItemDto Item, ItemCounts ItemCounts)> GetArtists(InternalItemsQuery filter)
  114. {
  115. return GetItemValues(filter, [ItemValueType.Artist], itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicArtist]!);
  116. }
  117. /// <inheritdoc />
  118. public QueryResult<(BaseItemDto Item, ItemCounts ItemCounts)> GetAlbumArtists(InternalItemsQuery filter)
  119. {
  120. return GetItemValues(filter, [ItemValueType.AlbumArtist], itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicArtist]!);
  121. }
  122. /// <inheritdoc />
  123. public QueryResult<(BaseItemDto Item, ItemCounts ItemCounts)> GetStudios(InternalItemsQuery filter)
  124. {
  125. return GetItemValues(filter, [ItemValueType.Studios], itemTypeLookup.BaseItemKindNames[BaseItemKind.Studio]!);
  126. }
  127. /// <inheritdoc />
  128. public QueryResult<(BaseItemDto Item, ItemCounts ItemCounts)> GetGenres(InternalItemsQuery filter)
  129. {
  130. return GetItemValues(filter, [ItemValueType.Genre], itemTypeLookup.BaseItemKindNames[BaseItemKind.Genre]!);
  131. }
  132. /// <inheritdoc />
  133. public QueryResult<(BaseItemDto Item, ItemCounts ItemCounts)> GetMusicGenres(InternalItemsQuery filter)
  134. {
  135. return GetItemValues(filter, [ItemValueType.Genre], itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicGenre]!);
  136. }
  137. /// <inheritdoc />
  138. public IReadOnlyList<string> GetStudioNames()
  139. {
  140. return GetItemValueNames([ItemValueType.Studios], [], []);
  141. }
  142. /// <inheritdoc />
  143. public IReadOnlyList<string> GetAllArtistNames()
  144. {
  145. return GetItemValueNames([ItemValueType.Artist, ItemValueType.AlbumArtist], [], []);
  146. }
  147. /// <inheritdoc />
  148. public IReadOnlyList<string> GetMusicGenreNames()
  149. {
  150. return GetItemValueNames(
  151. [ItemValueType.Genre],
  152. itemTypeLookup.MusicGenreTypes,
  153. []);
  154. }
  155. /// <inheritdoc />
  156. public IReadOnlyList<string> GetGenreNames()
  157. {
  158. return GetItemValueNames(
  159. [ItemValueType.Genre],
  160. [],
  161. itemTypeLookup.MusicGenreTypes);
  162. }
  163. /// <inheritdoc cref="IItemRepository"/>
  164. public QueryResult<BaseItemDto> GetItems(InternalItemsQuery filter)
  165. {
  166. ArgumentNullException.ThrowIfNull(filter);
  167. if (!filter.EnableTotalRecordCount || (!filter.Limit.HasValue && (filter.StartIndex ?? 0) == 0))
  168. {
  169. var returnList = GetItemList(filter);
  170. return new QueryResult<BaseItemDto>(
  171. filter.StartIndex,
  172. returnList.Count,
  173. returnList);
  174. }
  175. PrepareFilterQuery(filter);
  176. var result = new QueryResult<BaseItemDto>();
  177. using var context = dbProvider.CreateDbContext();
  178. IQueryable<BaseItemEntity> dbQuery = PrepareItemQuery(context, filter);
  179. dbQuery = TranslateQuery(dbQuery, context, filter);
  180. dbQuery = dbQuery.Distinct();
  181. if (filter.EnableTotalRecordCount)
  182. {
  183. result.TotalRecordCount = dbQuery.Count();
  184. }
  185. dbQuery = ApplyOrder(dbQuery, filter);
  186. dbQuery = ApplyQueryPageing(dbQuery, filter);
  187. result.Items = dbQuery.AsEnumerable().Where(e => e is not null).Select(w => DeserialiseBaseItem(w, filter.SkipDeserialization)).ToImmutableArray();
  188. result.StartIndex = filter.StartIndex ?? 0;
  189. return result;
  190. }
  191. /// <inheritdoc cref="IItemRepository"/>
  192. public IReadOnlyList<BaseItemDto> GetItemList(InternalItemsQuery filter)
  193. {
  194. ArgumentNullException.ThrowIfNull(filter);
  195. PrepareFilterQuery(filter);
  196. using var context = dbProvider.CreateDbContext();
  197. IQueryable<BaseItemEntity> dbQuery = PrepareItemQuery(context, filter);
  198. dbQuery = TranslateQuery(dbQuery, context, filter);
  199. dbQuery = dbQuery.Distinct();
  200. dbQuery = ApplyOrder(dbQuery, filter);
  201. dbQuery = ApplyGroupingFilter(dbQuery, filter);
  202. return dbQuery.AsEnumerable().Where(e => e is not null).Select(w => DeserialiseBaseItem(w, filter.SkipDeserialization)).ToImmutableArray();
  203. }
  204. private IQueryable<BaseItemEntity> ApplyGroupingFilter(IQueryable<BaseItemEntity> dbQuery, InternalItemsQuery filter)
  205. {
  206. var enableGroupByPresentationUniqueKey = EnableGroupByPresentationUniqueKey(filter);
  207. if (enableGroupByPresentationUniqueKey && filter.GroupBySeriesPresentationUniqueKey)
  208. {
  209. dbQuery = dbQuery.GroupBy(e => new { e.PresentationUniqueKey, e.SeriesPresentationUniqueKey }).Select(e => e.First());
  210. }
  211. else if (enableGroupByPresentationUniqueKey)
  212. {
  213. dbQuery = dbQuery.GroupBy(e => e.PresentationUniqueKey).Select(e => e.First());
  214. }
  215. else if (filter.GroupBySeriesPresentationUniqueKey)
  216. {
  217. dbQuery = dbQuery.GroupBy(e => e.SeriesPresentationUniqueKey).Select(e => e.First());
  218. }
  219. else
  220. {
  221. dbQuery = dbQuery.Distinct();
  222. }
  223. return dbQuery;
  224. }
  225. private IQueryable<BaseItemEntity> ApplyQueryPageing(IQueryable<BaseItemEntity> dbQuery, InternalItemsQuery filter)
  226. {
  227. if (filter.Limit.HasValue || filter.StartIndex.HasValue)
  228. {
  229. var offset = filter.StartIndex ?? 0;
  230. if (offset > 0)
  231. {
  232. dbQuery = dbQuery.Skip(offset);
  233. }
  234. if (filter.Limit.HasValue)
  235. {
  236. dbQuery = dbQuery.Take(filter.Limit.Value);
  237. }
  238. }
  239. return dbQuery;
  240. }
  241. private IQueryable<BaseItemEntity> ApplyQueryFilter(IQueryable<BaseItemEntity> dbQuery, JellyfinDbContext context, InternalItemsQuery filter)
  242. {
  243. dbQuery = TranslateQuery(dbQuery, context, filter);
  244. dbQuery = ApplyOrder(dbQuery, filter);
  245. dbQuery = ApplyGroupingFilter(dbQuery, filter);
  246. dbQuery = ApplyQueryPageing(dbQuery, filter);
  247. return dbQuery;
  248. }
  249. private IQueryable<BaseItemEntity> PrepareItemQuery(JellyfinDbContext context, InternalItemsQuery filter)
  250. {
  251. IQueryable<BaseItemEntity> dbQuery = context.BaseItems.AsNoTracking().AsSingleQuery()
  252. .Include(e => e.TrailerTypes)
  253. .Include(e => e.Provider)
  254. .Include(e => e.LockedFields);
  255. if (filter.DtoOptions.EnableImages)
  256. {
  257. dbQuery = dbQuery.Include(e => e.Images);
  258. }
  259. return dbQuery;
  260. }
  261. /// <inheritdoc/>
  262. public int GetCount(InternalItemsQuery filter)
  263. {
  264. ArgumentNullException.ThrowIfNull(filter);
  265. // Hack for right now since we currently don't support filtering out these duplicates within a query
  266. PrepareFilterQuery(filter);
  267. using var context = dbProvider.CreateDbContext();
  268. var dbQuery = TranslateQuery(context.BaseItems.AsNoTracking(), context, filter);
  269. return dbQuery.Count();
  270. }
  271. #pragma warning disable CA1307 // Specify StringComparison for clarity
  272. private IQueryable<BaseItemEntity> TranslateQuery(
  273. IQueryable<BaseItemEntity> baseQuery,
  274. JellyfinDbContext context,
  275. InternalItemsQuery filter)
  276. {
  277. var minWidth = filter.MinWidth;
  278. var maxWidth = filter.MaxWidth;
  279. var now = DateTime.UtcNow;
  280. if (filter.IsHD.HasValue)
  281. {
  282. const int Threshold = 1200;
  283. if (filter.IsHD.Value)
  284. {
  285. minWidth = Threshold;
  286. }
  287. else
  288. {
  289. maxWidth = Threshold - 1;
  290. }
  291. }
  292. if (filter.Is4K.HasValue)
  293. {
  294. const int Threshold = 3800;
  295. if (filter.Is4K.Value)
  296. {
  297. minWidth = Threshold;
  298. }
  299. else
  300. {
  301. maxWidth = Threshold - 1;
  302. }
  303. }
  304. if (minWidth.HasValue)
  305. {
  306. baseQuery = baseQuery.Where(e => e.Width >= minWidth);
  307. }
  308. if (filter.MinHeight.HasValue)
  309. {
  310. baseQuery = baseQuery.Where(e => e.Height >= filter.MinHeight);
  311. }
  312. if (maxWidth.HasValue)
  313. {
  314. baseQuery = baseQuery.Where(e => e.Width >= maxWidth);
  315. }
  316. if (filter.MaxHeight.HasValue)
  317. {
  318. baseQuery = baseQuery.Where(e => e.Height <= filter.MaxHeight);
  319. }
  320. if (filter.IsLocked.HasValue)
  321. {
  322. baseQuery = baseQuery.Where(e => e.IsLocked == filter.IsLocked);
  323. }
  324. var tags = filter.Tags.ToList();
  325. var excludeTags = filter.ExcludeTags.ToList();
  326. if (filter.IsMovie == true)
  327. {
  328. if (filter.IncludeItemTypes.Length == 0
  329. || filter.IncludeItemTypes.Contains(BaseItemKind.Movie)
  330. || filter.IncludeItemTypes.Contains(BaseItemKind.Trailer))
  331. {
  332. baseQuery = baseQuery.Where(e => e.IsMovie);
  333. }
  334. }
  335. else if (filter.IsMovie.HasValue)
  336. {
  337. baseQuery = baseQuery.Where(e => e.IsMovie == filter.IsMovie);
  338. }
  339. if (filter.IsSeries.HasValue)
  340. {
  341. baseQuery = baseQuery.Where(e => e.IsSeries == filter.IsSeries);
  342. }
  343. if (filter.IsSports.HasValue)
  344. {
  345. if (filter.IsSports.Value)
  346. {
  347. tags.Add("Sports");
  348. }
  349. else
  350. {
  351. excludeTags.Add("Sports");
  352. }
  353. }
  354. if (filter.IsNews.HasValue)
  355. {
  356. if (filter.IsNews.Value)
  357. {
  358. tags.Add("News");
  359. }
  360. else
  361. {
  362. excludeTags.Add("News");
  363. }
  364. }
  365. if (filter.IsKids.HasValue)
  366. {
  367. if (filter.IsKids.Value)
  368. {
  369. tags.Add("Kids");
  370. }
  371. else
  372. {
  373. excludeTags.Add("Kids");
  374. }
  375. }
  376. if (!string.IsNullOrEmpty(filter.SearchTerm))
  377. {
  378. baseQuery = baseQuery.Where(e => e.CleanName!.Contains(filter.SearchTerm) || (e.OriginalTitle != null && e.OriginalTitle.Contains(filter.SearchTerm)));
  379. }
  380. if (filter.IsFolder.HasValue)
  381. {
  382. baseQuery = baseQuery.Where(e => e.IsFolder == filter.IsFolder);
  383. }
  384. var includeTypes = filter.IncludeItemTypes;
  385. // Only specify excluded types if no included types are specified
  386. if (filter.IncludeItemTypes.Length == 0)
  387. {
  388. var excludeTypes = filter.ExcludeItemTypes;
  389. if (excludeTypes.Length == 1)
  390. {
  391. if (itemTypeLookup.BaseItemKindNames.TryGetValue(excludeTypes[0], out var excludeTypeName))
  392. {
  393. baseQuery = baseQuery.Where(e => e.Type != excludeTypeName);
  394. }
  395. }
  396. else if (excludeTypes.Length > 1)
  397. {
  398. var excludeTypeName = new List<string>();
  399. foreach (var excludeType in excludeTypes)
  400. {
  401. if (itemTypeLookup.BaseItemKindNames.TryGetValue(excludeType, out var baseItemKindName))
  402. {
  403. excludeTypeName.Add(baseItemKindName!);
  404. }
  405. }
  406. baseQuery = baseQuery.Where(e => !excludeTypeName.Contains(e.Type));
  407. }
  408. }
  409. else if (includeTypes.Length == 1)
  410. {
  411. if (itemTypeLookup.BaseItemKindNames.TryGetValue(includeTypes[0], out var includeTypeName))
  412. {
  413. baseQuery = baseQuery.Where(e => e.Type == includeTypeName);
  414. }
  415. }
  416. else if (includeTypes.Length > 1)
  417. {
  418. var includeTypeName = new List<string>();
  419. foreach (var includeType in includeTypes)
  420. {
  421. if (itemTypeLookup.BaseItemKindNames.TryGetValue(includeType, out var baseItemKindName))
  422. {
  423. includeTypeName.Add(baseItemKindName!);
  424. }
  425. }
  426. baseQuery = baseQuery.Where(e => includeTypeName.Contains(e.Type));
  427. }
  428. if (filter.ChannelIds.Count > 0)
  429. {
  430. var channelIds = filter.ChannelIds.Select(e => e.ToString("N", CultureInfo.InvariantCulture)).ToArray();
  431. baseQuery = baseQuery.Where(e => channelIds.Contains(e.ChannelId));
  432. }
  433. if (!filter.ParentId.IsEmpty())
  434. {
  435. baseQuery = baseQuery.Where(e => e.ParentId!.Value == filter.ParentId);
  436. }
  437. if (!string.IsNullOrWhiteSpace(filter.Path))
  438. {
  439. baseQuery = baseQuery.Where(e => e.Path == filter.Path);
  440. }
  441. if (!string.IsNullOrWhiteSpace(filter.PresentationUniqueKey))
  442. {
  443. baseQuery = baseQuery.Where(e => e.PresentationUniqueKey == filter.PresentationUniqueKey);
  444. }
  445. if (filter.MinCommunityRating.HasValue)
  446. {
  447. baseQuery = baseQuery.Where(e => e.CommunityRating >= filter.MinCommunityRating);
  448. }
  449. if (filter.MinIndexNumber.HasValue)
  450. {
  451. baseQuery = baseQuery.Where(e => e.IndexNumber >= filter.MinIndexNumber);
  452. }
  453. if (filter.MinParentAndIndexNumber.HasValue)
  454. {
  455. baseQuery = baseQuery
  456. .Where(e => (e.ParentIndexNumber == filter.MinParentAndIndexNumber.Value.ParentIndexNumber && e.IndexNumber >= filter.MinParentAndIndexNumber.Value.IndexNumber) || e.ParentIndexNumber > filter.MinParentAndIndexNumber.Value.ParentIndexNumber);
  457. }
  458. if (filter.MinDateCreated.HasValue)
  459. {
  460. baseQuery = baseQuery.Where(e => e.DateCreated >= filter.MinDateCreated);
  461. }
  462. if (filter.MinDateLastSaved.HasValue)
  463. {
  464. baseQuery = baseQuery.Where(e => e.DateLastSaved != null && e.DateLastSaved >= filter.MinDateLastSaved.Value);
  465. }
  466. if (filter.MinDateLastSavedForUser.HasValue)
  467. {
  468. baseQuery = baseQuery.Where(e => e.DateLastSaved != null && e.DateLastSaved >= filter.MinDateLastSavedForUser.Value);
  469. }
  470. if (filter.IndexNumber.HasValue)
  471. {
  472. baseQuery = baseQuery.Where(e => e.IndexNumber == filter.IndexNumber.Value);
  473. }
  474. if (filter.ParentIndexNumber.HasValue)
  475. {
  476. baseQuery = baseQuery.Where(e => e.ParentIndexNumber == filter.ParentIndexNumber.Value);
  477. }
  478. if (filter.ParentIndexNumberNotEquals.HasValue)
  479. {
  480. baseQuery = baseQuery.Where(e => e.ParentIndexNumber != filter.ParentIndexNumberNotEquals.Value || e.ParentIndexNumber == null);
  481. }
  482. var minEndDate = filter.MinEndDate;
  483. var maxEndDate = filter.MaxEndDate;
  484. if (filter.HasAired.HasValue)
  485. {
  486. if (filter.HasAired.Value)
  487. {
  488. maxEndDate = DateTime.UtcNow;
  489. }
  490. else
  491. {
  492. minEndDate = DateTime.UtcNow;
  493. }
  494. }
  495. if (minEndDate.HasValue)
  496. {
  497. baseQuery = baseQuery.Where(e => e.EndDate >= minEndDate);
  498. }
  499. if (maxEndDate.HasValue)
  500. {
  501. baseQuery = baseQuery.Where(e => e.EndDate <= maxEndDate);
  502. }
  503. if (filter.MinStartDate.HasValue)
  504. {
  505. baseQuery = baseQuery.Where(e => e.StartDate >= filter.MinStartDate.Value);
  506. }
  507. if (filter.MaxStartDate.HasValue)
  508. {
  509. baseQuery = baseQuery.Where(e => e.StartDate <= filter.MaxStartDate.Value);
  510. }
  511. if (filter.MinPremiereDate.HasValue)
  512. {
  513. baseQuery = baseQuery.Where(e => e.PremiereDate <= filter.MinPremiereDate.Value);
  514. }
  515. if (filter.MaxPremiereDate.HasValue)
  516. {
  517. baseQuery = baseQuery.Where(e => e.PremiereDate <= filter.MaxPremiereDate.Value);
  518. }
  519. if (filter.TrailerTypes.Length > 0)
  520. {
  521. var trailerTypes = filter.TrailerTypes.Select(e => (int)e).ToArray();
  522. baseQuery = baseQuery.Where(e => trailerTypes.Any(f => e.TrailerTypes!.Any(w => w.Id == f)));
  523. }
  524. if (filter.IsAiring.HasValue)
  525. {
  526. if (filter.IsAiring.Value)
  527. {
  528. baseQuery = baseQuery.Where(e => e.StartDate <= now && e.EndDate >= now);
  529. }
  530. else
  531. {
  532. baseQuery = baseQuery.Where(e => e.StartDate > now && e.EndDate < now);
  533. }
  534. }
  535. if (filter.PersonIds.Length > 0)
  536. {
  537. baseQuery = baseQuery
  538. .Where(e =>
  539. context.PeopleBaseItemMap.Where(w => context.BaseItems.Where(r => filter.PersonIds.Contains(r.Id)).Any(f => f.Name == w.People.Name))
  540. .Any(f => f.ItemId == e.Id));
  541. }
  542. if (!string.IsNullOrWhiteSpace(filter.Person))
  543. {
  544. baseQuery = baseQuery.Where(e => e.Peoples!.Any(f => f.People.Name == filter.Person));
  545. }
  546. if (!string.IsNullOrWhiteSpace(filter.MinSortName))
  547. {
  548. // this does not makes sense.
  549. // baseQuery = baseQuery.Where(e => e.SortName >= query.MinSortName);
  550. // whereClauses.Add("SortName>=@MinSortName");
  551. // statement?.TryBind("@MinSortName", query.MinSortName);
  552. }
  553. if (!string.IsNullOrWhiteSpace(filter.ExternalSeriesId))
  554. {
  555. baseQuery = baseQuery.Where(e => e.ExternalSeriesId == filter.ExternalSeriesId);
  556. }
  557. if (!string.IsNullOrWhiteSpace(filter.ExternalId))
  558. {
  559. baseQuery = baseQuery.Where(e => e.ExternalId == filter.ExternalId);
  560. }
  561. if (!string.IsNullOrWhiteSpace(filter.Name))
  562. {
  563. var cleanName = GetCleanValue(filter.Name);
  564. baseQuery = baseQuery.Where(e => e.CleanName == cleanName);
  565. }
  566. // These are the same, for now
  567. var nameContains = filter.NameContains;
  568. if (!string.IsNullOrWhiteSpace(nameContains))
  569. {
  570. baseQuery = baseQuery.Where(e =>
  571. e.CleanName == filter.NameContains
  572. || e.OriginalTitle!.Contains(filter.NameContains!));
  573. }
  574. if (!string.IsNullOrWhiteSpace(filter.NameStartsWith))
  575. {
  576. baseQuery = baseQuery.Where(e => e.SortName!.StartsWith(filter.NameStartsWith) || e.Name!.StartsWith(filter.NameStartsWith));
  577. }
  578. if (!string.IsNullOrWhiteSpace(filter.NameStartsWithOrGreater))
  579. {
  580. // i hate this
  581. baseQuery = baseQuery.Where(e => e.SortName!.FirstOrDefault() > filter.NameStartsWithOrGreater[0] || e.Name!.FirstOrDefault() > filter.NameStartsWithOrGreater[0]);
  582. }
  583. if (!string.IsNullOrWhiteSpace(filter.NameLessThan))
  584. {
  585. // i hate this
  586. baseQuery = baseQuery.Where(e => e.SortName!.FirstOrDefault() < filter.NameLessThan[0] || e.Name!.FirstOrDefault() < filter.NameLessThan[0]);
  587. }
  588. if (filter.ImageTypes.Length > 0)
  589. {
  590. var imgTypes = filter.ImageTypes.Select(e => (ImageInfoImageType)e).ToArray();
  591. baseQuery = baseQuery.Where(e => imgTypes.Any(f => e.Images!.Any(w => w.ImageType == f)));
  592. }
  593. if (filter.IsLiked.HasValue)
  594. {
  595. baseQuery = baseQuery
  596. .Where(e => e.UserData!.FirstOrDefault(f => f.UserId == filter.User!.Id)!.Rating >= UserItemData.MinLikeValue);
  597. }
  598. if (filter.IsFavoriteOrLiked.HasValue)
  599. {
  600. baseQuery = baseQuery
  601. .Where(e => e.UserData!.FirstOrDefault(f => f.UserId == filter.User!.Id)!.IsFavorite == filter.IsFavoriteOrLiked);
  602. }
  603. if (filter.IsFavorite.HasValue)
  604. {
  605. baseQuery = baseQuery
  606. .Where(e => e.UserData!.FirstOrDefault(f => f.UserId == filter.User!.Id)!.IsFavorite == filter.IsFavorite);
  607. }
  608. if (filter.IsPlayed.HasValue)
  609. {
  610. baseQuery = baseQuery
  611. .Where(e => e.UserData!.FirstOrDefault(f => f.UserId == filter.User!.Id)!.Played == filter.IsPlayed.Value);
  612. }
  613. if (filter.IsResumable.HasValue)
  614. {
  615. if (filter.IsResumable.Value)
  616. {
  617. baseQuery = baseQuery
  618. .Where(e => e.UserData!.FirstOrDefault(f => f.UserId == filter.User!.Id)!.PlaybackPositionTicks > 0);
  619. }
  620. else
  621. {
  622. baseQuery = baseQuery
  623. .Where(e => e.UserData!.FirstOrDefault(f => f.UserId == filter.User!.Id)!.PlaybackPositionTicks == 0);
  624. }
  625. }
  626. if (filter.ArtistIds.Length > 0)
  627. {
  628. baseQuery = baseQuery
  629. .Where(e => e.ItemValues!.Any(f => f.ItemValue.Type <= ItemValueType.Artist && filter.ArtistIds.Contains(f.ItemId)));
  630. }
  631. if (filter.AlbumArtistIds.Length > 0)
  632. {
  633. baseQuery = baseQuery
  634. .Where(e => e.ItemValues!.Any(f => f.ItemValue.Type == ItemValueType.Artist && filter.AlbumArtistIds.Contains(f.ItemId)));
  635. }
  636. if (filter.ContributingArtistIds.Length > 0)
  637. {
  638. baseQuery = baseQuery
  639. .Where(e => e.ItemValues!.Any(f => f.ItemValue.Type == ItemValueType.Artist && filter.ContributingArtistIds.Contains(f.ItemId)));
  640. }
  641. if (filter.AlbumIds.Length > 0)
  642. {
  643. baseQuery = baseQuery.Where(e => context.BaseItems.Where(f => filter.AlbumIds.Contains(f.Id)).Any(f => f.Name == e.Album));
  644. }
  645. if (filter.ExcludeArtistIds.Length > 0)
  646. {
  647. baseQuery = baseQuery
  648. .Where(e => !e.ItemValues!.Any(f => f.ItemValue.Type == ItemValueType.Artist && filter.ExcludeArtistIds.Contains(f.ItemId)));
  649. }
  650. if (filter.GenreIds.Count > 0)
  651. {
  652. baseQuery = baseQuery
  653. .Where(e => e.ItemValues!.Any(f => f.ItemValue.Type == ItemValueType.Genre && filter.GenreIds.Contains(f.ItemId)));
  654. }
  655. if (filter.Genres.Count > 0)
  656. {
  657. var cleanGenres = filter.Genres.Select(e => GetCleanValue(e)).ToArray();
  658. baseQuery = baseQuery
  659. .Where(e => e.ItemValues!.Any(f => f.ItemValue.Type == ItemValueType.Genre && cleanGenres.Contains(f.ItemValue.CleanValue)));
  660. }
  661. if (tags.Count > 0)
  662. {
  663. var cleanValues = tags.Select(e => GetCleanValue(e)).ToArray();
  664. baseQuery = baseQuery
  665. .Where(e => e.ItemValues!.Any(f => f.ItemValue.Type == ItemValueType.Tags && cleanValues.Contains(f.ItemValue.CleanValue)));
  666. }
  667. if (excludeTags.Count > 0)
  668. {
  669. var cleanValues = excludeTags.Select(e => GetCleanValue(e)).ToArray();
  670. baseQuery = baseQuery
  671. .Where(e => !e.ItemValues!.Any(f => f.ItemValue.Type == ItemValueType.Tags && cleanValues.Contains(f.ItemValue.CleanValue)));
  672. }
  673. if (filter.StudioIds.Length > 0)
  674. {
  675. baseQuery = baseQuery
  676. .Where(e => e.ItemValues!.Any(f => f.ItemValue.Type == ItemValueType.Studios && filter.StudioIds.Contains(f.ItemId)));
  677. }
  678. if (filter.OfficialRatings.Length > 0)
  679. {
  680. baseQuery = baseQuery
  681. .Where(e => filter.OfficialRatings.Contains(e.OfficialRating));
  682. }
  683. if (filter.HasParentalRating ?? false)
  684. {
  685. if (filter.MinParentalRating.HasValue)
  686. {
  687. baseQuery = baseQuery
  688. .Where(e => e.InheritedParentalRatingValue >= filter.MinParentalRating.Value);
  689. }
  690. if (filter.MaxParentalRating.HasValue)
  691. {
  692. baseQuery = baseQuery
  693. .Where(e => e.InheritedParentalRatingValue < filter.MaxParentalRating.Value);
  694. }
  695. }
  696. else if (filter.BlockUnratedItems.Length > 0)
  697. {
  698. var unratedItems = filter.BlockUnratedItems.Select(f => f.ToString()).ToArray();
  699. if (filter.MinParentalRating.HasValue)
  700. {
  701. if (filter.MaxParentalRating.HasValue)
  702. {
  703. baseQuery = baseQuery
  704. .Where(e => (e.InheritedParentalRatingValue == null && !unratedItems.Contains(e.UnratedType))
  705. || (e.InheritedParentalRatingValue >= filter.MinParentalRating && e.InheritedParentalRatingValue <= filter.MaxParentalRating));
  706. }
  707. else
  708. {
  709. baseQuery = baseQuery
  710. .Where(e => (e.InheritedParentalRatingValue == null && !unratedItems.Contains(e.UnratedType))
  711. || e.InheritedParentalRatingValue >= filter.MinParentalRating);
  712. }
  713. }
  714. else
  715. {
  716. baseQuery = baseQuery
  717. .Where(e => e.InheritedParentalRatingValue != null && !unratedItems.Contains(e.UnratedType));
  718. }
  719. }
  720. else if (filter.MinParentalRating.HasValue)
  721. {
  722. if (filter.MaxParentalRating.HasValue)
  723. {
  724. baseQuery = baseQuery
  725. .Where(e => e.InheritedParentalRatingValue != null && e.InheritedParentalRatingValue >= filter.MinParentalRating.Value && e.InheritedParentalRatingValue <= filter.MaxParentalRating.Value);
  726. }
  727. else
  728. {
  729. baseQuery = baseQuery
  730. .Where(e => e.InheritedParentalRatingValue != null && e.InheritedParentalRatingValue >= filter.MinParentalRating.Value);
  731. }
  732. }
  733. else if (filter.MaxParentalRating.HasValue)
  734. {
  735. baseQuery = baseQuery
  736. .Where(e => e.InheritedParentalRatingValue != null && e.InheritedParentalRatingValue >= filter.MaxParentalRating.Value);
  737. }
  738. else if (!filter.HasParentalRating ?? false)
  739. {
  740. baseQuery = baseQuery
  741. .Where(e => e.InheritedParentalRatingValue == null);
  742. }
  743. if (filter.HasOfficialRating.HasValue)
  744. {
  745. if (filter.HasOfficialRating.Value)
  746. {
  747. baseQuery = baseQuery
  748. .Where(e => e.OfficialRating != null && e.OfficialRating != string.Empty);
  749. }
  750. else
  751. {
  752. baseQuery = baseQuery
  753. .Where(e => e.OfficialRating == null || e.OfficialRating == string.Empty);
  754. }
  755. }
  756. if (filter.HasOverview.HasValue)
  757. {
  758. if (filter.HasOverview.Value)
  759. {
  760. baseQuery = baseQuery
  761. .Where(e => e.Overview != null && e.Overview != string.Empty);
  762. }
  763. else
  764. {
  765. baseQuery = baseQuery
  766. .Where(e => e.Overview == null || e.Overview == string.Empty);
  767. }
  768. }
  769. if (filter.HasOwnerId.HasValue)
  770. {
  771. if (filter.HasOwnerId.Value)
  772. {
  773. baseQuery = baseQuery
  774. .Where(e => e.OwnerId != null);
  775. }
  776. else
  777. {
  778. baseQuery = baseQuery
  779. .Where(e => e.OwnerId == null);
  780. }
  781. }
  782. if (!string.IsNullOrWhiteSpace(filter.HasNoAudioTrackWithLanguage))
  783. {
  784. baseQuery = baseQuery
  785. .Where(e => !e.MediaStreams!.Any(f => f.StreamType == MediaStreamTypeEntity.Audio && f.Language == filter.HasNoAudioTrackWithLanguage));
  786. }
  787. if (!string.IsNullOrWhiteSpace(filter.HasNoInternalSubtitleTrackWithLanguage))
  788. {
  789. baseQuery = baseQuery
  790. .Where(e => !e.MediaStreams!.Any(f => f.StreamType == MediaStreamTypeEntity.Subtitle && !f.IsExternal && f.Language == filter.HasNoInternalSubtitleTrackWithLanguage));
  791. }
  792. if (!string.IsNullOrWhiteSpace(filter.HasNoExternalSubtitleTrackWithLanguage))
  793. {
  794. baseQuery = baseQuery
  795. .Where(e => !e.MediaStreams!.Any(f => f.StreamType == MediaStreamTypeEntity.Subtitle && f.IsExternal && f.Language == filter.HasNoExternalSubtitleTrackWithLanguage));
  796. }
  797. if (!string.IsNullOrWhiteSpace(filter.HasNoSubtitleTrackWithLanguage))
  798. {
  799. baseQuery = baseQuery
  800. .Where(e => !e.MediaStreams!.Any(f => f.StreamType == MediaStreamTypeEntity.Subtitle && f.Language == filter.HasNoSubtitleTrackWithLanguage));
  801. }
  802. if (filter.HasSubtitles.HasValue)
  803. {
  804. baseQuery = baseQuery
  805. .Where(e => e.MediaStreams!.Any(f => f.StreamType == MediaStreamTypeEntity.Subtitle) == filter.HasSubtitles.Value);
  806. }
  807. if (filter.HasChapterImages.HasValue)
  808. {
  809. baseQuery = baseQuery
  810. .Where(e => e.Chapters!.Any(f => f.ImagePath != null) == filter.HasChapterImages.Value);
  811. }
  812. if (filter.HasDeadParentId.HasValue && filter.HasDeadParentId.Value)
  813. {
  814. baseQuery = baseQuery
  815. .Where(e => e.ParentId.HasValue && !context.BaseItems.Any(f => f.Id == e.ParentId.Value));
  816. }
  817. if (filter.IsDeadArtist.HasValue && filter.IsDeadArtist.Value)
  818. {
  819. baseQuery = baseQuery
  820. .Where(e => e.ItemValues!.Count(f => f.ItemValue.Type == ItemValueType.Artist || f.ItemValue.Type == ItemValueType.AlbumArtist) == 1);
  821. }
  822. if (filter.IsDeadStudio.HasValue && filter.IsDeadStudio.Value)
  823. {
  824. baseQuery = baseQuery
  825. .Where(e => e.ItemValues!.Count(f => f.ItemValue.Type == ItemValueType.Studios) == 1);
  826. }
  827. if (filter.IsDeadPerson.HasValue && filter.IsDeadPerson.Value)
  828. {
  829. baseQuery = baseQuery
  830. .Where(e => !context.Peoples.Any(f => f.Name == e.Name));
  831. }
  832. if (filter.Years.Length == 1)
  833. {
  834. baseQuery = baseQuery
  835. .Where(e => e.ProductionYear == filter.Years[0]);
  836. }
  837. else if (filter.Years.Length > 1)
  838. {
  839. baseQuery = baseQuery
  840. .Where(e => filter.Years.Any(f => f == e.ProductionYear));
  841. }
  842. var isVirtualItem = filter.IsVirtualItem ?? filter.IsMissing;
  843. if (isVirtualItem.HasValue)
  844. {
  845. baseQuery = baseQuery
  846. .Where(e => e.IsVirtualItem == isVirtualItem.Value);
  847. }
  848. if (filter.IsSpecialSeason.HasValue)
  849. {
  850. if (filter.IsSpecialSeason.Value)
  851. {
  852. baseQuery = baseQuery
  853. .Where(e => e.IndexNumber == 0);
  854. }
  855. else
  856. {
  857. baseQuery = baseQuery
  858. .Where(e => e.IndexNumber != 0);
  859. }
  860. }
  861. if (filter.IsUnaired.HasValue)
  862. {
  863. if (filter.IsUnaired.Value)
  864. {
  865. baseQuery = baseQuery
  866. .Where(e => e.PremiereDate >= now);
  867. }
  868. else
  869. {
  870. baseQuery = baseQuery
  871. .Where(e => e.PremiereDate < now);
  872. }
  873. }
  874. if (filter.MediaTypes.Length > 0)
  875. {
  876. var mediaTypes = filter.MediaTypes.Select(f => f.ToString()).ToArray();
  877. baseQuery = baseQuery
  878. .Where(e => mediaTypes.Contains(e.MediaType));
  879. }
  880. if (filter.ItemIds.Length > 0)
  881. {
  882. baseQuery = baseQuery
  883. .Where(e => filter.ItemIds.Contains(e.Id));
  884. }
  885. if (filter.ExcludeItemIds.Length > 0)
  886. {
  887. baseQuery = baseQuery
  888. .Where(e => !filter.ItemIds.Contains(e.Id));
  889. }
  890. if (filter.ExcludeProviderIds is not null && filter.ExcludeProviderIds.Count > 0)
  891. {
  892. baseQuery = baseQuery.Where(e => !e.Provider!.All(f => !filter.ExcludeProviderIds.All(w => f.ProviderId == w.Key && f.ProviderValue == w.Value)));
  893. }
  894. if (filter.HasAnyProviderId is not null && filter.HasAnyProviderId.Count > 0)
  895. {
  896. baseQuery = baseQuery.Where(e => e.Provider!.Any(f => !filter.HasAnyProviderId.Any(w => f.ProviderId == w.Key && f.ProviderValue == w.Value)));
  897. }
  898. if (filter.HasImdbId.HasValue)
  899. {
  900. baseQuery = baseQuery.Where(e => e.Provider!.Any(f => f.ProviderId == "imdb"));
  901. }
  902. if (filter.HasTmdbId.HasValue)
  903. {
  904. baseQuery = baseQuery.Where(e => e.Provider!.Any(f => f.ProviderId == "tmdb"));
  905. }
  906. if (filter.HasTvdbId.HasValue)
  907. {
  908. baseQuery = baseQuery.Where(e => e.Provider!.Any(f => f.ProviderId == "tvdb"));
  909. }
  910. var queryTopParentIds = filter.TopParentIds;
  911. if (queryTopParentIds.Length > 0)
  912. {
  913. var includedItemByNameTypes = GetItemByNameTypesInQuery(filter);
  914. var enableItemsByName = (filter.IncludeItemsByName ?? false) && includedItemByNameTypes.Count > 0;
  915. if (enableItemsByName && includedItemByNameTypes.Count > 0)
  916. {
  917. baseQuery = baseQuery.Where(e => includedItemByNameTypes.Contains(e.Type) || queryTopParentIds.Any(w => w == e.TopParentId!.Value));
  918. }
  919. else
  920. {
  921. baseQuery = baseQuery.Where(e => queryTopParentIds.Contains(e.TopParentId!.Value));
  922. }
  923. }
  924. if (filter.AncestorIds.Length > 0)
  925. {
  926. baseQuery = baseQuery.Where(e => e.Children!.Any(f => filter.AncestorIds.Contains(f.ParentItemId)));
  927. }
  928. if (!string.IsNullOrWhiteSpace(filter.AncestorWithPresentationUniqueKey))
  929. {
  930. baseQuery = baseQuery
  931. .Where(e => context.BaseItems.Where(f => f.PresentationUniqueKey == filter.AncestorWithPresentationUniqueKey).Any(f => f.ParentAncestors!.Any(w => w.ItemId == f.Id)));
  932. }
  933. if (!string.IsNullOrWhiteSpace(filter.SeriesPresentationUniqueKey))
  934. {
  935. baseQuery = baseQuery
  936. .Where(e => e.SeriesPresentationUniqueKey == filter.SeriesPresentationUniqueKey);
  937. }
  938. if (filter.ExcludeInheritedTags.Length > 0)
  939. {
  940. baseQuery = baseQuery
  941. .Where(e => !e.ItemValues!.Where(w => w.ItemValue.Type == ItemValueType.InheritedTags)
  942. .Any(f => filter.ExcludeInheritedTags.Contains(f.ItemValue.CleanValue)));
  943. }
  944. if (filter.IncludeInheritedTags.Length > 0)
  945. {
  946. // Episodes do not store inherit tags from their parents in the database, and the tag may be still required by the client.
  947. // In addtion to the tags for the episodes themselves, we need to manually query its parent (the season)'s tags as well.
  948. if (includeTypes.Length == 1 && includeTypes.FirstOrDefault() is BaseItemKind.Episode)
  949. {
  950. baseQuery = baseQuery
  951. .Where(e => e.ItemValues!.Where(f => f.ItemValue.Type == ItemValueType.InheritedTags)
  952. .Any(f => filter.IncludeInheritedTags.Contains(f.ItemValue.CleanValue))
  953. ||
  954. (e.ParentId.HasValue && context.ItemValuesMap.Where(w => w.ItemId == e.ParentId.Value)!.Where(w => w.ItemValue.Type == ItemValueType.InheritedTags)
  955. .Any(f => filter.IncludeInheritedTags.Contains(f.ItemValue.CleanValue))));
  956. }
  957. // A playlist should be accessible to its owner regardless of allowed tags.
  958. else if (includeTypes.Length == 1 && includeTypes.FirstOrDefault() is BaseItemKind.Playlist)
  959. {
  960. baseQuery = baseQuery
  961. .Where(e =>
  962. e.ParentAncestors!
  963. .Any(f =>
  964. f.ParentItem.ItemValues!.Any(w => w.ItemValue.Type == ItemValueType.Tags && filter.IncludeInheritedTags.Contains(w.ItemValue.CleanValue))
  965. || e.Data!.Contains($"OwnerUserId\":\"{filter.User!.Id:N}\"")));
  966. // d ^^ this is stupid it hate this.
  967. }
  968. else
  969. {
  970. baseQuery = baseQuery
  971. .Where(e => e.ParentAncestors!.Any(f => f.ParentItem.ItemValues!.Any(w => w.ItemValue.Type == ItemValueType.Tags && filter.IncludeInheritedTags.Contains(w.ItemValue.CleanValue))));
  972. }
  973. }
  974. if (filter.SeriesStatuses.Length > 0)
  975. {
  976. var seriesStatus = filter.SeriesStatuses.Select(e => e.ToString()).ToArray();
  977. baseQuery = baseQuery
  978. .Where(e => seriesStatus.Any(f => e.Data!.Contains(f)));
  979. }
  980. if (filter.BoxSetLibraryFolders.Length > 0)
  981. {
  982. var boxsetFolders = filter.BoxSetLibraryFolders.Select(e => e.ToString("N", CultureInfo.InvariantCulture)).ToArray();
  983. baseQuery = baseQuery
  984. .Where(e => boxsetFolders.Any(f => e.Data!.Contains(f)));
  985. }
  986. if (filter.VideoTypes.Length > 0)
  987. {
  988. var videoTypeBs = filter.VideoTypes.Select(e => $"\"VideoType\":\"{e}\"");
  989. baseQuery = baseQuery
  990. .Where(e => videoTypeBs.Any(f => e.Data!.Contains(f)));
  991. }
  992. if (filter.Is3D.HasValue)
  993. {
  994. if (filter.Is3D.Value)
  995. {
  996. baseQuery = baseQuery
  997. .Where(e => e.Data!.Contains("Video3DFormat"));
  998. }
  999. else
  1000. {
  1001. baseQuery = baseQuery
  1002. .Where(e => !e.Data!.Contains("Video3DFormat"));
  1003. }
  1004. }
  1005. if (filter.IsPlaceHolder.HasValue)
  1006. {
  1007. if (filter.IsPlaceHolder.Value)
  1008. {
  1009. baseQuery = baseQuery
  1010. .Where(e => e.Data!.Contains("IsPlaceHolder\":true"));
  1011. }
  1012. else
  1013. {
  1014. baseQuery = baseQuery
  1015. .Where(e => !e.Data!.Contains("IsPlaceHolder\":true"));
  1016. }
  1017. }
  1018. if (filter.HasSpecialFeature.HasValue)
  1019. {
  1020. if (filter.HasSpecialFeature.Value)
  1021. {
  1022. baseQuery = baseQuery
  1023. .Where(e => e.ExtraIds != null);
  1024. }
  1025. else
  1026. {
  1027. baseQuery = baseQuery
  1028. .Where(e => e.ExtraIds == null);
  1029. }
  1030. }
  1031. if (filter.HasTrailer.HasValue || filter.HasThemeSong.HasValue || filter.HasThemeVideo.HasValue)
  1032. {
  1033. if (filter.HasTrailer.GetValueOrDefault() || filter.HasThemeSong.GetValueOrDefault() || filter.HasThemeVideo.GetValueOrDefault())
  1034. {
  1035. baseQuery = baseQuery
  1036. .Where(e => e.ExtraIds != null);
  1037. }
  1038. else
  1039. {
  1040. baseQuery = baseQuery
  1041. .Where(e => e.ExtraIds == null);
  1042. }
  1043. }
  1044. return baseQuery;
  1045. }
  1046. /// <summary>
  1047. /// Gets the type.
  1048. /// </summary>
  1049. /// <param name="typeName">Name of the type.</param>
  1050. /// <returns>Type.</returns>
  1051. /// <exception cref="ArgumentNullException"><c>typeName</c> is null.</exception>
  1052. private static Type? GetType(string typeName)
  1053. {
  1054. ArgumentException.ThrowIfNullOrEmpty(typeName);
  1055. return _typeMap.GetOrAdd(typeName, k => AppDomain.CurrentDomain.GetAssemblies()
  1056. .Select(a => a.GetType(k))
  1057. .FirstOrDefault(t => t is not null));
  1058. }
  1059. /// <inheritdoc cref="IItemRepository" />
  1060. public void SaveImages(BaseItemDto item)
  1061. {
  1062. ArgumentNullException.ThrowIfNull(item);
  1063. var images = item.ImageInfos.Select(e => Map(item.Id, e));
  1064. using var context = dbProvider.CreateDbContext();
  1065. using var transaction = context.Database.BeginTransaction();
  1066. context.BaseItemImageInfos.Where(e => e.ItemId == item.Id).ExecuteDelete();
  1067. context.BaseItemImageInfos.AddRange(images);
  1068. context.SaveChanges();
  1069. transaction.Commit();
  1070. }
  1071. /// <inheritdoc cref="IItemRepository" />
  1072. public void SaveItems(IReadOnlyList<BaseItemDto> items, CancellationToken cancellationToken)
  1073. {
  1074. UpdateOrInsertItems(items, cancellationToken);
  1075. }
  1076. /// <inheritdoc cref="IItemRepository" />
  1077. public void UpdateOrInsertItems(IReadOnlyList<BaseItemDto> items, CancellationToken cancellationToken)
  1078. {
  1079. ArgumentNullException.ThrowIfNull(items);
  1080. cancellationToken.ThrowIfCancellationRequested();
  1081. var itemsLen = items.Count;
  1082. var tuples = new (BaseItemDto Item, List<Guid>? AncestorIds, BaseItemDto TopParent, string? UserDataKey, List<string> InheritedTags)[itemsLen];
  1083. for (int i = 0; i < itemsLen; i++)
  1084. {
  1085. var item = items[i];
  1086. var ancestorIds = item.SupportsAncestors ?
  1087. item.GetAncestorIds().Distinct().ToList() :
  1088. null;
  1089. var topParent = item.GetTopParent();
  1090. var userdataKey = item.GetUserDataKeys().FirstOrDefault();
  1091. var inheritedTags = item.GetInheritedTags();
  1092. tuples[i] = (item, ancestorIds, topParent, userdataKey, inheritedTags);
  1093. }
  1094. using var context = dbProvider.CreateDbContext();
  1095. using var transaction = context.Database.BeginTransaction();
  1096. foreach (var item in tuples)
  1097. {
  1098. var entity = Map(item.Item);
  1099. if (!context.BaseItems.Any(e => e.Id == entity.Id))
  1100. {
  1101. context.BaseItems.Add(entity);
  1102. }
  1103. else
  1104. {
  1105. context.BaseItemProviders.Where(e => e.ItemId == entity.Id).ExecuteDelete();
  1106. context.BaseItems.Attach(entity).State = EntityState.Modified;
  1107. }
  1108. context.AncestorIds.Where(e => e.ItemId == entity.Id).ExecuteDelete();
  1109. if (item.Item.SupportsAncestors && item.AncestorIds != null)
  1110. {
  1111. foreach (var ancestorId in item.AncestorIds)
  1112. {
  1113. context.AncestorIds.Add(new AncestorId()
  1114. {
  1115. ParentItemId = ancestorId,
  1116. ItemId = entity.Id,
  1117. Item = null!,
  1118. ParentItem = null!
  1119. });
  1120. }
  1121. }
  1122. var itemValuesToSave = GetItemValuesToSave(item.Item, item.InheritedTags);
  1123. context.ItemValuesMap.Where(e => e.ItemId == entity.Id).ExecuteDelete();
  1124. entity.ItemValues = new List<ItemValueMap>();
  1125. foreach (var itemValue in itemValuesToSave)
  1126. {
  1127. var refValue = context.ItemValues
  1128. .Where(f => f.CleanValue == GetCleanValue(itemValue.Value) && (int)f.Type == itemValue.MagicNumber)
  1129. .Select(e => e.ItemValueId)
  1130. .FirstOrDefault();
  1131. if (!refValue.IsEmpty())
  1132. {
  1133. entity.ItemValues.Add(new ItemValueMap()
  1134. {
  1135. Item = entity,
  1136. ItemId = entity.Id,
  1137. ItemValue = null!,
  1138. ItemValueId = refValue
  1139. });
  1140. }
  1141. else
  1142. {
  1143. entity.ItemValues.Add(new ItemValueMap()
  1144. {
  1145. Item = entity,
  1146. ItemId = entity.Id,
  1147. ItemValue = new ItemValue()
  1148. {
  1149. CleanValue = GetCleanValue(itemValue.Value),
  1150. Type = (ItemValueType)itemValue.MagicNumber,
  1151. ItemValueId = Guid.NewGuid(),
  1152. Value = itemValue.Value
  1153. },
  1154. ItemValueId = Guid.Empty
  1155. });
  1156. }
  1157. }
  1158. }
  1159. context.SaveChanges();
  1160. transaction.Commit();
  1161. }
  1162. /// <inheritdoc cref="IItemRepository" />
  1163. public BaseItemDto? RetrieveItem(Guid id)
  1164. {
  1165. if (id.IsEmpty())
  1166. {
  1167. throw new ArgumentException("Guid can't be empty", nameof(id));
  1168. }
  1169. using var context = dbProvider.CreateDbContext();
  1170. var item = context.BaseItems
  1171. .Include(e => e.TrailerTypes)
  1172. .Include(e => e.Provider)
  1173. .Include(e => e.Images)
  1174. .Include(e => e.LockedFields).AsNoTracking().AsSingleQuery().FirstOrDefault(e => e.Id == id);
  1175. if (item is null)
  1176. {
  1177. return null;
  1178. }
  1179. return DeserialiseBaseItem(item);
  1180. }
  1181. /// <summary>
  1182. /// Maps a Entity to the DTO.
  1183. /// </summary>
  1184. /// <param name="entity">The entity.</param>
  1185. /// <param name="dto">The dto base instance.</param>
  1186. /// <param name="appHost">The Application server Host.</param>
  1187. /// <returns>The dto to map.</returns>
  1188. public static BaseItemDto Map(BaseItemEntity entity, BaseItemDto dto, IServerApplicationHost? appHost)
  1189. {
  1190. dto.Id = entity.Id;
  1191. dto.ParentId = entity.ParentId.GetValueOrDefault();
  1192. dto.Path = appHost?.ExpandVirtualPath(entity.Path) ?? entity.Path;
  1193. dto.EndDate = entity.EndDate;
  1194. dto.CommunityRating = entity.CommunityRating;
  1195. dto.CustomRating = entity.CustomRating;
  1196. dto.IndexNumber = entity.IndexNumber;
  1197. dto.IsLocked = entity.IsLocked;
  1198. dto.Name = entity.Name;
  1199. dto.OfficialRating = entity.OfficialRating;
  1200. dto.Overview = entity.Overview;
  1201. dto.ParentIndexNumber = entity.ParentIndexNumber;
  1202. dto.PremiereDate = entity.PremiereDate;
  1203. dto.ProductionYear = entity.ProductionYear;
  1204. dto.SortName = entity.SortName;
  1205. dto.ForcedSortName = entity.ForcedSortName;
  1206. dto.RunTimeTicks = entity.RunTimeTicks;
  1207. dto.PreferredMetadataLanguage = entity.PreferredMetadataLanguage;
  1208. dto.PreferredMetadataCountryCode = entity.PreferredMetadataCountryCode;
  1209. dto.IsInMixedFolder = entity.IsInMixedFolder;
  1210. dto.InheritedParentalRatingValue = entity.InheritedParentalRatingValue;
  1211. dto.CriticRating = entity.CriticRating;
  1212. dto.PresentationUniqueKey = entity.PresentationUniqueKey;
  1213. dto.OriginalTitle = entity.OriginalTitle;
  1214. dto.Album = entity.Album;
  1215. dto.LUFS = entity.LUFS;
  1216. dto.NormalizationGain = entity.NormalizationGain;
  1217. dto.IsVirtualItem = entity.IsVirtualItem;
  1218. dto.ExternalSeriesId = entity.ExternalSeriesId;
  1219. dto.Tagline = entity.Tagline;
  1220. dto.TotalBitrate = entity.TotalBitrate;
  1221. dto.ExternalId = entity.ExternalId;
  1222. dto.Size = entity.Size;
  1223. dto.Genres = entity.Genres?.Split('|') ?? [];
  1224. dto.DateCreated = entity.DateCreated.GetValueOrDefault();
  1225. dto.DateModified = entity.DateModified.GetValueOrDefault();
  1226. dto.ChannelId = string.IsNullOrWhiteSpace(entity.ChannelId) ? Guid.Empty : (Guid.TryParse(entity.ChannelId, out var channelId) ? channelId : Guid.Empty);
  1227. dto.DateLastRefreshed = entity.DateLastRefreshed.GetValueOrDefault();
  1228. dto.DateLastSaved = entity.DateLastSaved.GetValueOrDefault();
  1229. dto.OwnerId = string.IsNullOrWhiteSpace(entity.OwnerId) ? Guid.Empty : (Guid.TryParse(entity.OwnerId, out var ownerId) ? ownerId : Guid.Empty);
  1230. dto.Width = entity.Width.GetValueOrDefault();
  1231. dto.Height = entity.Height.GetValueOrDefault();
  1232. if (entity.Provider is not null)
  1233. {
  1234. dto.ProviderIds = entity.Provider.ToDictionary(e => e.ProviderId, e => e.ProviderValue);
  1235. }
  1236. if (entity.ExtraType is not null)
  1237. {
  1238. dto.ExtraType = (ExtraType)entity.ExtraType;
  1239. }
  1240. if (entity.LockedFields is not null)
  1241. {
  1242. dto.LockedFields = entity.LockedFields?.Select(e => (MetadataField)e.Id).ToArray() ?? [];
  1243. }
  1244. if (entity.Audio is not null)
  1245. {
  1246. dto.Audio = (ProgramAudio)entity.Audio;
  1247. }
  1248. dto.ExtraIds = string.IsNullOrWhiteSpace(entity.ExtraIds) ? [] : entity.ExtraIds.Split('|').Select(e => Guid.Parse(e)).ToArray();
  1249. dto.ProductionLocations = entity.ProductionLocations?.Split('|') ?? [];
  1250. dto.Studios = entity.Studios?.Split('|') ?? [];
  1251. dto.Tags = entity.Tags?.Split('|') ?? [];
  1252. if (dto is IHasProgramAttributes hasProgramAttributes)
  1253. {
  1254. hasProgramAttributes.IsMovie = entity.IsMovie;
  1255. hasProgramAttributes.IsSeries = entity.IsSeries;
  1256. hasProgramAttributes.EpisodeTitle = entity.EpisodeTitle;
  1257. hasProgramAttributes.IsRepeat = entity.IsRepeat;
  1258. }
  1259. if (dto is LiveTvChannel liveTvChannel)
  1260. {
  1261. liveTvChannel.ServiceName = entity.ExternalServiceId;
  1262. }
  1263. if (dto is Trailer trailer)
  1264. {
  1265. trailer.TrailerTypes = entity.TrailerTypes?.Select(e => (TrailerType)e.Id).ToArray() ?? [];
  1266. }
  1267. if (dto is Video video)
  1268. {
  1269. video.PrimaryVersionId = entity.PrimaryVersionId;
  1270. }
  1271. if (dto is IHasSeries hasSeriesName)
  1272. {
  1273. hasSeriesName.SeriesName = entity.SeriesName;
  1274. hasSeriesName.SeriesId = entity.SeriesId.GetValueOrDefault();
  1275. hasSeriesName.SeriesPresentationUniqueKey = entity.SeriesPresentationUniqueKey;
  1276. }
  1277. if (dto is Episode episode)
  1278. {
  1279. episode.SeasonName = entity.SeasonName;
  1280. episode.SeasonId = entity.SeasonId.GetValueOrDefault();
  1281. }
  1282. if (dto is IHasArtist hasArtists)
  1283. {
  1284. hasArtists.Artists = entity.Artists?.Split('|', StringSplitOptions.RemoveEmptyEntries) ?? [];
  1285. }
  1286. if (dto is IHasAlbumArtist hasAlbumArtists)
  1287. {
  1288. hasAlbumArtists.AlbumArtists = entity.AlbumArtists?.Split('|', StringSplitOptions.RemoveEmptyEntries) ?? [];
  1289. }
  1290. if (dto is LiveTvProgram program)
  1291. {
  1292. program.ShowId = entity.ShowId;
  1293. }
  1294. if (entity.Images is not null)
  1295. {
  1296. dto.ImageInfos = entity.Images.Select(e => Map(e, appHost)).ToArray();
  1297. }
  1298. // dto.Type = entity.Type;
  1299. // dto.Data = entity.Data;
  1300. // dto.MediaType = entity.MediaType;
  1301. if (dto is IHasStartDate hasStartDate)
  1302. {
  1303. hasStartDate.StartDate = entity.StartDate;
  1304. }
  1305. // Fields that are present in the DB but are never actually used
  1306. // dto.UnratedType = entity.UnratedType;
  1307. // dto.TopParentId = entity.TopParentId;
  1308. // dto.CleanName = entity.CleanName;
  1309. // dto.UserDataKey = entity.UserDataKey;
  1310. if (dto is Folder folder)
  1311. {
  1312. folder.DateLastMediaAdded = entity.DateLastMediaAdded;
  1313. }
  1314. return dto;
  1315. }
  1316. /// <summary>
  1317. /// Maps a Entity to the DTO.
  1318. /// </summary>
  1319. /// <param name="dto">The entity.</param>
  1320. /// <returns>The dto to map.</returns>
  1321. public BaseItemEntity Map(BaseItemDto dto)
  1322. {
  1323. var dtoType = dto.GetType();
  1324. var entity = new BaseItemEntity()
  1325. {
  1326. Type = dtoType.ToString(),
  1327. Id = dto.Id
  1328. };
  1329. if (TypeRequiresDeserialization(dtoType))
  1330. {
  1331. entity.Data = JsonSerializer.Serialize(dto, dtoType, JsonDefaults.Options);
  1332. }
  1333. entity.ParentId = !dto.ParentId.IsEmpty() ? dto.ParentId : null;
  1334. entity.Path = GetPathToSave(dto.Path);
  1335. entity.EndDate = dto.EndDate.GetValueOrDefault();
  1336. entity.CommunityRating = dto.CommunityRating;
  1337. entity.CustomRating = dto.CustomRating;
  1338. entity.IndexNumber = dto.IndexNumber;
  1339. entity.IsLocked = dto.IsLocked;
  1340. entity.Name = dto.Name;
  1341. entity.OfficialRating = dto.OfficialRating;
  1342. entity.Overview = dto.Overview;
  1343. entity.ParentIndexNumber = dto.ParentIndexNumber;
  1344. entity.PremiereDate = dto.PremiereDate;
  1345. entity.ProductionYear = dto.ProductionYear;
  1346. entity.SortName = dto.SortName;
  1347. entity.ForcedSortName = dto.ForcedSortName;
  1348. entity.RunTimeTicks = dto.RunTimeTicks;
  1349. entity.PreferredMetadataLanguage = dto.PreferredMetadataLanguage;
  1350. entity.PreferredMetadataCountryCode = dto.PreferredMetadataCountryCode;
  1351. entity.IsInMixedFolder = dto.IsInMixedFolder;
  1352. entity.InheritedParentalRatingValue = dto.InheritedParentalRatingValue;
  1353. entity.CriticRating = dto.CriticRating;
  1354. entity.PresentationUniqueKey = dto.PresentationUniqueKey;
  1355. entity.OriginalTitle = dto.OriginalTitle;
  1356. entity.Album = dto.Album;
  1357. entity.LUFS = dto.LUFS;
  1358. entity.NormalizationGain = dto.NormalizationGain;
  1359. entity.IsVirtualItem = dto.IsVirtualItem;
  1360. entity.ExternalSeriesId = dto.ExternalSeriesId;
  1361. entity.Tagline = dto.Tagline;
  1362. entity.TotalBitrate = dto.TotalBitrate;
  1363. entity.ExternalId = dto.ExternalId;
  1364. entity.Size = dto.Size;
  1365. entity.Genres = string.Join('|', dto.Genres);
  1366. entity.DateCreated = dto.DateCreated;
  1367. entity.DateModified = dto.DateModified;
  1368. entity.ChannelId = dto.ChannelId.ToString();
  1369. entity.DateLastRefreshed = dto.DateLastRefreshed;
  1370. entity.DateLastSaved = dto.DateLastSaved;
  1371. entity.OwnerId = dto.OwnerId.ToString();
  1372. entity.Width = dto.Width;
  1373. entity.Height = dto.Height;
  1374. entity.Provider = dto.ProviderIds.Select(e => new BaseItemProvider()
  1375. {
  1376. Item = entity,
  1377. ProviderId = e.Key,
  1378. ProviderValue = e.Value
  1379. }).ToList();
  1380. if (dto.Audio.HasValue)
  1381. {
  1382. entity.Audio = (ProgramAudioEntity)dto.Audio;
  1383. }
  1384. if (dto.ExtraType.HasValue)
  1385. {
  1386. entity.ExtraType = (BaseItemExtraType)dto.ExtraType;
  1387. }
  1388. entity.ExtraIds = dto.ExtraIds is not null ? string.Join('|', dto.ExtraIds) : null;
  1389. entity.ProductionLocations = dto.ProductionLocations is not null ? string.Join('|', dto.ProductionLocations) : null;
  1390. entity.Studios = dto.Studios is not null ? string.Join('|', dto.Studios) : null;
  1391. entity.Tags = dto.Tags is not null ? string.Join('|', dto.Tags) : null;
  1392. entity.LockedFields = dto.LockedFields is not null ? dto.LockedFields
  1393. .Select(e => new BaseItemMetadataField()
  1394. {
  1395. Id = (int)e,
  1396. Item = entity,
  1397. ItemId = entity.Id
  1398. })
  1399. .ToArray() : null;
  1400. if (dto is IHasProgramAttributes hasProgramAttributes)
  1401. {
  1402. entity.IsMovie = hasProgramAttributes.IsMovie;
  1403. entity.IsSeries = hasProgramAttributes.IsSeries;
  1404. entity.EpisodeTitle = hasProgramAttributes.EpisodeTitle;
  1405. entity.IsRepeat = hasProgramAttributes.IsRepeat;
  1406. }
  1407. if (dto is LiveTvChannel liveTvChannel)
  1408. {
  1409. entity.ExternalServiceId = liveTvChannel.ServiceName;
  1410. }
  1411. if (dto is Video video)
  1412. {
  1413. entity.PrimaryVersionId = video.PrimaryVersionId;
  1414. }
  1415. if (dto is IHasSeries hasSeriesName)
  1416. {
  1417. entity.SeriesName = hasSeriesName.SeriesName;
  1418. entity.SeriesId = hasSeriesName.SeriesId;
  1419. entity.SeriesPresentationUniqueKey = hasSeriesName.SeriesPresentationUniqueKey;
  1420. }
  1421. if (dto is Episode episode)
  1422. {
  1423. entity.SeasonName = episode.SeasonName;
  1424. entity.SeasonId = episode.SeasonId;
  1425. }
  1426. if (dto is IHasArtist hasArtists)
  1427. {
  1428. entity.Artists = hasArtists.Artists is not null ? string.Join('|', hasArtists.Artists) : null;
  1429. }
  1430. if (dto is IHasAlbumArtist hasAlbumArtists)
  1431. {
  1432. entity.AlbumArtists = hasAlbumArtists.AlbumArtists is not null ? string.Join('|', hasAlbumArtists.AlbumArtists) : null;
  1433. }
  1434. if (dto is LiveTvProgram program)
  1435. {
  1436. entity.ShowId = program.ShowId;
  1437. }
  1438. if (dto.ImageInfos is not null)
  1439. {
  1440. entity.Images = dto.ImageInfos.Select(f => Map(dto.Id, f)).ToArray();
  1441. }
  1442. if (dto is Trailer trailer)
  1443. {
  1444. entity.TrailerTypes = trailer.TrailerTypes?.Select(e => new BaseItemTrailerType()
  1445. {
  1446. Id = (int)e,
  1447. Item = entity,
  1448. ItemId = entity.Id
  1449. }).ToArray() ?? [];
  1450. }
  1451. // dto.Type = entity.Type;
  1452. // dto.Data = entity.Data;
  1453. // dto.MediaType = entity.MediaType;
  1454. if (dto is IHasStartDate hasStartDate)
  1455. {
  1456. entity.StartDate = hasStartDate.StartDate;
  1457. }
  1458. // Fields that are present in the DB but are never actually used
  1459. // dto.UnratedType = entity.UnratedType;
  1460. // dto.TopParentId = entity.TopParentId;
  1461. // dto.CleanName = entity.CleanName;
  1462. // dto.UserDataKey = entity.UserDataKey;
  1463. if (dto is Folder folder)
  1464. {
  1465. entity.DateLastMediaAdded = folder.DateLastMediaAdded;
  1466. entity.IsFolder = folder.IsFolder;
  1467. }
  1468. return entity;
  1469. }
  1470. private IReadOnlyList<string> GetItemValueNames(ItemValueType[] itemValueTypes, IReadOnlyList<string> withItemTypes, IReadOnlyList<string> excludeItemTypes)
  1471. {
  1472. using var context = dbProvider.CreateDbContext();
  1473. var query = context.ItemValuesMap
  1474. .AsNoTracking()
  1475. .Where(e => itemValueTypes.Any(w => (ItemValueType)w == e.ItemValue.Type));
  1476. if (withItemTypes.Count > 0)
  1477. {
  1478. query = query.Where(e => withItemTypes.Contains(e.Item.Type));
  1479. }
  1480. if (excludeItemTypes.Count > 0)
  1481. {
  1482. query = query.Where(e => !excludeItemTypes.Contains(e.Item.Type));
  1483. }
  1484. // query = query.DistinctBy(e => e.CleanValue);
  1485. return query.Select(e => e.ItemValue.CleanValue).ToImmutableArray();
  1486. }
  1487. private static bool TypeRequiresDeserialization(Type type)
  1488. {
  1489. return type.GetCustomAttribute<RequiresSourceSerialisationAttribute>() == null;
  1490. }
  1491. private BaseItemDto DeserialiseBaseItem(BaseItemEntity baseItemEntity, bool skipDeserialization = false)
  1492. {
  1493. ArgumentNullException.ThrowIfNull(baseItemEntity, nameof(baseItemEntity));
  1494. if (serverConfigurationManager?.Configuration is null)
  1495. {
  1496. throw new InvalidOperationException("Server Configuration manager or configuration is null");
  1497. }
  1498. var typeToSerialise = GetType(baseItemEntity.Type);
  1499. return BaseItemRepository.DeserialiseBaseItem(
  1500. baseItemEntity,
  1501. logger,
  1502. appHost,
  1503. skipDeserialization || (serverConfigurationManager.Configuration.SkipDeserializationForBasicTypes && (typeToSerialise == typeof(Channel) || typeToSerialise == typeof(UserRootFolder))));
  1504. }
  1505. /// <summary>
  1506. /// Deserialises a BaseItemEntity and sets all properties.
  1507. /// </summary>
  1508. /// <param name="baseItemEntity">The DB entity.</param>
  1509. /// <param name="logger">Logger.</param>
  1510. /// <param name="appHost">The application server Host.</param>
  1511. /// <param name="skipDeserialization">If only mapping should be processed.</param>
  1512. /// <returns>A mapped BaseItem.</returns>
  1513. /// <exception cref="InvalidOperationException">Will be thrown if an invalid serialisation is requested.</exception>
  1514. public static BaseItemDto DeserialiseBaseItem(BaseItemEntity baseItemEntity, ILogger logger, IServerApplicationHost? appHost, bool skipDeserialization = false)
  1515. {
  1516. var type = GetType(baseItemEntity.Type) ?? throw new InvalidOperationException("Cannot deserialise unkown type.");
  1517. BaseItemDto? dto = null;
  1518. if (TypeRequiresDeserialization(type) && baseItemEntity.Data is not null && !skipDeserialization)
  1519. {
  1520. try
  1521. {
  1522. using var dataAsStream = new MemoryStream(Encoding.UTF8.GetBytes(baseItemEntity.Data!));
  1523. dto = JsonSerializer.Deserialize(dataAsStream, type, JsonDefaults.Options) as BaseItemDto;
  1524. }
  1525. catch (JsonException ex)
  1526. {
  1527. logger.LogError(ex, "Error deserializing item with JSON: {Data}", baseItemEntity.Data);
  1528. }
  1529. }
  1530. if (dto is null)
  1531. {
  1532. dto = Activator.CreateInstance(type) as BaseItemDto ?? throw new InvalidOperationException("Cannot deserialise unkown type.");
  1533. }
  1534. return Map(baseItemEntity, dto, appHost);
  1535. }
  1536. private QueryResult<(BaseItemDto Item, ItemCounts ItemCounts)> GetItemValues(InternalItemsQuery filter, ItemValueType[] itemValueTypes, string returnType)
  1537. {
  1538. ArgumentNullException.ThrowIfNull(filter);
  1539. if (!filter.Limit.HasValue)
  1540. {
  1541. filter.EnableTotalRecordCount = false;
  1542. }
  1543. using var context = dbProvider.CreateDbContext();
  1544. var innerQuery = new InternalItemsQuery(filter.User)
  1545. {
  1546. ExcludeItemTypes = filter.ExcludeItemTypes,
  1547. IncludeItemTypes = filter.IncludeItemTypes,
  1548. MediaTypes = filter.MediaTypes,
  1549. AncestorIds = filter.AncestorIds,
  1550. ItemIds = filter.ItemIds,
  1551. TopParentIds = filter.TopParentIds,
  1552. ParentId = filter.ParentId,
  1553. IsAiring = filter.IsAiring,
  1554. IsMovie = filter.IsMovie,
  1555. IsSports = filter.IsSports,
  1556. IsKids = filter.IsKids,
  1557. IsNews = filter.IsNews,
  1558. IsSeries = filter.IsSeries
  1559. };
  1560. var query = TranslateQuery(context.BaseItems.AsNoTracking(), context, innerQuery);
  1561. query = query.Where(e => e.Type == returnType && e.ItemValues!.Any(f => e.CleanName == f.ItemValue.CleanValue && itemValueTypes.Any(w => (ItemValueType)w == f.ItemValue.Type)));
  1562. if (filter.OrderBy.Count != 0
  1563. || !string.IsNullOrEmpty(filter.SearchTerm))
  1564. {
  1565. query = ApplyOrder(query, filter);
  1566. }
  1567. else
  1568. {
  1569. query = query.OrderBy(e => e.SortName);
  1570. }
  1571. if (filter.Limit.HasValue || filter.StartIndex.HasValue)
  1572. {
  1573. var offset = filter.StartIndex ?? 0;
  1574. if (offset > 0)
  1575. {
  1576. query = query.Skip(offset);
  1577. }
  1578. if (filter.Limit.HasValue)
  1579. {
  1580. query = query.Take(filter.Limit.Value);
  1581. }
  1582. }
  1583. var result = new QueryResult<(BaseItemDto, ItemCounts)>();
  1584. if (filter.EnableTotalRecordCount)
  1585. {
  1586. result.TotalRecordCount = query.DistinctBy(e => e.PresentationUniqueKey).Count();
  1587. }
  1588. var seriesTypeName = itemTypeLookup.BaseItemKindNames[BaseItemKind.Series];
  1589. var movieTypeName = itemTypeLookup.BaseItemKindNames[BaseItemKind.Movie];
  1590. var episodeTypeName = itemTypeLookup.BaseItemKindNames[BaseItemKind.Episode];
  1591. var musicAlbumTypeName = itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicAlbum];
  1592. var musicArtistTypeName = itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicArtist];
  1593. var audioTypeName = itemTypeLookup.BaseItemKindNames[BaseItemKind.Audio];
  1594. var trailerTypeName = itemTypeLookup.BaseItemKindNames[BaseItemKind.Trailer];
  1595. var resultQuery = query.Select(e => new
  1596. {
  1597. item = e,
  1598. // TODO: This is bad refactor!
  1599. itemCount = new ItemCounts()
  1600. {
  1601. SeriesCount = e.ItemValues!.Count(f => f.Item.Type == seriesTypeName),
  1602. EpisodeCount = e.ItemValues!.Count(f => f.Item.Type == episodeTypeName),
  1603. MovieCount = e.ItemValues!.Count(f => f.Item.Type == movieTypeName),
  1604. AlbumCount = e.ItemValues!.Count(f => f.Item.Type == musicAlbumTypeName),
  1605. ArtistCount = e.ItemValues!.Count(f => f.Item.Type == musicArtistTypeName),
  1606. SongCount = e.ItemValues!.Count(f => f.Item.Type == audioTypeName),
  1607. TrailerCount = e.ItemValues!.Count(f => f.Item.Type == trailerTypeName),
  1608. }
  1609. });
  1610. result.StartIndex = filter.StartIndex ?? 0;
  1611. result.Items = resultQuery.ToImmutableArray().Where(e => e is not null).Select(e =>
  1612. {
  1613. return (DeserialiseBaseItem(e.item, filter.SkipDeserialization), e.itemCount);
  1614. }).ToImmutableArray();
  1615. return result;
  1616. }
  1617. private static void PrepareFilterQuery(InternalItemsQuery query)
  1618. {
  1619. if (query.Limit.HasValue && query.EnableGroupByMetadataKey)
  1620. {
  1621. query.Limit = query.Limit.Value + 4;
  1622. }
  1623. if (query.IsResumable ?? false)
  1624. {
  1625. query.IsVirtualItem = false;
  1626. }
  1627. }
  1628. private string GetCleanValue(string value)
  1629. {
  1630. if (string.IsNullOrWhiteSpace(value))
  1631. {
  1632. return value;
  1633. }
  1634. return value.RemoveDiacritics().ToLowerInvariant();
  1635. }
  1636. private List<(int MagicNumber, string Value)> GetItemValuesToSave(BaseItemDto item, List<string> inheritedTags)
  1637. {
  1638. var list = new List<(int, string)>();
  1639. if (item is IHasArtist hasArtist)
  1640. {
  1641. list.AddRange(hasArtist.Artists.Select(i => (0, i)));
  1642. }
  1643. if (item is IHasAlbumArtist hasAlbumArtist)
  1644. {
  1645. list.AddRange(hasAlbumArtist.AlbumArtists.Select(i => (1, i)));
  1646. }
  1647. list.AddRange(item.Genres.Select(i => (2, i)));
  1648. list.AddRange(item.Studios.Select(i => (3, i)));
  1649. list.AddRange(item.Tags.Select(i => (4, i)));
  1650. // keywords was 5
  1651. list.AddRange(inheritedTags.Select(i => (6, i)));
  1652. // Remove all invalid values.
  1653. list.RemoveAll(i => string.IsNullOrWhiteSpace(i.Item2));
  1654. return list;
  1655. }
  1656. private static BaseItemImageInfo Map(Guid baseItemId, ItemImageInfo e)
  1657. {
  1658. return new BaseItemImageInfo()
  1659. {
  1660. ItemId = baseItemId,
  1661. Id = Guid.NewGuid(),
  1662. Path = e.Path,
  1663. Blurhash = e.BlurHash != null ? Encoding.UTF8.GetBytes(e.BlurHash) : null,
  1664. DateModified = e.DateModified,
  1665. Height = e.Height,
  1666. Width = e.Width,
  1667. ImageType = (ImageInfoImageType)e.Type,
  1668. Item = null!
  1669. };
  1670. }
  1671. private static ItemImageInfo Map(BaseItemImageInfo e, IServerApplicationHost? appHost)
  1672. {
  1673. return new ItemImageInfo()
  1674. {
  1675. Path = appHost?.ExpandVirtualPath(e.Path) ?? e.Path,
  1676. BlurHash = e.Blurhash != null ? Encoding.UTF8.GetString(e.Blurhash) : null,
  1677. DateModified = e.DateModified,
  1678. Height = e.Height,
  1679. Width = e.Width,
  1680. Type = (ImageType)e.ImageType
  1681. };
  1682. }
  1683. private string? GetPathToSave(string path)
  1684. {
  1685. if (path is null)
  1686. {
  1687. return null;
  1688. }
  1689. return appHost.ReverseVirtualPath(path);
  1690. }
  1691. private List<string> GetItemByNameTypesInQuery(InternalItemsQuery query)
  1692. {
  1693. var list = new List<string>();
  1694. if (IsTypeInQuery(BaseItemKind.Person, query))
  1695. {
  1696. list.Add(itemTypeLookup.BaseItemKindNames[BaseItemKind.Person]!);
  1697. }
  1698. if (IsTypeInQuery(BaseItemKind.Genre, query))
  1699. {
  1700. list.Add(itemTypeLookup.BaseItemKindNames[BaseItemKind.Genre]!);
  1701. }
  1702. if (IsTypeInQuery(BaseItemKind.MusicGenre, query))
  1703. {
  1704. list.Add(itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicGenre]!);
  1705. }
  1706. if (IsTypeInQuery(BaseItemKind.MusicArtist, query))
  1707. {
  1708. list.Add(itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicArtist]!);
  1709. }
  1710. if (IsTypeInQuery(BaseItemKind.Studio, query))
  1711. {
  1712. list.Add(itemTypeLookup.BaseItemKindNames[BaseItemKind.Studio]!);
  1713. }
  1714. return list;
  1715. }
  1716. private bool IsTypeInQuery(BaseItemKind type, InternalItemsQuery query)
  1717. {
  1718. if (query.ExcludeItemTypes.Contains(type))
  1719. {
  1720. return false;
  1721. }
  1722. return query.IncludeItemTypes.Length == 0 || query.IncludeItemTypes.Contains(type);
  1723. }
  1724. private Expression<Func<BaseItemEntity, object>> MapOrderByField(ItemSortBy sortBy, InternalItemsQuery query)
  1725. {
  1726. #pragma warning disable CS8603 // Possible null reference return.
  1727. return sortBy switch
  1728. {
  1729. ItemSortBy.AirTime => e => e.SortName, // TODO
  1730. ItemSortBy.Runtime => e => e.RunTimeTicks,
  1731. ItemSortBy.Random => e => EF.Functions.Random(),
  1732. ItemSortBy.DatePlayed => e => e.UserData!.FirstOrDefault(f => f.UserId == query.User!.Id)!.LastPlayedDate,
  1733. ItemSortBy.PlayCount => e => e.UserData!.FirstOrDefault(f => f.UserId == query.User!.Id)!.PlayCount,
  1734. ItemSortBy.IsFavoriteOrLiked => e => e.UserData!.FirstOrDefault(f => f.UserId == query.User!.Id)!.IsFavorite,
  1735. ItemSortBy.IsFolder => e => e.IsFolder,
  1736. ItemSortBy.IsPlayed => e => e.UserData!.FirstOrDefault(f => f.UserId == query.User!.Id)!.Played,
  1737. ItemSortBy.IsUnplayed => e => !e.UserData!.FirstOrDefault(f => f.UserId == query.User!.Id)!.Played,
  1738. ItemSortBy.DateLastContentAdded => e => e.DateLastMediaAdded,
  1739. ItemSortBy.Artist => e => e.ItemValues!.Where(f => f.ItemValue.Type == ItemValueType.Artist).Select(f => f.ItemValue.CleanValue),
  1740. ItemSortBy.AlbumArtist => e => e.ItemValues!.Where(f => f.ItemValue.Type == ItemValueType.AlbumArtist).Select(f => f.ItemValue.CleanValue),
  1741. ItemSortBy.Studio => e => e.ItemValues!.Where(f => f.ItemValue.Type == ItemValueType.Studios).Select(f => f.ItemValue.CleanValue),
  1742. ItemSortBy.OfficialRating => e => e.InheritedParentalRatingValue,
  1743. // ItemSortBy.SeriesDatePlayed => "(Select MAX(LastPlayedDate) from TypedBaseItems B" + GetJoinUserDataText(query) + " where Played=1 and B.SeriesPresentationUniqueKey=A.PresentationUniqueKey)",
  1744. ItemSortBy.SeriesSortName => e => e.SeriesName,
  1745. // ItemSortBy.AiredEpisodeOrder => "AiredEpisodeOrder",
  1746. ItemSortBy.Album => e => e.Album,
  1747. ItemSortBy.DateCreated => e => e.DateCreated,
  1748. ItemSortBy.PremiereDate => e => e.PremiereDate,
  1749. ItemSortBy.StartDate => e => e.StartDate,
  1750. ItemSortBy.Name => e => e.Name,
  1751. ItemSortBy.CommunityRating => e => e.CommunityRating,
  1752. ItemSortBy.ProductionYear => e => e.ProductionYear,
  1753. ItemSortBy.CriticRating => e => e.CriticRating,
  1754. ItemSortBy.VideoBitRate => e => e.TotalBitrate,
  1755. ItemSortBy.ParentIndexNumber => e => e.ParentIndexNumber,
  1756. ItemSortBy.IndexNumber => e => e.IndexNumber,
  1757. _ => e => e.SortName
  1758. };
  1759. #pragma warning restore CS8603 // Possible null reference return.
  1760. }
  1761. private bool EnableGroupByPresentationUniqueKey(InternalItemsQuery query)
  1762. {
  1763. if (!query.GroupByPresentationUniqueKey)
  1764. {
  1765. return false;
  1766. }
  1767. if (query.GroupBySeriesPresentationUniqueKey)
  1768. {
  1769. return false;
  1770. }
  1771. if (!string.IsNullOrWhiteSpace(query.PresentationUniqueKey))
  1772. {
  1773. return false;
  1774. }
  1775. if (query.User is null)
  1776. {
  1777. return false;
  1778. }
  1779. if (query.IncludeItemTypes.Length == 0)
  1780. {
  1781. return true;
  1782. }
  1783. return query.IncludeItemTypes.Contains(BaseItemKind.Episode)
  1784. || query.IncludeItemTypes.Contains(BaseItemKind.Video)
  1785. || query.IncludeItemTypes.Contains(BaseItemKind.Movie)
  1786. || query.IncludeItemTypes.Contains(BaseItemKind.MusicVideo)
  1787. || query.IncludeItemTypes.Contains(BaseItemKind.Series)
  1788. || query.IncludeItemTypes.Contains(BaseItemKind.Season);
  1789. }
  1790. private IQueryable<BaseItemEntity> ApplyOrder(IQueryable<BaseItemEntity> query, InternalItemsQuery filter)
  1791. {
  1792. var orderBy = filter.OrderBy;
  1793. bool hasSearch = !string.IsNullOrEmpty(filter.SearchTerm);
  1794. if (hasSearch)
  1795. {
  1796. List<(ItemSortBy, SortOrder)> prepend = new List<(ItemSortBy, SortOrder)>(4);
  1797. if (hasSearch)
  1798. {
  1799. prepend.Add((ItemSortBy.SortName, SortOrder.Ascending));
  1800. }
  1801. orderBy = filter.OrderBy = [.. prepend, .. orderBy];
  1802. }
  1803. else if (orderBy.Count == 0)
  1804. {
  1805. return query;
  1806. }
  1807. IOrderedQueryable<BaseItemEntity>? orderedQuery = null;
  1808. var firstOrdering = orderBy.FirstOrDefault();
  1809. if (firstOrdering != default)
  1810. {
  1811. var expression = MapOrderByField(firstOrdering.OrderBy, filter);
  1812. if (firstOrdering.SortOrder == SortOrder.Ascending)
  1813. {
  1814. orderedQuery = query.OrderBy(expression);
  1815. }
  1816. else
  1817. {
  1818. orderedQuery = query.OrderByDescending(expression);
  1819. }
  1820. if (firstOrdering.OrderBy is ItemSortBy.Default or ItemSortBy.SortName)
  1821. {
  1822. if (firstOrdering.SortOrder is SortOrder.Ascending)
  1823. {
  1824. orderedQuery = orderedQuery.ThenBy(e => e.Name);
  1825. }
  1826. else
  1827. {
  1828. orderedQuery = orderedQuery.ThenByDescending(e => e.Name);
  1829. }
  1830. }
  1831. }
  1832. foreach (var item in orderBy.Skip(1))
  1833. {
  1834. var expression = MapOrderByField(item.OrderBy, filter);
  1835. if (item.SortOrder == SortOrder.Ascending)
  1836. {
  1837. orderedQuery = orderedQuery!.ThenBy(expression);
  1838. }
  1839. else
  1840. {
  1841. orderedQuery = orderedQuery!.ThenByDescending(expression);
  1842. }
  1843. }
  1844. return orderedQuery ?? query;
  1845. }
  1846. }