using SurveyLib.Core.Models;
using SurveyLib.Core.Repositories;
using Microsoft.EntityFrameworkCore;
using SurveyLib.Infrastructure.EFCore.Data;

namespace SurveyLib.Infrastructure.EFCore.Repositories;

public class AnswerRepository : IAnswerRepository
{
    private readonly SurveyDbContext _context;

    public AnswerRepository(SurveyDbContext context)
    {
        _context = context;
    }

    public async Task<Answer?> GetByIdAsync(int id)
    {
        throw new NotImplementedException();
    }

    public async Task<IEnumerable<Answer>> GetAllAsync()
    {
        return await _context.Answers.ToListAsync();
    }

    public async Task AddAsync(Answer entity)
    {
        await _context.Answers.AddAsync(entity);
        await _context.SaveChangesAsync();
    }

    public async Task UpdateAsync(Answer entity)
    {
        _context.Answers.Update(entity);
        await _context.SaveChangesAsync();
    }

    public async Task DeleteAsync(Answer entity)
    {
        _context.Answers.Remove(entity);
        await _context.SaveChangesAsync();
    }

    public async Task<IEnumerable<Answer>> GetByAttemptIdAsync(int attemptId)
    {
        return await _context.Answers.Where(a => a.CompletionId == attemptId).ToListAsync();
    }

    public async Task<IEnumerable<Answer>> GetByQuestionIdAsync(int questionId)
    {
        return await _context.Answers.Where(a => a.QuestionId == questionId).ToListAsync();
    }
}