BaseItemRepository.cs 79 KB

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