using System.Collections.Generic; using System.Linq; using System.Threading.Tasks; using API.Constants; using API.DTOs; using API.Entities; using API.Interfaces; using API.Interfaces.Repositories; using AutoMapper; using AutoMapper.QueryableExtensions; using Microsoft.AspNetCore.Identity; using Microsoft.EntityFrameworkCore; namespace API.Data.Repositories { public class UserRepository : IUserRepository { private readonly DataContext _context; private readonly UserManager _userManager; private readonly IMapper _mapper; public UserRepository(DataContext context, UserManager userManager, IMapper mapper) { _context = context; _userManager = userManager; _mapper = mapper; } public void Update(AppUser user) { _context.Entry(user).State = EntityState.Modified; } public void Update(AppUserPreferences preferences) { _context.Entry(preferences).State = EntityState.Modified; } public void Delete(AppUser user) { _context.AppUser.Remove(user); } /// /// Gets an AppUser by username. Returns back Progress information. /// /// /// public async Task GetUserByUsernameAsync(string username) { return await _context.Users .Include(u => u.Progresses) .Include(u => u.Bookmarks) .SingleOrDefaultAsync(x => x.UserName == username); } /// /// This fetches the Id for a user. Use whenever you just need an ID. /// /// /// public async Task GetUserIdByUsernameAsync(string username) { return await _context.Users .Where(x => x.UserName == username) .Select(u => u.Id) .SingleOrDefaultAsync(); } /// /// Gets an AppUser by username. Returns back Reading List and their Items. /// /// /// public async Task GetUserWithReadingListsByUsernameAsync(string username) { return await _context.Users .Include(u => u.ReadingLists) .ThenInclude(l => l.Items) .SingleOrDefaultAsync(x => x.UserName == username); } /// /// Gets an AppUser by id. Returns back Progress information. /// /// /// public async Task GetUserByIdAsync(int id) { return await _context.Users .Include(u => u.Progresses) .Include(u => u.Bookmarks) .SingleOrDefaultAsync(x => x.Id == id); } public async Task> GetAdminUsersAsync() { return await _userManager.GetUsersInRoleAsync(PolicyConstants.AdminRole); } public async Task GetUserRating(int seriesId, int userId) { return await _context.AppUserRating.Where(r => r.SeriesId == seriesId && r.AppUserId == userId) .SingleOrDefaultAsync(); } public void AddRatingTracking(AppUserRating userRating) { _context.AppUserRating.Add(userRating); } public async Task GetPreferencesAsync(string username) { return await _context.AppUserPreferences .Include(p => p.AppUser) .SingleOrDefaultAsync(p => p.AppUser.UserName == username); } public async Task> GetBookmarkDtosForSeries(int userId, int seriesId) { return await _context.AppUserBookmark .Where(x => x.AppUserId == userId && x.SeriesId == seriesId) .OrderBy(x => x.Page) .AsNoTracking() .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); } public async Task> GetBookmarkDtosForVolume(int userId, int volumeId) { return await _context.AppUserBookmark .Where(x => x.AppUserId == userId && x.VolumeId == volumeId) .OrderBy(x => x.Page) .AsNoTracking() .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); } public async Task> GetBookmarkDtosForChapter(int userId, int chapterId) { return await _context.AppUserBookmark .Where(x => x.AppUserId == userId && x.ChapterId == chapterId) .OrderBy(x => x.Page) .AsNoTracking() .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); } public async Task> GetAllBookmarkDtos(int userId) { return await _context.AppUserBookmark .Where(x => x.AppUserId == userId) .OrderBy(x => x.Page) .AsNoTracking() .ProjectTo(_mapper.ConfigurationProvider) .ToListAsync(); } public async Task GetUserByApiKeyAsync(string apiKey) { return await _context.AppUser .SingleOrDefaultAsync(u => u.ApiKey.Equals(apiKey)); } public async Task> GetMembersAsync() { return await _context.Users .Include(x => x.Libraries) .Include(r => r.UserRoles) .ThenInclude(r => r.Role) .OrderBy(u => u.UserName) .Select(u => new MemberDto { Id = u.Id, Username = u.UserName, Created = u.Created, LastActive = u.LastActive, Roles = u.UserRoles.Select(r => r.Role.Name).ToList(), Libraries = u.Libraries.Select(l => new LibraryDto { Name = l.Name, CoverImage = l.CoverImage, Type = l.Type, Folders = l.Folders.Select(x => x.Path).ToList() }).ToList() }) .AsNoTracking() .ToListAsync(); } } }