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

2025-07-04 01: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/Queries/CollectionUserUpdateUsersQuery.cs
+++ b/src/Infrastructure.EntityFramework/Repositories/Queries/CollectionUserUpdateUsersQuery.cs
@ -0,0 +1,121 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using Bit.Core.Models.Data;
+using Bit.Infrastructure.EntityFramework.Models;
+using Microsoft.EntityFrameworkCore;
+
+namespace Bit.Infrastructure.EntityFramework.Repositories.Queries
+{
+    public class CollectionUserUpdateUsersQuery
+    {
+        public readonly CollectionUserUpdateUsersInsertQuery Insert;
+        public readonly CollectionUserUpdateUsersUpdateQuery Update;
+        public readonly CollectionUserUpdateUsersDeleteQuery Delete;
+
+        public CollectionUserUpdateUsersQuery(Guid collectionId, IEnumerable<SelectionReadOnly> users)
+        {
+            Insert = new CollectionUserUpdateUsersInsertQuery(collectionId, users);
+            Update = new CollectionUserUpdateUsersUpdateQuery(collectionId, users);
+            Delete = new CollectionUserUpdateUsersDeleteQuery(collectionId, users);
+        }
+    }
+
+    public class CollectionUserUpdateUsersInsertQuery : IQuery<OrganizationUser>
+    {
+        private readonly Guid _collectionId;
+        private readonly IEnumerable<SelectionReadOnly> _users;
+
+        public CollectionUserUpdateUsersInsertQuery(Guid collectionId, IEnumerable<SelectionReadOnly> users)
+        {
+            _collectionId = collectionId;
+            _users = users;
+        }
+
+        public IQueryable<OrganizationUser> Run(DatabaseContext dbContext)
+        {
+            var orgId = dbContext.Collections.FirstOrDefault(c => c.Id == _collectionId)?.OrganizationId;
+            var organizationUserIds = _users.Select(u => u.Id);
+            var insertQuery = from ou in dbContext.OrganizationUsers
+                              where
+                                  organizationUserIds.Contains(ou.Id) &&
+                                  ou.OrganizationId == orgId &&
+                                  !dbContext.CollectionUsers.Any(
+                                      x => x.CollectionId != _collectionId && x.OrganizationUserId == ou.Id)
+                              select ou;
+            return insertQuery;
+        }
+
+        public async Task<IEnumerable<CollectionUser>> BuildInMemory(DatabaseContext dbContext)
+        {
+            var data = await Run(dbContext).ToListAsync();
+            var collectionUsers = data.Select(x => new CollectionUser()
+            {
+                CollectionId = _collectionId,
+                OrganizationUserId = x.Id,
+                ReadOnly = _users.FirstOrDefault(u => u.Id.Equals(x.Id)).ReadOnly,
+                HidePasswords = _users.FirstOrDefault(u => u.Id.Equals(x.Id)).HidePasswords,
+            });
+            return collectionUsers;
+        }
+    }
+
+    public class CollectionUserUpdateUsersUpdateQuery : IQuery<CollectionUser>
+    {
+        private readonly Guid _collectionId;
+        private readonly IEnumerable<SelectionReadOnly> _users;
+
+        public CollectionUserUpdateUsersUpdateQuery(Guid collectionId, IEnumerable<SelectionReadOnly> users)
+        {
+            _collectionId = collectionId;
+            _users = users;
+        }
+
+        public IQueryable<CollectionUser> Run(DatabaseContext dbContext)
+        {
+            var orgId = dbContext.Collections.FirstOrDefault(c => c.Id == _collectionId)?.OrganizationId;
+            var ids = _users.Select(x => x.Id);
+            var updateQuery = from target in dbContext.CollectionUsers
+                              where target.CollectionId == _collectionId &&
+                                  ids.Contains(target.OrganizationUserId)
+                              select target;
+            return updateQuery;
+        }
+
+        public async Task<IEnumerable<CollectionUser>> BuildInMemory(DatabaseContext dbContext)
+        {
+            var data = await Run(dbContext).ToListAsync();
+            var collectionUsers = data.Select(x => new CollectionUser
+            {
+                CollectionId = _collectionId,
+                OrganizationUserId = x.OrganizationUserId,
+                ReadOnly = _users.FirstOrDefault(u => u.Id.Equals(x.OrganizationUserId)).ReadOnly,
+                HidePasswords = _users.FirstOrDefault(u => u.Id.Equals(x.OrganizationUserId)).HidePasswords,
+            });
+            return collectionUsers;
+        }
+    }
+
+    public class CollectionUserUpdateUsersDeleteQuery : IQuery<CollectionUser>
+    {
+        private readonly Guid _collectionId;
+        private readonly IEnumerable<SelectionReadOnly> _users;
+
+        public CollectionUserUpdateUsersDeleteQuery(Guid collectionId, IEnumerable<SelectionReadOnly> users)
+        {
+            _collectionId = collectionId;
+            _users = users;
+        }
+
+        public IQueryable<CollectionUser> Run(DatabaseContext dbContext)
+        {
+            var orgId = dbContext.Collections.FirstOrDefault(c => c.Id == _collectionId)?.OrganizationId;
+            var deleteQuery = from cu in dbContext.CollectionUsers
+                              where !dbContext.Users.Any(
+                                  u => u.Id == cu.OrganizationUserId)
+                              select cu;
+            return deleteQuery;
+        }
+    }
+}