AB4227 - Moved Persistence Repository implementations

2019-12-17 16:30:26 +01:00
parent 03631a1731
commit a22824e33a
77 changed files with 111 additions and 116 deletions
--- a/src/Umbraco.Infrastructure/Persistence/Repositories/Implement/NotificationsRepository.cs
+++ b/src/Umbraco.Infrastructure/Persistence/Repositories/Implement/NotificationsRepository.cs
@@ -0,0 +1,113 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using Umbraco.Core.Models;
+using Umbraco.Core.Models.Entities;
+using Umbraco.Core.Models.Membership;
+using Umbraco.Core.Persistence.Dtos;
+using Umbraco.Core.Scoping;
+
+namespace Umbraco.Core.Persistence.Repositories.Implement
+{
+    public class NotificationsRepository : INotificationsRepository
+    {
+        private readonly IScopeAccessor _scopeAccessor;
+
+        public NotificationsRepository(IScopeAccessor scopeAccessor)
+        {
+            _scopeAccessor = scopeAccessor;
+        }
+
+        private IScope AmbientScope => _scopeAccessor.AmbientScope;
+
+        public IEnumerable<Notification> GetUsersNotifications(IEnumerable<int> userIds, string action, IEnumerable<int> nodeIds, Guid objectType)
+        {
+            var nodeIdsA = nodeIds.ToArray();
+            var sql = AmbientScope.SqlContext.Sql()
+                    .Select("DISTINCT umbracoNode.id nodeId, umbracoUser.id userId, umbracoNode.nodeObjectType, umbracoUser2NodeNotify.action")
+                    .From<User2NodeNotifyDto>()
+                    .InnerJoin<NodeDto>().On<User2NodeNotifyDto, NodeDto>(left => left.NodeId, right => right.NodeId)
+                    .InnerJoin<UserDto>().On<User2NodeNotifyDto, UserDto>(left => left.UserId, right => right.Id)
+                    .Where<NodeDto>(x => x.NodeObjectType == objectType)
+                    .Where<UserDto>(x => x.Disabled == false) // only approved users
+                    .Where<User2NodeNotifyDto>(x => x.Action == action); // on the specified action
+            if (nodeIdsA.Length > 0)
+                sql
+                    .WhereIn<NodeDto>(x => x.NodeId, nodeIdsA); // for the specified nodes
+            sql
+                .OrderBy<UserDto>(x => x.Id)
+                .OrderBy<NodeDto>(dto => dto.NodeId);
+            return AmbientScope.Database.Fetch<dynamic>(sql).Select(x => new Notification(x.nodeId, x.userId, x.action, objectType));
+        }
+
+        public IEnumerable<Notification> GetUserNotifications(IUser user)
+        {
+            var sql = AmbientScope.SqlContext.Sql()
+                .Select("DISTINCT umbracoNode.id, umbracoUser2NodeNotify.userId, umbracoNode.nodeObjectType, umbracoUser2NodeNotify.action")
+                .From<User2NodeNotifyDto>()
+                .InnerJoin<NodeDto>()
+                .On<User2NodeNotifyDto, NodeDto>(dto => dto.NodeId, dto => dto.NodeId)
+                .Where<User2NodeNotifyDto>(dto => dto.UserId == (int)user.Id)
+                .OrderBy<NodeDto>(dto => dto.NodeId);
+
+            var dtos = AmbientScope.Database.Fetch<dynamic>(sql);
+            //need to map the results
+            return dtos.Select(d => new Notification(d.id, d.userId, d.action, d.nodeObjectType)).ToList();
+        }
+
+        public IEnumerable<Notification> SetNotifications(IUser user, IEntity entity, string[] actions)
+        {
+            DeleteNotifications(user, entity);
+            return actions.Select(action => CreateNotification(user, entity, action)).ToList();
+        }
+
+        public IEnumerable<Notification> GetEntityNotifications(IEntity entity)
+        {
+            var sql = AmbientScope.SqlContext.Sql()
+                .Select("DISTINCT umbracoNode.id, umbracoUser2NodeNotify.userId, umbracoNode.nodeObjectType, umbracoUser2NodeNotify.action")
+                .From<User2NodeNotifyDto>()
+                .InnerJoin<NodeDto>()
+                .On<User2NodeNotifyDto, NodeDto>(dto => dto.NodeId, dto => dto.NodeId)
+                .Where<User2NodeNotifyDto>(dto => dto.NodeId == entity.Id)
+                .OrderBy<NodeDto>(dto => dto.NodeId);
+
+            var dtos = AmbientScope.Database.Fetch<dynamic>(sql);
+            //need to map the results
+            return dtos.Select(d => new Notification(d.id, d.userId, d.action, d.nodeObjectType)).ToList();
+        }
+
+        public int DeleteNotifications(IEntity entity)
+        {
+            return AmbientScope.Database.Delete<User2NodeNotifyDto>("WHERE nodeId = @nodeId", new { nodeId = entity.Id });
+        }
+
+        public int DeleteNotifications(IUser user)
+        {
+            return AmbientScope.Database.Delete<User2NodeNotifyDto>("WHERE userId = @userId", new { userId = user.Id });
+        }
+
+        public int DeleteNotifications(IUser user, IEntity entity)
+        {
+            // delete all settings on the node for this user
+            return AmbientScope.Database.Delete<User2NodeNotifyDto>("WHERE userId = @userId AND nodeId = @nodeId", new { userId = user.Id, nodeId = entity.Id });
+        }
+
+        public Notification CreateNotification(IUser user, IEntity entity, string action)
+        {
+            var sql = AmbientScope.SqlContext.Sql()
+                .Select("DISTINCT nodeObjectType")
+                .From<NodeDto>()
+                .Where<NodeDto>(nodeDto => nodeDto.NodeId == entity.Id);
+            var nodeType = AmbientScope.Database.ExecuteScalar<Guid>(sql);
+
+            var dto = new User2NodeNotifyDto
+                {
+                    Action = action,
+                    NodeId = entity.Id,
+                    UserId = user.Id
+                };
+            AmbientScope.Database.Insert(dto);
+            return new Notification(dto.NodeId, dto.UserId, dto.Action, nodeType);
+        }
+    }
+}