aws_client 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: eb9f2d7d1282f3fa8fff4efee7c147769e4ca528
+  data.tar.gz: f918751ee5ee64bc06ebc258c2c9cce6e1e99072
+SHA512:
+  metadata.gz: b2eb460108a486dc98ee0434e767204bac1948b25d867b09f5270de038d66b1a5dc1ff22defbe3601f9abf3b09217286e7cb95251d62051559b396776d99f557
+  data.tar.gz: dfa5c5959e58e8807fad8f865253bcfed5a739c6b0c3fa9262e277e97c48d464817b7b5daf86272c49fe9f21efb0d9f6d8b20643c9114ad2963c98600bc89e9f

data/.gitignore

@@ -0,0 +1,9 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in aws_client.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2015 TODO: Write your name
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,4 @@
+# AwsClient
+
+Very light-weight wrapper for AWS interactions
+

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+

data/aws_client.gemspec

@@ -0,0 +1,25 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'aws_client/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "aws_client"
+  spec.version       = AwsClient::VERSION
+  spec.authors       = ["Webzakimbo"]
+  spec.email         = ["guillaume@webzakimbo.com"]
+
+  spec.summary       = %q{Light-weight AWS wrapper.}
+  spec.description   = %q{A very light-weight wrapper for my most commonly used AWS functions}
+  spec.homepage      = "http://webzakimbo.com"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency 'aws-sdk', '~> 2'
+  spec.add_development_dependency "bundler", "~> 1.8"
+  spec.add_development_dependency "rake", "~> 10.0"
+end

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "aws_client"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start

data/bin/setup

@@ -0,0 +1,7 @@
+#!/bin/bash
+set -euo pipefail
+IFS=$'\n\t'
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/aws_client/version.rb

@@ -0,0 +1,3 @@
+module AwsClient
+  VERSION = "0.0.1"
+end

data/lib/aws_client.rb

@@ -0,0 +1,60 @@
+require "aws-sdk"
+require "aws_client/version"
+require "wrapper"
+require "s3_wrapper"
+require "cf_wrapper"
+require "rds_wrapper"
+require "ec2_wrapper"
+
+module AwsClient
+
+  CREDENTIALS_FILE = ".fog"
+  PATH_TO_CREDENTIALS = File.join(Dir.home, CREDENTIALS_FILE)
+
+  class Container
+
+    def initialize(credentials_key, aws_region, provisioning_region)
+      @aws_account_credentials_key = credentials_key
+      @aws_region = aws_region
+      @provisioning_region = provisioning_region
+      validate
+    end
+
+    def rds
+      @rds ||= ::AwsClient::RdsWrapper.new(client: client_connect(::Aws::RDS::Client))
+    end
+
+    def cloudformation
+      @cloudformation ||= ::AwsClient::CfWrapper.new(client: client_connect(::Aws::CloudFormation::Client))
+    end
+
+    def s3
+      @s3 ||= ::AwsClient::S3Wrapper.new(client: client_connect(::Aws::S3::Client), credentials_key: @aws_account_credentials_key, region: @aws_region)
+    end
+
+    def ec2
+      @ec2 ||= ::AwsClient::Ec2Wrapper.new(client: client_connect(::Aws::EC2::Client))
+    end
+
+    protected
+
+    def credentials
+      all_credentials = YAML.load_file(::AwsClient::PATH_TO_CREDENTIALS)
+      raise "Cannot find credentials for '#{@aws_account_credentials_key}'" unless all_credentials.has_key?(@aws_account_credentials_key)
+      creds_for_key = all_credentials[@aws_account_credentials_key]
+      return Aws::Credentials.new(creds_for_key["aws_access_key_id"], creds_for_key["aws_secret_access_key"])
+    end
+
+    def validate
+      raise "Missing Aws Account Credentials Key" unless @aws_account_credentials_key
+      raise "Missing credentials file: cannot find #{::AwsClient::CREDENTIALS_FILE} in your home directory" unless File.exists?(::AwsClient::PATH_TO_CREDENTIALS)
+    end
+
+    private
+
+    def client_connect(client_klass)
+      return client_klass.new(credentials: credentials, region: @provisioning_region)
+    end
+
+  end
+end

