Split out repositories to Infrastructure.Dapper / EntityFramework (#1759)

2025-07-02 00:22:50 -05:00 · 2022-01-11 10:40:51 +01:00
parent e2c6fc81f4
commit e4a10aae27
536 changed files with 1629 additions and 1589 deletions
--- a/src/Infrastructure.EntityFramework/Repositories/OrganizationRepository.cs
+++ b/src/Infrastructure.EntityFramework/Repositories/OrganizationRepository.cs
@ -0,0 +1,123 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using AutoMapper;
+using Bit.Core.Repositories;
+using Bit.Infrastructure.EntityFramework.Models;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.Extensions.DependencyInjection;
+using DataModel = Bit.Core.Models.Data;
+
+namespace Bit.Infrastructure.EntityFramework.Repositories
+{
+    public class OrganizationRepository : Repository<Core.Entities.Organization, Organization, Guid>, IOrganizationRepository
+    {
+        public OrganizationRepository(IServiceScopeFactory serviceScopeFactory, IMapper mapper)
+            : base(serviceScopeFactory, mapper, (DatabaseContext context) => context.Organizations)
+        { }
+
+        public async Task<Core.Entities.Organization> GetByIdentifierAsync(string identifier)
+        {
+            using (var scope = ServiceScopeFactory.CreateScope())
+            {
+                var dbContext = GetDatabaseContext(scope);
+                var organization = await GetDbSet(dbContext).Where(e => e.Identifier == identifier)
+                    .FirstOrDefaultAsync();
+                return organization;
+            }
+        }
+
+        public async Task<ICollection<Core.Entities.Organization>> GetManyByEnabledAsync()
+        {
+            using (var scope = ServiceScopeFactory.CreateScope())
+            {
+                var dbContext = GetDatabaseContext(scope);
+                var organizations = await GetDbSet(dbContext).Where(e => e.Enabled).ToListAsync();
+                return Mapper.Map<List<Core.Entities.Organization>>(organizations);
+            }
+        }
+
+        public async Task<ICollection<Core.Entities.Organization>> GetManyByUserIdAsync(Guid userId)
+        {
+            using (var scope = ServiceScopeFactory.CreateScope())
+            {
+                var dbContext = GetDatabaseContext(scope);
+                var organizations = await GetDbSet(dbContext)
+                    .Select(e => e.OrganizationUsers
+                        .Where(ou => ou.UserId == userId)
+                        .Select(ou => ou.Organization))
+                    .ToListAsync();
+                return Mapper.Map<List<Core.Entities.Organization>>(organizations);
+            }
+        }
+
+        public async Task<ICollection<Core.Entities.Organization>> SearchAsync(string name, string userEmail,
+            bool? paid, int skip, int take)
+        {
+            using (var scope = ServiceScopeFactory.CreateScope())
+            {
+                var dbContext = GetDatabaseContext(scope);
+                var organizations = await GetDbSet(dbContext)
+                    .Where(e => name == null || e.Name.Contains(name))
+                    .Where(e => userEmail == null || e.OrganizationUsers.Any(u => u.Email == userEmail))
+                    .Where(e => paid == null ||
+                            (paid == true && !string.IsNullOrWhiteSpace(e.GatewaySubscriptionId)) ||
+                            (paid == false && e.GatewaySubscriptionId == null))
+                    .OrderBy(e => e.CreationDate)
+                    .Skip(skip).Take(take)
+                    .ToListAsync();
+                return Mapper.Map<List<Core.Entities.Organization>>(organizations);
+            }
+        }
+
+        public async Task<ICollection<DataModel.OrganizationAbility>> GetManyAbilitiesAsync()
+        {
+            using (var scope = ServiceScopeFactory.CreateScope())
+            {
+                var dbContext = GetDatabaseContext(scope);
+                return await GetDbSet(dbContext)
+                .Select(e => new DataModel.OrganizationAbility
+                {
+                    Enabled = e.Enabled,
+                    Id = e.Id,
+                    Use2fa = e.Use2fa,
+                    UseEvents = e.UseEvents,
+                    UsersGetPremium = e.UsersGetPremium,
+                    Using2fa = e.Use2fa && e.TwoFactorProviders != null,
+                    UseSso = e.UseSso,
+                    UseKeyConnector = e.UseKeyConnector,
+                }).ToListAsync();
+            }
+        }
+
+        public async Task UpdateStorageAsync(Guid id)
+        {
+            await OrganizationUpdateStorage(id);
+        }
+
+        public override async Task DeleteAsync(Core.Entities.Organization organization)
+        {
+            using (var scope = ServiceScopeFactory.CreateScope())
+            {
+                var dbContext = GetDatabaseContext(scope);
+                var orgEntity = await dbContext.FindAsync<Organization>(organization.Id);
+                var sponsorships = dbContext.OrganizationSponsorships
+                    .Where(os =>
+                        os.SponsoringOrganizationId == organization.Id ||
+                        os.SponsoredOrganizationId == organization.Id);
+
+                Guid? UpdatedOrgId(Guid? orgId) => orgId == organization.Id ? null : organization.Id;
+                foreach (var sponsorship in sponsorships)
+                {
+                    sponsorship.SponsoredOrganizationId = UpdatedOrgId(sponsorship.SponsoredOrganizationId);
+                    sponsorship.SponsoringOrganizationId = UpdatedOrgId(sponsorship.SponsoringOrganizationId);
+                    sponsorship.FriendlyName = null;
+                }
+
+                dbContext.Remove(orgEntity);
+                await dbContext.SaveChangesAsync();
+            }
+        }
+    }
+}