12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327 |
- #pragma warning disable RS0030 // Do not use banned APIs
- // Do not enforce that because EFCore cannot deal with cultures well.
- #pragma warning disable CA1304 // Specify CultureInfo
- #pragma warning disable CA1311 // Specify a culture or use an invariant version
- #pragma warning disable CA1862 // Use the 'StringComparison' method overloads to perform case-insensitive string comparisons
- using System;
- using System.Collections.Concurrent;
- using System.Collections.Generic;
- using System.Globalization;
- using System.Linq;
- using System.Linq.Expressions;
- using System.Reflection;
- using System.Text;
- using System.Text.Json;
- using System.Threading;
- using Jellyfin.Data.Enums;
- using Jellyfin.Database.Implementations;
- using Jellyfin.Database.Implementations.Entities;
- using Jellyfin.Database.Implementations.Enums;
- using Jellyfin.Extensions;
- using Jellyfin.Extensions.Json;
- using Jellyfin.Server.Implementations.Extensions;
- using MediaBrowser.Common;
- using MediaBrowser.Controller;
- using MediaBrowser.Controller.Channels;
- using MediaBrowser.Controller.Configuration;
- using MediaBrowser.Controller.Entities;
- using MediaBrowser.Controller.Entities.Audio;
- using MediaBrowser.Controller.Entities.TV;
- using MediaBrowser.Controller.LiveTv;
- using MediaBrowser.Controller.Persistence;
- using MediaBrowser.Model.Dto;
- using MediaBrowser.Model.Entities;
- using MediaBrowser.Model.LiveTv;
- using MediaBrowser.Model.Querying;
- using Microsoft.EntityFrameworkCore;
- using Microsoft.Extensions.Logging;
- using BaseItemDto = MediaBrowser.Controller.Entities.BaseItem;
- using BaseItemEntity = Jellyfin.Database.Implementations.Entities.BaseItemEntity;
- namespace Jellyfin.Server.Implementations.Item;
- /*
- All queries in this class and all other nullable enabled EFCore repository classes will make liberal use of the null-forgiving operator "!".
- This is done as the code isn't actually executed client side, but only the expressions are interpret and the compiler cannot know that.
- This is your only warning/message regarding this topic.
- */
- /// <summary>
- /// Handles all storage logic for BaseItems.
- /// </summary>
- public sealed class BaseItemRepository
- : IItemRepository
- {
- /// <summary>
- /// This holds all the types in the running assemblies
- /// so that we can de-serialize properly when we don't have strong types.
- /// </summary>
- private static readonly ConcurrentDictionary<string, Type?> _typeMap = new ConcurrentDictionary<string, Type?>();
- private readonly IDbContextFactory<JellyfinDbContext> _dbProvider;
- private readonly IServerApplicationHost _appHost;
- private readonly IItemTypeLookup _itemTypeLookup;
- private readonly IServerConfigurationManager _serverConfigurationManager;
- private readonly ILogger<BaseItemRepository> _logger;
- private static readonly IReadOnlyList<ItemValueType> _getAllArtistsValueTypes = [ItemValueType.Artist, ItemValueType.AlbumArtist];
- private static readonly IReadOnlyList<ItemValueType> _getArtistValueTypes = [ItemValueType.Artist];
- private static readonly IReadOnlyList<ItemValueType> _getAlbumArtistValueTypes = [ItemValueType.AlbumArtist];
- private static readonly IReadOnlyList<ItemValueType> _getStudiosValueTypes = [ItemValueType.Studios];
- private static readonly IReadOnlyList<ItemValueType> _getGenreValueTypes = [ItemValueType.Genre];
- /// <summary>
- /// Initializes a new instance of the <see cref="BaseItemRepository"/> class.
- /// </summary>
- /// <param name="dbProvider">The db factory.</param>
- /// <param name="appHost">The Application host.</param>
- /// <param name="itemTypeLookup">The static type lookup.</param>
- /// <param name="serverConfigurationManager">The server Configuration manager.</param>
- /// <param name="logger">System logger.</param>
- public BaseItemRepository(
- IDbContextFactory<JellyfinDbContext> dbProvider,
- IServerApplicationHost appHost,
- IItemTypeLookup itemTypeLookup,
- IServerConfigurationManager serverConfigurationManager,
- ILogger<BaseItemRepository> logger)
- {
- _dbProvider = dbProvider;
- _appHost = appHost;
- _itemTypeLookup = itemTypeLookup;
- _serverConfigurationManager = serverConfigurationManager;
- _logger = logger;
- }
- /// <inheritdoc />
- public void DeleteItem(Guid id)
- {
- if (id.IsEmpty())
- {
- throw new ArgumentException("Guid can't be empty", nameof(id));
- }
- using var context = _dbProvider.CreateDbContext();
- using var transaction = context.Database.BeginTransaction();
- context.AncestorIds.Where(e => e.ItemId == id || e.ParentItemId == id).ExecuteDelete();
- context.AttachmentStreamInfos.Where(e => e.ItemId == id).ExecuteDelete();
- context.BaseItemImageInfos.Where(e => e.ItemId == id).ExecuteDelete();
- context.BaseItemMetadataFields.Where(e => e.ItemId == id).ExecuteDelete();
- context.BaseItemProviders.Where(e => e.ItemId == id).ExecuteDelete();
- context.BaseItemTrailerTypes.Where(e => e.ItemId == id).ExecuteDelete();
- context.BaseItems.Where(e => e.Id == id).ExecuteDelete();
- context.Chapters.Where(e => e.ItemId == id).ExecuteDelete();
- context.CustomItemDisplayPreferences.Where(e => e.ItemId == id).ExecuteDelete();
- context.ItemDisplayPreferences.Where(e => e.ItemId == id).ExecuteDelete();
- context.ItemValues.Where(e => e.BaseItemsMap!.Count == 0).ExecuteDelete();
- context.ItemValuesMap.Where(e => e.ItemId == id).ExecuteDelete();
- context.KeyframeData.Where(e => e.ItemId == id).ExecuteDelete();
- context.MediaSegments.Where(e => e.ItemId == id).ExecuteDelete();
- context.MediaStreamInfos.Where(e => e.ItemId == id).ExecuteDelete();
- context.PeopleBaseItemMap.Where(e => e.ItemId == id).ExecuteDelete();
- context.Peoples.Where(e => e.BaseItems!.Count == 0).ExecuteDelete();
- context.TrickplayInfos.Where(e => e.ItemId == id).ExecuteDelete();
- context.SaveChanges();
- transaction.Commit();
- }
- /// <inheritdoc />
- public void UpdateInheritedValues()
- {
- using var context = _dbProvider.CreateDbContext();
- using var transaction = context.Database.BeginTransaction();
- context.ItemValuesMap.Where(e => e.ItemValue.Type == ItemValueType.InheritedTags).ExecuteDelete();
- // ItemValue Inheritance is now correctly mapped via AncestorId on demand
- context.SaveChanges();
- transaction.Commit();
- }
- /// <inheritdoc />
- public IReadOnlyList<Guid> GetItemIdsList(InternalItemsQuery filter)
- {
- ArgumentNullException.ThrowIfNull(filter);
- PrepareFilterQuery(filter);
- using var context = _dbProvider.CreateDbContext();
- return ApplyQueryFilter(context.BaseItems.AsNoTracking(), context, filter).Select(e => e.Id).ToArray();
- }
- /// <inheritdoc />
- public QueryResult<(BaseItemDto Item, ItemCounts? ItemCounts)> GetAllArtists(InternalItemsQuery filter)
- {
- return GetItemValues(filter, _getAllArtistsValueTypes, _itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicArtist]);
- }
- /// <inheritdoc />
- public QueryResult<(BaseItemDto Item, ItemCounts? ItemCounts)> GetArtists(InternalItemsQuery filter)
- {
- return GetItemValues(filter, _getArtistValueTypes, _itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicArtist]);
- }
- /// <inheritdoc />
- public QueryResult<(BaseItemDto Item, ItemCounts? ItemCounts)> GetAlbumArtists(InternalItemsQuery filter)
- {
- return GetItemValues(filter, _getAlbumArtistValueTypes, _itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicArtist]);
- }
- /// <inheritdoc />
- public QueryResult<(BaseItemDto Item, ItemCounts? ItemCounts)> GetStudios(InternalItemsQuery filter)
- {
- return GetItemValues(filter, _getStudiosValueTypes, _itemTypeLookup.BaseItemKindNames[BaseItemKind.Studio]);
- }
- /// <inheritdoc />
- public QueryResult<(BaseItemDto Item, ItemCounts? ItemCounts)> GetGenres(InternalItemsQuery filter)
- {
- return GetItemValues(filter, _getGenreValueTypes, _itemTypeLookup.BaseItemKindNames[BaseItemKind.Genre]);
- }
- /// <inheritdoc />
- public QueryResult<(BaseItemDto Item, ItemCounts? ItemCounts)> GetMusicGenres(InternalItemsQuery filter)
- {
- return GetItemValues(filter, _getGenreValueTypes, _itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicGenre]);
- }
- /// <inheritdoc />
- public IReadOnlyList<string> GetStudioNames()
- {
- return GetItemValueNames(_getStudiosValueTypes, [], []);
- }
- /// <inheritdoc />
- public IReadOnlyList<string> GetAllArtistNames()
- {
- return GetItemValueNames(_getAllArtistsValueTypes, [], []);
- }
- /// <inheritdoc />
- public IReadOnlyList<string> GetMusicGenreNames()
- {
- return GetItemValueNames(
- _getGenreValueTypes,
- _itemTypeLookup.MusicGenreTypes,
- []);
- }
- /// <inheritdoc />
- public IReadOnlyList<string> GetGenreNames()
- {
- return GetItemValueNames(
- _getGenreValueTypes,
- [],
- _itemTypeLookup.MusicGenreTypes);
- }
- /// <inheritdoc />
- public QueryResult<BaseItemDto> GetItems(InternalItemsQuery filter)
- {
- ArgumentNullException.ThrowIfNull(filter);
- if (!filter.EnableTotalRecordCount || (!filter.Limit.HasValue && (filter.StartIndex ?? 0) == 0))
- {
- var returnList = GetItemList(filter);
- return new QueryResult<BaseItemDto>(
- filter.StartIndex,
- returnList.Count,
- returnList);
- }
- PrepareFilterQuery(filter);
- var result = new QueryResult<BaseItemDto>();
- using var context = _dbProvider.CreateDbContext();
- IQueryable<BaseItemEntity> dbQuery = PrepareItemQuery(context, filter);
- dbQuery = TranslateQuery(dbQuery, context, filter);
- if (filter.EnableTotalRecordCount)
- {
- result.TotalRecordCount = dbQuery.Count();
- }
- dbQuery = ApplyGroupingFilter(dbQuery, filter);
- dbQuery = ApplyQueryPaging(dbQuery, filter);
- result.Items = dbQuery.AsEnumerable().Where(e => e is not null).Select(w => DeserialiseBaseItem(w, filter.SkipDeserialization)).ToArray();
- result.StartIndex = filter.StartIndex ?? 0;
- return result;
- }
- /// <inheritdoc />
- public IReadOnlyList<BaseItemDto> GetItemList(InternalItemsQuery filter)
- {
- ArgumentNullException.ThrowIfNull(filter);
- PrepareFilterQuery(filter);
- using var context = _dbProvider.CreateDbContext();
- IQueryable<BaseItemEntity> dbQuery = PrepareItemQuery(context, filter);
- dbQuery = TranslateQuery(dbQuery, context, filter);
- dbQuery = ApplyGroupingFilter(dbQuery, filter);
- dbQuery = ApplyQueryPaging(dbQuery, filter);
- return dbQuery.AsEnumerable().Where(e => e is not null).Select(w => DeserialiseBaseItem(w, filter.SkipDeserialization)).ToArray();
- }
- /// <inheritdoc/>
- public IReadOnlyList<BaseItem> GetLatestItemList(InternalItemsQuery filter, CollectionType collectionType)
- {
- ArgumentNullException.ThrowIfNull(filter);
- PrepareFilterQuery(filter);
- // Early exit if collection type is not tvshows or music
- if (collectionType != CollectionType.tvshows && collectionType != CollectionType.music)
- {
- return Array.Empty<BaseItem>();
- }
- using var context = _dbProvider.CreateDbContext();
- // Subquery to group by SeriesNames/Album and get the max Date Created for each group.
- var subquery = PrepareItemQuery(context, filter);
- subquery = TranslateQuery(subquery, context, filter);
- var subqueryGrouped = subquery.GroupBy(g => collectionType == CollectionType.tvshows ? g.SeriesName : g.Album)
- .Select(g => new
- {
- Key = g.Key,
- MaxDateCreated = g.Max(a => a.DateCreated)
- })
- .OrderByDescending(g => g.MaxDateCreated)
- .Select(g => g);
- if (filter.Limit.HasValue)
- {
- subqueryGrouped = subqueryGrouped.Take(filter.Limit.Value);
- }
- filter.Limit = null;
- var mainquery = PrepareItemQuery(context, filter);
- mainquery = TranslateQuery(mainquery, context, filter);
- mainquery = mainquery.Where(g => g.DateCreated >= subqueryGrouped.Min(s => s.MaxDateCreated));
- mainquery = ApplyGroupingFilter(mainquery, filter);
- mainquery = ApplyQueryPaging(mainquery, filter);
- return mainquery.AsEnumerable().Where(e => e is not null).Select(w => DeserialiseBaseItem(w, filter.SkipDeserialization)).ToArray();
- }
- /// <inheritdoc />
- public IReadOnlyList<string> GetNextUpSeriesKeys(InternalItemsQuery filter, DateTime dateCutoff)
- {
- ArgumentNullException.ThrowIfNull(filter);
- ArgumentNullException.ThrowIfNull(filter.User);
- using var context = _dbProvider.CreateDbContext();
- var query = context.BaseItems
- .AsNoTracking()
- .Where(i => filter.TopParentIds.Contains(i.TopParentId!.Value))
- .Where(i => i.Type == _itemTypeLookup.BaseItemKindNames[BaseItemKind.Episode])
- .Join(
- context.UserData.AsNoTracking(),
- i => new { UserId = filter.User.Id, ItemId = i.Id },
- u => new { UserId = u.UserId, ItemId = u.ItemId },
- (entity, data) => new { Item = entity, UserData = data })
- .GroupBy(g => g.Item.SeriesPresentationUniqueKey)
- .Select(g => new { g.Key, LastPlayedDate = g.Max(u => u.UserData.LastPlayedDate) })
- .Where(g => g.Key != null && g.LastPlayedDate != null && g.LastPlayedDate >= dateCutoff)
- .OrderByDescending(g => g.LastPlayedDate)
- .Select(g => g.Key!);
- if (filter.Limit.HasValue)
- {
- query = query.Take(filter.Limit.Value);
- }
- return query.ToArray();
- }
- private IQueryable<BaseItemEntity> ApplyGroupingFilter(IQueryable<BaseItemEntity> dbQuery, InternalItemsQuery filter)
- {
- // This whole block is needed to filter duplicate entries on request
- // for the time being it cannot be used because it would destroy the ordering
- // this results in "duplicate" responses for queries that try to lookup individual series or multiple versions but
- // for that case the invoker has to run a DistinctBy(e => e.PresentationUniqueKey) on their own
- // var enableGroupByPresentationUniqueKey = EnableGroupByPresentationUniqueKey(filter);
- // if (enableGroupByPresentationUniqueKey && filter.GroupBySeriesPresentationUniqueKey)
- // {
- // dbQuery = ApplyOrder(dbQuery, filter);
- // dbQuery = dbQuery.GroupBy(e => new { e.PresentationUniqueKey, e.SeriesPresentationUniqueKey }).Select(e => e.First());
- // }
- // else if (enableGroupByPresentationUniqueKey)
- // {
- // dbQuery = ApplyOrder(dbQuery, filter);
- // dbQuery = dbQuery.GroupBy(e => e.PresentationUniqueKey).Select(e => e.First());
- // }
- // else if (filter.GroupBySeriesPresentationUniqueKey)
- // {
- // dbQuery = ApplyOrder(dbQuery, filter);
- // dbQuery = dbQuery.GroupBy(e => e.SeriesPresentationUniqueKey).Select(e => e.First());
- // }
- // else
- // {
- // dbQuery = dbQuery.Distinct();
- // dbQuery = ApplyOrder(dbQuery, filter);
- // }
- dbQuery = dbQuery.Distinct();
- dbQuery = ApplyOrder(dbQuery, filter);
- return dbQuery;
- }
- private IQueryable<BaseItemEntity> ApplyQueryPaging(IQueryable<BaseItemEntity> dbQuery, InternalItemsQuery filter)
- {
- if (filter.Limit.HasValue || filter.StartIndex.HasValue)
- {
- var offset = filter.StartIndex ?? 0;
- if (offset > 0)
- {
- dbQuery = dbQuery.Skip(offset);
- }
- if (filter.Limit.HasValue)
- {
- dbQuery = dbQuery.Take(filter.Limit.Value);
- }
- }
- return dbQuery;
- }
- private IQueryable<BaseItemEntity> ApplyQueryFilter(IQueryable<BaseItemEntity> dbQuery, JellyfinDbContext context, InternalItemsQuery filter)
- {
- dbQuery = TranslateQuery(dbQuery, context, filter);
- dbQuery = ApplyOrder(dbQuery, filter);
- dbQuery = ApplyGroupingFilter(dbQuery, filter);
- dbQuery = ApplyQueryPaging(dbQuery, filter);
- return dbQuery;
- }
- private IQueryable<BaseItemEntity> PrepareItemQuery(JellyfinDbContext context, InternalItemsQuery filter)
- {
- IQueryable<BaseItemEntity> dbQuery = context.BaseItems.AsNoTracking();
- dbQuery = dbQuery.AsSingleQuery()
- .Include(e => e.TrailerTypes)
- .Include(e => e.Provider)
- .Include(e => e.LockedFields);
- if (filter.DtoOptions.EnableImages)
- {
- dbQuery = dbQuery.Include(e => e.Images);
- }
- return dbQuery;
- }
- /// <inheritdoc/>
- public int GetCount(InternalItemsQuery filter)
- {
- ArgumentNullException.ThrowIfNull(filter);
- // Hack for right now since we currently don't support filtering out these duplicates within a query
- PrepareFilterQuery(filter);
- using var context = _dbProvider.CreateDbContext();
- var dbQuery = TranslateQuery(context.BaseItems.AsNoTracking(), context, filter);
- return dbQuery.Count();
- }
- #pragma warning disable CA1307 // Specify StringComparison for clarity
- /// <summary>
- /// Gets the type.
- /// </summary>
- /// <param name="typeName">Name of the type.</param>
- /// <returns>Type.</returns>
- /// <exception cref="ArgumentNullException"><c>typeName</c> is null.</exception>
- private static Type? GetType(string typeName)
- {
- ArgumentException.ThrowIfNullOrEmpty(typeName);
- // TODO: this isn't great. Refactor later to be both globally handled by a dedicated service not just an static variable and be loaded eagerly.
- // currently this is done so that plugins may introduce their own type of baseitems as we dont know when we are first called, before or after plugins are loaded
- return _typeMap.GetOrAdd(typeName, k => AppDomain.CurrentDomain.GetAssemblies()
- .Select(a => a.GetType(k))
- .FirstOrDefault(t => t is not null));
- }
- /// <inheritdoc />
- public void SaveImages(BaseItemDto item)
- {
- ArgumentNullException.ThrowIfNull(item);
- var images = item.ImageInfos.Select(e => Map(item.Id, e));
- using var context = _dbProvider.CreateDbContext();
- context.BaseItemImageInfos.Where(e => e.ItemId == item.Id).ExecuteDelete();
- context.BaseItemImageInfos.AddRange(images);
- context.SaveChanges();
- }
- /// <inheritdoc />
- public void SaveItems(IReadOnlyList<BaseItemDto> items, CancellationToken cancellationToken)
- {
- UpdateOrInsertItems(items, cancellationToken);
- }
- /// <inheritdoc cref="IItemRepository"/>
- public void UpdateOrInsertItems(IReadOnlyList<BaseItemDto> items, CancellationToken cancellationToken)
- {
- ArgumentNullException.ThrowIfNull(items);
- cancellationToken.ThrowIfCancellationRequested();
- var tuples = new List<(BaseItemDto Item, List<Guid>? AncestorIds, BaseItemDto TopParent, IEnumerable<string> UserDataKey, List<string> InheritedTags)>();
- foreach (var item in items.GroupBy(e => e.Id).Select(e => e.Last()))
- {
- var ancestorIds = item.SupportsAncestors ?
- item.GetAncestorIds().Distinct().ToList() :
- null;
- var topParent = item.GetTopParent();
- var userdataKey = item.GetUserDataKeys();
- var inheritedTags = item.GetInheritedTags();
- tuples.Add((item, ancestorIds, topParent, userdataKey, inheritedTags));
- }
- using var context = _dbProvider.CreateDbContext();
- using var transaction = context.Database.BeginTransaction();
- var ids = tuples.Select(f => f.Item.Id).ToArray();
- var existingItems = context.BaseItems.Where(e => ids.Contains(e.Id)).Select(f => f.Id).ToArray();
- foreach (var item in tuples)
- {
- var entity = Map(item.Item);
- // TODO: refactor this "inconsistency"
- entity.TopParentId = item.TopParent?.Id;
- if (!existingItems.Any(e => e == entity.Id))
- {
- context.BaseItems.Add(entity);
- }
- else
- {
- context.BaseItemProviders.Where(e => e.ItemId == entity.Id).ExecuteDelete();
- context.BaseItems.Attach(entity).State = EntityState.Modified;
- }
- }
- context.SaveChanges();
- var itemValueMaps = tuples
- .Select(e => (Item: e.Item, Values: GetItemValuesToSave(e.Item, e.InheritedTags)))
- .ToArray();
- var allListedItemValues = itemValueMaps
- .SelectMany(f => f.Values)
- .Distinct()
- .ToArray();
- var existingValues = context.ItemValues
- .Select(e => new
- {
- item = e,
- Key = e.Type + "+" + e.Value
- })
- .Where(f => allListedItemValues.Select(e => $"{(int)e.MagicNumber}+{e.Value}").Contains(f.Key))
- .Select(e => e.item)
- .ToArray();
- var missingItemValues = allListedItemValues.Except(existingValues.Select(f => (MagicNumber: f.Type, f.Value))).Select(f => new ItemValue()
- {
- CleanValue = GetCleanValue(f.Value),
- ItemValueId = Guid.NewGuid(),
- Type = f.MagicNumber,
- Value = f.Value
- }).ToArray();
- context.ItemValues.AddRange(missingItemValues);
- context.SaveChanges();
- var itemValuesStore = existingValues.Concat(missingItemValues).ToArray();
- var valueMap = itemValueMaps
- .Select(f => (Item: f.Item, Values: f.Values.Select(e => itemValuesStore.First(g => g.Value == e.Value && g.Type == e.MagicNumber)).ToArray()))
- .ToArray();
- var mappedValues = context.ItemValuesMap.Where(e => ids.Contains(e.ItemId)).ToList();
- foreach (var item in valueMap)
- {
- var itemMappedValues = mappedValues.Where(e => e.ItemId == item.Item.Id).ToList();
- foreach (var itemValue in item.Values)
- {
- var existingItem = itemMappedValues.FirstOrDefault(f => f.ItemValueId == itemValue.ItemValueId);
- if (existingItem is null)
- {
- context.ItemValuesMap.Add(new ItemValueMap()
- {
- Item = null!,
- ItemId = item.Item.Id,
- ItemValue = null!,
- ItemValueId = itemValue.ItemValueId
- });
- }
- else
- {
- // map exists, remove from list so its been handled.
- itemMappedValues.Remove(existingItem);
- }
- }
- // all still listed values are not in the new list so remove them.
- context.ItemValuesMap.RemoveRange(itemMappedValues);
- }
- context.SaveChanges();
- foreach (var item in tuples)
- {
- if (item.Item.SupportsAncestors && item.AncestorIds != null)
- {
- var existingAncestorIds = context.AncestorIds.Where(e => e.ItemId == item.Item.Id).ToList();
- var validAncestorIds = context.BaseItems.Where(e => item.AncestorIds.Contains(e.Id)).Select(f => f.Id).ToArray();
- foreach (var ancestorId in validAncestorIds)
- {
- var existingAncestorId = existingAncestorIds.FirstOrDefault(e => e.ParentItemId == ancestorId);
- if (existingAncestorId is null)
- {
- context.AncestorIds.Add(new AncestorId()
- {
- ParentItemId = ancestorId,
- ItemId = item.Item.Id,
- Item = null!,
- ParentItem = null!
- });
- }
- else
- {
- existingAncestorIds.Remove(existingAncestorId);
- }
- }
- context.AncestorIds.RemoveRange(existingAncestorIds);
- }
- }
- context.SaveChanges();
- transaction.Commit();
- }
- /// <inheritdoc />
- public BaseItemDto? RetrieveItem(Guid id)
- {
- if (id.IsEmpty())
- {
- throw new ArgumentException("Guid can't be empty", nameof(id));
- }
- using var context = _dbProvider.CreateDbContext();
- var item = PrepareItemQuery(context, new()
- {
- DtoOptions = new()
- {
- EnableImages = true
- }
- }).FirstOrDefault(e => e.Id == id);
- if (item is null)
- {
- return null;
- }
- return DeserialiseBaseItem(item);
- }
- /// <summary>
- /// Maps a Entity to the DTO.
- /// </summary>
- /// <param name="entity">The entity.</param>
- /// <param name="dto">The dto base instance.</param>
- /// <param name="appHost">The Application server Host.</param>
- /// <returns>The dto to map.</returns>
- public static BaseItemDto Map(BaseItemEntity entity, BaseItemDto dto, IServerApplicationHost? appHost)
- {
- dto.Id = entity.Id;
- dto.ParentId = entity.ParentId.GetValueOrDefault();
- dto.Path = appHost?.ExpandVirtualPath(entity.Path) ?? entity.Path;
- dto.EndDate = entity.EndDate;
- dto.CommunityRating = entity.CommunityRating;
- dto.CustomRating = entity.CustomRating;
- dto.IndexNumber = entity.IndexNumber;
- dto.IsLocked = entity.IsLocked;
- dto.Name = entity.Name;
- dto.OfficialRating = entity.OfficialRating;
- dto.Overview = entity.Overview;
- dto.ParentIndexNumber = entity.ParentIndexNumber;
- dto.PremiereDate = entity.PremiereDate;
- dto.ProductionYear = entity.ProductionYear;
- dto.SortName = entity.SortName;
- dto.ForcedSortName = entity.ForcedSortName;
- dto.RunTimeTicks = entity.RunTimeTicks;
- dto.PreferredMetadataLanguage = entity.PreferredMetadataLanguage;
- dto.PreferredMetadataCountryCode = entity.PreferredMetadataCountryCode;
- dto.IsInMixedFolder = entity.IsInMixedFolder;
- dto.InheritedParentalRatingValue = entity.InheritedParentalRatingValue;
- dto.InheritedParentalRatingSubValue = entity.InheritedParentalRatingSubValue;
- dto.CriticRating = entity.CriticRating;
- dto.PresentationUniqueKey = entity.PresentationUniqueKey;
- dto.OriginalTitle = entity.OriginalTitle;
- dto.Album = entity.Album;
- dto.LUFS = entity.LUFS;
- dto.NormalizationGain = entity.NormalizationGain;
- dto.IsVirtualItem = entity.IsVirtualItem;
- dto.ExternalSeriesId = entity.ExternalSeriesId;
- dto.Tagline = entity.Tagline;
- dto.TotalBitrate = entity.TotalBitrate;
- dto.ExternalId = entity.ExternalId;
- dto.Size = entity.Size;
- dto.Genres = entity.Genres?.Split('|') ?? [];
- dto.DateCreated = entity.DateCreated.GetValueOrDefault();
- dto.DateModified = entity.DateModified.GetValueOrDefault();
- dto.ChannelId = entity.ChannelId ?? Guid.Empty;
- dto.DateLastRefreshed = entity.DateLastRefreshed.GetValueOrDefault();
- dto.DateLastSaved = entity.DateLastSaved.GetValueOrDefault();
- dto.OwnerId = string.IsNullOrWhiteSpace(entity.OwnerId) ? Guid.Empty : (Guid.TryParse(entity.OwnerId, out var ownerId) ? ownerId : Guid.Empty);
- dto.Width = entity.Width.GetValueOrDefault();
- dto.Height = entity.Height.GetValueOrDefault();
- if (entity.Provider is not null)
- {
- dto.ProviderIds = entity.Provider.ToDictionary(e => e.ProviderId, e => e.ProviderValue);
- }
- if (entity.ExtraType is not null)
- {
- dto.ExtraType = (ExtraType)entity.ExtraType;
- }
- if (entity.LockedFields is not null)
- {
- dto.LockedFields = entity.LockedFields?.Select(e => (MetadataField)e.Id).ToArray() ?? [];
- }
- if (entity.Audio is not null)
- {
- dto.Audio = (ProgramAudio)entity.Audio;
- }
- dto.ExtraIds = string.IsNullOrWhiteSpace(entity.ExtraIds) ? [] : entity.ExtraIds.Split('|').Select(e => Guid.Parse(e)).ToArray();
- dto.ProductionLocations = entity.ProductionLocations?.Split('|') ?? [];
- dto.Studios = entity.Studios?.Split('|') ?? [];
- dto.Tags = entity.Tags?.Split('|') ?? [];
- if (dto is IHasProgramAttributes hasProgramAttributes)
- {
- hasProgramAttributes.IsMovie = entity.IsMovie;
- hasProgramAttributes.IsSeries = entity.IsSeries;
- hasProgramAttributes.EpisodeTitle = entity.EpisodeTitle;
- hasProgramAttributes.IsRepeat = entity.IsRepeat;
- }
- if (dto is LiveTvChannel liveTvChannel)
- {
- liveTvChannel.ServiceName = entity.ExternalServiceId;
- }
- if (dto is Trailer trailer)
- {
- trailer.TrailerTypes = entity.TrailerTypes?.Select(e => (TrailerType)e.Id).ToArray() ?? [];
- }
- if (dto is Video video)
- {
- video.PrimaryVersionId = entity.PrimaryVersionId;
- }
- if (dto is IHasSeries hasSeriesName)
- {
- hasSeriesName.SeriesName = entity.SeriesName;
- hasSeriesName.SeriesId = entity.SeriesId.GetValueOrDefault();
- hasSeriesName.SeriesPresentationUniqueKey = entity.SeriesPresentationUniqueKey;
- }
- if (dto is Episode episode)
- {
- episode.SeasonName = entity.SeasonName;
- episode.SeasonId = entity.SeasonId.GetValueOrDefault();
- }
- if (dto is IHasArtist hasArtists)
- {
- hasArtists.Artists = entity.Artists?.Split('|', StringSplitOptions.RemoveEmptyEntries) ?? [];
- }
- if (dto is IHasAlbumArtist hasAlbumArtists)
- {
- hasAlbumArtists.AlbumArtists = entity.AlbumArtists?.Split('|', StringSplitOptions.RemoveEmptyEntries) ?? [];
- }
- if (dto is LiveTvProgram program)
- {
- program.ShowId = entity.ShowId;
- }
- if (entity.Images is not null)
- {
- dto.ImageInfos = entity.Images.Select(e => Map(e, appHost)).ToArray();
- }
- // dto.Type = entity.Type;
- // dto.Data = entity.Data;
- // dto.MediaType = Enum.TryParse<MediaType>(entity.MediaType);
- if (dto is IHasStartDate hasStartDate)
- {
- hasStartDate.StartDate = entity.StartDate.GetValueOrDefault();
- }
- // Fields that are present in the DB but are never actually used
- // dto.UnratedType = entity.UnratedType;
- // dto.TopParentId = entity.TopParentId;
- // dto.CleanName = entity.CleanName;
- // dto.UserDataKey = entity.UserDataKey;
- if (dto is Folder folder)
- {
- folder.DateLastMediaAdded = entity.DateLastMediaAdded;
- }
- return dto;
- }
- /// <summary>
- /// Maps a Entity to the DTO.
- /// </summary>
- /// <param name="dto">The entity.</param>
- /// <returns>The dto to map.</returns>
- public BaseItemEntity Map(BaseItemDto dto)
- {
- var dtoType = dto.GetType();
- var entity = new BaseItemEntity()
- {
- Type = dtoType.ToString(),
- Id = dto.Id
- };
- if (TypeRequiresDeserialization(dtoType))
- {
- entity.Data = JsonSerializer.Serialize(dto, dtoType, JsonDefaults.Options);
- }
- entity.ParentId = !dto.ParentId.IsEmpty() ? dto.ParentId : null;
- entity.Path = GetPathToSave(dto.Path);
- entity.EndDate = dto.EndDate;
- entity.CommunityRating = dto.CommunityRating;
- entity.CustomRating = dto.CustomRating;
- entity.IndexNumber = dto.IndexNumber;
- entity.IsLocked = dto.IsLocked;
- entity.Name = dto.Name;
- entity.CleanName = GetCleanValue(dto.Name);
- entity.OfficialRating = dto.OfficialRating;
- entity.Overview = dto.Overview;
- entity.ParentIndexNumber = dto.ParentIndexNumber;
- entity.PremiereDate = dto.PremiereDate;
- entity.ProductionYear = dto.ProductionYear;
- entity.SortName = dto.SortName;
- entity.ForcedSortName = dto.ForcedSortName;
- entity.RunTimeTicks = dto.RunTimeTicks;
- entity.PreferredMetadataLanguage = dto.PreferredMetadataLanguage;
- entity.PreferredMetadataCountryCode = dto.PreferredMetadataCountryCode;
- entity.IsInMixedFolder = dto.IsInMixedFolder;
- entity.InheritedParentalRatingValue = dto.InheritedParentalRatingValue;
- entity.InheritedParentalRatingSubValue = dto.InheritedParentalRatingSubValue;
- entity.CriticRating = dto.CriticRating;
- entity.PresentationUniqueKey = dto.PresentationUniqueKey;
- entity.OriginalTitle = dto.OriginalTitle;
- entity.Album = dto.Album;
- entity.LUFS = dto.LUFS;
- entity.NormalizationGain = dto.NormalizationGain;
- entity.IsVirtualItem = dto.IsVirtualItem;
- entity.ExternalSeriesId = dto.ExternalSeriesId;
- entity.Tagline = dto.Tagline;
- entity.TotalBitrate = dto.TotalBitrate;
- entity.ExternalId = dto.ExternalId;
- entity.Size = dto.Size;
- entity.Genres = string.Join('|', dto.Genres);
- entity.DateCreated = dto.DateCreated;
- entity.DateModified = dto.DateModified;
- entity.ChannelId = dto.ChannelId;
- entity.DateLastRefreshed = dto.DateLastRefreshed;
- entity.DateLastSaved = dto.DateLastSaved;
- entity.OwnerId = dto.OwnerId.ToString();
- entity.Width = dto.Width;
- entity.Height = dto.Height;
- entity.Provider = dto.ProviderIds.Select(e => new BaseItemProvider()
- {
- Item = entity,
- ProviderId = e.Key,
- ProviderValue = e.Value
- }).ToList();
- if (dto.Audio.HasValue)
- {
- entity.Audio = (ProgramAudioEntity)dto.Audio;
- }
- if (dto.ExtraType.HasValue)
- {
- entity.ExtraType = (BaseItemExtraType)dto.ExtraType;
- }
- entity.ExtraIds = dto.ExtraIds is not null ? string.Join('|', dto.ExtraIds) : null;
- entity.ProductionLocations = dto.ProductionLocations is not null ? string.Join('|', dto.ProductionLocations) : null;
- entity.Studios = dto.Studios is not null ? string.Join('|', dto.Studios) : null;
- entity.Tags = dto.Tags is not null ? string.Join('|', dto.Tags) : null;
- entity.LockedFields = dto.LockedFields is not null ? dto.LockedFields
- .Select(e => new BaseItemMetadataField()
- {
- Id = (int)e,
- Item = entity,
- ItemId = entity.Id
- })
- .ToArray() : null;
- if (dto is IHasProgramAttributes hasProgramAttributes)
- {
- entity.IsMovie = hasProgramAttributes.IsMovie;
- entity.IsSeries = hasProgramAttributes.IsSeries;
- entity.EpisodeTitle = hasProgramAttributes.EpisodeTitle;
- entity.IsRepeat = hasProgramAttributes.IsRepeat;
- }
- if (dto is LiveTvChannel liveTvChannel)
- {
- entity.ExternalServiceId = liveTvChannel.ServiceName;
- }
- if (dto is Video video)
- {
- entity.PrimaryVersionId = video.PrimaryVersionId;
- }
- if (dto is IHasSeries hasSeriesName)
- {
- entity.SeriesName = hasSeriesName.SeriesName;
- entity.SeriesId = hasSeriesName.SeriesId;
- entity.SeriesPresentationUniqueKey = hasSeriesName.SeriesPresentationUniqueKey;
- }
- if (dto is Episode episode)
- {
- entity.SeasonName = episode.SeasonName;
- entity.SeasonId = episode.SeasonId;
- }
- if (dto is IHasArtist hasArtists)
- {
- entity.Artists = hasArtists.Artists is not null ? string.Join('|', hasArtists.Artists) : null;
- }
- if (dto is IHasAlbumArtist hasAlbumArtists)
- {
- entity.AlbumArtists = hasAlbumArtists.AlbumArtists is not null ? string.Join('|', hasAlbumArtists.AlbumArtists) : null;
- }
- if (dto is LiveTvProgram program)
- {
- entity.ShowId = program.ShowId;
- }
- if (dto.ImageInfos is not null)
- {
- entity.Images = dto.ImageInfos.Select(f => Map(dto.Id, f)).ToArray();
- }
- if (dto is Trailer trailer)
- {
- entity.TrailerTypes = trailer.TrailerTypes?.Select(e => new BaseItemTrailerType()
- {
- Id = (int)e,
- Item = entity,
- ItemId = entity.Id
- }).ToArray() ?? [];
- }
- // dto.Type = entity.Type;
- // dto.Data = entity.Data;
- entity.MediaType = dto.MediaType.ToString();
- if (dto is IHasStartDate hasStartDate)
- {
- entity.StartDate = hasStartDate.StartDate;
- }
- entity.UnratedType = dto.GetBlockUnratedType().ToString();
- // Fields that are present in the DB but are never actually used
- // dto.UserDataKey = entity.UserDataKey;
- if (dto is Folder folder)
- {
- entity.DateLastMediaAdded = folder.DateLastMediaAdded;
- entity.IsFolder = folder.IsFolder;
- }
- return entity;
- }
- private string[] GetItemValueNames(IReadOnlyList<ItemValueType> itemValueTypes, IReadOnlyList<string> withItemTypes, IReadOnlyList<string> excludeItemTypes)
- {
- using var context = _dbProvider.CreateDbContext();
- var query = context.ItemValuesMap
- .AsNoTracking()
- .Where(e => itemValueTypes.Any(w => (ItemValueType)w == e.ItemValue.Type));
- if (withItemTypes.Count > 0)
- {
- query = query.Where(e => withItemTypes.Contains(e.Item.Type));
- }
- if (excludeItemTypes.Count > 0)
- {
- query = query.Where(e => !excludeItemTypes.Contains(e.Item.Type));
- }
- // query = query.DistinctBy(e => e.CleanValue);
- return query.Select(e => e.ItemValue)
- .GroupBy(e => e.CleanValue)
- .Select(e => e.First().Value)
- .ToArray();
- }
- private static bool TypeRequiresDeserialization(Type type)
- {
- return type.GetCustomAttribute<RequiresSourceSerialisationAttribute>() == null;
- }
- private BaseItemDto DeserialiseBaseItem(BaseItemEntity baseItemEntity, bool skipDeserialization = false)
- {
- ArgumentNullException.ThrowIfNull(baseItemEntity, nameof(baseItemEntity));
- if (_serverConfigurationManager?.Configuration is null)
- {
- throw new InvalidOperationException("Server Configuration manager or configuration is null");
- }
- var typeToSerialise = GetType(baseItemEntity.Type);
- return BaseItemRepository.DeserialiseBaseItem(
- baseItemEntity,
- _logger,
- _appHost,
- skipDeserialization || (_serverConfigurationManager.Configuration.SkipDeserializationForBasicTypes && (typeToSerialise == typeof(Channel) || typeToSerialise == typeof(UserRootFolder))));
- }
- /// <summary>
- /// Deserialises a BaseItemEntity and sets all properties.
- /// </summary>
- /// <param name="baseItemEntity">The DB entity.</param>
- /// <param name="logger">Logger.</param>
- /// <param name="appHost">The application server Host.</param>
- /// <param name="skipDeserialization">If only mapping should be processed.</param>
- /// <returns>A mapped BaseItem.</returns>
- /// <exception cref="InvalidOperationException">Will be thrown if an invalid serialisation is requested.</exception>
- public static BaseItemDto DeserialiseBaseItem(BaseItemEntity baseItemEntity, ILogger logger, IServerApplicationHost? appHost, bool skipDeserialization = false)
- {
- var type = GetType(baseItemEntity.Type) ?? throw new InvalidOperationException("Cannot deserialise unknown type.");
- BaseItemDto? dto = null;
- if (TypeRequiresDeserialization(type) && baseItemEntity.Data is not null && !skipDeserialization)
- {
- try
- {
- dto = JsonSerializer.Deserialize(baseItemEntity.Data, type, JsonDefaults.Options) as BaseItemDto;
- }
- catch (JsonException ex)
- {
- logger.LogError(ex, "Error deserializing item with JSON: {Data}", baseItemEntity.Data);
- }
- }
- if (dto is null)
- {
- dto = Activator.CreateInstance(type) as BaseItemDto ?? throw new InvalidOperationException("Cannot deserialise unknown type.");
- }
- return Map(baseItemEntity, dto, appHost);
- }
- private QueryResult<(BaseItemDto Item, ItemCounts? ItemCounts)> GetItemValues(InternalItemsQuery filter, IReadOnlyList<ItemValueType> itemValueTypes, string returnType)
- {
- ArgumentNullException.ThrowIfNull(filter);
- if (!filter.Limit.HasValue)
- {
- filter.EnableTotalRecordCount = false;
- }
- using var context = _dbProvider.CreateDbContext();
- var innerQueryFilter = TranslateQuery(context.BaseItems, context, new InternalItemsQuery(filter.User)
- {
- ExcludeItemTypes = filter.ExcludeItemTypes,
- IncludeItemTypes = filter.IncludeItemTypes,
- MediaTypes = filter.MediaTypes,
- AncestorIds = filter.AncestorIds,
- ItemIds = filter.ItemIds,
- TopParentIds = filter.TopParentIds,
- ParentId = filter.ParentId,
- IsAiring = filter.IsAiring,
- IsMovie = filter.IsMovie,
- IsSports = filter.IsSports,
- IsKids = filter.IsKids,
- IsNews = filter.IsNews,
- IsSeries = filter.IsSeries
- });
- var innerQuery = PrepareItemQuery(context, filter)
- .Where(e => e.Type == returnType)
- .Where(e => context.ItemValues!
- .Where(f => itemValueTypes.Contains(f.Type))
- .Where(f => innerQueryFilter.Any(g => f.BaseItemsMap!.Any(w => w.ItemId == g.Id)))
- .Select(f => f.CleanValue)
- .Contains(e.CleanName));
- var outerQueryFilter = new InternalItemsQuery(filter.User)
- {
- IsPlayed = filter.IsPlayed,
- IsFavorite = filter.IsFavorite,
- IsFavoriteOrLiked = filter.IsFavoriteOrLiked,
- IsLiked = filter.IsLiked,
- IsLocked = filter.IsLocked,
- NameLessThan = filter.NameLessThan,
- NameStartsWith = filter.NameStartsWith,
- NameStartsWithOrGreater = filter.NameStartsWithOrGreater,
- Tags = filter.Tags,
- OfficialRatings = filter.OfficialRatings,
- StudioIds = filter.StudioIds,
- GenreIds = filter.GenreIds,
- Genres = filter.Genres,
- Years = filter.Years,
- NameContains = filter.NameContains,
- SearchTerm = filter.SearchTerm,
- ExcludeItemIds = filter.ExcludeItemIds
- };
- var query = TranslateQuery(innerQuery, context, outerQueryFilter)
- .GroupBy(e => e.PresentationUniqueKey);
- var result = new QueryResult<(BaseItemDto, ItemCounts?)>();
- if (filter.EnableTotalRecordCount)
- {
- result.TotalRecordCount = query.Count();
- }
- if (filter.Limit.HasValue || filter.StartIndex.HasValue)
- {
- var offset = filter.StartIndex ?? 0;
- if (offset > 0)
- {
- query = query.Skip(offset);
- }
- if (filter.Limit.HasValue)
- {
- query = query.Take(filter.Limit.Value);
- }
- }
- IQueryable<BaseItemEntity>? itemCountQuery = null;
- if (filter.IncludeItemTypes.Length > 0)
- {
- // if we are to include more then one type, sub query those items beforehand.
- var typeSubQuery = new InternalItemsQuery(filter.User)
- {
- ExcludeItemTypes = filter.ExcludeItemTypes,
- IncludeItemTypes = filter.IncludeItemTypes,
- MediaTypes = filter.MediaTypes,
- AncestorIds = filter.AncestorIds,
- ExcludeItemIds = filter.ExcludeItemIds,
- ItemIds = filter.ItemIds,
- TopParentIds = filter.TopParentIds,
- ParentId = filter.ParentId,
- IsPlayed = filter.IsPlayed
- };
- itemCountQuery = TranslateQuery(context.BaseItems.AsNoTracking(), context, typeSubQuery)
- .Where(e => e.ItemValues!.Any(f => itemValueTypes!.Contains(f.ItemValue.Type)));
- var seriesTypeName = _itemTypeLookup.BaseItemKindNames[BaseItemKind.Series];
- var movieTypeName = _itemTypeLookup.BaseItemKindNames[BaseItemKind.Movie];
- var episodeTypeName = _itemTypeLookup.BaseItemKindNames[BaseItemKind.Episode];
- var musicAlbumTypeName = _itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicAlbum];
- var musicArtistTypeName = _itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicArtist];
- var audioTypeName = _itemTypeLookup.BaseItemKindNames[BaseItemKind.Audio];
- var trailerTypeName = _itemTypeLookup.BaseItemKindNames[BaseItemKind.Trailer];
- var resultQuery = query.Select(e => new
- {
- item = e.AsQueryable()
- .Include(e => e.TrailerTypes)
- .Include(e => e.Provider)
- .Include(e => e.LockedFields)
- .Include(e => e.Images)
- .AsSingleQuery().First(),
- // TODO: This is bad refactor!
- itemCount = new ItemCounts()
- {
- SeriesCount = itemCountQuery!.Count(f => f.Type == seriesTypeName),
- EpisodeCount = itemCountQuery!.Count(f => f.Type == episodeTypeName),
- MovieCount = itemCountQuery!.Count(f => f.Type == movieTypeName),
- AlbumCount = itemCountQuery!.Count(f => f.Type == musicAlbumTypeName),
- ArtistCount = itemCountQuery!.Count(f => f.Type == musicArtistTypeName),
- SongCount = itemCountQuery!.Count(f => f.Type == audioTypeName),
- TrailerCount = itemCountQuery!.Count(f => f.Type == trailerTypeName),
- }
- });
- result.StartIndex = filter.StartIndex ?? 0;
- result.Items =
- [
- .. resultQuery
- .AsEnumerable()
- .Where(e => e is not null)
- .Select(e =>
- {
- return (DeserialiseBaseItem(e.item, filter.SkipDeserialization), e.itemCount);
- })
- ];
- }
- else
- {
- result.StartIndex = filter.StartIndex ?? 0;
- result.Items =
- [
- .. query
- .Select(e => e.First())
- .AsEnumerable()
- .Where(e => e is not null)
- .Select<BaseItemEntity, (BaseItemDto, ItemCounts?)>(e =>
- {
- return (DeserialiseBaseItem(e, filter.SkipDeserialization), null);
- })
- ];
- }
- return result;
- }
- private static void PrepareFilterQuery(InternalItemsQuery query)
- {
- if (query.Limit.HasValue && query.EnableGroupByMetadataKey)
- {
- query.Limit = query.Limit.Value + 4;
- }
- if (query.IsResumable ?? false)
- {
- query.IsVirtualItem = false;
- }
- }
- private string GetCleanValue(string value)
- {
- if (string.IsNullOrWhiteSpace(value))
- {
- return value;
- }
- return value.RemoveDiacritics().ToLowerInvariant();
- }
- private List<(ItemValueType MagicNumber, string Value)> GetItemValuesToSave(BaseItemDto item, List<string> inheritedTags)
- {
- var list = new List<(ItemValueType, string)>();
- if (item is IHasArtist hasArtist)
- {
- list.AddRange(hasArtist.Artists.Select(i => ((ItemValueType)0, i)));
- }
- if (item is IHasAlbumArtist hasAlbumArtist)
- {
- list.AddRange(hasAlbumArtist.AlbumArtists.Select(i => (ItemValueType.AlbumArtist, i)));
- }
- list.AddRange(item.Genres.Select(i => (ItemValueType.Genre, i)));
- list.AddRange(item.Studios.Select(i => (ItemValueType.Studios, i)));
- list.AddRange(item.Tags.Select(i => (ItemValueType.Tags, i)));
- // keywords was 5
- list.AddRange(inheritedTags.Select(i => (ItemValueType.InheritedTags, i)));
- // Remove all invalid values.
- list.RemoveAll(i => string.IsNullOrWhiteSpace(i.Item2));
- return list;
- }
- private static BaseItemImageInfo Map(Guid baseItemId, ItemImageInfo e)
- {
- return new BaseItemImageInfo()
- {
- ItemId = baseItemId,
- Id = Guid.NewGuid(),
- Path = e.Path,
- Blurhash = e.BlurHash is null ? null : Encoding.UTF8.GetBytes(e.BlurHash),
- DateModified = e.DateModified,
- Height = e.Height,
- Width = e.Width,
- ImageType = (ImageInfoImageType)e.Type,
- Item = null!
- };
- }
- private static ItemImageInfo Map(BaseItemImageInfo e, IServerApplicationHost? appHost)
- {
- return new ItemImageInfo()
- {
- Path = appHost?.ExpandVirtualPath(e.Path) ?? e.Path,
- BlurHash = e.Blurhash is null ? null : Encoding.UTF8.GetString(e.Blurhash),
- DateModified = e.DateModified,
- Height = e.Height,
- Width = e.Width,
- Type = (ImageType)e.ImageType
- };
- }
- private string? GetPathToSave(string path)
- {
- if (path is null)
- {
- return null;
- }
- return _appHost.ReverseVirtualPath(path);
- }
- private List<string> GetItemByNameTypesInQuery(InternalItemsQuery query)
- {
- var list = new List<string>();
- if (IsTypeInQuery(BaseItemKind.Person, query))
- {
- list.Add(_itemTypeLookup.BaseItemKindNames[BaseItemKind.Person]!);
- }
- if (IsTypeInQuery(BaseItemKind.Genre, query))
- {
- list.Add(_itemTypeLookup.BaseItemKindNames[BaseItemKind.Genre]!);
- }
- if (IsTypeInQuery(BaseItemKind.MusicGenre, query))
- {
- list.Add(_itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicGenre]!);
- }
- if (IsTypeInQuery(BaseItemKind.MusicArtist, query))
- {
- list.Add(_itemTypeLookup.BaseItemKindNames[BaseItemKind.MusicArtist]!);
- }
- if (IsTypeInQuery(BaseItemKind.Studio, query))
- {
- list.Add(_itemTypeLookup.BaseItemKindNames[BaseItemKind.Studio]!);
- }
- return list;
- }
- private bool IsTypeInQuery(BaseItemKind type, InternalItemsQuery query)
- {
- if (query.ExcludeItemTypes.Contains(type))
- {
- return false;
- }
- return query.IncludeItemTypes.Length == 0 || query.IncludeItemTypes.Contains(type);
- }
- private bool EnableGroupByPresentationUniqueKey(InternalItemsQuery query)
- {
- if (!query.GroupByPresentationUniqueKey)
- {
- return false;
- }
- if (query.GroupBySeriesPresentationUniqueKey)
- {
- return false;
- }
- if (!string.IsNullOrWhiteSpace(query.PresentationUniqueKey))
- {
- return false;
- }
- if (query.User is null)
- {
- return false;
- }
- if (query.IncludeItemTypes.Length == 0)
- {
- return true;
- }
- return query.IncludeItemTypes.Contains(BaseItemKind.Episode)
- || query.IncludeItemTypes.Contains(BaseItemKind.Video)
- || query.IncludeItemTypes.Contains(BaseItemKind.Movie)
- || query.IncludeItemTypes.Contains(BaseItemKind.MusicVideo)
- || query.IncludeItemTypes.Contains(BaseItemKind.Series)
- || query.IncludeItemTypes.Contains(BaseItemKind.Season);
- }
- private IQueryable<BaseItemEntity> ApplyOrder(IQueryable<BaseItemEntity> query, InternalItemsQuery filter)
- {
- var orderBy = filter.OrderBy;
- var hasSearch = !string.IsNullOrEmpty(filter.SearchTerm);
- if (hasSearch)
- {
- orderBy = filter.OrderBy = [(ItemSortBy.SortName, SortOrder.Ascending), .. orderBy];
- }
- else if (orderBy.Count == 0)
- {
- return query.OrderBy(e => e.SortName);
- }
- IOrderedQueryable<BaseItemEntity>? orderedQuery = null;
- var firstOrdering = orderBy.FirstOrDefault();
- if (firstOrdering != default)
- {
- var expression = OrderMapper.MapOrderByField(firstOrdering.OrderBy, filter);
- if (firstOrdering.SortOrder == SortOrder.Ascending)
- {
- orderedQuery = query.OrderBy(expression);
- }
- else
- {
- orderedQuery = query.OrderByDescending(expression);
- }
- if (firstOrdering.OrderBy is ItemSortBy.Default or ItemSortBy.SortName)
- {
- if (firstOrdering.SortOrder is SortOrder.Ascending)
- {
- orderedQuery = orderedQuery.ThenBy(e => e.Name);
- }
- else
- {
- orderedQuery = orderedQuery.ThenByDescending(e => e.Name);
- }
- }
- }
- foreach (var item in orderBy.Skip(1))
- {
- var expression = OrderMapper.MapOrderByField(item.OrderBy, filter);
- if (item.SortOrder == SortOrder.Ascending)
- {
- orderedQuery = orderedQuery!.ThenBy(expression);
- }
- else
- {
- orderedQuery = orderedQuery!.ThenByDescending(expression);
- }
- }
- return orderedQuery ?? query;
- }
- private IQueryable<BaseItemEntity> TranslateQuery(
- IQueryable<BaseItemEntity> baseQuery,
- JellyfinDbContext context,
- InternalItemsQuery filter)
- {
- const int HDWidth = 1200;
- const int UHDWidth = 3800;
- const int UHDHeight = 2100;
- var minWidth = filter.MinWidth;
- var maxWidth = filter.MaxWidth;
- var now = DateTime.UtcNow;
- if (filter.IsHD.HasValue || filter.Is4K.HasValue)
- {
- bool includeSD = false;
- bool includeHD = false;
- bool include4K = false;
- if (filter.IsHD.HasValue && !filter.IsHD.Value)
- {
- includeSD = true;
- }
- if (filter.IsHD.HasValue && filter.IsHD.Value)
- {
- includeHD = true;
- }
- if (filter.Is4K.HasValue && filter.Is4K.Value)
- {
- include4K = true;
- }
- baseQuery = baseQuery.Where(e =>
- (includeSD && e.Width < HDWidth) ||
- (includeHD && e.Width >= HDWidth && !(e.Width >= UHDWidth || e.Height >= UHDHeight)) ||
- (include4K && (e.Width >= UHDWidth || e.Height >= UHDHeight)));
- }
- if (minWidth.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.Width >= minWidth);
- }
- if (filter.MinHeight.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.Height >= filter.MinHeight);
- }
- if (maxWidth.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.Width >= maxWidth);
- }
- if (filter.MaxHeight.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.Height <= filter.MaxHeight);
- }
- if (filter.IsLocked.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.IsLocked == filter.IsLocked);
- }
- var tags = filter.Tags.ToList();
- var excludeTags = filter.ExcludeTags.ToList();
- if (filter.IsMovie == true)
- {
- if (filter.IncludeItemTypes.Length == 0
- || filter.IncludeItemTypes.Contains(BaseItemKind.Movie)
- || filter.IncludeItemTypes.Contains(BaseItemKind.Trailer))
- {
- baseQuery = baseQuery.Where(e => e.IsMovie);
- }
- }
- else if (filter.IsMovie.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.IsMovie == filter.IsMovie);
- }
- if (filter.IsSeries.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.IsSeries == filter.IsSeries);
- }
- if (filter.IsSports.HasValue)
- {
- if (filter.IsSports.Value)
- {
- tags.Add("Sports");
- }
- else
- {
- excludeTags.Add("Sports");
- }
- }
- if (filter.IsNews.HasValue)
- {
- if (filter.IsNews.Value)
- {
- tags.Add("News");
- }
- else
- {
- excludeTags.Add("News");
- }
- }
- if (filter.IsKids.HasValue)
- {
- if (filter.IsKids.Value)
- {
- tags.Add("Kids");
- }
- else
- {
- excludeTags.Add("Kids");
- }
- }
- if (!string.IsNullOrEmpty(filter.SearchTerm))
- {
- var searchTerm = filter.SearchTerm.ToLower();
- baseQuery = baseQuery.Where(e => e.CleanName!.ToLower().Contains(searchTerm) || (e.OriginalTitle != null && e.OriginalTitle.ToLower().Contains(searchTerm)));
- }
- if (filter.IsFolder.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.IsFolder == filter.IsFolder);
- }
- var includeTypes = filter.IncludeItemTypes;
- // Only specify excluded types if no included types are specified
- if (filter.IncludeItemTypes.Length == 0)
- {
- var excludeTypes = filter.ExcludeItemTypes;
- if (excludeTypes.Length == 1)
- {
- if (_itemTypeLookup.BaseItemKindNames.TryGetValue(excludeTypes[0], out var excludeTypeName))
- {
- baseQuery = baseQuery.Where(e => e.Type != excludeTypeName);
- }
- }
- else if (excludeTypes.Length > 1)
- {
- var excludeTypeName = new List<string>();
- foreach (var excludeType in excludeTypes)
- {
- if (_itemTypeLookup.BaseItemKindNames.TryGetValue(excludeType, out var baseItemKindName))
- {
- excludeTypeName.Add(baseItemKindName!);
- }
- }
- baseQuery = baseQuery.Where(e => !excludeTypeName.Contains(e.Type));
- }
- }
- else
- {
- string[] types = includeTypes.Select(f => _itemTypeLookup.BaseItemKindNames.GetValueOrDefault(f)).Where(e => e != null).ToArray()!;
- baseQuery = baseQuery.WhereOneOrMany(types, f => f.Type);
- }
- if (filter.ChannelIds.Count > 0)
- {
- baseQuery = baseQuery.Where(e => e.ChannelId != null && filter.ChannelIds.Contains(e.ChannelId.Value));
- }
- if (!filter.ParentId.IsEmpty())
- {
- baseQuery = baseQuery.Where(e => e.ParentId!.Value == filter.ParentId);
- }
- if (!string.IsNullOrWhiteSpace(filter.Path))
- {
- baseQuery = baseQuery.Where(e => e.Path == filter.Path);
- }
- if (!string.IsNullOrWhiteSpace(filter.PresentationUniqueKey))
- {
- baseQuery = baseQuery.Where(e => e.PresentationUniqueKey == filter.PresentationUniqueKey);
- }
- if (filter.MinCommunityRating.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.CommunityRating >= filter.MinCommunityRating);
- }
- if (filter.MinIndexNumber.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.IndexNumber >= filter.MinIndexNumber);
- }
- if (filter.MinParentAndIndexNumber.HasValue)
- {
- baseQuery = baseQuery
- .Where(e => (e.ParentIndexNumber == filter.MinParentAndIndexNumber.Value.ParentIndexNumber && e.IndexNumber >= filter.MinParentAndIndexNumber.Value.IndexNumber) || e.ParentIndexNumber > filter.MinParentAndIndexNumber.Value.ParentIndexNumber);
- }
- if (filter.MinDateCreated.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.DateCreated >= filter.MinDateCreated);
- }
- if (filter.MinDateLastSaved.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.DateLastSaved != null && e.DateLastSaved >= filter.MinDateLastSaved.Value);
- }
- if (filter.MinDateLastSavedForUser.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.DateLastSaved != null && e.DateLastSaved >= filter.MinDateLastSavedForUser.Value);
- }
- if (filter.IndexNumber.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.IndexNumber == filter.IndexNumber.Value);
- }
- if (filter.ParentIndexNumber.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.ParentIndexNumber == filter.ParentIndexNumber.Value);
- }
- if (filter.ParentIndexNumberNotEquals.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.ParentIndexNumber != filter.ParentIndexNumberNotEquals.Value || e.ParentIndexNumber == null);
- }
- var minEndDate = filter.MinEndDate;
- var maxEndDate = filter.MaxEndDate;
- if (filter.HasAired.HasValue)
- {
- if (filter.HasAired.Value)
- {
- maxEndDate = DateTime.UtcNow;
- }
- else
- {
- minEndDate = DateTime.UtcNow;
- }
- }
- if (minEndDate.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.EndDate >= minEndDate);
- }
- if (maxEndDate.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.EndDate <= maxEndDate);
- }
- if (filter.MinStartDate.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.StartDate >= filter.MinStartDate.Value);
- }
- if (filter.MaxStartDate.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.StartDate <= filter.MaxStartDate.Value);
- }
- if (filter.MinPremiereDate.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.PremiereDate <= filter.MinPremiereDate.Value);
- }
- if (filter.MaxPremiereDate.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.PremiereDate <= filter.MaxPremiereDate.Value);
- }
- if (filter.TrailerTypes.Length > 0)
- {
- var trailerTypes = filter.TrailerTypes.Select(e => (int)e).ToArray();
- baseQuery = baseQuery.Where(e => trailerTypes.Any(f => e.TrailerTypes!.Any(w => w.Id == f)));
- }
- if (filter.IsAiring.HasValue)
- {
- if (filter.IsAiring.Value)
- {
- baseQuery = baseQuery.Where(e => e.StartDate <= now && e.EndDate >= now);
- }
- else
- {
- baseQuery = baseQuery.Where(e => e.StartDate > now && e.EndDate < now);
- }
- }
- if (filter.PersonIds.Length > 0)
- {
- baseQuery = baseQuery
- .Where(e =>
- context.PeopleBaseItemMap.Where(w => context.BaseItems.Where(r => filter.PersonIds.Contains(r.Id)).Any(f => f.Name == w.People.Name))
- .Any(f => f.ItemId == e.Id));
- }
- if (!string.IsNullOrWhiteSpace(filter.Person))
- {
- baseQuery = baseQuery.Where(e => e.Peoples!.Any(f => f.People.Name == filter.Person));
- }
- if (!string.IsNullOrWhiteSpace(filter.MinSortName))
- {
- // this does not makes sense.
- // baseQuery = baseQuery.Where(e => e.SortName >= query.MinSortName);
- // whereClauses.Add("SortName>=@MinSortName");
- // statement?.TryBind("@MinSortName", query.MinSortName);
- }
- if (!string.IsNullOrWhiteSpace(filter.ExternalSeriesId))
- {
- baseQuery = baseQuery.Where(e => e.ExternalSeriesId == filter.ExternalSeriesId);
- }
- if (!string.IsNullOrWhiteSpace(filter.ExternalId))
- {
- baseQuery = baseQuery.Where(e => e.ExternalId == filter.ExternalId);
- }
- if (!string.IsNullOrWhiteSpace(filter.Name))
- {
- var cleanName = GetCleanValue(filter.Name);
- baseQuery = baseQuery.Where(e => e.CleanName == cleanName);
- }
- // These are the same, for now
- var nameContains = filter.NameContains;
- if (!string.IsNullOrWhiteSpace(nameContains))
- {
- baseQuery = baseQuery.Where(e =>
- e.CleanName!.Contains(nameContains)
- || e.OriginalTitle!.ToLower().Contains(nameContains!));
- }
- if (!string.IsNullOrWhiteSpace(filter.NameStartsWith))
- {
- baseQuery = baseQuery.Where(e => e.SortName!.StartsWith(filter.NameStartsWith) || e.Name!.StartsWith(filter.NameStartsWith));
- }
- if (!string.IsNullOrWhiteSpace(filter.NameStartsWithOrGreater))
- {
- // i hate this
- baseQuery = baseQuery.Where(e => e.SortName!.FirstOrDefault() > filter.NameStartsWithOrGreater[0] || e.Name!.FirstOrDefault() > filter.NameStartsWithOrGreater[0]);
- }
- if (!string.IsNullOrWhiteSpace(filter.NameLessThan))
- {
- // i hate this
- baseQuery = baseQuery.Where(e => e.SortName!.FirstOrDefault() < filter.NameLessThan[0] || e.Name!.FirstOrDefault() < filter.NameLessThan[0]);
- }
- if (filter.ImageTypes.Length > 0)
- {
- var imgTypes = filter.ImageTypes.Select(e => (ImageInfoImageType)e).ToArray();
- baseQuery = baseQuery.Where(e => imgTypes.Any(f => e.Images!.Any(w => w.ImageType == f)));
- }
- if (filter.IsLiked.HasValue)
- {
- baseQuery = baseQuery
- .Where(e => e.UserData!.FirstOrDefault(f => f.UserId == filter.User!.Id)!.Rating >= UserItemData.MinLikeValue);
- }
- if (filter.IsFavoriteOrLiked.HasValue)
- {
- baseQuery = baseQuery
- .Where(e => e.UserData!.FirstOrDefault(f => f.UserId == filter.User!.Id)!.IsFavorite == filter.IsFavoriteOrLiked);
- }
- if (filter.IsFavorite.HasValue)
- {
- baseQuery = baseQuery
- .Where(e => e.UserData!.FirstOrDefault(f => f.UserId == filter.User!.Id)!.IsFavorite == filter.IsFavorite);
- }
- if (filter.IsPlayed.HasValue)
- {
- // We should probably figure this out for all folders, but for right now, this is the only place where we need it
- if (filter.IncludeItemTypes.Length == 1 && filter.IncludeItemTypes[0] == BaseItemKind.Series)
- {
- baseQuery = baseQuery.Where(e => context.BaseItems
- .Where(e => e.IsFolder == false && e.IsVirtualItem == false)
- .Where(f => f.UserData!.FirstOrDefault(e => e.UserId == filter.User!.Id && e.Played)!.Played)
- .Any(f => f.SeriesPresentationUniqueKey == e.PresentationUniqueKey) == filter.IsPlayed);
- }
- else
- {
- baseQuery = baseQuery
- .Select(e => new
- {
- IsPlayed = e.UserData!.Where(f => f.UserId == filter.User!.Id).Select(f => (bool?)f.Played).FirstOrDefault() ?? false,
- Item = e
- })
- .Where(e => e.IsPlayed == filter.IsPlayed)
- .Select(f => f.Item);
- }
- }
- if (filter.IsResumable.HasValue)
- {
- if (filter.IsResumable.Value)
- {
- baseQuery = baseQuery
- .Where(e => e.UserData!.FirstOrDefault(f => f.UserId == filter.User!.Id)!.PlaybackPositionTicks > 0);
- }
- else
- {
- baseQuery = baseQuery
- .Where(e => e.UserData!.FirstOrDefault(f => f.UserId == filter.User!.Id)!.PlaybackPositionTicks == 0);
- }
- }
- if (filter.ArtistIds.Length > 0)
- {
- baseQuery = baseQuery.WhereReferencedItem(context, ItemValueType.Artist, filter.ArtistIds);
- }
- if (filter.AlbumArtistIds.Length > 0)
- {
- baseQuery = baseQuery.WhereReferencedItem(context, ItemValueType.Artist, filter.AlbumArtistIds);
- }
- if (filter.ContributingArtistIds.Length > 0)
- {
- baseQuery = baseQuery.WhereReferencedItem(context, ItemValueType.Artist, filter.ContributingArtistIds);
- }
- if (filter.AlbumIds.Length > 0)
- {
- var subQuery = context.BaseItems.WhereOneOrMany(filter.AlbumIds, f => f.Id);
- baseQuery = baseQuery.Where(e => subQuery.Any(f => f.Name == e.Album));
- }
- if (filter.ExcludeArtistIds.Length > 0)
- {
- baseQuery = baseQuery.WhereReferencedItem(context, ItemValueType.Artist, filter.ExcludeArtistIds, true);
- }
- if (filter.GenreIds.Count > 0)
- {
- baseQuery = baseQuery.WhereReferencedItem(context, ItemValueType.Genre, filter.GenreIds.ToArray());
- }
- if (filter.Genres.Count > 0)
- {
- var cleanGenres = filter.Genres.Select(e => GetCleanValue(e)).ToArray().OneOrManyExpressionBuilder<ItemValueMap, string>(f => f.ItemValue.CleanValue);
- baseQuery = baseQuery
- .Where(e => e.ItemValues!.AsQueryable().Where(f => f.ItemValue.Type == ItemValueType.Genre).Any(cleanGenres));
- }
- if (tags.Count > 0)
- {
- var cleanValues = tags.Select(e => GetCleanValue(e)).ToArray().OneOrManyExpressionBuilder<ItemValueMap, string>(f => f.ItemValue.CleanValue);
- baseQuery = baseQuery
- .Where(e => e.ItemValues!.AsQueryable().Where(f => f.ItemValue.Type == ItemValueType.Tags).Any(cleanValues));
- }
- if (excludeTags.Count > 0)
- {
- var cleanValues = excludeTags.Select(e => GetCleanValue(e)).ToArray().OneOrManyExpressionBuilder<ItemValueMap, string>(f => f.ItemValue.CleanValue);
- baseQuery = baseQuery
- .Where(e => !e.ItemValues!.AsQueryable().Where(f => f.ItemValue.Type == ItemValueType.Tags).Any(cleanValues));
- }
- if (filter.StudioIds.Length > 0)
- {
- baseQuery = baseQuery.WhereReferencedItem(context, ItemValueType.Studios, filter.StudioIds.ToArray());
- }
- if (filter.OfficialRatings.Length > 0)
- {
- baseQuery = baseQuery
- .Where(e => filter.OfficialRatings.Contains(e.OfficialRating));
- }
- Expression<Func<BaseItemEntity, bool>>? minParentalRatingFilter = null;
- if (filter.MinParentalRating != null)
- {
- var min = filter.MinParentalRating;
- minParentalRatingFilter = e => e.InheritedParentalRatingValue >= min.Score || e.InheritedParentalRatingValue == null;
- if (min.SubScore != null)
- {
- minParentalRatingFilter = minParentalRatingFilter.And(e => e.InheritedParentalRatingValue >= min.SubScore || e.InheritedParentalRatingValue == null);
- }
- }
- Expression<Func<BaseItemEntity, bool>>? maxParentalRatingFilter = null;
- if (filter.MaxParentalRating != null)
- {
- var max = filter.MaxParentalRating;
- maxParentalRatingFilter = e => e.InheritedParentalRatingValue <= max.Score || e.InheritedParentalRatingValue == null;
- if (max.SubScore != null)
- {
- maxParentalRatingFilter = maxParentalRatingFilter.And(e => e.InheritedParentalRatingValue <= max.SubScore || e.InheritedParentalRatingValue == null);
- }
- }
- if (filter.HasParentalRating ?? false)
- {
- if (minParentalRatingFilter != null)
- {
- baseQuery = baseQuery.Where(minParentalRatingFilter);
- }
- if (maxParentalRatingFilter != null)
- {
- baseQuery = baseQuery.Where(maxParentalRatingFilter);
- }
- }
- else if (filter.BlockUnratedItems.Length > 0)
- {
- var unratedItemTypes = filter.BlockUnratedItems.Select(f => f.ToString()).ToArray();
- Expression<Func<BaseItemEntity, bool>> unratedItemFilter = e => e.InheritedParentalRatingValue != null || !unratedItemTypes.Contains(e.UnratedType);
- if (minParentalRatingFilter != null && maxParentalRatingFilter != null)
- {
- baseQuery = baseQuery.Where(unratedItemFilter.And(minParentalRatingFilter.And(maxParentalRatingFilter)));
- }
- else if (minParentalRatingFilter != null)
- {
- baseQuery = baseQuery.Where(unratedItemFilter.And(minParentalRatingFilter));
- }
- else if (maxParentalRatingFilter != null)
- {
- baseQuery = baseQuery.Where(unratedItemFilter.And(maxParentalRatingFilter));
- }
- else
- {
- baseQuery = baseQuery.Where(unratedItemFilter);
- }
- }
- else if (minParentalRatingFilter != null || maxParentalRatingFilter != null)
- {
- if (minParentalRatingFilter != null)
- {
- baseQuery = baseQuery.Where(minParentalRatingFilter);
- }
- if (maxParentalRatingFilter != null)
- {
- baseQuery = baseQuery.Where(maxParentalRatingFilter);
- }
- }
- else if (!filter.HasParentalRating ?? false)
- {
- baseQuery = baseQuery
- .Where(e => e.InheritedParentalRatingValue == null);
- }
- if (filter.HasOfficialRating.HasValue)
- {
- if (filter.HasOfficialRating.Value)
- {
- baseQuery = baseQuery
- .Where(e => e.OfficialRating != null && e.OfficialRating != string.Empty);
- }
- else
- {
- baseQuery = baseQuery
- .Where(e => e.OfficialRating == null || e.OfficialRating == string.Empty);
- }
- }
- if (filter.HasOverview.HasValue)
- {
- if (filter.HasOverview.Value)
- {
- baseQuery = baseQuery
- .Where(e => e.Overview != null && e.Overview != string.Empty);
- }
- else
- {
- baseQuery = baseQuery
- .Where(e => e.Overview == null || e.Overview == string.Empty);
- }
- }
- if (filter.HasOwnerId.HasValue)
- {
- if (filter.HasOwnerId.Value)
- {
- baseQuery = baseQuery
- .Where(e => e.OwnerId != null);
- }
- else
- {
- baseQuery = baseQuery
- .Where(e => e.OwnerId == null);
- }
- }
- if (!string.IsNullOrWhiteSpace(filter.HasNoAudioTrackWithLanguage))
- {
- baseQuery = baseQuery
- .Where(e => !e.MediaStreams!.Any(f => f.StreamType == MediaStreamTypeEntity.Audio && f.Language == filter.HasNoAudioTrackWithLanguage));
- }
- if (!string.IsNullOrWhiteSpace(filter.HasNoInternalSubtitleTrackWithLanguage))
- {
- baseQuery = baseQuery
- .Where(e => !e.MediaStreams!.Any(f => f.StreamType == MediaStreamTypeEntity.Subtitle && !f.IsExternal && f.Language == filter.HasNoInternalSubtitleTrackWithLanguage));
- }
- if (!string.IsNullOrWhiteSpace(filter.HasNoExternalSubtitleTrackWithLanguage))
- {
- baseQuery = baseQuery
- .Where(e => !e.MediaStreams!.Any(f => f.StreamType == MediaStreamTypeEntity.Subtitle && f.IsExternal && f.Language == filter.HasNoExternalSubtitleTrackWithLanguage));
- }
- if (!string.IsNullOrWhiteSpace(filter.HasNoSubtitleTrackWithLanguage))
- {
- baseQuery = baseQuery
- .Where(e => !e.MediaStreams!.Any(f => f.StreamType == MediaStreamTypeEntity.Subtitle && f.Language == filter.HasNoSubtitleTrackWithLanguage));
- }
- if (filter.HasSubtitles.HasValue)
- {
- baseQuery = baseQuery
- .Where(e => e.MediaStreams!.Any(f => f.StreamType == MediaStreamTypeEntity.Subtitle) == filter.HasSubtitles.Value);
- }
- if (filter.HasChapterImages.HasValue)
- {
- baseQuery = baseQuery
- .Where(e => e.Chapters!.Any(f => f.ImagePath != null) == filter.HasChapterImages.Value);
- }
- if (filter.HasDeadParentId.HasValue && filter.HasDeadParentId.Value)
- {
- baseQuery = baseQuery
- .Where(e => e.ParentId.HasValue && !context.BaseItems.Any(f => f.Id == e.ParentId.Value));
- }
- if (filter.IsDeadArtist.HasValue && filter.IsDeadArtist.Value)
- {
- baseQuery = baseQuery
- .Where(e => !context.ItemValues.Where(f => _getAllArtistsValueTypes.Contains(f.Type)).Any(f => f.Value == e.Name));
- }
- if (filter.IsDeadStudio.HasValue && filter.IsDeadStudio.Value)
- {
- baseQuery = baseQuery
- .Where(e => !context.ItemValues.Where(f => _getStudiosValueTypes.Contains(f.Type)).Any(f => f.Value == e.Name));
- }
- if (filter.IsDeadGenre.HasValue && filter.IsDeadGenre.Value)
- {
- baseQuery = baseQuery
- .Where(e => !context.ItemValues.Where(f => _getGenreValueTypes.Contains(f.Type)).Any(f => f.Value == e.Name));
- }
- if (filter.IsDeadPerson.HasValue && filter.IsDeadPerson.Value)
- {
- baseQuery = baseQuery
- .Where(e => !context.Peoples.Any(f => f.Name == e.Name));
- }
- if (filter.Years.Length > 0)
- {
- baseQuery = baseQuery.WhereOneOrMany(filter.Years, e => e.ProductionYear!.Value);
- }
- var isVirtualItem = filter.IsVirtualItem ?? filter.IsMissing;
- if (isVirtualItem.HasValue)
- {
- baseQuery = baseQuery
- .Where(e => e.IsVirtualItem == isVirtualItem.Value);
- }
- if (filter.IsSpecialSeason.HasValue)
- {
- if (filter.IsSpecialSeason.Value)
- {
- baseQuery = baseQuery
- .Where(e => e.IndexNumber == 0);
- }
- else
- {
- baseQuery = baseQuery
- .Where(e => e.IndexNumber != 0);
- }
- }
- if (filter.IsUnaired.HasValue)
- {
- if (filter.IsUnaired.Value)
- {
- baseQuery = baseQuery
- .Where(e => e.PremiereDate >= now);
- }
- else
- {
- baseQuery = baseQuery
- .Where(e => e.PremiereDate < now);
- }
- }
- if (filter.MediaTypes.Length > 0)
- {
- var mediaTypes = filter.MediaTypes.Select(f => f.ToString()).ToArray();
- baseQuery = baseQuery.WhereOneOrMany(mediaTypes, e => e.MediaType);
- }
- if (filter.ItemIds.Length > 0)
- {
- baseQuery = baseQuery.WhereOneOrMany(filter.ItemIds, e => e.Id);
- }
- if (filter.ExcludeItemIds.Length > 0)
- {
- baseQuery = baseQuery
- .Where(e => !filter.ItemIds.Contains(e.Id));
- }
- if (filter.ExcludeProviderIds is not null && filter.ExcludeProviderIds.Count > 0)
- {
- baseQuery = baseQuery.Where(e => !e.Provider!.All(f => !filter.ExcludeProviderIds.All(w => f.ProviderId == w.Key && f.ProviderValue == w.Value)));
- }
- if (filter.HasAnyProviderId is not null && filter.HasAnyProviderId.Count > 0)
- {
- baseQuery = baseQuery.Where(e => e.Provider!.Any(f => !filter.HasAnyProviderId.Any(w => f.ProviderId == w.Key && f.ProviderValue == w.Value)));
- }
- if (filter.HasImdbId.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.Provider!.Any(f => f.ProviderId == "imdb"));
- }
- if (filter.HasTmdbId.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.Provider!.Any(f => f.ProviderId == "tmdb"));
- }
- if (filter.HasTvdbId.HasValue)
- {
- baseQuery = baseQuery.Where(e => e.Provider!.Any(f => f.ProviderId == "tvdb"));
- }
- var queryTopParentIds = filter.TopParentIds;
- if (queryTopParentIds.Length > 0)
- {
- var includedItemByNameTypes = GetItemByNameTypesInQuery(filter);
- var enableItemsByName = (filter.IncludeItemsByName ?? false) && includedItemByNameTypes.Count > 0;
- if (enableItemsByName && includedItemByNameTypes.Count > 0)
- {
- baseQuery = baseQuery.Where(e => includedItemByNameTypes.Contains(e.Type) || queryTopParentIds.Any(w => w == e.TopParentId!.Value));
- }
- else
- {
- baseQuery = baseQuery.WhereOneOrMany(queryTopParentIds, e => e.TopParentId!.Value);
- }
- }
- if (filter.AncestorIds.Length > 0)
- {
- baseQuery = baseQuery.Where(e => e.Parents!.Any(f => filter.AncestorIds.Contains(f.ParentItemId)));
- }
- if (!string.IsNullOrWhiteSpace(filter.AncestorWithPresentationUniqueKey))
- {
- baseQuery = baseQuery
- .Where(e => context.BaseItems.Where(f => f.PresentationUniqueKey == filter.AncestorWithPresentationUniqueKey).Any(f => f.Children!.Any(w => w.ItemId == e.Id)));
- }
- if (!string.IsNullOrWhiteSpace(filter.SeriesPresentationUniqueKey))
- {
- baseQuery = baseQuery
- .Where(e => e.SeriesPresentationUniqueKey == filter.SeriesPresentationUniqueKey);
- }
- if (filter.ExcludeInheritedTags.Length > 0)
- {
- baseQuery = baseQuery
- .Where(e => !e.ItemValues!.Where(w => w.ItemValue.Type == ItemValueType.InheritedTags)
- .Any(f => filter.ExcludeInheritedTags.Contains(f.ItemValue.CleanValue)));
- }
- if (filter.IncludeInheritedTags.Length > 0)
- {
- // Episodes do not store inherit tags from their parents in the database, and the tag may be still required by the client.
- // In addition to the tags for the episodes themselves, we need to manually query its parent (the season)'s tags as well.
- if (includeTypes.Length == 1 && includeTypes.FirstOrDefault() is BaseItemKind.Episode)
- {
- baseQuery = baseQuery
- .Where(e => e.ItemValues!.Where(f => f.ItemValue.Type == ItemValueType.InheritedTags)
- .Any(f => filter.IncludeInheritedTags.Contains(f.ItemValue.CleanValue))
- ||
- (e.ParentId.HasValue && context.ItemValuesMap.Where(w => w.ItemId == e.ParentId.Value)!.Where(w => w.ItemValue.Type == ItemValueType.InheritedTags)
- .Any(f => filter.IncludeInheritedTags.Contains(f.ItemValue.CleanValue))));
- }
- // A playlist should be accessible to its owner regardless of allowed tags.
- else if (includeTypes.Length == 1 && includeTypes.FirstOrDefault() is BaseItemKind.Playlist)
- {
- baseQuery = baseQuery
- .Where(e =>
- e.Parents!
- .Any(f =>
- f.ParentItem.ItemValues!.Any(w => w.ItemValue.Type == ItemValueType.Tags && filter.IncludeInheritedTags.Contains(w.ItemValue.CleanValue))
- || e.Data!.Contains($"OwnerUserId\":\"{filter.User!.Id:N}\"")));
- // d ^^ this is stupid it hate this.
- }
- else
- {
- baseQuery = baseQuery
- .Where(e => e.Parents!.Any(f => f.ParentItem.ItemValues!.Any(w => w.ItemValue.Type == ItemValueType.Tags && filter.IncludeInheritedTags.Contains(w.ItemValue.CleanValue))));
- }
- }
- if (filter.SeriesStatuses.Length > 0)
- {
- var seriesStatus = filter.SeriesStatuses.Select(e => e.ToString()).ToArray();
- baseQuery = baseQuery
- .Where(e => seriesStatus.Any(f => e.Data!.Contains(f)));
- }
- if (filter.BoxSetLibraryFolders.Length > 0)
- {
- var boxsetFolders = filter.BoxSetLibraryFolders.Select(e => e.ToString("N", CultureInfo.InvariantCulture)).ToArray();
- baseQuery = baseQuery
- .Where(e => boxsetFolders.Any(f => e.Data!.Contains(f)));
- }
- if (filter.VideoTypes.Length > 0)
- {
- var videoTypeBs = filter.VideoTypes.Select(e => $"\"VideoType\":\"{e}\"");
- baseQuery = baseQuery
- .Where(e => videoTypeBs.Any(f => e.Data!.Contains(f)));
- }
- if (filter.Is3D.HasValue)
- {
- if (filter.Is3D.Value)
- {
- baseQuery = baseQuery
- .Where(e => e.Data!.Contains("Video3DFormat"));
- }
- else
- {
- baseQuery = baseQuery
- .Where(e => !e.Data!.Contains("Video3DFormat"));
- }
- }
- if (filter.IsPlaceHolder.HasValue)
- {
- if (filter.IsPlaceHolder.Value)
- {
- baseQuery = baseQuery
- .Where(e => e.Data!.Contains("IsPlaceHolder\":true"));
- }
- else
- {
- baseQuery = baseQuery
- .Where(e => !e.Data!.Contains("IsPlaceHolder\":true"));
- }
- }
- if (filter.HasSpecialFeature.HasValue)
- {
- if (filter.HasSpecialFeature.Value)
- {
- baseQuery = baseQuery
- .Where(e => e.ExtraIds != null);
- }
- else
- {
- baseQuery = baseQuery
- .Where(e => e.ExtraIds == null);
- }
- }
- if (filter.HasTrailer.HasValue || filter.HasThemeSong.HasValue || filter.HasThemeVideo.HasValue)
- {
- if (filter.HasTrailer.GetValueOrDefault() || filter.HasThemeSong.GetValueOrDefault() || filter.HasThemeVideo.GetValueOrDefault())
- {
- baseQuery = baseQuery
- .Where(e => e.ExtraIds != null);
- }
- else
- {
- baseQuery = baseQuery
- .Where(e => e.ExtraIds == null);
- }
- }
- return baseQuery;
- }
- }
|