data/lib/cf_wrapper.rb

@@ -0,0 +1,66 @@
+module AwsClient
+  class CfWrapper < Wrapper
+
+    STACK_COMPLETE_STATUS = "CREATE_COMPLETE"
+    STACK_ROLLLBACK_STATUS = "ROLLBACK_IN_PROGRESS"
+    STACK_WAIT_SLEEP_TIME = 3
+
+    def create_stack(stack_params)
+      puts "Creating stack with the following parameters: #{stack_params.inspect}"
+      response = client.create_stack(stack_params)
+      wait_for_stack(stack_params[:stack_name])
+    end
+
+    def stack_status_for(stack_name)
+      stack_info = stack_info_for(stack_name)
+      return stack_info_for(stack_name).stack_status
+    end
+
+    def stack_info_for(stack_name)
+      page = client.describe_stacks  
+      stack_info = page.data.stacks.select{|stack| stack["stack_name"] == stack_name }
+      raise "Cannot find CF stack named '#{stack_name}'" unless stack_info.any?
+      return stack_info.last
+    end
+
+    def stack_resources_by_stack_name_and_resource_type(stack_name, resource_type)
+      return stack_resources_for_stack(stack_name).select{|r| r.resource_type == resource_type }  
+    end
+
+    def ec2_server_stack_resource_for_node_name(stack_name, node_name)
+      server_resources = ec2_server_stack_resources(stack_name)
+      server_resource = server_resources.select{|s| s.logical_resource_id == node_name}
+      return server_resource
+    end
+
+    def ec2_server_stack_resources(stack_name)
+      stack_resources_by_stack_name_and_resource_type(stack_name, "AWS::EC2::Instance")
+    end
+
+    def stack_resources_for_stack(stack_name)
+      @resources ||= client.describe_stack_resources("stack_name" => stack_name).data[:stack_resources]
+    end 
+
+    def resource_data_for_resource_name(stack_name, resource_name)
+      stack_resources_for_stack(stack_name).select{|r| r.logical_resource_id == resource_name }.first
+    end 
+
+    def physical_resource_id_for(stack_name, resource_name)
+      resource_data_for_resource_name(stack_name, resource_name).physical_resource_id
+    end
+
+    def wait_for_stack(stack_name)
+      puts "Waiting for stack to spin up"
+      status = ""
+      while status != STACK_COMPLETE_STATUS
+        status = stack_status_for(stack_name)
+        puts "'#{stack_name}' stack status: #{status}"
+ 
+        raise "Rollback in progress - CF build failed" if status == STACK_ROLLLBACK_STATUS 
+        sleep(STACK_WAIT_SLEEP_TIME)     
+      end
+      puts "Stack is up. Ready for bootstrapping"
+    end
+
+  end
+end

data/lib/ec2_wrapper.rb

@@ -0,0 +1,19 @@
+module AwsClient
+  class Ec2Wrapper < AwsClient::Wrapper
+
+    def ec2_instance_public_address(node_name)
+      return ec2_instance_via_cloudformation_lookup(node_name).public_ip_address
+    end
+ 
+    def ec2_instance_via_cloudformation_lookup(node_name)
+      instance = cloudformation_resource_for_server(node_name).first
+      instances_result = client.describe_instances(filters: [{name:'instance-id', values:[instance.physical_resource_id]}])
+      return instances_result.reservations[0].instances[0]
+    end
+
+    def cloudformation_resource_for_server(node_name)
+      return CLIENTS.cloudformation.ec2_server_stack_resource_for_node_name(BOOT_CONFIG.cf_stack_name, node_name)
+    end
+
+  end
+end

data/lib/rds_wrapper.rb

