using System; using System.Collections.Generic; using System.Globalization; using System.Linq; using System.Text.RegularExpressions; using System.Threading.Tasks; using API.Data.Scanner; using API.DTOs; using API.DTOs.CollectionTags; using API.DTOs.Filtering; using API.DTOs.Metadata; using API.DTOs.ReadingLists; using API.DTOs.Search; using API.DTOs.SeriesDetail; using API.Entities; using API.Entities.Enums; using API.Entities.Metadata; using API.Extensions; using API.Helpers; using API.Services.Tasks; using AutoMapper; using AutoMapper.QueryableExtensions; using Kavita.Common.Extensions; using Microsoft.AspNetCore.Mvc; using Microsoft.EntityFrameworkCore; using SQLitePCL; namespace API.Data.Repositories; [Flags] public enum SeriesIncludes { None = 1, Volumes = 2, Metadata = 4, Related = 8, //Related = 16, //UserPreferences = 32 } internal class RecentlyAddedSeries { public int LibraryId { get; init; } public LibraryType LibraryType { get; init; } public DateTime Created { get; init; } public int SeriesId { get; init; } public string SeriesName { get; init; } public MangaFormat Format { get; init; } public int ChapterId { get; init; } public int VolumeId { get; init; } public string ChapterNumber { get; init; } public string ChapterRange { get; init; } public string ChapterTitle { get; init; } public bool IsSpecial { get; init; } public int VolumeNumber { get; init; } } public interface ISeriesRepository { void Attach(Series series); void Update(Series series); void Remove(Series series); void Remove(IEnumerable series); Task DoesSeriesNameExistInLibrary(string name, int libraryId, MangaFormat format); /// /// Adds user information like progress, ratings, etc /// /// /// /// /// Task> GetSeriesDtoForLibraryIdAsync(int libraryId, int userId, UserParams userParams, FilterDto filter); /// /// Does not add user information like progress, ratings, etc. /// /// /// /// /// /// Task SearchSeries(int userId, bool isAdmin, int[] libraryIds, string searchQuery); Task> GetSeriesForLibraryIdAsync(int libraryId); Task GetSeriesDtoByIdAsync(int seriesId, int userId); Task DeleteSeriesAsync(int seriesId); Task GetSeriesByIdAsync(int seriesId, SeriesIncludes includes = SeriesIncludes.Volumes | SeriesIncludes.Metadata); Task> GetSeriesByIdsAsync(IList seriesIds); Task GetChapterIdsForSeriesAsync(IList seriesIds); Task>> GetChapterIdWithSeriesIdForSeriesAsync(int[] seriesIds); /// /// Used to add Progress/Rating information to series list. /// /// /// /// Task AddSeriesModifiers(int userId, List series); Task GetSeriesCoverImageAsync(int seriesId); Task> GetOnDeck(int userId, int libraryId, UserParams userParams, FilterDto filter); Task> GetRecentlyAdded(int libraryId, int userId, UserParams userParams, FilterDto filter); Task GetSeriesMetadata(int seriesId); Task> GetSeriesDtoForCollectionAsync(int collectionId, int userId, UserParams userParams); Task> GetFilesForSeries(int seriesId); Task> GetSeriesDtoForIdsAsync(IEnumerable seriesIds, int userId); Task> GetAllCoverImagesAsync(); Task> GetLockedCoverImagesAsync(); Task> GetFullSeriesForLibraryIdAsync(int libraryId, UserParams userParams); Task GetFullSeriesForSeriesIdAsync(int seriesId); Task GetChunkInfo(int libraryId = 0); Task> GetSeriesMetadataForIdsAsync(IEnumerable seriesIds); Task> GetAllAgeRatingsDtosForLibrariesAsync(List libraryIds); // TODO: Move to LibraryRepository Task> GetAllLanguagesForLibrariesAsync(List libraryIds); // TODO: Move to LibraryRepository IEnumerable GetAllPublicationStatusesDtosForLibrariesAsync(List libraryIds); // TODO: Move to LibraryRepository Task> GetRecentlyUpdatedSeries(int userId, int pageSize = 30); Task GetRelatedSeries(int userId, int seriesId); Task> GetSeriesForRelationKind(int userId, int seriesId, RelationKind kind); Task> GetQuickReads(int userId, int libraryId, UserParams userParams); Task> GetHighlyRated(int userId, int libraryId, UserParams userParams); Task> GetMoreIn(int userId, int libraryId, int genreId, UserParams userParams); Task> GetRediscover(int userId, int libraryId, UserParams userParams); Task GetSeriesForMangaFile(int mangaFileId, int userId); Task GetSeriesForChapter(int chapterId, int userId); } public class SeriesRepository : ISeriesRepository { private readonly DataContext _context; private readonly IMapper _mapper; public SeriesRepository(DataContext context, IMapper mapper) { _context = context; _mapper = mapper; } public void Attach(Series series) { _context.Series.Attach(series); } public void Update(Series series) { _context.Entry(series).State = EntityState.Modified; } public void Remove(Series series) { _context.Series.Remove(series); } public void Remove(IEnumerable series) { _context.Series.RemoveRange(series); } /// /// Returns if a series name and format exists already in a library /// /// Name of series /// Format of series /// public async Task DoesSeriesNameExistInLibrary(string name, int libraryId, MangaFormat format) { return await _context.Series .AsNoTracking() .Where(s => s.LibraryId == libraryId && s.Name.Equals(name) && s.Format == format) .AnyAsync(); } public async Task> GetSeriesForLibraryIdAsync(int libraryId) { return await _context.Series .Where(s => s.LibraryId == libraryId) .OrderBy(s => s.SortName) .ToListAsync(); } /// /// Used for to /// /// /// public async Task> GetFullSeriesForLibraryIdAsync(int libraryId, UserParams userParams) { var query = _context.Series .Where(s => s.LibraryId == libraryId) .Include(s => s.Metadata) .ThenInclude(m => m.People) .Include(s => s.Metadata) .ThenInclude(m => m.Genres) .Include(s => s.Metadata) .ThenInclude(m => m.Tags) .Include(s => s.Volumes) .ThenInclude(v => v.Chapters) .ThenInclude(cm => cm.People) .Include(s => s.Volumes) .ThenInclude(v => v.Chapters) .ThenInclude(c => c.Genres) .Include(s => s.Volumes) .ThenInclude(v => v.Chapters) .ThenInclude(c => c.Tags) .Include(s => s.Volumes) .ThenInclude(v => v.Chapters) .ThenInclude(c => c.Files) .AsSplitQuery() .OrderBy(s => s.SortName); return await PagedList.CreateAsync(query, userParams.PageNumber, userParams.PageSize); } /// /// This is a heavy call. Returns all entities down to Files and Library and Series Metadata. /// /// /// public async Task GetFullSeriesForSeriesIdAsync(int seriesId) { return await _context.Series .Where(s => s.Id == seriesId) .Include(s => s.Metadata) .ThenInclude(m => m.People) .Include(s => s.Metadata) .ThenInclude(m => m.Genres) .Include(s => s.Library) .Include(s => s.Volumes) .ThenInclude(v => v.Chapters) .ThenInclude(cm => cm.People) .Include(s => s.Volumes) .ThenInclude(v => v.Chapters) .ThenInclude(c => c.Tags) .Include(s => s.Volumes) .ThenInclude(v => v.Chapters) .ThenInclude(c => c.Genres) .Include(s => s.Metadata) .ThenInclude(m => m.Tags) .Include(s => s.Volumes) .ThenInclude(v => v.Chapters) .ThenInclude(c => c.Files) .AsSplitQuery() .SingleOrDefaultAsync(); } /// /// Gets all series /// /// Restricts to just one library /// /// /// /// public async Task> GetSeriesDtoForLibraryIdAsync(int libraryId, int userId, UserParams userParams, FilterDto filter) { var query = await CreateFilteredSearchQueryable(userId, libraryId, filter); var retSeries = query .ProjectTo(_mapper.ConfigurationProvider) .AsSplitQuery() .AsNoTracking(); return await PagedList.CreateAsync(retSeries, userParams.PageNumber, userParams.PageSize); } private async Task> GetUserLibraries(int libraryId, int userId) { if (libraryId == 0) { return await _context.Library .Include(l => l.AppUsers) .Where(library => library.AppUsers.Any(user => user.Id == userId)) .AsNoTracking() .Select(library => library.Id) .ToListAsync(); } return new List() { libraryId }; } public async Task SearchSeries(int userId, bool isAdmin, int[] libraryIds, string searchQuery) { const int maxRecords = 15; var result = new SearchResultGroupDto(); var searchQueryNormalized = Parser.Parser.Normalize(searchQuery); var seriesIds = _context.Series .Where(s => libraryIds.Contains(s.LibraryId)) .Select(s => s.Id) .ToList(); result.Libraries = await _context.Library .Where(l => libraryIds.Contains(l.Id)) .Where(l => EF.Functions.Like(l.Name, $"%{searchQuery}%")) .OrderBy(l => l.Name) .AsSplitQuery() .Take(maxRecords) .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); var justYear = Regex.Match(searchQuery, @"\d{4}").Value; var hasYearInQuery = !string.IsNullOrEmpty(justYear); var yearComparison = hasYearInQuery ? int.Parse(justYear) : 0; result.Series = _context.Series .Where(s => libraryIds.Contains(s.LibraryId)) .Where(s => EF.Functions.Like(s.Name, $"%{searchQuery}%") || EF.Functions.Like(s.OriginalName, $"%{searchQuery}%") || EF.Functions.Like(s.LocalizedName, $"%{searchQuery}%") || EF.Functions.Like(s.NormalizedName, $"%{searchQueryNormalized}%") || (hasYearInQuery && s.Metadata.ReleaseYear == yearComparison)) .Include(s => s.Library) .OrderBy(s => s.SortName) .AsNoTracking() .AsSplitQuery() .Take(maxRecords) .ProjectTo(_mapper.ConfigurationProvider) .AsEnumerable(); result.ReadingLists = await _context.ReadingList .Where(rl => rl.AppUserId == userId || rl.Promoted) .Where(rl => EF.Functions.Like(rl.Title, $"%{searchQuery}%")) .AsSplitQuery() .Take(maxRecords) .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); result.Collections = await _context.CollectionTag .Where(s => EF.Functions.Like(s.Title, $"%{searchQuery}%") || EF.Functions.Like(s.NormalizedTitle, $"%{searchQueryNormalized}%")) .Where(s => s.Promoted || isAdmin) .OrderBy(s => s.Title) .AsNoTracking() .AsSplitQuery() .Take(maxRecords) .OrderBy(c => c.NormalizedTitle) .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); result.Persons = await _context.SeriesMetadata .Where(sm => seriesIds.Contains(sm.SeriesId)) .SelectMany(sm => sm.People.Where(t => EF.Functions.Like(t.Name, $"%{searchQuery}%"))) .AsSplitQuery() .Take(maxRecords) .Distinct() .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); result.Genres = await _context.SeriesMetadata .Where(sm => seriesIds.Contains(sm.SeriesId)) .SelectMany(sm => sm.Genres.Where(t => EF.Functions.Like(t.Title, $"%{searchQuery}%"))) .AsSplitQuery() .OrderBy(t => t.Title) .Distinct() .Take(maxRecords) .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); result.Tags = await _context.SeriesMetadata .Where(sm => seriesIds.Contains(sm.SeriesId)) .SelectMany(sm => sm.Tags.Where(t => EF.Functions.Like(t.Title, $"%{searchQuery}%"))) .AsSplitQuery() .OrderBy(t => t.Title) .Distinct() .Take(maxRecords) .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); var fileIds = _context.Series .Where(s => libraryIds.Contains(s.LibraryId)) .AsSplitQuery() .SelectMany(s => s.Volumes) .SelectMany(v => v.Chapters) .SelectMany(c => c.Files.Select(f => f.Id)); result.Files = await _context.MangaFile .Where(m => EF.Functions.Like(m.FilePath, $"%{searchQuery}%") && fileIds.Contains(m.Id)) .AsSplitQuery() .Take(maxRecords) .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); result.Chapters = await _context.Chapter .Include(c => c.Files) .Where(c => EF.Functions.Like(c.TitleName, $"%{searchQuery}%")) .Where(c => c.Files.All(f => fileIds.Contains(f.Id))) .AsSplitQuery() .Take(maxRecords) .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); return result; } public async Task GetSeriesDtoByIdAsync(int seriesId, int userId) { var series = await _context.Series.Where(x => x.Id == seriesId) .ProjectTo(_mapper.ConfigurationProvider) .SingleAsync(); var seriesList = new List() {series}; await AddSeriesModifiers(userId, seriesList); return seriesList[0]; } public async Task DeleteSeriesAsync(int seriesId) { var series = await _context.Series.Where(s => s.Id == seriesId).SingleOrDefaultAsync(); if (series != null) _context.Series.Remove(series); return await _context.SaveChangesAsync() > 0; } /// /// Returns Volumes, Metadata (Incl Genres and People), and Collection Tags /// /// /// public async Task GetSeriesByIdAsync(int seriesId, SeriesIncludes includes = SeriesIncludes.Volumes | SeriesIncludes.Metadata) { var query = _context.Series .Where(s => s.Id == seriesId) .AsSplitQuery(); if (includes.HasFlag(SeriesIncludes.Volumes)) { query = query.Include(s => s.Volumes); } if (includes.HasFlag(SeriesIncludes.Related)) { query = query.Include(s => s.Relations) .ThenInclude(r => r.TargetSeries) .Include(s => s.RelationOf); } if (includes.HasFlag(SeriesIncludes.Metadata)) { query = query.Include(s => s.Metadata) .ThenInclude(m => m.CollectionTags) .Include(s => s.Metadata) .ThenInclude(m => m.Genres) .Include(s => s.Metadata) .ThenInclude(m => m.People) .Include(s => s.Metadata) .ThenInclude(m => m.Tags); } return await query.SingleOrDefaultAsync(); } /// /// Returns Volumes, Metadata, and Collection Tags /// /// /// public async Task> GetSeriesByIdsAsync(IList seriesIds) { return await _context.Series .Include(s => s.Volumes) .Include(s => s.Metadata) .ThenInclude(m => m.CollectionTags) .Where(s => seriesIds.Contains(s.Id)) .AsSplitQuery() .ToListAsync(); } public async Task GetChapterIdsForSeriesAsync(IList seriesIds) { var volumes = await _context.Volume .Where(v => seriesIds.Contains(v.SeriesId)) .Include(v => v.Chapters) .ToListAsync(); IList chapterIds = new List(); foreach (var v in volumes) { foreach (var c in v.Chapters) { chapterIds.Add(c.Id); } } return chapterIds.ToArray(); } /// /// This returns a dictionary mapping seriesId -> list of chapters back for each series id passed /// /// /// public async Task>> GetChapterIdWithSeriesIdForSeriesAsync(int[] seriesIds) { var volumes = await _context.Volume .Where(v => seriesIds.Contains(v.SeriesId)) .Include(v => v.Chapters) .ToListAsync(); var seriesChapters = new Dictionary>(); foreach (var v in volumes) { foreach (var c in v.Chapters) { if (!seriesChapters.ContainsKey(v.SeriesId)) { var list = new List(); seriesChapters.Add(v.SeriesId, list); } seriesChapters[v.SeriesId].Add(c.Id); } } return seriesChapters; } public async Task AddSeriesModifiers(int userId, List series) { var userProgress = await _context.AppUserProgresses .Where(p => p.AppUserId == userId && series.Select(s => s.Id).Contains(p.SeriesId)) .ToListAsync(); var userRatings = await _context.AppUserRating .Where(r => r.AppUserId == userId && series.Select(s => s.Id).Contains(r.SeriesId)) .ToListAsync(); foreach (var s in series) { s.PagesRead = userProgress.Where(p => p.SeriesId == s.Id).Sum(p => p.PagesRead); var rating = userRatings.SingleOrDefault(r => r.SeriesId == s.Id); if (rating != null) { s.UserRating = rating.Rating; s.UserReview = rating.Review; } if (userProgress.Count > 0) { s.LatestReadDate = userProgress.Max(p => p.LastModified); } } } public async Task GetSeriesCoverImageAsync(int seriesId) { return await _context.Series .Where(s => s.Id == seriesId) .Select(s => s.CoverImage) .AsNoTracking() .SingleOrDefaultAsync(); } /// /// Returns a list of Series that were added, ordered by Created desc /// /// /// Library to restrict to, if 0, will apply to all libraries /// Contains pagination information /// Optional filter on query /// public async Task> GetRecentlyAdded(int libraryId, int userId, UserParams userParams, FilterDto filter) { var query = await CreateFilteredSearchQueryable(userId, libraryId, filter); var retSeries = query .OrderByDescending(s => s.Created) .ProjectTo(_mapper.ConfigurationProvider) .AsSplitQuery() .AsNoTracking(); return await PagedList.CreateAsync(retSeries, userParams.PageNumber, userParams.PageSize); } private IList ExtractFilters(int libraryId, int userId, FilterDto filter, ref List userLibraries, out List allPeopleIds, out bool hasPeopleFilter, out bool hasGenresFilter, out bool hasCollectionTagFilter, out bool hasRatingFilter, out bool hasProgressFilter, out IList seriesIds, out bool hasAgeRating, out bool hasTagsFilter, out bool hasLanguageFilter, out bool hasPublicationFilter, out bool hasSeriesNameFilter) { var formats = filter.GetSqlFilter(); if (filter.Libraries.Count > 0) { userLibraries = userLibraries.Where(l => filter.Libraries.Contains(l)).ToList(); } else if (libraryId > 0) { userLibraries = userLibraries.Where(l => l == libraryId).ToList(); } allPeopleIds = new List(); allPeopleIds.AddRange(filter.Writers); allPeopleIds.AddRange(filter.Character); allPeopleIds.AddRange(filter.Colorist); allPeopleIds.AddRange(filter.Editor); allPeopleIds.AddRange(filter.Inker); allPeopleIds.AddRange(filter.Letterer); allPeopleIds.AddRange(filter.Penciller); allPeopleIds.AddRange(filter.Publisher); allPeopleIds.AddRange(filter.CoverArtist); allPeopleIds.AddRange(filter.Translators); hasPeopleFilter = allPeopleIds.Count > 0; hasGenresFilter = filter.Genres.Count > 0; hasCollectionTagFilter = filter.CollectionTags.Count > 0; hasRatingFilter = filter.Rating > 0; hasProgressFilter = !filter.ReadStatus.Read || !filter.ReadStatus.InProgress || !filter.ReadStatus.NotRead; hasAgeRating = filter.AgeRating.Count > 0; hasTagsFilter = filter.Tags.Count > 0; hasLanguageFilter = filter.Languages.Count > 0; hasPublicationFilter = filter.PublicationStatus.Count > 0; bool ProgressComparison(int pagesRead, int totalPages) { var result = false; if (filter.ReadStatus.NotRead) { result = (pagesRead == 0); } if (filter.ReadStatus.Read) { result = result || (pagesRead == totalPages); } if (filter.ReadStatus.InProgress) { result = result || (pagesRead > 0 && pagesRead < totalPages); } return result; } seriesIds = new List(); if (hasProgressFilter) { seriesIds = _context.Series .Include(s => s.Progress) .Select(s => new { Series = s, PagesRead = s.Progress.Where(p => p.AppUserId == userId).Sum(p => p.PagesRead), }) .AsEnumerable() .Where(s => ProgressComparison(s.PagesRead, s.Series.Pages)) .Select(s => s.Series.Id) .ToList(); } hasSeriesNameFilter = !string.IsNullOrEmpty(filter.SeriesNameQuery); return formats; } /// /// Returns Series that the user has some partial progress on. Sorts based on activity. Sort first by User progress, then /// by when chapters have been added to series. Restricts progress in the past 30 days and chapters being added to last 7. /// /// /// Library to restrict to, if 0, will apply to all libraries /// Pagination information /// Optional (default null) filter on query /// public async Task> GetOnDeck(int userId, int libraryId, UserParams userParams, FilterDto filter) { var cutoffProgressPoint = DateTime.Now - TimeSpan.FromDays(30); var cutoffLastAddedPoint = DateTime.Now - TimeSpan.FromDays(7); var libraryIds = GetLibraryIdsForUser(userId, libraryId); var usersSeriesIds = GetSeriesIdsForLibraryIds(libraryIds); var query = _context.Series .Where(s => usersSeriesIds.Contains(s.Id)) .Select(s => new { Series = s, PagesRead = _context.AppUserProgresses.Where(p => p.SeriesId == s.Id && p.AppUserId == userId) .Sum(s1 => s1.PagesRead), LatestReadDate = _context.AppUserProgresses .Where(p => p.SeriesId == s.Id && p.AppUserId == userId) .Max(p => p.LastModified), s.LastChapterAdded, }) .Where(s => s.PagesRead > 0 && s.PagesRead < s.Series.Pages) .Where(d => d.LatestReadDate >= cutoffProgressPoint || d.LastChapterAdded >= cutoffLastAddedPoint).OrderByDescending(s => s.LatestReadDate) .ThenByDescending(s => s.LastChapterAdded) .Select(s => s.Series) .ProjectTo(_mapper.ConfigurationProvider) .AsSplitQuery() .AsNoTracking(); return await PagedList.CreateAsync(query, userParams.PageNumber, userParams.PageSize); } private async Task> CreateFilteredSearchQueryable(int userId, int libraryId, FilterDto filter) { var userLibraries = await GetUserLibraries(libraryId, userId); var formats = ExtractFilters(libraryId, userId, filter, ref userLibraries, out var allPeopleIds, out var hasPeopleFilter, out var hasGenresFilter, out var hasCollectionTagFilter, out var hasRatingFilter, out var hasProgressFilter, out var seriesIds, out var hasAgeRating, out var hasTagsFilter, out var hasLanguageFilter, out var hasPublicationFilter, out var hasSeriesNameFilter); var query = _context.Series .Where(s => userLibraries.Contains(s.LibraryId) && formats.Contains(s.Format) && (!hasGenresFilter || s.Metadata.Genres.Any(g => filter.Genres.Contains(g.Id))) && (!hasPeopleFilter || s.Metadata.People.Any(p => allPeopleIds.Contains(p.Id))) && (!hasCollectionTagFilter || s.Metadata.CollectionTags.Any(t => filter.CollectionTags.Contains(t.Id))) && (!hasRatingFilter || s.Ratings.Any(r => r.Rating >= filter.Rating && r.AppUserId == userId)) && (!hasProgressFilter || seriesIds.Contains(s.Id)) && (!hasAgeRating || filter.AgeRating.Contains(s.Metadata.AgeRating)) && (!hasTagsFilter || s.Metadata.Tags.Any(t => filter.Tags.Contains(t.Id))) && (!hasLanguageFilter || filter.Languages.Contains(s.Metadata.Language)) && (!hasPublicationFilter || filter.PublicationStatus.Contains(s.Metadata.PublicationStatus))) .Where(s => !hasSeriesNameFilter || EF.Functions.Like(s.Name, $"%{filter.SeriesNameQuery}%") || EF.Functions.Like(s.OriginalName, $"%{filter.SeriesNameQuery}%") || EF.Functions.Like(s.LocalizedName, $"%{filter.SeriesNameQuery}%")) .AsNoTracking(); // If no sort options, default to using SortName filter.SortOptions ??= new SortOptions() { IsAscending = true, SortField = SortField.SortName }; if (filter.SortOptions.IsAscending) { query = filter.SortOptions.SortField switch { SortField.SortName => query.OrderBy(s => s.SortName), SortField.CreatedDate => query.OrderBy(s => s.Created), SortField.LastModifiedDate => query.OrderBy(s => s.LastModified), SortField.LastChapterAdded => query.OrderBy(s => s.LastChapterAdded), _ => query }; } else { query = filter.SortOptions.SortField switch { SortField.SortName => query.OrderByDescending(s => s.SortName), SortField.CreatedDate => query.OrderByDescending(s => s.Created), SortField.LastModifiedDate => query.OrderByDescending(s => s.LastModified), SortField.LastChapterAdded => query.OrderByDescending(s => s.LastChapterAdded), _ => query }; } return query; } public async Task GetSeriesMetadata(int seriesId) { var metadataDto = await _context.SeriesMetadata .Where(metadata => metadata.SeriesId == seriesId) .Include(m => m.Genres) .Include(m => m.Tags) .Include(m => m.People) .AsNoTracking() .ProjectTo(_mapper.ConfigurationProvider) .AsSplitQuery() .SingleOrDefaultAsync(); if (metadataDto != null) { metadataDto.CollectionTags = await _context.CollectionTag .Include(t => t.SeriesMetadatas) .Where(t => t.SeriesMetadatas.Select(s => s.SeriesId).Contains(seriesId)) .ProjectTo(_mapper.ConfigurationProvider) .AsNoTracking() .AsSplitQuery() .ToListAsync(); } return metadataDto; } public async Task> GetSeriesDtoForCollectionAsync(int collectionId, int userId, UserParams userParams) { var userLibraries = _context.Library .Include(l => l.AppUsers) .Where(library => library.AppUsers.Any(user => user.Id == userId)) .AsNoTracking() .Select(library => library.Id) .ToList(); var query = _context.CollectionTag .Where(s => s.Id == collectionId) .Include(c => c.SeriesMetadatas) .ThenInclude(m => m.Series) .SelectMany(c => c.SeriesMetadatas.Select(sm => sm.Series).Where(s => userLibraries.Contains(s.LibraryId))) .OrderBy(s => s.LibraryId) .ThenBy(s => s.SortName) .ProjectTo(_mapper.ConfigurationProvider) .AsSplitQuery() .AsNoTracking(); return await PagedList.CreateAsync(query, userParams.PageNumber, userParams.PageSize); } public async Task> GetFilesForSeries(int seriesId) { return await _context.Volume .Where(v => v.SeriesId == seriesId) .Include(v => v.Chapters) .ThenInclude(c => c.Files) .SelectMany(v => v.Chapters.SelectMany(c => c.Files)) .AsNoTracking() .ToListAsync(); } public async Task> GetSeriesDtoForIdsAsync(IEnumerable seriesIds, int userId) { var allowedLibraries = _context.Library .Include(l => l.AppUsers) .Where(library => library.AppUsers.Any(x => x.Id == userId)) .Select(l => l.Id); return await _context.Series .Where(s => seriesIds.Contains(s.Id) && allowedLibraries.Contains(s.LibraryId)) .OrderBy(s => s.SortName) .ProjectTo(_mapper.ConfigurationProvider) .AsNoTracking() .AsSplitQuery() .ToListAsync(); } public async Task> GetAllCoverImagesAsync() { return await _context.Series .Select(s => s.CoverImage) .Where(t => !string.IsNullOrEmpty(t)) .AsNoTracking() .ToListAsync(); } public async Task> GetLockedCoverImagesAsync() { return await _context.Series .Where(s => s.CoverImageLocked && !string.IsNullOrEmpty(s.CoverImage)) .Select(s => s.CoverImage) .AsNoTracking() .ToListAsync(); } /// /// Returns the number of series for a given library (or all libraries if libraryId is 0) /// /// Defaults to 0, library to restrict count to /// private async Task GetSeriesCount(int libraryId = 0) { if (libraryId > 0) { return await _context.Series .Where(s => s.LibraryId == libraryId) .CountAsync(); } return await _context.Series.CountAsync(); } /// /// Returns the number of series that should be processed in parallel to optimize speed and memory. Minimum of 50 /// /// Defaults to 0 meaning no library /// private async Task> GetChunkSize(int libraryId = 0) { var totalSeries = await GetSeriesCount(libraryId); return new Tuple(totalSeries, 50); } public async Task GetChunkInfo(int libraryId = 0) { var (totalSeries, chunkSize) = await GetChunkSize(libraryId); if (totalSeries == 0) return new Chunk() { TotalChunks = 0, TotalSize = 0, ChunkSize = 0 }; var totalChunks = Math.Max((int) Math.Ceiling((totalSeries * 1.0) / chunkSize), 1); return new Chunk() { TotalSize = totalSeries, ChunkSize = chunkSize, TotalChunks = totalChunks }; } public async Task> GetSeriesMetadataForIdsAsync(IEnumerable seriesIds) { return await _context.SeriesMetadata .Where(sm => seriesIds.Contains(sm.SeriesId)) .Include(sm => sm.CollectionTags) .ToListAsync(); } public async Task> GetAllAgeRatingsDtosForLibrariesAsync(List libraryIds) { return await _context.Series .Where(s => libraryIds.Contains(s.LibraryId)) .Select(s => s.Metadata.AgeRating) .Distinct() .Select(s => new AgeRatingDto() { Value = s, Title = s.ToDescription() }) .ToListAsync(); } public async Task> GetAllLanguagesForLibrariesAsync(List libraryIds) { var ret = await _context.Series .Where(s => libraryIds.Contains(s.LibraryId)) .Select(s => s.Metadata.Language) .AsNoTracking() .Distinct() .ToListAsync(); return ret .Where(s => !string.IsNullOrEmpty(s)) .Select(s => new LanguageDto() { Title = CultureInfo.GetCultureInfo(s).DisplayName, IsoCode = s }) .OrderBy(s => s.Title) .ToList(); } public IEnumerable GetAllPublicationStatusesDtosForLibrariesAsync(List libraryIds) { return _context.Series .Where(s => libraryIds.Contains(s.LibraryId)) .Select(s => s.Metadata.PublicationStatus) .Distinct() .AsEnumerable() .Select(s => new PublicationStatusDto() { Value = s, Title = s.ToDescription() }) .OrderBy(s => s.Title); } /// /// Return recently updated series, regardless of read progress, and group the number of volume or chapters added. /// /// This provides 2 levels of pagination. Fetching the individual chapters only looks at 3000. Then when performing grouping /// in memory, we stop after 30 series. /// Used to ensure user has access to libraries /// public async Task> GetRecentlyUpdatedSeries(int userId, int pageSize = 30) { var seriesMap = new Dictionary(); var index = 0; foreach (var item in await GetRecentlyAddedChaptersQuery(userId)) { if (seriesMap.Keys.Count == pageSize) break; if (seriesMap.ContainsKey(item.SeriesName)) { seriesMap[item.SeriesName].Count += 1; } else { seriesMap[item.SeriesName] = new GroupedSeriesDto() { LibraryId = item.LibraryId, LibraryType = item.LibraryType, SeriesId = item.SeriesId, SeriesName = item.SeriesName, Created = item.Created, Id = index, Format = item.Format, Count = 1, }; index += 1; } } return seriesMap.Values.AsEnumerable(); } public async Task> GetSeriesForRelationKind(int userId, int seriesId, RelationKind kind) { var libraryIds = GetLibraryIdsForUser(userId); var usersSeriesIds = _context.Series .Where(s => libraryIds.Contains(s.LibraryId)) .Select(s => s.Id); var targetSeries = _context.SeriesRelation .Where(sr => sr.SeriesId == seriesId && sr.RelationKind == kind && usersSeriesIds.Contains(sr.TargetSeriesId)) .Include(sr => sr.TargetSeries) .AsSplitQuery() .AsNoTracking() .Select(sr => sr.TargetSeriesId); return await _context.Series .Where(s => targetSeries.Contains(s.Id)) .AsSplitQuery() .AsNoTracking() .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); } private IQueryable GetLibraryIdsForUser(int userId) { return _context.AppUser .Where(u => u.Id == userId) .SelectMany(l => l.Libraries.Select(lib => lib.Id)); } public async Task> GetMoreIn(int userId, int libraryId, int genreId, UserParams userParams) { var libraryIds = GetLibraryIdsForUser(userId, libraryId); var usersSeriesIds = GetSeriesIdsForLibraryIds(libraryIds); var query = _context.Series .Where(s => s.Metadata.Genres.Select(g => g.Id).Contains(genreId)) .Where(s => usersSeriesIds.Contains(s.Id)) .AsSplitQuery() .ProjectTo(_mapper.ConfigurationProvider); return await PagedList.CreateAsync(query, userParams.PageNumber, userParams.PageSize); } public async Task> GetRediscover(int userId, int libraryId, UserParams userParams) { var libraryIds = GetLibraryIdsForUser(userId, libraryId); var usersSeriesIds = GetSeriesIdsForLibraryIds(libraryIds); var distinctSeriesIdsWithProgress = _context.AppUserProgresses .Where(s => usersSeriesIds.Contains(s.SeriesId)) .Select(p => p.SeriesId) .Distinct(); var query = _context.Series .Where(s => distinctSeriesIdsWithProgress.Contains(s.Id) && _context.AppUserProgresses.Where(s1 => s1.SeriesId == s.Id && s1.AppUserId == userId) .Sum(s1 => s1.PagesRead) >= s.Pages) .AsSplitQuery() .ProjectTo(_mapper.ConfigurationProvider); return await PagedList.CreateAsync(query, userParams.PageNumber, userParams.PageSize); } public async Task GetSeriesForMangaFile(int mangaFileId, int userId) { var libraryIds = GetLibraryIdsForUser(userId); return await _context.MangaFile .Where(m => m.Id == mangaFileId) .AsSplitQuery() .Select(f => f.Chapter) .Select(c => c.Volume) .Select(v => v.Series) .Where(s => libraryIds.Contains(s.LibraryId)) .ProjectTo(_mapper.ConfigurationProvider) .SingleOrDefaultAsync(); } public async Task GetSeriesForChapter(int chapterId, int userId) { var libraryIds = GetLibraryIdsForUser(userId); return await _context.Chapter .Where(m => m.Id == chapterId) .AsSplitQuery() .Select(c => c.Volume) .Select(v => v.Series) .Where(s => libraryIds.Contains(s.LibraryId)) .ProjectTo(_mapper.ConfigurationProvider) .SingleOrDefaultAsync(); } public async Task> GetHighlyRated(int userId, int libraryId, UserParams userParams) { var libraryIds = GetLibraryIdsForUser(userId, libraryId); var usersSeriesIds = GetSeriesIdsForLibraryIds(libraryIds); var distinctSeriesIdsWithHighRating = _context.AppUserRating .Where(s => usersSeriesIds.Contains(s.SeriesId) && s.Rating > 4) .Select(p => p.SeriesId) .Distinct(); var query = _context.Series .Where(s => distinctSeriesIdsWithHighRating.Contains(s.Id)) .AsSplitQuery() .OrderByDescending(s => _context.AppUserRating.Where(r => r.SeriesId == s.Id).Select(r => r.Rating).Average()) .ProjectTo(_mapper.ConfigurationProvider); return await PagedList.CreateAsync(query, userParams.PageNumber, userParams.PageSize); } public async Task> GetQuickReads(int userId, int libraryId, UserParams userParams) { var libraryIds = GetLibraryIdsForUser(userId, libraryId); var usersSeriesIds = GetSeriesIdsForLibraryIds(libraryIds); var distinctSeriesIdsWithProgress = _context.AppUserProgresses .Where(s => usersSeriesIds.Contains(s.SeriesId)) .Select(p => p.SeriesId) .Distinct(); var query = _context.Series .Where(s => s.Pages < 2000 && !distinctSeriesIdsWithProgress.Contains(s.Id) && usersSeriesIds.Contains(s.Id)) .Where(s => s.Metadata.PublicationStatus != PublicationStatus.OnGoing) .AsSplitQuery() .ProjectTo(_mapper.ConfigurationProvider); return await PagedList.CreateAsync(query, userParams.PageNumber, userParams.PageSize); } /// /// Returns all library ids for a user /// /// /// 0 for no library filter /// private IQueryable GetLibraryIdsForUser(int userId, int libraryId) { return _context.AppUser .Where(u => u.Id == userId) .SelectMany(l => l.Libraries.Where(l => l.Id == libraryId || libraryId == 0).Select(lib => lib.Id)); } public async Task GetRelatedSeries(int userId, int seriesId) { var libraryIds = GetLibraryIdsForUser(userId); var usersSeriesIds = GetSeriesIdsForLibraryIds(libraryIds); return new RelatedSeriesDto() { SourceSeriesId = seriesId, Adaptations = await GetRelatedSeriesQuery(seriesId, usersSeriesIds, RelationKind.Adaptation), Characters = await GetRelatedSeriesQuery(seriesId, usersSeriesIds, RelationKind.Character), Prequels = await GetRelatedSeriesQuery(seriesId, usersSeriesIds, RelationKind.Prequel), Sequels = await GetRelatedSeriesQuery(seriesId, usersSeriesIds, RelationKind.Sequel), Contains = await GetRelatedSeriesQuery(seriesId, usersSeriesIds, RelationKind.Contains), SideStories = await GetRelatedSeriesQuery(seriesId, usersSeriesIds, RelationKind.SideStory), SpinOffs = await GetRelatedSeriesQuery(seriesId, usersSeriesIds, RelationKind.SpinOff), Others = await GetRelatedSeriesQuery(seriesId, usersSeriesIds, RelationKind.Other), AlternativeSettings = await GetRelatedSeriesQuery(seriesId, usersSeriesIds, RelationKind.AlternativeSetting), AlternativeVersions = await GetRelatedSeriesQuery(seriesId, usersSeriesIds, RelationKind.AlternativeVersion), Doujinshis = await GetRelatedSeriesQuery(seriesId, usersSeriesIds, RelationKind.Doujinshi), Parent = await _context.Series .SelectMany(s => s.RelationOf.Where(r => r.TargetSeriesId == seriesId && usersSeriesIds.Contains(r.TargetSeriesId) && r.RelationKind != RelationKind.Prequel && r.RelationKind != RelationKind.Sequel) .Select(sr => sr.Series)) .AsSplitQuery() .AsNoTracking() .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync() }; } private IQueryable GetSeriesIdsForLibraryIds(IQueryable libraryIds) { return _context.Series .Where(s => libraryIds.Contains(s.LibraryId)) .Select(s => s.Id); } private async Task> GetRelatedSeriesQuery(int seriesId, IEnumerable usersSeriesIds, RelationKind kind) { return await _context.Series.SelectMany(s => s.Relations.Where(sr => sr.RelationKind == kind && sr.SeriesId == seriesId && usersSeriesIds.Contains(sr.TargetSeriesId)) .Select(sr => sr.TargetSeries)) .AsSplitQuery() .AsNoTracking() .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); } private async Task> GetRecentlyAddedChaptersQuery(int userId, int maxRecords = 3000) { var libraries = await _context.AppUser .Where(u => u.Id == userId) .SelectMany(u => u.Libraries.Select(l => new {LibraryId = l.Id, LibraryType = l.Type})) .ToListAsync(); var libraryIds = libraries.Select(l => l.LibraryId).ToList(); var withinLastWeek = DateTime.Now - TimeSpan.FromDays(12); var ret = _context.Chapter .Where(c => c.Created >= withinLastWeek) .AsNoTracking() .Include(c => c.Volume) .ThenInclude(v => v.Series) .ThenInclude(s => s.Library) .OrderByDescending(c => c.Created) .Select(c => new RecentlyAddedSeries() { LibraryId = c.Volume.Series.LibraryId, LibraryType = c.Volume.Series.Library.Type, Created = c.Created, SeriesId = c.Volume.Series.Id, SeriesName = c.Volume.Series.Name, VolumeId = c.VolumeId, ChapterId = c.Id, Format = c.Volume.Series.Format, ChapterNumber = c.Number, ChapterRange = c.Range, IsSpecial = c.IsSpecial, VolumeNumber = c.Volume.Number, ChapterTitle = c.Title }) //.Take(maxRecords) .AsSplitQuery() .Where(c => c.Created >= withinLastWeek && libraryIds.Contains(c.LibraryId)) .AsEnumerable(); return ret; } }