EnvelopeGenerator/EnvelopeGenerator.Infrastructure/Repositories/EnvelopeHistoryRepository.cs

using DigitalData.Core.Infrastructure;
using EnvelopeGenerator.Domain.Entities;
using EnvelopeGenerator.Infrastructure.Contracts;
using Microsoft.EntityFrameworkCore;

namespace EnvelopeGenerator.Infrastructure.Repositories
{
    public class EnvelopeHistoryRepository : CRUDRepository<EnvelopeHistory, long, EGDbContext>, IEnvelopeHistoryRepository
    {
        public EnvelopeHistoryRepository(EGDbContext dbContext) : base(dbContext, dbContext.EnvelopeHistories)
        {
        }

        private IQueryable<EnvelopeHistory> By(int? envelopeId = null, string? userReference = null, int? status = null, bool withSender = false, bool withReceiver = false)
        {
            var query = _dbSet.AsQueryable();

            if (envelopeId is not null)
                query = query.Where(eh => eh.EnvelopeId == envelopeId);

            if (userReference is not null)
                query = query.Where(eh => eh.UserReference == userReference);

            if (status is not null)
                query = query.Where(eh => eh.Status == status);

            if (withSender)
                query = query.Include(eh => eh.Sender);

            if (withReceiver)
                query = query.Include(eh => eh.Receiver);

            return query;
        }

        public async Task<int> CountAsync(int? envelopeId = null, string? userReference = null, int? status = null) => await By(
            envelopeId: envelopeId,
            userReference: userReference,
            status: status).CountAsync();

        public async Task<IEnumerable<EnvelopeHistory>> ReadAsync(int? envelopeId = null, string? userReference = null, int? status = null, bool withSender = false, bool withReceiver = false)
            => await By(
            envelopeId: envelopeId,
            userReference: userReference,
            status: status,
            withSender: withSender,
            withReceiver: withReceiver).ToListAsync();
    }
}