aswi20220x00-gitlab/Backend/Core/Services/UserService/UserServiceEF.cs @ 7dcdd235

using Core.Contexts;
using Core.Entities;
using Models.Enums;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Serilog;
using BCrypt.Net;
using Models.Users;
using AutoMapper;
using Microsoft.EntityFrameworkCore;

namespace Core.Services
{
    public class UserServiceEF : IUserService
    {
        // DI

        private readonly DatabaseContext _databaseContext;
        private readonly ILogger _logger;
        private readonly IMapper _mapper;

        public UserServiceEF(DatabaseContext context, ILogger logger, IMapper mapper)
        {
            _databaseContext = context;
            _logger = logger;
            _mapper = mapper;
        }

        public User ChangePassword(User user, string newPassword)
        {
            if (newPassword == "")
            {
                throw new InvalidOperationException("Empty password.");
            }

            // Check if the user is tracked by EF
            if (!_databaseContext.Users.Local.Any(u => u.Id == user.Id))
            {
                _logger.Information($"User {user.Username} with Guid {user.Id} is untracked by EF and password change cannot be done.");
                throw new InvalidOperationException("User is untracked by EF.");
            }

            user.Password = BCrypt.Net.BCrypt.HashPassword(newPassword);
            _databaseContext.SaveChanges();
            return user;
        }

        public User? ChangePassword(Guid userId, string newPassword)
        {
            User? user = null;
            user = _databaseContext.Users.First(u => u.Id == userId);

            return ChangePassword(user, newPassword);
        }

        public User? CheckUsernamePassword(string username, string password)
        {
            try
            {
                // Throws exception if user does not exist
                User u = _databaseContext.Users.First(u => u.Username == username);
                if (!BCrypt.Net.BCrypt.Verify(password, u.Password))
                {
                    _logger.Information($"Password for user {username} doesn't match.");
                    return null;
                }
                return u;
            }
            catch (Exception)
            {
                _logger.Information($"No user with username {username} found.");
                return null;
            }
        }

        public User? CreateUser(string username, string name, string surname, string password, ERole role)
        {
            if (password == "")
            {
                return null;
            }

            // Check if username already used
            if (_databaseContext.Users.Any(u => u.Username == username))
            {
                _logger.Information($"Username {username} is already used. Cannot create new user");
                return null;
            }

            User user = new User()
            {
                Username = username,
                Name = name,
                Surname = surname,
                Password = BCrypt.Net.BCrypt.HashPassword(password),
                Role = role
            };
            _databaseContext.Users.Add(user);
            _databaseContext.SaveChanges();
            return user;
        }

        public User? GetUserByUsername(string username)
        {
            try
            {
                // Throws exception on no user found
                User user = _databaseContext.Users.First(u => u.Username == username);
                return user;
            }
            catch (InvalidOperationException)
            {
                _logger.Warning($"No user with the username {username} found.");
                return null;
            }
        }

        public User? GetUserById(Guid id)
        {
            try
            {
                // Throws exception on no user found
                User user = _databaseContext.Users.First(u => u.Id == id);
                return user;
            }
            catch (InvalidOperationException)
            {
                _logger.Warning($"No user with the GUID {id} found.");
                return null;
            }
        }

        public User UpdateUser(User user, string? username = null, string? name = null, string? surname = null, ERole? role = null)
        {
            // Check if the user is tracked by EF
            if (!_databaseContext.Users.Local.Any(u => u.Id == user.Id))
            {
                _logger.Information($"User {user.Username} with Guid {user.Id} is untracked by EF and update cannot be done.");
                throw new InvalidOperationException("User is untracked by EF.");
            }

            if (username is not null && username != "")
            {
                user.Username = username;
            }

            if (name is not null && name != "")
            {
                user.Name = name;
            }

            if (surname is not null && surname != "")
            {
                user.Surname = surname;
            }

            if (role is not null)
            {
                // We know it is not null
                user.Role = (ERole)role;
            }

            _databaseContext.SaveChanges();
            return user;
        }

        public User? UpdateUser(Guid userId, string? username = null, string? name = null, string? surname = null, ERole? role = null)
        {
            try
            {
                User user = _databaseContext.Users.First(u => u.Id == userId);
                return UpdateUser(user, username, name, surname, role);
            }
            catch (Exception ex)
            {
                return null;
            }
        }


        public UserList GetUsers()
        {
            var userList = new UserList();
            var users = _databaseContext.Users
                .OrderBy(u => u.Surname)
                .ToList();
            foreach (var user in users)
            {
                var userInfo = _mapper.Map<UserInfo>(user);
                var annotations = _databaseContext.Annotations
                    .Include(a => a.User)
                    .Where(a => !(a is FinalAnnotation))
                    .Where(a => a.User == user)
                    .ToList();
                userInfo.NewDocumentsCount = annotations.Where(a => a.State == EState.NEW).Count();
                userInfo.InProgressDocumentsCount = annotations.Where(a => a.State == EState.IN_PROGRESS).Count();
                userInfo.FinalizedDocumentsCount = annotations.Where(a => a.State == EState.DONE).Count();
                userList.Users.Add(userInfo);
            }

            return userList;
        }


        public void DeleteUser(Guid userId)
        {
            try
            {
                var user = _databaseContext.Users.Where(u => u.Id == userId).First();
                _databaseContext.Users.Remove(user);
                _databaseContext.SaveChanges();
            }
            catch (Exception)
            {
                throw new InvalidOperationException("User not found");
            }
        }
    }
}