territory claiming

2026-05-18 15:40:15 -05:00
parent 37a46a7b36
commit 3167558e04
5 changed files with 127 additions and 4 deletions
@@ -78,6 +78,33 @@ public class UnitTrainingService
        return training;
    }

+    public async Task<bool> TryConsumeCompletedScoutAsync(string username)
+    {
+        await CompleteFinishedTrainingAsync(username);
+
+        var now = DateTime.UtcNow;
+        var update = Builders<UnitTraining>.Update
+            .Set(x => x.Status, "Consumed")
+            .Set(x => x.UpdatedAtUtc, now);
+
+        var consumed = await _training.FindOneAndUpdateAsync(
+            x => x.Username == username && x.UnitType == ScoutUnitType && x.Status == "Completed",
+            update,
+            new FindOneAndUpdateOptions<UnitTraining>
+            {
+                Sort = Builders<UnitTraining>.Sort.Ascending(x => x.CompletedAtUtc).Ascending(x => x.CreatedAtUtc),
+                ReturnDocument = ReturnDocument.After
+            });
+
+        return consumed is not null;
+    }
+
+    public async Task<bool> HasCompletedScoutAsync(string username)
+    {
+        await CompleteFinishedTrainingAsync(username);
+        return await _training.Find(x => x.Username == username && x.UnitType == ScoutUnitType && x.Status == "Completed").AnyAsync();
+    }
+
    private async Task CompleteFinishedTrainingAsync(string username)
    {
        var now = DateTime.UtcNow;