From 3829dc6a119dc6abb36472944f3bcfd0e5b86985 Mon Sep 17 00:00:00 2001
From: r888888888 <r888888888@gmail.com>
Date: Wed, 30 Dec 2015 11:46:54 -0800
Subject: [PATCH] further level-gate saved search updates on listbooru

---
 app/logical/user_promotion.rb                | 13 +++++++++++++
 app/models/saved_search.rb                   |  2 ++
 app/views/users/upgrade_information.html.erb |  2 +-
 3 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/app/logical/user_promotion.rb b/app/logical/user_promotion.rb
index bae317121..7b99ae2e2 100644
--- a/app/logical/user_promotion.rb
+++ b/app/logical/user_promotion.rb
@@ -22,6 +22,7 @@ class UserPromotion
     create_transaction_log_item
     create_user_feedback unless options[:skip_feedback]
     create_dmail unless options[:skip_dmail]
+    update_saved_searches
 
     user.save
   end
@@ -84,4 +85,16 @@ private
       :disable_dmail_notification => true
     )
   end
+
+  def update_saved_searches
+    if user.is_gold?
+      user.saved_searches.each do |ss|
+        ss.update_listbooru_on_create
+      end
+    else
+      user.saved_searches.each do |ss|
+        ss.update_listbooru_on_destroy
+      end
+    end
+  end
 end
diff --git a/app/models/saved_search.rb b/app/models/saved_search.rb
index d98367f5b..4cc6b1c78 100644
--- a/app/models/saved_search.rb
+++ b/app/models/saved_search.rb
@@ -28,6 +28,7 @@ class SavedSearch < ActiveRecord::Base
 
     def update_listbooru_on_create
       return unless Danbooru.config.listbooru_enabled?
+      return unless user.is_gold?
 
       sqs = SqsService.new(Danbooru.config.aws_sqs_queue_url)
       sqs.send_message("create\n#{user_id}\n#{category}\n#{tag_query}")
@@ -42,6 +43,7 @@ class SavedSearch < ActiveRecord::Base
 
     def update_listbooru_on_update
       return unless Danbooru.config.listbooru_enabled?
+      return unless user.is_gold?
 
       sqs = SqsService.new(Danbooru.config.aws_sqs_queue_url)
       sqs.send_message("update\n#{user_id}\n#{category_was}\n#{tag_query_was}\n#{category}\n#{tag_query}")
diff --git a/app/views/users/upgrade_information.html.erb b/app/views/users/upgrade_information.html.erb
index a3ae6d0fc..fdbd2d797 100644
--- a/app/views/users/upgrade_information.html.erb
+++ b/app/views/users/upgrade_information.html.erb
@@ -44,7 +44,7 @@
             <td>5,000</td>
           </tr>
           <tr>
-            <td>Tag Subscriptions</td>
+            <td>Saved Searches</td>
             <td>No</td>
             <td>Yes</td>
             <td>Yes</td>