From 57095f0bb7bf89488bd416c4b60f5f11e4a2d6bb Mon Sep 17 00:00:00 2001
From: Gerhard Schlager <gerhard.schlager@discourse.org>
Date: Tue, 13 Oct 2020 17:40:21 +0200
Subject: [PATCH] FIX: Killing a Unicorn worker shouldn't kill a running backup
 or restore process
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

By spawning and forking the backup and restore, the process owner changes from 🦄 to the init process.
---
 lib/backup_restore.rb          |  4 +--
 script/spawn_backup_restore.rb | 58 ++++++++++++++++++----------------
 2 files changed, 32 insertions(+), 30 deletions(-)

diff --git a/lib/backup_restore.rb b/lib/backup_restore.rb
index 6528b1728e9..b8a19efe4a1 100644
--- a/lib/backup_restore.rb
+++ b/lib/backup_restore.rb
@@ -172,9 +172,9 @@ module BackupRestore
 
   def self.spawn_process!(type, user_id, opts)
     script = File.join(Rails.root, "script", "spawn_backup_restore.rb")
-    command = ["bundle", "exec", "ruby", script, type, user_id, opts.to_json].shelljoin
+    command = ["bundle", "exec", "ruby", script, type, user_id, opts.to_json].map(&:to_s)
 
-    pid = spawn({ "RAILS_DB" => RailsMultisite::ConnectionManagement.current_db }, command)
+    pid = spawn({ "RAILS_DB" => RailsMultisite::ConnectionManagement.current_db }, *command)
     Process.detach(pid)
   end
 
diff --git a/script/spawn_backup_restore.rb b/script/spawn_backup_restore.rb
index 57e0de5a163..4af4f667c63 100644
--- a/script/spawn_backup_restore.rb
+++ b/script/spawn_backup_restore.rb
@@ -1,38 +1,40 @@
 # frozen_string_literal: true
 # This script is used by BackupRestore.backup! and BackupRestore.restore!
 
-require File.expand_path("../../config/environment", __FILE__)
+fork do
+  require File.expand_path("../../config/environment", __FILE__)
 
-def backup
-  user_id, opts = parse_params
-  BackupRestore::Backuper.new(user_id, opts).run
-end
+  def backup
+    user_id, opts = parse_params
+    BackupRestore::Backuper.new(user_id, opts).run
+  end
 
-def restore
-  user_id, opts = parse_params
+  def restore
+    user_id, opts = parse_params
 
-  BackupRestore::Restorer.new(
-    user_id: user_id,
-    filename: opts[:filename],
-    factory: BackupRestore::Factory.new(
+    BackupRestore::Restorer.new(
       user_id: user_id,
-      client_id: opts[:client_id]
-    ),
-    disable_emails: opts.fetch(:disable_emails, true)
-  ).run
-end
+      filename: opts[:filename],
+      factory: BackupRestore::Factory.new(
+        user_id: user_id,
+        client_id: opts[:client_id]
+      ),
+      disable_emails: opts.fetch(:disable_emails, true)
+    ).run
+  end
 
-def parse_params
-  user_id = ARGV[1].to_i
-  opts = JSON.parse(ARGV[2], symbolize_names: true)
-  [user_id, opts]
-end
+  def parse_params
+    user_id = ARGV[1].to_i
+    opts = JSON.parse(ARGV[2], symbolize_names: true)
+    [user_id, opts]
+  end
 
-case ARGV[0]
-when "backup"
-  backup
-when "restore"
-  restore
-else
-  raise "Unknown argument: #{ARGV[0]}"
+  case ARGV[0]
+  when "backup"
+    backup
+  when "restore"
+    restore
+  else
+    raise "Unknown argument: #{ARGV[0]}"
+  end
 end