Add linked suggestions with synced voting

Endpoints/VoteEndpoints.cs

@@ -41,28 +41,40 @@ public static class VoteEndpoints
             if (string.IsNullOrWhiteSpace(player.DisplayName))
                 return Results.BadRequest(new { error = "Set a display name before voting." });
 
-            var suggestionExists = await db.Suggestions.AnyAsync(s => s.Id == request.SuggestionId);
-            if (!suggestionExists)
+            var linkMap = await db.Suggestions.AsNoTracking()
+                .Select(s => new { s.Id, s.ParentSuggestionId })
+                .ToListAsync();
+            var rootIndex = EndpointHelpers.BuildLinkRoots(linkMap.Select(s => (s.Id, s.ParentSuggestionId)));
+            if (!rootIndex.ContainsKey(request.SuggestionId))
                 return Results.BadRequest(new { error = "Suggestion not found." });
+            var linkedIds = EndpointHelpers.LinkedIdsFor(request.SuggestionId, rootIndex);
+            if (linkedIds.Count == 0)
+                linkedIds.Add(request.SuggestionId);
 
-            var vote = await db.Votes.FirstOrDefaultAsync(v => v.PlayerId == player.Id && v.SuggestionId == request.SuggestionId);
-            if (vote == null)
+            var existingVotes = await db.Votes
+                .Where(v => v.PlayerId == player.Id && linkedIds.Contains(v.SuggestionId))
+                .ToListAsync();
+
+            foreach (var suggestionId in linkedIds)
             {
-                vote = new Vote
+                var vote = existingVotes.FirstOrDefault(v => v.SuggestionId == suggestionId);
+                if (vote == null)
                 {
-                    PlayerId = player.Id,
-                    SuggestionId = request.SuggestionId,
-                    Score = request.Score
-                };
-                db.Votes.Add(vote);
-            }
-            else
-            {
-                vote.Score = request.Score;
+                    db.Votes.Add(new Vote
+                    {
+                        PlayerId = player.Id,
+                        SuggestionId = suggestionId,
+                        Score = request.Score
+                    });
+                }
+                else
+                {
+                    vote.Score = request.Score;
+                }
             }
 
             await db.SaveChangesAsync();
-            return Results.Ok(new { vote.Id, vote.Score });
+            return Results.Ok(new { SuggestionIds = linkedIds, request.Score });
         });
 
         app.MapPost("/api/votes/finalize", async ([FromBody] VoteFinalizeRequest request, HttpContext ctx, AppDbContext db) =>