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

src/Infrastructure.EntityFramework/Repositories/Queries/CipherUpdateCollectionsQuery.cs

@@ -0,0 +1,69 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using Bit.Core.Entities;
+using Bit.Core.Enums;
+using CollectionCipher = Bit.Infrastructure.EntityFramework.Models.CollectionCipher;
+
+namespace Bit.Infrastructure.EntityFramework.Repositories.Queries
+{
+    public class CipherUpdateCollectionsQuery : IQuery<CollectionCipher>
+    {
+        private readonly Cipher _cipher;
+        private readonly IEnumerable<Guid> _collectionIds;
+
+        public CipherUpdateCollectionsQuery(Cipher cipher, IEnumerable<Guid> collectionIds)
+        {
+            _cipher = cipher;
+            _collectionIds = collectionIds;
+        }
+
+        public virtual IQueryable<CollectionCipher> Run(DatabaseContext dbContext)
+        {
+            if (!_cipher.OrganizationId.HasValue || !_collectionIds.Any())
+            {
+                return null;
+            }
+
+            var availibleCollections = !_cipher.UserId.HasValue ?
+                from c in dbContext.Collections
+                where c.OrganizationId == _cipher.OrganizationId
+                select c.Id :
+                from c in dbContext.Collections
+                join o in dbContext.Organizations
+                    on c.OrganizationId equals o.Id
+                join ou in dbContext.OrganizationUsers
+                    on o.Id equals ou.OrganizationId
+                where ou.UserId == _cipher.UserId
+                join cu in dbContext.CollectionUsers
+                    on c.Id equals cu.CollectionId into cu_g
+                from cu in cu_g.DefaultIfEmpty()
+                where !ou.AccessAll && cu.OrganizationUserId == ou.Id
+                join gu in dbContext.GroupUsers
+                    on ou.Id equals gu.OrganizationUserId into gu_g
+                from gu in gu_g.DefaultIfEmpty()
+                where cu.CollectionId == null && !ou.AccessAll
+                join g in dbContext.Groups
+                    on gu.GroupId equals g.Id into g_g
+                from g in g_g.DefaultIfEmpty()
+                join cg in dbContext.CollectionGroups
+                    on c.Id equals cg.CollectionId into cg_g
+                from cg in cg_g.DefaultIfEmpty()
+                where !g.AccessAll && gu.GroupId == cg.GroupId &&
+                    o.Id == _cipher.OrganizationId &&
+                    o.Enabled &&
+                    ou.Status == OrganizationUserStatusType.Confirmed &&
+                    (ou.AccessAll || !cu.ReadOnly || g.AccessAll || !cg.ReadOnly)
+                select c.Id;
+
+            if (!availibleCollections.Any())
+            {
+                return null;
+            }
+
+            var query = from c in availibleCollections
+                        select new CollectionCipher { CollectionId = c, CipherId = _cipher.Id };
+            return query;
+        }
+    }
+}