@@ -0,0 +1,38 @@
+module AwsClient
+  class RdsWrapper < AwsClient::Wrapper
+
+    def instance_by_database_name(database_name)
+      return all_instances.select{|db_instance| db_instance.db_name == database_name }.last
+    end
+
+    def instance_data_by_vpc_id(vpc_id)
+      return all_instances.select{|instance| instance.db_subnet_group.vpc_id = vpc_id }
+    end
+
+    def instance_endpoint(instance)
+      return "UNDEFINED" if instance.nil?
+      return instance.endpoint.address
+    end
+
+    def instance_endpoint_for_named_database(database_name)
+      instance = instance_by_database_name(database_name)
+      return instance_endpoint(instance)
+    end
+
+    def all_instances
+      @all_instances ||= get_all_instances.collect{|raw_instance| raw_instance.db_instances.flatten }.flatten
+    end
+
+    def get_all_instances
+      all_instances = []
+      pages = client.describe_db_instances
+      all_instances << pages.data
+      while pages.next_page?
+        pages = pages.next_page
+        @all_instances << pages.data
+      end
+      return all_instances    
+    end
+
+  end
+end

data/lib/s3_wrapper.rb

@@ -0,0 +1,68 @@
+module AwsClient
+  class S3Wrapper < AwsClient::Wrapper
+
+    BUCKET_DEFINITION_FILE = ".aws_buckets"
+    PATH_TO_BUCKETS_DEFINITION = File.join(Dir.home, BUCKET_DEFINITION_FILE)
+
+    def initialize(*params)
+      super
+      @credentials_key = params[0][:credentials_key]
+      @aws_region = params[0][:region]
+    end
+ 
+    def resource
+      @resource ||= Aws::S3::Resource.new(client: client)
+    end
+
+    def bucket_object_for_bucket_prefix(bucket_prefix_code)
+      bucket = find_or_create_bucket_for(bucket_prefix_code)
+      return bucket.objects.collect{|object| object.object } # 'object.object' The S3 object belonging to the S3 resource    
+    end
+
+    def find_bucket_by(bucket_name)
+      buckets.select{|bucket| bucket.name == bucket_name }.first
+    end
+
+    def write_file_to_bucket(bucket_prefix_code, file_path)
+      bucket = find_or_create_bucket_for(bucket_prefix_code)
+      s3_object = bucket.put_object(
+        :key => File.basename(file_path),
+        :body => IO.read(file_path)
+      )
+      puts "\nRemote file"
+      puts "#{s3_object.public_url}"
+      return s3_object
+    end
+
+    def buckets
+      return resource.buckets
+    end
+
+    def find_or_create_bucket_for(bucket_prefix_code)
+      bucket_prefix = bucket_prefixes[bucket_prefix_code]
+      full_bucket_name = bucket_name_for_function(bucket_prefix) 
+      return find_or_create_bucket(full_bucket_name)
+    end
+  
+    def bucket_name_for_function(bucket_function)
+      return "#{bucket_function}.#{@credentials_key}.#{@aws_region}"  
+    end
+
+    def bucket_prefixes
+      YAML.load_file(PATH_TO_BUCKETS_DEFINITION)
+    end
+
+    private
+
+    def find_or_create_bucket(bucket_name)
+      begin 
+        bucket = resource.create_bucket(:bucket => bucket_name)
+      rescue Aws::S3::Errors::BucketAlreadyOwnedByYou
+        bucket = find_bucket_by(bucket_name)
+      end
+
+      return bucket
+    end
+
+  end
+end

data/lib/wrapper.rb

@@ -0,0 +1,11 @@
+module AwsClient
+
+  class Wrapper
+    attr_reader :client
+
+    def initialize(*params)
+      @client = params[0][:client]
+    end
+  end
+
+end

metadata

@@ -0,0 +1,101 @@
+--- !ruby/object:Gem::Specification
+name: aws_client
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Webzakimbo
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2015-12-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: A very light-weight wrapper for my most commonly used AWS functions
+email:
+- guillaume@webzakimbo.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- aws_client.gemspec
+- bin/console
+- bin/setup
+- lib/aws_client.rb
+- lib/aws_client/version.rb
+- lib/cf_wrapper.rb
+- lib/ec2_wrapper.rb
+- lib/rds_wrapper.rb
+- lib/s3_wrapper.rb
+- lib/wrapper.rb
+homepage: http://webzakimbo.com
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Light-weight AWS wrapper.
+test_files: []