using System.Collections.Generic; using System.Linq; using System.Threading.Tasks; using API.Constants; using API.DTOs; using API.Entities; using API.Interfaces; using AutoMapper; using AutoMapper.QueryableExtensions; using Microsoft.AspNetCore.Identity; using Microsoft.EntityFrameworkCore; namespace API.Data { public class UserRepository : IUserRepository { private readonly DataContext _context; private readonly IMapper _mapper; private readonly UserManager _userManager; public UserRepository(DataContext context, IMapper mapper, UserManager userManager) { _context = context; _mapper = mapper; _userManager = userManager; } public void Update(AppUser user) { _context.Entry(user).State = EntityState.Modified; } public void Delete(AppUser user) { _context.Users.Remove(user); } public async Task SaveAllAsync() { return await _context.SaveChangesAsync() > 0; } public async Task> GetUsersAsync() { return await _context.Users.ToListAsync(); } public async Task GetUserByIdAsync(int id) { return await _context.Users.FindAsync(id); } public async Task GetUserByUsernameAsync(string username) { return await _context.Users .SingleOrDefaultAsync(x => x.UserName == username); } public async Task> GetAdminUsersAsync() { return await _userManager.GetUsersInRoleAsync(PolicyConstants.AdminRole); } public async Task> GetMembersAsync() { return await _userManager.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() }) .ToListAsync(); } public async Task GetMemberAsync(string username) { return await _context.Users.Where(x => x.UserName == username) .Include(x => x.Libraries) .ProjectTo(_mapper.ConfigurationProvider) .SingleOrDefaultAsync(); } } }