mirror of
https://github.com/jellyfin/jellyfin.git
synced 2026-06-18 13:40:45 +01:00
Split BaseItemRepository and IItemRepository
This commit is contained in:
Shadowghost
345
Jellyfin.Server.Implementations/Item/NextUpService.cs
Normal file
345
Jellyfin.Server.Implementations/Item/NextUpService.cs
Normal file
@@ -0,0 +1,345 @@
|
||||
#pragma warning disable RS0030 // Do not use banned APIs
|
||||
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using Jellyfin.Data.Enums;
|
||||
using Jellyfin.Database.Implementations;
|
||||
using Jellyfin.Database.Implementations.Entities;
|
||||
using MediaBrowser.Controller.Entities;
|
||||
using MediaBrowser.Controller.Persistence;
|
||||
using Microsoft.EntityFrameworkCore;
|
||||
using BaseItemDto = MediaBrowser.Controller.Entities.BaseItem;
|
||||
|
||||
namespace Jellyfin.Server.Implementations.Item;
|
||||
|
||||
/// <summary>
|
||||
/// Provides next-up episode query operations.
|
||||
/// </summary>
|
||||
public class NextUpService : INextUpService
|
||||
{
|
||||
private readonly IDbContextFactory<JellyfinDbContext> _dbProvider;
|
||||
private readonly IItemTypeLookup _itemTypeLookup;
|
||||
private readonly IItemQueryHelpers _queryHelpers;
|
||||
|
||||
/// <summary>
|
||||
/// Initializes a new instance of the <see cref="NextUpService"/> class.
|
||||
/// </summary>
|
||||
/// <param name="dbProvider">The database context factory.</param>
|
||||
/// <param name="itemTypeLookup">The item type lookup.</param>
|
||||
/// <param name="queryHelpers">The shared query helpers.</param>
|
||||
public NextUpService(
|
||||
IDbContextFactory<JellyfinDbContext> dbProvider,
|
||||
IItemTypeLookup itemTypeLookup,
|
||||
IItemQueryHelpers queryHelpers)
|
||||
{
|
||||
_dbProvider = dbProvider;
|
||||
_itemTypeLookup = itemTypeLookup;
|
||||
_queryHelpers = queryHelpers;
|
||||
}
|
||||
|
||||
/// <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().Where(e => e.ItemId != EF.Constant(BaseItemRepository.PlaceholderId)),
|
||||
i => new { UserId = filter.User.Id, ItemId = i.Id },
|
||||
u => new { u.UserId, 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();
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
public IReadOnlyDictionary<string, NextUpEpisodeBatchResult> GetNextUpEpisodesBatch(
|
||||
InternalItemsQuery filter,
|
||||
IReadOnlyList<string> seriesKeys,
|
||||
bool includeSpecials,
|
||||
bool includeWatchedForRewatching)
|
||||
{
|
||||
ArgumentNullException.ThrowIfNull(filter);
|
||||
ArgumentNullException.ThrowIfNull(filter.User);
|
||||
|
||||
if (seriesKeys.Count == 0)
|
||||
{
|
||||
return new Dictionary<string, NextUpEpisodeBatchResult>();
|
||||
}
|
||||
|
||||
_queryHelpers.PrepareFilterQuery(filter);
|
||||
using var context = _dbProvider.CreateDbContext();
|
||||
|
||||
var userId = filter.User.Id;
|
||||
var episodeTypeName = _itemTypeLookup.BaseItemKindNames[BaseItemKind.Episode];
|
||||
|
||||
var lastWatchedBase = context.BaseItems
|
||||
.AsNoTracking()
|
||||
.Where(e => e.Type == episodeTypeName)
|
||||
.Where(e => e.SeriesPresentationUniqueKey != null && seriesKeys.Contains(e.SeriesPresentationUniqueKey))
|
||||
.Where(e => e.ParentIndexNumber != 0)
|
||||
.Where(e => e.UserData!.Any(ud => ud.UserId == userId && ud.Played));
|
||||
lastWatchedBase = _queryHelpers.ApplyAccessFiltering(context, lastWatchedBase, filter);
|
||||
var lastWatchedInfo = lastWatchedBase
|
||||
.GroupBy(e => e.SeriesPresentationUniqueKey)
|
||||
.Select(g => new
|
||||
{
|
||||
SeriesKey = g.Key!,
|
||||
LastWatchedId = g.OrderByDescending(e => e.ParentIndexNumber)
|
||||
.ThenByDescending(e => e.IndexNumber)
|
||||
.Select(e => e.Id)
|
||||
.FirstOrDefault()
|
||||
})
|
||||
.ToDictionary(x => x.SeriesKey, x => x.LastWatchedId);
|
||||
|
||||
Dictionary<string, Guid> lastWatchedByDateInfo = new();
|
||||
if (includeWatchedForRewatching)
|
||||
{
|
||||
var lastWatchedByDateBase = context.BaseItems
|
||||
.AsNoTracking()
|
||||
.Where(e => e.Type == episodeTypeName)
|
||||
.Where(e => e.SeriesPresentationUniqueKey != null && seriesKeys.Contains(e.SeriesPresentationUniqueKey))
|
||||
.Where(e => e.ParentIndexNumber != 0)
|
||||
.Where(e => e.UserData!.Any(ud => ud.UserId == userId && ud.Played));
|
||||
lastWatchedByDateBase = _queryHelpers.ApplyAccessFiltering(context, lastWatchedByDateBase, filter);
|
||||
lastWatchedByDateInfo = lastWatchedByDateBase
|
||||
.SelectMany(e => e.UserData!.Where(ud => ud.UserId == userId && ud.Played)
|
||||
.Select(ud => new { Episode = e, ud.LastPlayedDate }))
|
||||
.GroupBy(x => x.Episode.SeriesPresentationUniqueKey)
|
||||
.Select(g => new
|
||||
{
|
||||
SeriesKey = g.Key!,
|
||||
LastWatchedId = g.OrderByDescending(x => x.LastPlayedDate)
|
||||
.Select(x => x.Episode.Id)
|
||||
.FirstOrDefault()
|
||||
})
|
||||
.ToDictionary(x => x.SeriesKey, x => x.LastWatchedId);
|
||||
}
|
||||
|
||||
var allLastWatchedIds = lastWatchedInfo.Values
|
||||
.Concat(lastWatchedByDateInfo.Values)
|
||||
.Where(id => id != Guid.Empty)
|
||||
.Distinct()
|
||||
.ToList();
|
||||
var lastWatchedEpisodes = new Dictionary<Guid, BaseItemEntity>();
|
||||
if (allLastWatchedIds.Count > 0)
|
||||
{
|
||||
var lwQuery = context.BaseItems.AsNoTracking().Where(e => allLastWatchedIds.Contains(e.Id));
|
||||
lwQuery = _queryHelpers.ApplyNavigations(lwQuery, filter);
|
||||
lastWatchedEpisodes = lwQuery.ToDictionary(e => e.Id);
|
||||
}
|
||||
|
||||
Dictionary<string, List<BaseItemEntity>> specialsBySeriesKey = new();
|
||||
if (includeSpecials)
|
||||
{
|
||||
var specialsQuery = context.BaseItems
|
||||
.AsNoTracking()
|
||||
.Where(e => e.Type == episodeTypeName)
|
||||
.Where(e => e.SeriesPresentationUniqueKey != null && seriesKeys.Contains(e.SeriesPresentationUniqueKey))
|
||||
.Where(e => e.ParentIndexNumber == 0)
|
||||
.Where(e => !e.IsVirtualItem);
|
||||
specialsQuery = _queryHelpers.ApplyAccessFiltering(context, specialsQuery, filter);
|
||||
specialsQuery = _queryHelpers.ApplyNavigations(specialsQuery, filter).AsSingleQuery();
|
||||
|
||||
foreach (var special in specialsQuery)
|
||||
{
|
||||
var key = special.SeriesPresentationUniqueKey!;
|
||||
if (!specialsBySeriesKey.TryGetValue(key, out var list))
|
||||
{
|
||||
list = new List<BaseItemEntity>();
|
||||
specialsBySeriesKey[key] = list;
|
||||
}
|
||||
|
||||
list.Add(special);
|
||||
}
|
||||
}
|
||||
|
||||
var positionLookup = new Dictionary<string, (int Season, int Episode)>();
|
||||
foreach (var kvp in lastWatchedInfo)
|
||||
{
|
||||
if (kvp.Value != Guid.Empty
|
||||
&& lastWatchedEpisodes.TryGetValue(kvp.Value, out var lw)
|
||||
&& lw.ParentIndexNumber.HasValue
|
||||
&& lw.IndexNumber.HasValue)
|
||||
{
|
||||
positionLookup[kvp.Key] = (lw.ParentIndexNumber.Value, lw.IndexNumber.Value);
|
||||
}
|
||||
}
|
||||
|
||||
var allUnplayedBase = context.BaseItems
|
||||
.AsNoTracking()
|
||||
.Where(e => e.Type == episodeTypeName)
|
||||
.Where(e => e.SeriesPresentationUniqueKey != null && seriesKeys.Contains(e.SeriesPresentationUniqueKey))
|
||||
.Where(e => e.ParentIndexNumber != 0)
|
||||
.Where(e => !e.IsVirtualItem)
|
||||
.Where(e => !e.UserData!.Any(ud => ud.UserId == userId && ud.Played));
|
||||
allUnplayedBase = _queryHelpers.ApplyAccessFiltering(context, allUnplayedBase, filter);
|
||||
var allUnplayedCandidates = allUnplayedBase
|
||||
.Select(e => new
|
||||
{
|
||||
e.Id,
|
||||
e.SeriesPresentationUniqueKey,
|
||||
e.ParentIndexNumber,
|
||||
EpisodeNumber = e.IndexNumber
|
||||
})
|
||||
.ToList();
|
||||
|
||||
var nextEpisodeIds = new HashSet<Guid>();
|
||||
var seriesNextIdMap = new Dictionary<string, Guid>();
|
||||
|
||||
foreach (var seriesKey in seriesKeys)
|
||||
{
|
||||
var candidates = allUnplayedCandidates
|
||||
.Where(c => c.SeriesPresentationUniqueKey == seriesKey);
|
||||
|
||||
if (positionLookup.TryGetValue(seriesKey, out var pos))
|
||||
{
|
||||
candidates = candidates.Where(c =>
|
||||
c.ParentIndexNumber > pos.Season
|
||||
|| (c.ParentIndexNumber == pos.Season && c.EpisodeNumber > pos.Episode));
|
||||
}
|
||||
|
||||
var nextCandidate = candidates
|
||||
.OrderBy(c => c.ParentIndexNumber)
|
||||
.ThenBy(c => c.EpisodeNumber)
|
||||
.FirstOrDefault();
|
||||
|
||||
if (nextCandidate is not null && nextCandidate.Id != Guid.Empty)
|
||||
{
|
||||
nextEpisodeIds.Add(nextCandidate.Id);
|
||||
seriesNextIdMap[seriesKey] = nextCandidate.Id;
|
||||
}
|
||||
}
|
||||
|
||||
var seriesNextPlayedIdMap = new Dictionary<string, Guid>();
|
||||
if (includeWatchedForRewatching)
|
||||
{
|
||||
var allPlayedBase = context.BaseItems
|
||||
.AsNoTracking()
|
||||
.Where(e => e.Type == episodeTypeName)
|
||||
.Where(e => e.SeriesPresentationUniqueKey != null && seriesKeys.Contains(e.SeriesPresentationUniqueKey))
|
||||
.Where(e => e.ParentIndexNumber != 0)
|
||||
.Where(e => !e.IsVirtualItem)
|
||||
.Where(e => e.UserData!.Any(ud => ud.UserId == userId && ud.Played));
|
||||
allPlayedBase = _queryHelpers.ApplyAccessFiltering(context, allPlayedBase, filter);
|
||||
var allPlayedCandidates = allPlayedBase
|
||||
.Select(e => new
|
||||
{
|
||||
e.Id,
|
||||
e.SeriesPresentationUniqueKey,
|
||||
e.ParentIndexNumber,
|
||||
EpisodeNumber = e.IndexNumber
|
||||
})
|
||||
.ToList();
|
||||
|
||||
foreach (var seriesKey in seriesKeys)
|
||||
{
|
||||
if (!lastWatchedByDateInfo.TryGetValue(seriesKey, out var lastByDateId))
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
var lastByDateEntity = lastWatchedEpisodes.GetValueOrDefault(lastByDateId);
|
||||
if (lastByDateEntity is null)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
var playedCandidates = allPlayedCandidates
|
||||
.Where(c => c.SeriesPresentationUniqueKey == seriesKey);
|
||||
|
||||
if (lastByDateEntity.ParentIndexNumber.HasValue && lastByDateEntity.IndexNumber.HasValue)
|
||||
{
|
||||
var lastSeason = lastByDateEntity.ParentIndexNumber.Value;
|
||||
var lastEp = lastByDateEntity.IndexNumber.Value;
|
||||
playedCandidates = playedCandidates.Where(c =>
|
||||
c.ParentIndexNumber > lastSeason
|
||||
|| (c.ParentIndexNumber == lastSeason && c.EpisodeNumber > lastEp));
|
||||
}
|
||||
|
||||
var nextPlayedCandidate = playedCandidates
|
||||
.OrderBy(c => c.ParentIndexNumber)
|
||||
.ThenBy(c => c.EpisodeNumber)
|
||||
.FirstOrDefault();
|
||||
|
||||
if (nextPlayedCandidate is not null && nextPlayedCandidate.Id != Guid.Empty)
|
||||
{
|
||||
nextEpisodeIds.Add(nextPlayedCandidate.Id);
|
||||
seriesNextPlayedIdMap[seriesKey] = nextPlayedCandidate.Id;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
var nextEpisodes = new Dictionary<Guid, BaseItemEntity>();
|
||||
if (nextEpisodeIds.Count > 0)
|
||||
{
|
||||
var nextQuery = context.BaseItems.AsNoTracking().Where(e => nextEpisodeIds.Contains(e.Id));
|
||||
nextQuery = _queryHelpers.ApplyNavigations(nextQuery, filter).AsSingleQuery();
|
||||
nextEpisodes = nextQuery.ToDictionary(e => e.Id);
|
||||
}
|
||||
|
||||
var result = new Dictionary<string, NextUpEpisodeBatchResult>();
|
||||
foreach (var seriesKey in seriesKeys)
|
||||
{
|
||||
var batchResult = new NextUpEpisodeBatchResult();
|
||||
|
||||
if (lastWatchedInfo.TryGetValue(seriesKey, out var lwId) && lwId != Guid.Empty)
|
||||
{
|
||||
if (lastWatchedEpisodes.TryGetValue(lwId, out var entity))
|
||||
{
|
||||
batchResult.LastWatched = _queryHelpers.DeserializeBaseItem(entity, filter.SkipDeserialization);
|
||||
}
|
||||
}
|
||||
|
||||
if (seriesNextIdMap.TryGetValue(seriesKey, out var nextId) && nextEpisodes.TryGetValue(nextId, out var nextEntity))
|
||||
{
|
||||
batchResult.NextUp = _queryHelpers.DeserializeBaseItem(nextEntity, filter.SkipDeserialization);
|
||||
}
|
||||
|
||||
if (includeSpecials && specialsBySeriesKey.TryGetValue(seriesKey, out var specials))
|
||||
{
|
||||
batchResult.Specials = specials.Select(s => _queryHelpers.DeserializeBaseItem(s, filter.SkipDeserialization)!).ToList();
|
||||
}
|
||||
else
|
||||
{
|
||||
batchResult.Specials = Array.Empty<BaseItemDto>();
|
||||
}
|
||||
|
||||
if (includeWatchedForRewatching)
|
||||
{
|
||||
if (lastWatchedByDateInfo.TryGetValue(seriesKey, out var lastByDateId) &&
|
||||
lastWatchedEpisodes.TryGetValue(lastByDateId, out var lastByDateEntity))
|
||||
{
|
||||
batchResult.LastWatchedForRewatching = _queryHelpers.DeserializeBaseItem(lastByDateEntity, filter.SkipDeserialization);
|
||||
}
|
||||
|
||||
if (seriesNextPlayedIdMap.TryGetValue(seriesKey, out var nextPlayedId) &&
|
||||
nextEpisodes.TryGetValue(nextPlayedId, out var nextPlayedEntity))
|
||||
{
|
||||
batchResult.NextPlayedForRewatching = _queryHelpers.DeserializeBaseItem(nextPlayedEntity, filter.SkipDeserialization);
|
||||
}
|
||||
}
|
||||
|
||||
result[seriesKey] = batchResult;
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user