ConnectionsAPI/Database/Repository/PuzzleRepository.cs

using ConnectionsAPI.Database.Entities;
using Microsoft.EntityFrameworkCore;

namespace ConnectionsAPI.Database.Repository
{
    public class PuzzleRepository(ConnectionsContext _db)
    {
        private readonly ConnectionsContext _db = _db;

        public async Task<CategoriesPuzzle?> GetPuzzleByDateAsync(string printDate, bool includeSolutions = true)
        {
            // query for the puzzle
            var query = _db.CategoriesPuzzles.AsNoTracking();

            if (includeSolutions)
            {
                query = query
                    .Include(x => x.Categories)
                    .ThenInclude(x => x.CategoriesPuzzleCards);
            }

            var puzzle = await query.FirstOrDefaultAsync(x => x.PrintDate == printDate);

            // if not found, we're done here
            if (puzzle == null)
            {
                return null;
            }

            await EnhancePuzzleWithDatesAsync(puzzle);

            return puzzle;
        }

        public async Task<IEnumerable<CategoriesPuzzle>> GetAllPuzzlesAsync(bool includeSolutions = true)
        {
            // query all, ordered by print date
            var query = _db.CategoriesPuzzles
                .AsNoTracking();

            if (includeSolutions)
            {
                query = query
                    .Include(x => x.Categories)
                    .ThenInclude(x => x.CategoriesPuzzleCards);
            }

            var result = (await query.OrderBy(x => x.PrintDate).ToListAsync()) ?? [];

            foreach (var puzzle in result)
            {
                await EnhancePuzzleWithDatesAsync(puzzle);
            }

            return result;
        }

        private async Task EnhancePuzzleWithDatesAsync(CategoriesPuzzle puzzle)
        {
            string? previousPuzzleDate = await _db.CategoriesPuzzles.AsNoTracking()
                .Where(x => x.PrintDate.CompareTo(puzzle.PrintDate) < 0)
                .OrderByDescending(x => x.PrintDate)
                .Select(x => x.PrintDate)
                .FirstOrDefaultAsync();

            string? nextPuzzleDate = await _db.CategoriesPuzzles.AsNoTracking()
                .Where(x => x.PrintDate.CompareTo(puzzle.PrintDate) > 0)
                .OrderBy(x => x.PrintDate)
                .Select(x => x.PrintDate)
                .FirstOrDefaultAsync();

            puzzle.PrevPrintDate = previousPuzzleDate;
            puzzle.NextPrintDate = nextPuzzleDate;
        }
    }
}