Adding units

2026-05-18 15:30:46 -05:00
parent 6366ca46c6
commit 37a46a7b36
10 changed files with 529 additions and 3 deletions
@@ -0,0 +1,58 @@
+using LocationsApi.Models;
+using LocationsApi.Services;
+using Microsoft.AspNetCore.Authorization;
+using Microsoft.AspNetCore.Mvc;
+using System.Security.Claims;
+
+namespace LocationsApi.Controllers;
+
+[ApiController]
+[Route("api/[controller]")]
+public class UnitsController : ControllerBase
+{
+    private readonly UnitTrainingService _units;
+
+    public UnitsController(UnitTrainingService units)
+    {
+        _units = units;
+    }
+
+    [Authorize(Roles = "USER,SUPER")]
+    [HttpGet("me")]
+    public async Task<ActionResult<PlayerUnitsResponse>> GetMine()
+    {
+        var username = User.FindFirstValue(ClaimTypes.Name);
+        if (string.IsNullOrWhiteSpace(username))
+        {
+            return Unauthorized();
+        }
+
+        return Ok(await _units.GetMineAsync(username));
+    }
+
+    [Authorize(Roles = "USER,SUPER")]
+    [HttpPost("train")]
+    public async Task<ActionResult<UnitTrainingResponse>> Train([FromBody] TrainUnitRequest request)
+    {
+        var username = User.FindFirstValue(ClaimTypes.Name);
+        if (string.IsNullOrWhiteSpace(username))
+        {
+            return Unauthorized();
+        }
+
+        try
+        {
+            var training = await _units.TrainAsync(username, request.UnitType);
+            if (training is null)
+            {
+                return BadRequest(new { error = "Not enough gold." });
+            }
+
+            return Ok(UnitTrainingResponse.FromTraining(training));
+        }
+        catch (ArgumentException ex)
+        {
+            return BadRequest(new { error = ex.Message });
+        }
+    }
+}