using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Threading.Tasks; using API.Comparators; using API.DTOs; using API.DTOs.Reader; using API.Entities; using API.Extensions; using API.Interfaces; using API.Interfaces.Services; using Microsoft.AspNetCore.Mvc; using Microsoft.Extensions.Logging; namespace API.Controllers { public class ReaderController : BaseApiController { private readonly IDirectoryService _directoryService; private readonly ICacheService _cacheService; private readonly ILogger _logger; private readonly IUnitOfWork _unitOfWork; private readonly ChapterSortComparer _chapterSortComparer = new ChapterSortComparer(); public ReaderController(IDirectoryService directoryService, ICacheService cacheService, ILogger logger, IUnitOfWork unitOfWork) { _directoryService = directoryService; _cacheService = cacheService; _logger = logger; _unitOfWork = unitOfWork; } [HttpGet("image")] public async Task GetImage(int chapterId, int page) { if (page < 0) return BadRequest("Page cannot be less than 0"); var chapter = await _cacheService.Ensure(chapterId); if (chapter == null) return BadRequest("There was an issue finding image file for reading"); var (path, _) = await _cacheService.GetCachedPagePath(chapter, page); if (string.IsNullOrEmpty(path) || !System.IO.File.Exists(path)) return BadRequest($"No such image for page {page}"); var content = await _directoryService.ReadFileAsync(path); var format = Path.GetExtension(path).Replace(".", ""); // Calculates SHA1 Hash for byte[] Response.AddCacheHeader(content); return File(content, "image/" + format); } [HttpGet("chapter-info")] public async Task> GetChapterInfo(int chapterId) { var chapter = await _cacheService.Ensure(chapterId); if (chapter == null) return BadRequest("Could not find Chapter"); var volume = await _unitOfWork.SeriesRepository.GetVolumeAsync(chapter.VolumeId); if (volume == null) return BadRequest("Could not find Volume"); var (_, mangaFile) = await _cacheService.GetCachedPagePath(chapter, 0); var series = await _unitOfWork.SeriesRepository.GetSeriesByIdAsync(volume.SeriesId); return Ok(new ChapterInfoDto() { ChapterNumber = chapter.Range, VolumeNumber = volume.Number + string.Empty, VolumeId = volume.Id, FileName = Path.GetFileName(mangaFile.FilePath), SeriesName = series?.Name, IsSpecial = chapter.IsSpecial, Pages = chapter.Pages, }); } [HttpGet("get-bookmark")] public async Task> GetBookmark(int chapterId) { var user = await _unitOfWork.UserRepository.GetUserByUsernameAsync(User.GetUsername()); var bookmark = new BookmarkDto() { PageNum = 0, ChapterId = chapterId, VolumeId = 0, SeriesId = 0 }; if (user.Progresses == null) return Ok(bookmark); var progress = user.Progresses.SingleOrDefault(x => x.AppUserId == user.Id && x.ChapterId == chapterId); if (progress != null) { bookmark.SeriesId = progress.SeriesId; bookmark.VolumeId = progress.VolumeId; bookmark.PageNum = progress.PagesRead; bookmark.BookScrollId = progress.BookScrollId; } return Ok(bookmark); } [HttpPost("mark-read")] public async Task MarkRead(MarkReadDto markReadDto) { var user = await _unitOfWork.UserRepository.GetUserByUsernameAsync(User.GetUsername()); var volumes = await _unitOfWork.SeriesRepository.GetVolumes(markReadDto.SeriesId); user.Progresses ??= new List(); foreach (var volume in volumes) { foreach (var chapter in volume.Chapters) { var userProgress = user.Progresses.SingleOrDefault(x => x.ChapterId == chapter.Id && x.AppUserId == user.Id); if (userProgress == null) // I need to get all chapters and generate new user progresses for them? { user.Progresses.Add(new AppUserProgress { PagesRead = chapter.Pages, VolumeId = volume.Id, SeriesId = markReadDto.SeriesId, ChapterId = chapter.Id }); } else { userProgress.PagesRead = chapter.Pages; userProgress.SeriesId = markReadDto.SeriesId; userProgress.VolumeId = volume.Id; } } } _unitOfWork.UserRepository.Update(user); if (await _unitOfWork.CommitAsync()) { return Ok(); } return BadRequest("There was an issue saving progress"); } [HttpPost("mark-unread")] public async Task MarkUnread(MarkReadDto markReadDto) { var user = await _unitOfWork.UserRepository.GetUserByUsernameAsync(User.GetUsername()); var volumes = await _unitOfWork.SeriesRepository.GetVolumes(markReadDto.SeriesId); user.Progresses ??= new List(); foreach (var volume in volumes) { foreach (var chapter in volume.Chapters) { var userProgress = user.Progresses.SingleOrDefault(x => x.ChapterId == chapter.Id && x.AppUserId == user.Id); if (userProgress == null) { user.Progresses.Add(new AppUserProgress { PagesRead = 0, VolumeId = volume.Id, SeriesId = markReadDto.SeriesId, ChapterId = chapter.Id }); } else { userProgress.PagesRead = 0; userProgress.SeriesId = markReadDto.SeriesId; userProgress.VolumeId = volume.Id; } } } _unitOfWork.UserRepository.Update(user); if (await _unitOfWork.CommitAsync()) { return Ok(); } return BadRequest("There was an issue saving progress"); } [HttpPost("mark-volume-read")] public async Task MarkVolumeAsRead(MarkVolumeReadDto markVolumeReadDto) { var user = await _unitOfWork.UserRepository.GetUserByUsernameAsync(User.GetUsername()); _logger.LogDebug("Saving {UserName} progress for Volume {VolumeID} to read", user.UserName, markVolumeReadDto.VolumeId); var chapters = await _unitOfWork.VolumeRepository.GetChaptersAsync(markVolumeReadDto.VolumeId); foreach (var chapter in chapters) { user.Progresses ??= new List(); var userProgress = user.Progresses.SingleOrDefault(x => x.ChapterId == chapter.Id && x.AppUserId == user.Id); if (userProgress == null) { user.Progresses.Add(new AppUserProgress { PagesRead = chapter.Pages, VolumeId = markVolumeReadDto.VolumeId, SeriesId = markVolumeReadDto.SeriesId, ChapterId = chapter.Id }); } else { userProgress.PagesRead = chapter.Pages; userProgress.SeriesId = markVolumeReadDto.SeriesId; userProgress.VolumeId = markVolumeReadDto.VolumeId; } } _unitOfWork.UserRepository.Update(user); if (await _unitOfWork.CommitAsync()) { return Ok(); } return BadRequest("Could not save progress"); } [HttpPost("bookmark")] public async Task Bookmark(BookmarkDto bookmarkDto) { var user = await _unitOfWork.UserRepository.GetUserByUsernameAsync(User.GetUsername()); _logger.LogDebug("Saving {UserName} progress for Chapter {ChapterId} to page {PageNum}", user.UserName, bookmarkDto.ChapterId, bookmarkDto.PageNum); // Don't let user bookmark past total pages. var chapter = await _unitOfWork.VolumeRepository.GetChapterAsync(bookmarkDto.ChapterId); if (bookmarkDto.PageNum > chapter.Pages) { return BadRequest("Can't bookmark past max pages"); } if (bookmarkDto.PageNum < 0) { return BadRequest("Can't bookmark less than 0"); } user.Progresses ??= new List(); var userProgress = user.Progresses.SingleOrDefault(x => x.ChapterId == bookmarkDto.ChapterId && x.AppUserId == user.Id); if (userProgress == null) { user.Progresses.Add(new AppUserProgress { PagesRead = bookmarkDto.PageNum, VolumeId = bookmarkDto.VolumeId, SeriesId = bookmarkDto.SeriesId, ChapterId = bookmarkDto.ChapterId, BookScrollId = bookmarkDto.BookScrollId, LastModified = DateTime.Now }); } else { userProgress.PagesRead = bookmarkDto.PageNum; userProgress.SeriesId = bookmarkDto.SeriesId; userProgress.VolumeId = bookmarkDto.VolumeId; userProgress.BookScrollId = bookmarkDto.BookScrollId; userProgress.LastModified = DateTime.Now; } _unitOfWork.UserRepository.Update(user); if (await _unitOfWork.CommitAsync()) { return Ok(); } return BadRequest("Could not save progress"); } /// /// Returns the next logical chapter from the series. /// /// /// /// /// chapter id for next manga [HttpGet("next-chapter")] public async Task> GetNextChapter(int seriesId, int volumeId, int currentChapterId) { var user = await _unitOfWork.UserRepository.GetUserByUsernameAsync(User.GetUsername()); var volumes = await _unitOfWork.SeriesRepository.GetVolumesDtoAsync(seriesId, user.Id); var currentVolume = await _unitOfWork.SeriesRepository.GetVolumeAsync(volumeId); var currentChapter = await _unitOfWork.VolumeRepository.GetChapterAsync(currentChapterId); if (currentVolume.Number == 0) { var chapterId = GetNextChapterId(currentVolume.Chapters.OrderBy(x => double.Parse(x.Number), _chapterSortComparer), currentChapter.Number); if (chapterId > 0) return Ok(chapterId); } foreach (var volume in volumes) { if (volume.Number == currentVolume.Number && volume.Chapters.Count > 1) { var chapterId = GetNextChapterId(currentVolume.Chapters.OrderBy(x => double.Parse(x.Number), _chapterSortComparer), currentChapter.Number); if (chapterId > 0) return Ok(chapterId); } if (volume.Number == currentVolume.Number + 1) { return Ok(volume.Chapters.OrderBy(x => double.Parse(x.Number), _chapterSortComparer).FirstOrDefault()?.Id); } } return Ok(-1); } private static int GetNextChapterId(IEnumerable chapters, string currentChapterNumber) { var next = false; foreach (var chapter in chapters) { if (next) { return chapter.Id; } if (currentChapterNumber.Equals(chapter.Number)) next = true; } return -1; } /// /// Returns the previous logical chapter from the series. /// /// /// /// /// chapter id for next manga [HttpGet("prev-chapter")] public async Task> GetPreviousChapter(int seriesId, int volumeId, int currentChapterId) { var user = await _unitOfWork.UserRepository.GetUserByUsernameAsync(User.GetUsername()); var volumes = await _unitOfWork.SeriesRepository.GetVolumesDtoAsync(seriesId, user.Id); var currentVolume = await _unitOfWork.SeriesRepository.GetVolumeAsync(volumeId); var currentChapter = await _unitOfWork.VolumeRepository.GetChapterAsync(currentChapterId); if (currentVolume.Number == 0) { var chapterId = GetNextChapterId(currentVolume.Chapters.OrderBy(x => double.Parse(x.Number), _chapterSortComparer).Reverse(), currentChapter.Number); if (chapterId > 0) return Ok(chapterId); } foreach (var volume in volumes.Reverse()) { if (volume.Number == currentVolume.Number) { var chapterId = GetNextChapterId(currentVolume.Chapters.OrderBy(x => double.Parse(x.Number), _chapterSortComparer).Reverse(), currentChapter.Number); if (chapterId > 0) return Ok(chapterId); } if (volume.Number == currentVolume.Number - 1) { return Ok(volume.Chapters.OrderBy(x => double.Parse(x.Number), _chapterSortComparer).LastOrDefault()?.Id); } } return Ok(-1); } } }