#21: Implemented auto-follow ticket when creating a reply.

sfmskywalker · Apr 11, 2014 · bce3eab · bce3eab
1 parent 93a2421
commit bce3eab
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/src/Orchard.Web/Modules/OrchardPros/Handlers/ReplyPartHandler.cs b/src/Orchard.Web/Modules/OrchardPros/Handlers/ReplyPartHandler.cs
@@ -7,6 +7,7 @@
 using Orchard.JobsQueue.Services;
 using Orchard.Localization;
 using OrchardPros.Models;
+using OrchardPros.Services.Content;
 using OrchardPros.Services.User;
 
 namespace OrchardPros.Handlers {
@@ -15,15 +16,24 @@ public class ReplyPartHandler : ContentHandler {
         private readonly IShapeFactory _shapeFactory;
         private readonly IShapeDisplay _shapeDisplay;
         private readonly IJobsQueueService _jobsQueueService;
+        private readonly ISubscriptionService _subscriptionService;
+
+        public ReplyPartHandler(
+            IUserManager userManager, 
+            IShapeFactory shapeFactory, 
+            IShapeDisplay shapeDisplay, 
+            IJobsQueueService jobsQueueService, 
+            ISubscriptionService subscriptionService) {
 
-        public ReplyPartHandler(IUserManager userManager, IShapeFactory shapeFactory, IShapeDisplay shapeDisplay, IJobsQueueService jobsQueueService) {
             _userManager = userManager;
             _shapeFactory = shapeFactory;
             _shapeDisplay = shapeDisplay;
             _jobsQueueService = jobsQueueService;
+            _subscriptionService = subscriptionService;
             T = NullLocalizer.Instance;
 
             OnCreated<ReplyPart>(UpdateStats);
+            OnCreated<ReplyPart>(Subscribe);
             OnCreated<ReplyPart>(SendNotifications);
         }
 
@@ -37,6 +47,16 @@ public class ReplyPartHandler : ContentHandler {
             _userManager.AddActivityPoints(user, 5);
         }
 
+        private void Subscribe(CreateContentContext context, ReplyPart part)
+        {
+            var user = part.User;
+            var ticket = part.ContainingContent.As<TicketPart>();
+            var subscriptionSource = ticket.As<SubscriptionSourcePart>();
+
+            if(!_subscriptionService.HasSubscription(subscriptionSource, user))
+                _subscriptionService.Subscribe(subscriptionSource, user);
+        }
+
         private void SendNotifications(CreateContentContext context, ReplyPart part) {
             var ticket = part.ContainingContent.As<TicketPart>();
             var subscriptionSource = ticket.As<SubscriptionSourcePart>();

diff --git a/src/Orchard.Web/Modules/OrchardPros/Services/Content/ISubscriptionService.cs b/src/Orchard.Web/Modules/OrchardPros/Services/Content/ISubscriptionService.cs
@@ -9,5 +9,6 @@ public interface ISubscriptionService : IDependency {
         Subscription Subscribe(SubscriptionSourcePart subscriptionSource, IUser user);
         void Unsubscribe(SubscriptionSourcePart subscriptionSource, IUser user);
         IPagedList<SubscriptionSourcePart> GetSubscriptionSourcesByUser(int userId, int? skip = null, int? take = null);
+        bool HasSubscription(SubscriptionSourcePart subscriptionSource, IUser user);
     }
 }
diff --git a/src/Orchard.Web/Modules/OrchardPros/Services/Content/SubscriptionService.cs b/src/Orchard.Web/Modules/OrchardPros/Services/Content/SubscriptionService.cs
@@ -60,5 +60,9 @@ public class SubscriptionService : ISubscriptionService {
             var ids = query.ToArray();
             return _contentManager.GetMany<SubscriptionSourcePart>(ids, VersionOptions.Published, QueryHints.Empty).ToPagedList(count);
         }
+
+        public bool HasSubscription(SubscriptionSourcePart subscriptionSource, IUser user) {
+            return subscriptionSource.Subscribers.Any(x => x.Id == user.Id);
+        }
     }
 }