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

src/Infrastructure.EntityFramework/Repositories/DeviceRepository.cs

@@ -0,0 +1,75 @@
+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;
+
+namespace Bit.Infrastructure.EntityFramework.Repositories
+{
+    public class DeviceRepository : Repository<Core.Entities.Device, Device, Guid>, IDeviceRepository
+    {
+        public DeviceRepository(IServiceScopeFactory serviceScopeFactory, IMapper mapper)
+            : base(serviceScopeFactory, mapper, (DatabaseContext context) => context.Devices)
+        { }
+
+        public async Task ClearPushTokenAsync(Guid id)
+        {
+            using (var scope = ServiceScopeFactory.CreateScope())
+            {
+                var dbContext = GetDatabaseContext(scope);
+                var query = dbContext.Devices.Where(d => d.Id == id);
+                dbContext.AttachRange(query);
+                await query.ForEachAsync(x => x.PushToken = null);
+                await dbContext.SaveChangesAsync();
+            }
+        }
+
+        public async Task<Core.Entities.Device> GetByIdAsync(Guid id, Guid userId)
+        {
+            var device = await base.GetByIdAsync(id);
+            if (device == null || device.UserId != userId)
+            {
+                return null;
+            }
+
+            return Mapper.Map<Core.Entities.Device>(device);
+        }
+
+        public async Task<Core.Entities.Device> GetByIdentifierAsync(string identifier)
+        {
+            using (var scope = ServiceScopeFactory.CreateScope())
+            {
+                var dbContext = GetDatabaseContext(scope);
+                var query = dbContext.Devices.Where(d => d.Identifier == identifier);
+                var device = await query.FirstOrDefaultAsync();
+                return Mapper.Map<Core.Entities.Device>(device);
+            }
+        }
+
+        public async Task<Core.Entities.Device> GetByIdentifierAsync(string identifier, Guid userId)
+        {
+            using (var scope = ServiceScopeFactory.CreateScope())
+            {
+                var dbContext = GetDatabaseContext(scope);
+                var query = dbContext.Devices.Where(d => d.Identifier == identifier && d.UserId == userId);
+                var device = await query.FirstOrDefaultAsync();
+                return Mapper.Map<Core.Entities.Device>(device);
+            }
+        }
+
+        public async Task<ICollection<Core.Entities.Device>> GetManyByUserIdAsync(Guid userId)
+        {
+            using (var scope = ServiceScopeFactory.CreateScope())
+            {
+                var dbContext = GetDatabaseContext(scope);
+                var query = dbContext.Devices.Where(d => d.UserId == userId);
+                var devices = await query.ToListAsync();
+                return Mapper.Map<List<Core.Entities.Device>>(devices);
+            }
+        }
+    }
+}