Spaces:
Paused
Paused
File size: 2,953 Bytes
8a2241a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | using Application.Abstractions.Interfaces;
using Entities.Interfaces;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using System.Security.Claims;
namespace RobDeliveryAPI.Controllers;
[Route("api/[controller]")]
[ApiController]
[Authorize]
public class WalletController : ControllerBase
{
private readonly IUserRepository _userRepository;
public WalletController(IUserRepository userRepository)
{
_userRepository = userRepository;
}
private int GetAuthenticatedUserId()
{
var userIdClaim = User.FindFirst("Id") ?? User.FindFirst(ClaimTypes.NameIdentifier);
if (userIdClaim == null || !int.TryParse(userIdClaim.Value, out int userId))
{
throw new UnauthorizedAccessException("User ID not found in token");
}
return userId;
}
[HttpGet("balance")]
public async Task<ActionResult<object>> GetBalance()
{
try
{
int userId = GetAuthenticatedUserId();
var user = await _userRepository.GetByIdAsync(userId);
if (user == null)
{
return NotFound(new { success = false, message = "User not found" });
}
return Ok(new { success = true, balance = user.Balance });
}
catch (UnauthorizedAccessException ex)
{
return StatusCode(403, new { success = false, message = ex.Message });
}
catch (Exception ex)
{
return StatusCode(500, new { success = false, message = $"Internal server error: {ex.Message}" });
}
}
[HttpPost("withdraw")]
public async Task<ActionResult<object>> Withdraw()
{
try
{
int userId = GetAuthenticatedUserId();
var user = await _userRepository.GetByIdAsync(userId);
if (user == null)
{
return NotFound(new { success = false, message = "User not found" });
}
if (user.Balance <= 0)
{
return BadRequest(new { success = false, message = "No funds to withdraw" });
}
decimal withdrawnAmount = user.Balance;
// Simulate Payout - just reset to 0
user.Balance = 0;
await _userRepository.UpdateAsync(user);
return Ok(new
{
success = true,
message = "Withdrawal successful",
withdrawnAmount = withdrawnAmount,
newBalance = user.Balance,
processedAt = DateTime.UtcNow
});
}
catch (UnauthorizedAccessException ex)
{
return StatusCode(403, new { success = false, message = ex.Message });
}
catch (Exception ex)
{
return StatusCode(500, new { success = false, message = $"Internal server error: {ex.Message}" });
}
}
}
|