From d91a72a791b72bcaaa49de10ce74c61f379d1445 Mon Sep 17 00:00:00 2001
From: Neil Lalonde <neillalonde@gmail.com>
Date: Fri, 4 Apr 2014 11:10:47 -0400
Subject: [PATCH] Allow rebaking posts for only one site in multisite. Use
 RAILS_DB.

---
 lib/tasks/posts.rake | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/lib/tasks/posts.rake b/lib/tasks/posts.rake
index 198829083af..ddc716bbe97 100644
--- a/lib/tasks/posts.rake
+++ b/lib/tasks/posts.rake
@@ -1,11 +1,11 @@
 desc 'Update each post with latest markdown'
 task 'posts:rebake' => :environment do
-  rebake_posts
+  ENV['RAILS_DB'] ? rebake_posts : rebake_posts_all_sites
 end
 
 desc 'Update each post with latest markdown and refresh oneboxes'
 task 'posts:refresh_oneboxes' => :environment do
-  rebake_posts invalidate_oneboxes: true
+  ENV['RAILS_DB'] ? rebake_posts(invalidate_oneboxes: true) : rebake_posts_all_sites(invalidate_oneboxes: true)
 end
 
 def rebake_post(post,opts)
@@ -32,16 +32,20 @@ rescue => e
   puts "\n\nFailed to bake topic_id #{post.topic_id} post_id #{post.id} #{e}\n#{e.backtrace.join("\n")} \n\n"
 end
 
-def rebake_posts(opts = {})
+def rebake_posts_all_sites(opts = {})
   RailsMultisite::ConnectionManagement.each_connection do |db|
-    puts "Re baking post markdown for #{db}, changes are denoted with #, no change with ."
-
-    total = 0
-    Post.find_each do |post|
-      rebake_post(post,opts)
-      total += 1
-    end
-
-    puts "\n\n#{total} posts done!\n#{'-' * 50}\n"
+    rebake_posts(opts)
   end
 end
+
+def rebake_posts(opts = {})
+  puts "Re baking post markdown for #{RailsMultisite::ConnectionManagement.current_db}, changes are denoted with #, no change with ."
+
+  total = 0
+  Post.find_each do |post|
+    rebake_post(post,opts)
+    total += 1
+  end
+
+  puts "\n\n#{total} posts done!\n#{'-' * 50}\n"
+end