using Indotalent.Infrastructure.Database; using MediatR; using Microsoft.EntityFrameworkCore; namespace Indotalent.Features.Leave.LeaveBalance.Cqrs; public class UpdateLeaveBalanceRequest : CreateLeaveBalanceRequest { public string? Id { get; set; } public string? EmployeeName { get; set; } public string? LeaveCategoryName { get; set; } public DateTimeOffset? CreatedAt { get; set; } public string? CreatedBy { get; set; } public DateTimeOffset? UpdatedAt { get; set; } public string? UpdatedBy { get; set; } } public class UpdateLeaveBalanceResponse { public string? Id { get; set; } public bool Success { get; set; } } public record UpdateLeaveBalanceCommand(UpdateLeaveBalanceRequest Data) : IRequest; public class UpdateLeaveBalanceHandler : IRequestHandler { private readonly AppDbContext _context; public UpdateLeaveBalanceHandler(AppDbContext context) => _context = context; public async Task Handle(UpdateLeaveBalanceCommand request, CancellationToken cancellationToken) { var entity = await _context.LeaveBalance .FirstOrDefaultAsync(x => x.Id == request.Data.Id, cancellationToken); if (entity == null) { return new UpdateLeaveBalanceResponse { Id = request.Data.Id, Success = false }; } entity.EmployeeId = request.Data.EmployeeId; entity.LeaveCategoryId = request.Data.LeaveCategoryId; entity.Entitlement = request.Data.Entitlement; entity.Used = request.Data.Used; entity.Pending = request.Data.Pending; entity.Remaining = request.Data.Remaining; entity.Year = request.Data.Year; entity.ValidFrom = request.Data.ValidFrom; entity.ValidTo = request.Data.ValidTo; entity.CarryForward = request.Data.CarryForward; await _context.SaveChangesAsync(cancellationToken); return new UpdateLeaveBalanceResponse { Id = entity.Id, Success = true }; } }