Allow rebaking posts for only one site in multisite. Use RAILS_DB.

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