bitwarden/src/Infrastructure.EntityFramework/Repositories/EmergencyAccessRepository.cs

using AutoMapper;
using Bit.Core.Enums;
using Bit.Core.Models.Data;
using Bit.Core.Repositories;
using Bit.Infrastructure.EntityFramework.Models;
using Bit.Infrastructure.EntityFramework.Repositories.Queries;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.DependencyInjection;

namespace Bit.Infrastructure.EntityFramework.Repositories
{
    public class EmergencyAccessRepository : Repository<Core.Entities.EmergencyAccess, EmergencyAccess, Guid>, IEmergencyAccessRepository
    {
        public EmergencyAccessRepository(IServiceScopeFactory serviceScopeFactory, IMapper mapper)
            : base(serviceScopeFactory, mapper, (DatabaseContext context) => context.EmergencyAccesses)
        { }

        public async Task<int> GetCountByGrantorIdEmailAsync(Guid grantorId, string email, bool onlyRegisteredUsers)
        {
            var query = new EmergencyAccessReadCountByGrantorIdEmailQuery(grantorId, email, onlyRegisteredUsers);
            return await GetCountFromQuery(query);
        }

        public async Task<EmergencyAccessDetails> GetDetailsByIdGrantorIdAsync(Guid id, Guid grantorId)
        {
            using (var scope = ServiceScopeFactory.CreateScope())
            {
                var dbContext = GetDatabaseContext(scope);
                var view = new EmergencyAccessDetailsViewQuery();
                var query = view.Run(dbContext).Where(ea =>
                    ea.Id == id &&
                    ea.GrantorId == grantorId
                );
                return await query.FirstOrDefaultAsync();
            }
        }

        public async Task<ICollection<EmergencyAccessDetails>> GetExpiredRecoveriesAsync()
        {
            using (var scope = ServiceScopeFactory.CreateScope())
            {
                var dbContext = GetDatabaseContext(scope);
                var view = new EmergencyAccessDetailsViewQuery();
                var query = view.Run(dbContext).Where(ea =>
                    ea.Status == EmergencyAccessStatusType.RecoveryInitiated
                );
                return await query.ToListAsync();
            }
        }

        public async Task<ICollection<EmergencyAccessDetails>> GetManyDetailsByGranteeIdAsync(Guid granteeId)
        {
            using (var scope = ServiceScopeFactory.CreateScope())
            {
                var dbContext = GetDatabaseContext(scope);
                var view = new EmergencyAccessDetailsViewQuery();
                var query = view.Run(dbContext).Where(ea =>
                    ea.GranteeId == granteeId
                );
                return await query.ToListAsync();
            }
        }

        public async Task<ICollection<EmergencyAccessDetails>> GetManyDetailsByGrantorIdAsync(Guid grantorId)
        {
            using (var scope = ServiceScopeFactory.CreateScope())
            {
                var dbContext = GetDatabaseContext(scope);
                var view = new EmergencyAccessDetailsViewQuery();
                var query = view.Run(dbContext).Where(ea =>
                    ea.GrantorId == grantorId
                );
                return await query.ToListAsync();
            }
        }

        public async Task<ICollection<EmergencyAccessNotify>> GetManyToNotifyAsync()
        {
            using (var scope = ServiceScopeFactory.CreateScope())
            {
                var dbContext = GetDatabaseContext(scope);
                var view = new EmergencyAccessDetailsViewQuery();
                var query = view.Run(dbContext).Where(ea =>
                    ea.Status == EmergencyAccessStatusType.RecoveryInitiated
                );
                var notifies = await query.Select(ea => new EmergencyAccessNotify
                {
                    Id = ea.Id,
                    GrantorId = ea.GrantorId,
                    GranteeId = ea.GranteeId,
                    Email = ea.Email,
                    KeyEncrypted = ea.KeyEncrypted,
                    Type = ea.Type,
                    Status = ea.Status,
                    WaitTimeDays = ea.WaitTimeDays,
                    RecoveryInitiatedDate = ea.RecoveryInitiatedDate,
                    LastNotificationDate = ea.LastNotificationDate,
                    CreationDate = ea.CreationDate,
                    RevisionDate = ea.RevisionDate,
                    GranteeName = ea.GranteeName,
                    GranteeEmail = ea.GranteeEmail,
                    GrantorEmail = ea.GrantorEmail,
                }).ToListAsync();
                return notifies;
            }
        }
    }
}