Clean up

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

@@ -0,0 +1,41 @@
+using AutoMapper;
+using Imprink.Application.Exceptions;
+using Imprink.Application.Users.Dtos;
+using Imprink.Domain.Entities.Users;
+using MediatR;
+
+namespace Imprink.Application.Domains.Users;
+
+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;
+        }
+    }
+}