using ChapterX.Domain.Entities;
using ChapterX.Domain.Repositories;
using ChapterX.Infrastructure.Data.DataContext;
using Microsoft.EntityFrameworkCore;

namespace ChapterX.Infrastructure.Repositories
{
    public class ReadingListRepository : GenericRepository<ReadingList>, IReadingListRepository
    {
        public ReadingListRepository(ApplicationDbContext context) : base(context)
        {
        }

        public override async Task<IEnumerable<ReadingList>> GetAllAsync(CancellationToken cancellationToken = default)
        {
            return await _dbSet
                .Include(r => r.User)
                .Include(r => r.ReadingListItems)
                    .ThenInclude(i => i.Story)
                        .ThenInclude(s => s.Writer)
                            .ThenInclude(w => w!.User)
                .Include(r => r.ReadingListItems)
                    .ThenInclude(i => i.Story)
                        .ThenInclude(s => s.HasGenres)
                            .ThenInclude(hg => hg.Genre)
                .ToListAsync(cancellationToken);
        }

        public async Task<IEnumerable<ReadingList>> GetByUserIdAsync(int userId, CancellationToken cancellationToken = default)
        {
            return await _dbSet
                .Where(r => r.UserId == userId)
                .Include(r => r.User)
                .Include(r => r.ReadingListItems)
                    .ThenInclude(i => i.Story)
                        .ThenInclude(s => s.Writer)
                            .ThenInclude(w => w!.User)
                .Include(r => r.ReadingListItems)
                    .ThenInclude(i => i.Story)
                        .ThenInclude(s => s.HasGenres)
                            .ThenInclude(hg => hg.Genre)
                .ToListAsync(cancellationToken);
        }
    }
}