[PM-10560] Create notification database storage (#4688)

* Add new tables * Add stored procedures * Add core entities and models * Setup EF * Add repository interfaces * Add dapper repos * Add EF repos * Add order by * EF updates * PM-10560: Notifications repository matching requirements. * PM-10560: Notifications repository matching requirements. * PM-10560: Migration scripts * PM-10560: EF index optimizations * PM-10560: Cleanup * PM-10560: Priority in natural order, Repository, sql simplifications * PM-10560: Title column update * PM-10560: Incorrect EF migration removal * PM-10560: EF migrations * PM-10560: Added views, SP naming simplification * PM-10560: Notification entity Title update, EF migrations * PM-10560: Removing Notification_ReadByUserId * PM-10560: Notification ReadByUserIdAndStatus fix * PM-10560: Notification ReadByUserIdAndStatus fix to be in line with requirements and EF --------- Co-authored-by: Maciej Zieniuk <mzieniuk@bitwarden.com> Co-authored-by: Matt Bishop <mbishop@bitwarden.com>
2025-12-12 22:33:45 +00:00 · 2024-09-09 14:52:12 -05:00
parent 55bf815050
commit 4c0f8d54f3
39 changed files with 9983 additions and 0 deletions
--- a/src/Infrastructure.EntityFramework/NotificationCenter/Repositories/NotificationRepository.cs
+++ b/src/Infrastructure.EntityFramework/NotificationCenter/Repositories/NotificationRepository.cs
@@ -0,0 +1,92 @@
+#nullable enable
+using AutoMapper;
+using Bit.Core.NotificationCenter.Enums;
+using Bit.Core.NotificationCenter.Models.Filter;
+using Bit.Core.NotificationCenter.Repositories;
+using Bit.Infrastructure.EntityFramework.NotificationCenter.Models;
+using Bit.Infrastructure.EntityFramework.Repositories;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace Bit.Infrastructure.EntityFramework.NotificationCenter.Repositories;
+
+public class NotificationRepository : Repository<Core.NotificationCenter.Entities.Notification, Notification, Guid>,
+    INotificationRepository
+{
+    public NotificationRepository(IServiceScopeFactory serviceScopeFactory, IMapper mapper)
+        : base(serviceScopeFactory, mapper, context => context.Notifications)
+    {
+    }
+
+    public async Task<IEnumerable<Core.NotificationCenter.Entities.Notification>> GetByUserIdAsync(Guid userId,
+        ClientType clientType)
+    {
+        return await GetByUserIdAndStatusAsync(userId, clientType, new NotificationStatusFilter());
+    }
+
+    public async Task<IEnumerable<Core.NotificationCenter.Entities.Notification>> GetByUserIdAndStatusAsync(Guid userId,
+        ClientType clientType, NotificationStatusFilter? statusFilter)
+    {
+        await using var scope = ServiceScopeFactory.CreateAsyncScope();
+        var dbContext = GetDatabaseContext(scope);
+
+        var notificationQuery = BuildNotificationQuery(dbContext, userId, clientType);
+
+        if (statusFilter != null && (statusFilter.Read != null || statusFilter.Deleted != null))
+        {
+            notificationQuery = from n in notificationQuery
+                                join ns in dbContext.NotificationStatuses on n.Id equals ns.NotificationId
+                                where
+                                    ns.UserId == userId &&
+                                    (
+                                        statusFilter.Read == null ||
+                                        (statusFilter.Read == true ? ns.ReadDate != null : ns.ReadDate == null) ||
+                                        statusFilter.Deleted == null ||
+                                        (statusFilter.Deleted == true ? ns.DeletedDate != null : ns.DeletedDate == null)
+                                    )
+                                select n;
+        }
+
+        var notifications = await notificationQuery
+            .OrderByDescending(n => n.Priority)
+            .ThenByDescending(n => n.CreationDate)
+            .ToListAsync();
+
+        return Mapper.Map<List<Core.NotificationCenter.Entities.Notification>>(notifications);
+    }
+
+    private static IQueryable<Notification> BuildNotificationQuery(DatabaseContext dbContext, Guid userId,
+        ClientType clientType)
+    {
+        var clientTypes = new[] { ClientType.All };
+        if (clientType != ClientType.All)
+        {
+            clientTypes = [ClientType.All, clientType];
+        }
+
+        return from n in dbContext.Notifications
+               join ou in dbContext.OrganizationUsers.Where(ou => ou.UserId == userId)
+                   on n.OrganizationId equals ou.OrganizationId into grouping
+               from ou in grouping.DefaultIfEmpty()
+               where
+                   clientTypes.Contains(n.ClientType) &&
+                   (
+                       (
+                           n.Global &&
+                           n.UserId == null &&
+                           n.OrganizationId == null
+                       ) ||
+                       (
+                           !n.Global &&
+                           n.UserId == userId &&
+                           (n.OrganizationId == null || ou != null)
+                       ) ||
+                       (
+                           !n.Global &&
+                           n.UserId == null &&
+                           ou != null
+                       )
+                   )
+               select n;
+    }
+}