FIX: Don't use backticks that take in inputs.

lib/backup_restore/backup_restore.rb

@@ -1,4 +1,3 @@
-require "backup_restore/utils"
 require "backup_restore/backuper"
 require "backup_restore/restorer"
 

lib/backup_restore/backuper.rb

@@ -1,8 +1,6 @@
 module BackupRestore
 
   class Backuper
-    include BackupRestore::Utils
-
     attr_reader :success
 
     def initialize(user_id, opts={})
@@ -198,7 +196,7 @@ module BackupRestore
     def move_dump_backup
       log "Finalizing database dump file: #{@backup_filename}"
 
-      execute_command(
+      Discourse::Utils.execute_command(
         'mv', @dump_filename, File.join(@archive_directory, @backup_filename),
         failure_message: "Failed to move database dump file."
       )
@@ -212,15 +210,15 @@ module BackupRestore
       tar_filename = "#{@archive_basename}.tar"
 
       log "Making sure archive does not already exist..."
-      execute_command('rm', '-f', tar_filename)
-      execute_command('rm', '-f', "#{tar_filename}.gz")
+      Discourse::Utils.execute_command('rm', '-f', tar_filename)
+      Discourse::Utils.execute_command('rm', '-f', "#{tar_filename}.gz")
 
       log "Creating empty archive..."
-      execute_command('tar', '--create', '--file', tar_filename, '--files-from', '/dev/null')
+      Discourse::Utils.execute_command('tar', '--create', '--file', tar_filename, '--files-from', '/dev/null')
 
       log "Archiving data dump..."
       FileUtils.cd(File.dirname(@dump_filename)) do
-        execute_command(
+        Discourse::Utils.execute_command(
           'tar', '--append', '--dereference', '--file', tar_filename, File.basename(@dump_filename),
           failure_message: "Failed to archive data dump."
         )
@@ -231,7 +229,7 @@ module BackupRestore
       log "Archiving uploads..."
       FileUtils.cd(File.join(Rails.root, "public")) do
         if File.directory?(upload_directory)
-          execute_command(
+          Discourse::Utils.execute_command(
             'tar', '--append', '--dereference', '--file', tar_filename, upload_directory,
             failure_message: "Failed to archive uploads."
           )
@@ -243,7 +241,7 @@ module BackupRestore
       remove_tmp_directory
 
       log "Gzipping archive, this may take a while..."
-      execute_command('gzip', '-5', tar_filename, failure_message: "Failed to gzip archive.")
+      Discourse::Utils.execute_command('gzip', '-5', tar_filename, failure_message: "Failed to gzip archive.")
     end
 
     def after_create_hook
@@ -259,11 +257,11 @@ module BackupRestore
 
     def notify_user
       log "Notifying '#{@user.username}' of the end of the backup..."
-      if @success
-        SystemMessage.create_from_system_user(@user, :backup_succeeded, logs: pretty_logs(@logs))
-      else
-        SystemMessage.create_from_system_user(@user, :backup_failed, logs: pretty_logs(@logs))
-      end
+      status = @success ? :backup_succeeded : :backup_failed
+
+      SystemMessage.create_from_system_user(@user, status,
+        logs: Discouse::Utils.pretty_logs(@logs)
+      )
     end
 
     def clean_up

lib/backup_restore/restorer.rb

@@ -6,8 +6,6 @@ module BackupRestore
   class FilenameMissingError < RuntimeError; end
 
   class Restorer
-    include BackupRestore::Utils
-
     attr_reader :success
 
     def initialize(user_id, opts={})
@@ -166,7 +164,7 @@ module BackupRestore
 
     def copy_archive_to_tmp_directory
       log "Copying archive to tmp directory..."
-      execute_command('cp', @source_filename, @archive_filename, failure_message: "Failed to copy archive to tmp directory.")
+      Discourse::Utils.execute_command('cp', @source_filename, @archive_filename, failure_message: "Failed to copy archive to tmp directory.")
     end
 
     def unzip_archive
@@ -175,7 +173,7 @@ module BackupRestore
       log "Unzipping archive, this may take a while..."
 
       FileUtils.cd(@tmp_directory) do
-        execute_command('gzip', '--decompress', @archive_filename, failure_message: "Failed to unzip archive.")
+        Discourse::Utils.execute_command('gzip', '--decompress', @archive_filename, failure_message: "Failed to unzip archive.")
       end
     end
 
@@ -185,7 +183,7 @@ module BackupRestore
       @metadata =
         if system('tar', '--list', '--file', @tar_filename, BackupRestore::METADATA_FILE)
           FileUtils.cd(@tmp_directory) do
-            execute_command(
+            Discourse::Utils.execute_command(
               'tar', '--extract', '--file', @tar_filename, BackupRestore::METADATA_FILE,
               failure_message: "Failed to extract metadata file."
             )
@@ -233,7 +231,7 @@ module BackupRestore
       log "Extracting dump file..."
 
       FileUtils.cd(@tmp_directory) do
-        execute_command(
+        Discourse::Utils.execute_command(
           'tar', '--extract', '--file', @tar_filename, File.basename(@dump_filename),
           failure_message: "Failed to extract dump file."
         )
@@ -364,7 +362,7 @@ module BackupRestore
         log "Extracting uploads..."
 
         FileUtils.cd(@tmp_directory) do
-          execute_command(
+          Discourse::Utils.execute_command(
             'tar', '--extract', '--keep-newer-files', '--file', @tar_filename, 'uploads/',
             failure_message: "Failed to extract uploads."
           )
@@ -379,7 +377,7 @@ module BackupRestore
           previous_db_name = File.basename(tmp_uploads_path)
           current_db_name = RailsMultisite::ConnectionManagement.current_db
 
-          execute_command(
+          Discourse::Utils.execute_command(
             'rsync', '-avp', '--safe-links', "#{tmp_uploads_path}/", "uploads/#{current_db_name}/",
             failure_message: "Failed to restore uploads."
           )
@@ -404,11 +402,11 @@ module BackupRestore
     def notify_user
       if user = User.find_by(email: @user_info[:email])
         log "Notifying '#{user.username}' of the end of the restore..."
-        if @success
-          SystemMessage.create_from_system_user(user, :restore_succeeded, logs: pretty_logs(@logs))
-        else
-          SystemMessage.create_from_system_user(user, :restore_failed, logs: pretty_logs(@logs))
-        end
+        status = @success ? :restore_succeeded : :restore_failed
+
+        SystemMessage.create_from_system_user(user, status,
+          logs: Discourse::Utils.pretty_logs(@logs)
+        )
       else
         log "Could not send notification to '#{@user_info[:username]}' (#{@user_info[:email]}), because the user does not exists..."
       end

lib/backup_restore/utils.rb

@@ -1,20 +0,0 @@
-require 'open3'
-
-module BackupRestore
-  module Utils
-    def execute_command(*command, failure_message: "")
-      stdout, stderr, status = Open3.capture3(*command)
-
-      if !status.success?
-        failure_message = "#{failure_message}\n" if !failure_message.blank?
-        raise "#{failure_message}#{stderr}"
-      end
-
-      stdout
-    end
-
-    def pretty_logs(logs)
-      logs.join("\n")
-    end
-  end
-end