Mark spammers comments when banned

seuros · seuros · commit b27fbb8e2121 · 2015-08-23T17:17:19.000+01:00
diff --git a/app/models/protip.rb b/app/models/protip.rb
@@ -136,6 +136,10 @@ class Protip < ActiveRecord::Base
     event :mark_as_spam do
       transition any => :marked_as_spam
     end
+
+    after_transition any => :marked_as_spam do |protip|
+      protip.spam!
+    end
   end
 
   class << self
diff --git a/app/services/deindex_user_protips_service.rb b/app/services/deindex_user_protips_service.rb
diff --git a/app/services/index_user_protips_service.rb b/app/services/index_user_protips_service.rb
diff --git a/app/services/user_banner_service.rb b/app/services/user_banner_service.rb
@@ -1,11 +1,12 @@
 class UserBannerService
   def self.ban(user)
     user.update_attribute(:banned_at, Time.now.utc)
-    DeindexUserProtipsService.run(user)
+    UserProtipsService.deindex_all_for(user)
+    UserCommentsService.deindex_all_for(user)
   end
 
   def self.unban(user)
     user.update_attribute(:banned_at, nil)
-    IndexUserProtipsService.run(user)
+    UserProtipsService.reindex_all_for(user)
   end
 end
diff --git a/app/services/user_comments_service.rb b/app/services/user_comments_service.rb
@@ -0,0 +1,8 @@
+module UserCommentsService
+  def self.deindex_all_for(user)
+    user.comments.each do |comment|
+      comment.mark_as_spam
+    end
+  end
+end
+
diff --git a/app/services/user_protips_service.rb b/app/services/user_protips_service.rb
@@ -0,0 +1,15 @@
+module UserProtipsService
+  def self.deindex_all_for(user)
+    user.protips.each do |protip|
+      protip.mark_as_spam
+      ProtipIndexer.new(protip).remove
+    end
+  end
+
+  def self.reindex_all_for(user)
+    user.protips.each do |protip|
+      ProtipIndexer.new(protip).store
+    end
+  end
+end
+
