luticate2-api/Luticate2.Auth/Business/LuUsersBusiness.cs

using System;
using System.Collections.Generic;
using System.Security.Cryptography;
using System.Text;
using Luticate2.Auth.DataAccess;
using Luticate2.Auth.Dbo.Users;
using Luticate2.Auth.Interfaces.Tokens;
using Luticate2.Auth.Interfaces.Users;
using Luticate2.Utils.Business;
using Luticate2.Utils.Dbo.Result;
using Luticate2.Utils.Interfaces;
using Luticate2.Utils.Utils;
using Microsoft.AspNetCore.Cryptography.KeyDerivation;

namespace Luticate2.Auth.Business
{
    public class LuUsersBusiness : LuCrudBusiness<LuUsersDataAccess, LuUsersAddFullDbo, LuUsersFullDbo, LuUsersEditFullDbo, string>, ILuUsersBusiness
    {
        private readonly ILuTokensBusiness _luTokensBusiness;
        private readonly ILuLoggedUserAccessor _luLoggedUserAccessor;

        public LuUsersBusiness(LuUsersDataAccess dataAccess, ILuNotificationsBusiness notificationsBusiness,
            ILuTokensBusiness luTokensBusiness, ILuLoggedUserAccessor luLoggedUserAccessor) : base(dataAccess, notificationsBusiness)
        {
            _luTokensBusiness = luTokensBusiness;
            _luLoggedUserAccessor = luLoggedUserAccessor;
        }

        public string GenerateSalt()
        {
            var salt = new byte[128 / 8];
            using (var rng = RandomNumberGenerator.Create())
            {
                rng.GetBytes(salt);
            }
            return Convert.ToBase64String(salt);
        }

        public string HashPassword(string password, string salt)
        {
            var hashed = Convert.ToBase64String(KeyDerivation.Pbkdf2(password, Encoding.ASCII.GetBytes(salt),
                KeyDerivationPrf.HMACSHA1, 10000, 256 / 8));
            return hashed;
        }

        public bool VerifyPasswordHash(string password, string hash, string salt)
        {
            var newHash = HashPassword(password, salt);
            return newHash == hash;
        }

        public LuResult<LuUsersFullDbo> FindByUsername(string username)
        {
            return DataAccess.FindByUsername(username);
        }

        public LuResult<LuUsersFullDbo> Register(LuUsersAddDbo user)
        {
            var userRes = FindByUsername(user.Username);
            if (userRes)
            {
                return LuResult<LuUsersFullDbo>.Error(LuStatus.InputError,
                    $"username: {user.Username}", "Username already exists");
            }
            if (userRes.Status != LuStatus.NotFound)
            {
                return userRes;
            }
            var salt = GenerateSalt();
            var password = HashPassword(user.Password, salt);
            return this.AddDbo(new LuUsersAddFullDbo
            {
                AuthenticationSourceId = Guid.Empty.ToDbo(),
                Data = null,
                Password = password,
                Salt = salt,
                Username = user.Username
            });
        }

        public LuResult<LuUsersFullDbo> Edit(string id, LuUsersEditDbo user)
        {
            throw new NotImplementedException();
        }

        public LuResult<LuUsersLoginResultDbo> Login(string username, string password)
        {
            var userRes = FindByUsername(username);
            if (userRes.Status == LuStatus.NotFound)
            {
                return LuResult<LuUsersLoginResultDbo>.Error(LuStatus.LoginError,
                    $"unknown username; username: {username}", "Invalid username or password");
            }
            if (!userRes)
            {
                return userRes.To<LuUsersLoginResultDbo>();
            }
            if (userRes.Data.Password == null)
            {
                return LuResult<LuUsersLoginResultDbo>.Error(LuStatus.LoginError,
                    $"null password; username: {username}", "Invalid username or password");
            }
            if (!VerifyPasswordHash(password, userRes.Data.Password, userRes.Data.Salt))
            {
                return LuResult<LuUsersLoginResultDbo>.Error(LuStatus.LoginError,
                    $"invalid password; username: {username}", "Invalid username or password");
            }

            var tokenRes = _luTokensBusiness.GenerateToken(userRes.Data.ToLite());
            if (!tokenRes)
            {
                return tokenRes.To<LuUsersLoginResultDbo>();
            }

            return LuResult<LuUsersLoginResultDbo>.Ok(new LuUsersLoginResultDbo
            {
                Token = tokenRes.Data,
                User = userRes.Data.ToLite()
            });
        }

        public LuResult<bool> Logout(string token)
        {
            if (token == null)
            {
                return LuResult<bool>.Ok(true);
            }
            return _luTokensBusiness.UnRegisterToken(token).To(usersToken => true);
        }

        public LuResult<LuUsersFullDbo> Me()
        {
            return LuResult<LuUsersFullDbo>.Ok(_luLoggedUserAccessor.GetLoggedUser());
        }

        LuResult<T> ILuCrudInterface<LuUsersAddFullDbo, LuUsersFullDbo, LuUsersEditFullDbo, string>.DeleteSingleById<T>(string id, Func<LuUsersFullDbo, T> returnFunc)
        {
            var loggedUser = _luLoggedUserAccessor.GetLoggedUser();
            if (id == Guid.Empty.ToDbo())
            {
                return LuResult<T>.Error(LuStatus.InputError,
                    $"loggedUser: {loggedUser.Username}", "Can not remove anonymous user");
            }
            if (id == loggedUser.Id)
            {
                return LuResult<T>.Error(LuStatus.InputError,
                    $"loggedUser: {loggedUser.Username}", "Can not remove yourself");
            }
            return base.DeleteSingleById(id, returnFunc);
        }
    }
}