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 override async Task DeleteAsync(Core.Entities.EmergencyAccess emergencyAccess)
    {
        using (var scope = ServiceScopeFactory.CreateScope())
        {
            var dbContext = GetDatabaseContext(scope);
            await dbContext.UserBumpAccountRevisionDateByEmergencyAccessGranteeIdAsync(emergencyAccess.Id);
            await dbContext.SaveChangesAsync();
        }
        await base.DeleteAsync(emergencyAccess);
    }

    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;
        }
    }
}