require 'aws-sdk'

module Brillo
  module Transferrer
    class S3
      include Helpers::ExecHelper
      include Logger
      attr_reader :bucket, :filename, :region, :path, :enabled

      def initialize(config)
        @enabled              = config.transfer_config.enabled
        @bucket               = config.transfer_config.bucket
        @region               = config.transfer_config.region
        @filename             = config.compressed_filename
        @path                 = config.compressed_dump_path
        Aws.config.update(
          credentials: Aws::Credentials.new(
            config.transfer_config.access_key_id,
            config.transfer_config.secret_access_key
          ),
          region:             config.transfer_config.region
        )
      end

      def download
        return unless enabled
        logger.info("download from s3 #{bucket} #{path}")
        FileUtils.rm path, force: true
        client.get_object({bucket: bucket, key: path.to_s}, target: path)
      rescue Aws::S3::Errors::NoSuchBucket
        create_bucket
        retry
      end

      def upload
        return unless enabled
        logger.info("uploading to s3 #{bucket} #{path}")
        object = resource.bucket(bucket).object(path.to_s)
        object.upload_file(path)
      rescue Aws::S3::Errors::NoSuchBucket
        create_bucket
        retry
      end

      private

      def create_bucket
        client.create_bucket(bucket: bucket)
      end

      def client
        Aws::S3::Client.new
      end

      def resource
        Aws::S3::Resource.new
      end
    end
  end
end