1 files changed, 8 insertions, 0 deletions
diff --git a/app/models/comment.rb b/app/models/comment.rb
index 79eb59748..1716537af 100644
--- a/app/models/comment.rb
+++ b/app/models/comment.rb
@@ -19,6 +19,8 @@
 
 class Comment < ApplicationRecord
   include Redmine::SafeAttributes
+  include Redmine::Reaction::Reactable
+
   belongs_to :commented, :polymorphic => true, :counter_cache => true
   belongs_to :author, :class_name => 'User'
 
@@ -28,6 +30,8 @@ class Comment < ApplicationRecord
 
   safe_attributes 'comments'
 
+  delegate :visible?, to: :commented
+
   def comments=(arg)
     self.content = arg
   end
@@ -36,6 +40,10 @@ class Comment < ApplicationRecord
     content
   end
 
+  def project
+    commented.respond_to?(:project) ? commented.project : nil
+  end
+
   private
 
   def send_notification