From 68637f216474a415344533ffdcba5c26588985b1 Mon Sep 17 00:00:00 2001
From: Guo Xiang Tan <tgx_world@hotmail.com>
Date: Thu, 15 Sep 2016 21:45:36 +0800
Subject: [PATCH] FIX: Uploads being restored into the wrong directory for
 multisite.

---
 lib/backup_restore/restorer.rb | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/lib/backup_restore/restorer.rb b/lib/backup_restore/restorer.rb
index cf9ff1367..039936119 100644
--- a/lib/backup_restore/restorer.rb
+++ b/lib/backup_restore/restorer.rb
@@ -1,3 +1,5 @@
+require_dependency "db_helper"
+
 module BackupRestore
 
   class RestoreDisabledError < RuntimeError; end
@@ -360,12 +362,32 @@ module BackupRestore
     def extract_uploads
       if system('tar', '--exclude=*/*', '--list', '--file', @tar_filename, 'uploads')
         log "Extracting uploads..."
-        FileUtils.cd(File.join(Rails.root, "public")) do
+
+        FileUtils.cd(@tmp_directory) do
           execute_command(
             'tar', '--extract', '--keep-newer-files', '--file', @tar_filename, 'uploads/',
             failure_message: "Failed to extract uploadsd."
           )
         end
+
+        public_uploads_path = File.join(Rails.root, "public")
+
+        FileUtils.cd(public_uploads_path) do
+          FileUtils.mkdir_p("uploads")
+
+          tmp_uploads_path = Dir.glob(File.join(@tmp_directory, "uploads", "*")).first
+          previous_db_name = File.basename(tmp_uploads_path)
+          current_db_name = RailsMultisite::ConnectionManagement.current_db
+
+          execute_command(
+            'rsync', '-avp', "#{tmp_uploads_path}/", "uploads/#{current_db_name}/",
+            failure_message: "Failed to restore uploads."
+          )
+
+          if previous_db_name != current_db_name
+            DbHelper.remap("uploads/#{previous_db_name}", "uploads/#{current_db_name}")
+          end
+        end
       end
     end