From f4d4f7871eee00bc65777b6aa26b6ab42222bf9d Mon Sep 17 00:00:00 2001
From: Sam Saffron <sam.saffron@gmail.com>
Date: Tue, 21 May 2019 12:45:51 +1000
Subject: [PATCH] FEATURE: make posts:missing_uploads multisite friendly

Previously this rake job would only run on a single site which is a bit
misleading

This also adds `VERBOSE=1 rake posts:missing_uploads` that will provide a
full report of missing uploads
---
 lib/tasks/posts.rake   | 35 ++++++++++++++++++++++++++++++++---
 lib/tasks/uploads.rake |  2 +-
 2 files changed, 33 insertions(+), 4 deletions(-)

diff --git a/lib/tasks/posts.rake b/lib/tasks/posts.rake
index e15b1ec8275..c982d5d38bb 100644
--- a/lib/tasks/posts.rake
+++ b/lib/tasks/posts.rake
@@ -392,8 +392,7 @@ task 'posts:reorder_posts', [:topic_id] => [:environment] do |_, args|
   puts "", "Done.", ""
 end
 
-desc 'Finds missing post upload records from cooked HTML content'
-task 'posts:missing_uploads' => :environment do
+def missing_uploads
   old_scheme_upload_count = 0
 
   missing = Post.find_missing_uploads do |post, src, path, sha1|
@@ -440,15 +439,45 @@ task 'posts:missing_uploads' => :environment do
     upload_id
   end
 
+  puts "Database name: #{RailsMultisite::ConnectionManagement.current_db}"
   puts "", "#{missing[:count]} post uploads are missing.", ""
 
   if missing[:count] > 0
     puts "#{missing[:uploads].count} uploads are missing."
     puts "#{old_scheme_upload_count} of #{missing[:uploads].count} are old scheme uploads." if old_scheme_upload_count > 0
     puts "#{missing[:post_uploads].count} of #{Post.count} posts are affected.", ""
+
+    if ENV['VERBOSE'] == "1"
+      puts "missing uploads!"
+      missing[:uploads].each do |path|
+        puts "#{path}"
+      end
+
+      if missing[:post_uploads].count > 0
+        puts
+        puts "Posts with missing uploads"
+        missing[:post_uploads].each do |id, uploads|
+          post = Post.with_deleted.find_by(id: id)
+          if post
+            puts "#{post.url} missing #{uploads.join(", ")}"
+          else
+            puts "could not find post #{id}"
+          end
+        end
+      end
+    end
   end
+
+  missing[:count] == 0
 end
 
 desc 'Finds missing post upload records from cooked HTML content'
-task 'posts:missing_uploads' => :environment do
+task 'posts:missing_uploads', [:single_site] => :environment do |_, args|
+  if args[:single_site].to_s.downcase == "single_site"
+    missing_uploads
+  else
+    RailsMultisite::ConnectionManagement.each_connection do
+      missing_uploads
+    end
+  end
 end
diff --git a/lib/tasks/uploads.rake b/lib/tasks/uploads.rake
index 0ca34b4bb6f..c5bcb6514c1 100644
--- a/lib/tasks/uploads.rake
+++ b/lib/tasks/uploads.rake
@@ -236,7 +236,7 @@ def migration_successful?(db, should_raise = false)
   raise "#{count} posts are not remapped to new S3 upload URL. #{failure_message}" if count > 0 && should_raise
   return false if count > 0
 
-  Rake::Task['posts:missing_uploads'].invoke
+  Rake::Task['posts:missing_uploads'].invoke('single_site')
   count = PostCustomField.where(name: Post::MISSING_UPLOADS).count
   raise "rake posts:missing_uploads identified #{count} issues. #{failure_message}" if count > 0 && should_raise
   return false if count > 0