imprink/src/Imprink.Application/Users/Commands/SetUserRoleHandler.cs

using AutoMapper;
using Imprink.Application.Exceptions;
using Imprink.Application.Users.Dtos;
using Imprink.Domain.Entities.Users;
using MediatR;

namespace Imprink.Application.Users.Commands;

public record SetUserRoleCommand(string Sub, Guid RoleId) : IRequest<UserRoleDto?>;

public class SetUserRoleHandler(IUnitOfWork uw, IMapper mapper) : IRequestHandler<SetUserRoleCommand, UserRoleDto?>
{
    public async Task<UserRoleDto?> Handle(SetUserRoleCommand request, CancellationToken cancellationToken)
    {
        await uw.BeginTransactionAsync(cancellationToken);

        try
        {
            if (!await uw.UserRepository.UserExistsAsync(request.Sub, cancellationToken))
                throw new NotFoundException("User with ID: " + request.Sub + " does not exist.");

            var userRole = new UserRole
            {
                UserId = request.Sub,
                RoleId = request.RoleId
            };

            var addedRole = await uw.UserRoleRepository.AddUserRoleAsync(userRole, cancellationToken);

            await uw.SaveAsync(cancellationToken);
            await uw.CommitTransactionAsync(cancellationToken);

            return mapper.Map<UserRoleDto>(addedRole);
        }
        catch
        {
            await uw.RollbackTransactionAsync(cancellationToken);
            throw;
        }
    }
}