EnvelopeGenerator/EnvelopeGenerator.Infrastructure/Repositories/EnvlopeReceiverRepository.cs

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

namespace EnvelopeGenerator.Infrastructure.Repositories
{
    public class EnvelopeReceiverRepository : CRUDRepository<EnvelopeReceiver, (int Envelope, int Receiver), EGDbContext>, IEnvelopeReceiverRepository
    {
        public EnvelopeReceiverRepository(EGDbContext dbContext) : base(dbContext, dbContext.EnvelopeReceivers)
        {
        }

        private IQueryable<EnvelopeReceiver> ReadWhere(string? uuid = null, string? signature = null, bool withEnvelope = false, bool withReceiver = false, bool readOnly = false)
        {
            var query = readOnly ? _dbSet.AsNoTracking() : _dbSet;

            if (uuid is not null)
                query = query.Where(er => er.Envelope != null && er.Envelope.Uuid == uuid);

            if (signature is not null)
                query = query.Where(er => er.Receiver != null && er.Receiver.Signature == signature);

            if (withEnvelope)
                query = query
                    .Include(er => er.Envelope).ThenInclude(e => e!.Documents!).ThenInclude(d => d.Elements!.Where(e => signature == null || e.Receiver!.Signature == signature))
                    .Include(er => er.Envelope).ThenInclude(e => e!.History)
                    .Include(er => er.Envelope).ThenInclude(e => e!.User);

            if (withReceiver)
                query = query.Include(er => er.Receiver);
            return query;
        }

        public async Task<IEnumerable<EnvelopeReceiver>> ReadByUuidAsync(string uuid, bool withEnvelope = true, bool withReceiver = false, bool readOnly = false)
            => await ReadWhere(uuid: uuid, withEnvelope: withEnvelope, withReceiver: withReceiver, readOnly: readOnly).ToListAsync();

        public async Task<IEnumerable<EnvelopeReceiver>> ReadBySignatureAsync(string signature, bool withEnvelope = false, bool withReceiver = true, bool readOnly = false)
            => await ReadWhere(signature: signature, withEnvelope: withEnvelope, withReceiver: withReceiver, readOnly: readOnly).ToListAsync();

        public async Task<EnvelopeReceiver?> ReadByUuidSignatureAsync(string uuid, string signature, bool withEnvelope = true, bool withReceiver = true, bool readOnly = false)
            => await ReadWhere(uuid: uuid, signature: signature, withEnvelope: withEnvelope, withReceiver: withReceiver, readOnly: readOnly).FirstOrDefaultAsync();

        public async Task<string?> ReadAccessCodeAsync(string uuid, string signature, bool readOnly = false)
            => await ReadWhere(uuid: uuid, signature: signature, readOnly: readOnly)
                .Select(er => er.AccessCode)
                .FirstOrDefaultAsync();

        public async Task<int> CountAsync(string uuid, string signature) => await ReadWhere(uuid: uuid, signature: signature).CountAsync();

        private IQueryable<EnvelopeReceiver> ReadById(int envelopeId, int receiverId, bool readOnly = false)
        {
            var query = readOnly ? _dbSet.AsNoTracking() : _dbSet;
            return query.Where(er => er.EnvelopeId == envelopeId && er.ReceiverId == receiverId);
        }

        public async Task<EnvelopeReceiver?> ReadByIdAsync(int envelopeId, int receiverId, bool readOnly = false)
            => await ReadById(envelopeId: envelopeId, receiverId: receiverId, readOnly: readOnly)
            .FirstOrDefaultAsync();

        public async Task<string?> ReadAccessCodeByIdAsync(int envelopeId, int receiverId, bool readOnly = false)
            => await ReadById(envelopeId: envelopeId, receiverId: receiverId, readOnly: readOnly)
            .Select(er => er.AccessCode)
            .FirstOrDefaultAsync();

        public async Task<IEnumerable<EnvelopeReceiver>> ReadByUsernameAsync(string username, int? min_status = null, int? max_status = null, params int[] ignore_statuses)
        {
            var query = _dbSet.AsNoTracking().Where(er => er.Envelope!.User!.Username == username);

            if (min_status is not null)
                query = query.Where(er => er.Envelope!.Status >= min_status);

            if (max_status is not null)
                query = query.Where(er => er.Envelope!.Status <= max_status);

            foreach (var ignore_status in ignore_statuses)
                query = query.Where(er => er.Envelope!.Status != ignore_status);

            return await query.Include(er => er.Envelope).Include(er => er.Receiver).ToListAsync();
        }

        public async Task<EnvelopeReceiver?> ReadLastByReceiver(string email)
        {
            return await _dbSet.Where(er => er.Receiver!.EmailAddress == email).OrderBy(er => er.EnvelopeId).LastOrDefaultAsync();
        }      
    }
}