Contents

require 'scalingo_backups_manager/notification'
require 'net/sftp'

module ScalingoBackupsManager
  class SftpTools
    attr_accessor :ftp_host

    def initialize(ftp_host)
      @ftp_host = ftp_host
    end

    def start
      Net::SFTP.start(@ftp_host[:host], @ftp_host[:user], password: @ftp_host[:password], port: @ftp_host[:port]) do |sftp|
        yield(sftp) if block_given?
      end
    end

    def list_files(path)
      files = []
      start do |sftp|
        sftp.dir.glob("#{path}", "*.tar.gz").each do |file|
          files << file
        end
      end
      files
    end

    def remove!(path)
      start do |sftp|
        sftp.remove!(path)
      end
    end

    def mkdir!(path)
      start do |sftp|
        folder_tree = []
        path.split("/").each do |folder_name|
          next if folder_name.blank?

          folder_tree << folder_name
          begin
            sftp.mkdir!(folder_tree.join("/"))
          rescue
          end
        end
      end
    end

    def upload_file(filepath, remote_dir, options: {})
      filename = filepath.split('/').last
      start do |sftp|
        begin
          sftp.upload!(filepath, "#{remote_dir}/#{filename}")
        rescue
          if options.dig(:webhooks, :slack_webhook_url)
            ScalingoBackupsManager::Notification.send_slack_notification(options.dig(:webhooks, :slack_webhook_url), "An error has occured while uploading backup, see the logs for more information")
          end
          if options.dig(:webhooks, :discord_webhook_url)
            ScalingoBackupsManager::Notification.send_discord_notification(options.dig(:webhooks, :discord_webhook_url), "An error has occured while uploading backup, see the logs for more information")
          end
        end
      end
    end
  end

end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
scalingo_backups_manager-0.6.5	lib/scalingo_backups_manager/sftp_tools.rb
scalingo_backups_manager-0.6.4	lib/scalingo_backups_manager/sftp_tools.rb
scalingo_backups_manager-0.6.3	lib/scalingo_backups_manager/sftp_tools.rb
scalingo_backups_manager-0.6.2	lib/scalingo_backups_manager/sftp_tools.rb
scalingo_backups_manager-0.6.1	lib/scalingo_backups_manager/sftp_tools.rb
scalingo_backups_manager-0.6.0	lib/scalingo_backups_manager/sftp_tools.rb