backup_minister 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: af89f01465c72c52b25a8e609c50519a15fe6ab6
-  data.tar.gz: 62c62db31e310a4e6b22abdab8bfb69c2732f69f
+  metadata.gz: b944e992d64bd98fe8c5843d9a061a063ce2d159
+  data.tar.gz: e2c999388a8ce746da8be7398fc81d9c4f8c85b4
 SHA512:
-  metadata.gz: 411b3d4bcb3c1a841d7440f5bc8d3904fc26b7da9afce45feee939116c4e9b5b51259a79d16a95e1df3a39b0225e0bce8437470b53d9b3cde5557ea6dcd65204
-  data.tar.gz: 1bd6a245ba4a09435506db87cf257251ebe873bdbd897a63136452d18114f480dc9219f10902f5fb11fcf0eed571928528bbe48436a7a71db7c2f841ec5d1185
+  metadata.gz: 788d174df306e30204ac0535aab390d815c14b8946473d8d8bf1eef9de322a6156bc75a7ec2a2a67d85baa5370b960db1019295833e117c9b6107cbfb9b50500
+  data.tar.gz: 0a1eef9e35f4b30d3ba6e9f058d412baa7f5e98ec7348ba64feb45e51b5f3c5da9fba7da1560fd5b2a7d98300efc47dda0b1552f7402e1d61aaca2a757fadeb3

data/bin/backup_minister

@@ -3,12 +3,38 @@ require 'backup_minister'
 require 'thor'
 
 class CLI < Thor
+desc 'backup_database', 'Backup DB and place it one remote server.'
+  option :project_name, type: :string, required: true, aliases: :name
+  def backup_database
+    agent = BackupMinister::Agent.new
+    backup_file_path = agent.backup_database(options[:project_name])
+    if backup_file_path
+      digest = agent.file_sha256_hash(backup_file_path)
+      placed_file_path = agent.place_database_backup(backup_file_path)
+      if agent.process_remote_database_backup(options[:project_name], placed_file_path, digest)
+        LOGGER.info "Database for project #{options[:project_name]} placed on remote server."
+      else
+        LOGGER.error "Could not place database backup for project #{options[:project_name]} on remote server."
+        exit!(true)
+      end
+    else
+      LOGGER.error "Could not create database backup for project #{options[:project_name]}."
+      exit!(true)
+    end
+  end
+
   desc 'backup_database', 'Move database TAR GZ archive to destination directory'
   option :project_name, type: :string, required: true, aliases: :name
   option :file, type: :string, required: true, banner: '</Path/to/database/backup.tar.gz>'
-  def backup_database
+  option :sha256, type: :string, banner: 'Hash of uploaded file'
+  def store_database_backup
     server = BackupMinister::Server.new
-    server.place_database_backup(options[:project_name], options[:file])
+    if server.place_database_backup(options[:project_name], options[:file])
+      LOGGER.info "Database for project #{options[:project_name]} backed up."
+    else
+      LOGGER.error "Could not complete backup #{options[:project_name]}."
+      exit!(true)
+    end
   end
 
   desc 'sync_path', 'Server\'s path for files'

data/lib/backup_minister.rb

@@ -2,6 +2,7 @@ require 'net/ssh'
 require 'open3'
 require 'logger'
 require 'fileutils'
+require 'digest'
 
 LOGGER = Logger.new(STDOUT)
 APP_NAME = 'backup_minister'
@@ -87,6 +88,16 @@ class BackupMinister
     out
   end
 
+  # Get SHA256 Hash of file
+  #
+  # @param file_path [String] path to file
+  #
+  # @return [String, nil]
+  def file_sha256_hash(file_path)
+    file = File.read(file_path)
+    Digest::SHA256.hexdigest(file) if file
+  end
+
   private
 
   attr_accessor :config

data/lib/backup_minister/agent.rb

@@ -8,6 +8,7 @@ DATABASE_DRIVERS = %i(docker)
 class BackupMinister::Agent < BackupMinister
   @projects = nil
   @config_file_name = nil
+  @remote_server_connection = nil
 
   def initialize(file_name = nil)
     super
@@ -68,7 +69,7 @@ class BackupMinister::Agent < BackupMinister
         LOGGER.info "Connection with server #{user}@#{host} established."
 
         remote_install = connection.exec!("gem list -i #{APP_NAME}")
-        if remote_install == 'true'
+        if remote_install.to_s.strip == 'true'
           LOGGER.debug "#{APP_NAME} installed on remote server."
           result = true
         else
@@ -150,6 +151,45 @@ class BackupMinister::Agent < BackupMinister
     result
   end
 
+  # Move database backup to server
+  #
+  # @return [String, nil] path to backup on remote server
+  def place_database_backup(file_path)
+    result = nil
+    destination_directory = '/tmp/'
+    scp_command = "scp #{file_path} #{system_config('server', 'user')}@#{system_config('server', 'host')}:#{destination_directory}"
+    if execute(scp_command)
+      LOGGER.debug "Backup #{file_path} placed remotely to #{destination_directory}."
+
+      FileUtils.rm(file_path)
+      LOGGER.debug "Local file #{file_path} removed."
+
+      result = destination_directory + File.basename(file_path)
+    else
+      LOGGER.error "Could not move #{file_path} to server."
+    end
+    result
+  end
+
+  # Execute command on remote server for processing database backup
+  #
+  # @param project_name [String]
+  # @param remote_file_path [String]
+  # @param sha256 [String, nil]
+  #
+  # @return [Bool] is operation success?
+  def process_remote_database_backup(project_name, remote_file_path, sha256 = nil)
+    result = false
+    command = "#{APP_NAME} store_database_backup"
+    command += " --project_name=#{project_name}"
+    command += " --file=#{remote_file_path}"
+    command += " --sha256=#{sha256}" unless sha256.nil?
+
+    execute_remotely { result = (@remote_server_connection.exec!(command).exitstatus == 0) }
+
+    result
+  end
+
   # Create TAR GZ single file archive
   #
   # @param file_name [String] file name for new archive (including extension)
@@ -195,4 +235,27 @@ class BackupMinister::Agent < BackupMinister
       nil
     end
   end
+
+  def remote_server_connect
+    if check_server_requirements
+      begin
+        host = system_config('server', 'host')
+        user = system_config('server', 'user')
+
+        @remote_server_connection = Net::SSH.start(host, user)
+      rescue Exception => error
+        LOGGER.error "Could not establish connection: #{error.message}"
+      end
+    end
+  end
+
+  def remote_server_close_connection
+    @remote_server_connection.close if !@remote_server_connection.nil? and !@remote_server_connection.closed?
+  end
+
+  def execute_remotely
+    remote_server_connect
+    yield if block_given?
+    remote_server_close_connection
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: backup_minister
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Ilya Krigouzov