aws_docker_utils 0.0.1

data/lib/collectors/to_s3.rb

@@ -0,0 +1,31 @@
+require "./utils"
+require "./docker_compose/client"
+require "./s3/client"
+
+module Collectors
+  class ToS3
+
+    def initialize(opts={})
+      @bucket_name    = opts[:bucket_name]
+      @db_name        = opts[:db_name]
+      @task_name      = opts[:task] || "file"
+      @container_name = opts[:container_name]
+
+      @file_name   = "#{`date '+#{@task_name}_%y-%m-%d_%H-%M'`.chop}.sql"
+    end
+
+    def push
+      DockerCompose::Client.new(container_name: @container_name, cmd: "pg_dump -U postgres -O #{@db_name} > #{@file_name}").exec
+      file_name = Utils.compress(@file_name)
+
+      s3 = S3::Client.new(file_name: file_name, bucket_name: @bucket_name)
+      if s3.put(file_name)
+        Utils.rm(file_name)
+        puts "Pushed to S3."
+      else
+        puts "Unable to do the job."
+      end
+    end
+
+  end
+end

data/lib/docker_compose/client.rb

@@ -0,0 +1,14 @@
+module DockerCompose
+  class Client
+
+    def initialize(opts={})
+      @container_name = opts[:container_name]
+      @cmd            = opts[:cmd]
+    end
+
+    def exec
+      `docker exec $(docker ps -q --filter 'name=#{@container_name}') #{@cmd}`
+    end
+
+  end
+end

data/lib/s3/client.rb

@@ -0,0 +1,49 @@
+require 'yaml'
+require 'aws-sdk'
+
+module S3
+  class Client
+
+    DEFAULT_REGION = "us-east-1"
+
+    def initialize(opts={})
+      @bucket_name = opts[:bucket_name]
+
+      config = YAML.load(File.read(File.join(File.dirname(__FILE__), "credentials.yml")))
+      client = Aws::S3::Client.new(
+        region: (config["region"] || DEFAULT_REGION),
+        access_key_id: config["access_key"],
+        secret_access_key: config["secret_key"]
+      )
+      @s3 = Aws::S3::Resource.new(client: client)
+    end
+
+    def put(file_path)
+      raise "Please set bucket name with constructor." if @bucket_name.nil?
+
+      bucket = create_bucket(@bucket_name)
+      obj    = bucket.object(File.basename(file_path))
+
+      if obj.upload_file(file_path)
+        return true
+      else
+        puts "could not upload file #{@file_path} to S3."
+      end
+
+      false
+    end
+
+    private
+
+    def create_bucket(bucket_name)
+      bucket = @s3.bucket(bucket_name)
+      unless bucket.exists?
+        puts 'Bucket does not exist. Creating...'
+        bucket.create
+        puts 'Done.'
+      end
+      bucket
+    end
+
+  end
+end

data/lib/utils.rb

@@ -0,0 +1,15 @@
+module Utils
+
+  extend self
+
+  def rm(file_path)
+    `rm -f #{file_path.gsub(".tar.gz", '')}`
+    `rm -f #{file_path}`
+  end
+
+  def compress(source)
+    `tar -pcvzf #{source}.tar.gz #{source}`
+    "#{source}.tar.gz"
+  end
+
+end

metadata

@@ -0,0 +1,65 @@
+--- !ruby/object:Gem::Specification
+name: aws_docker_utils
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Bogdan Kulbida
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-04-15 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Utils package for working with AWS and Docker.
+email: bkulbida@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/s3/client.rb
+- lib/utils.rb
+- lib/collectors/to_s3.rb
+- lib/docker_compose/client.rb
+homepage: http://rubygems.org/gems/aws_docker_utils
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23.2
+signing_key: 
+specification_version: 3
+summary: Utils package for working with AWS and Docker.
+test_files: []