infopark-aws_utils 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 78f958024570261c6c789338b7e9d35ffef622e6
+  data.tar.gz: 0494bb77b8f40b3db3e2ca8e9dd4a1941ff6fc4b
+SHA512:
+  metadata.gz: a2a277c0579471d4b08f0cfac7894365eea0ece83baeefd6d80a324138d16a38964a3f6a9e3d69bb5864d301784b92d8cfc68551c14685edc485b2f3ee36c158
+  data.tar.gz: 21dfbc376fb2a28b6980a34c80614d37395576f8889a6cc36f34b8735545a08d2f90350812052bbae61e15ff64dd272b131e4143e1859a5120496d3ce6255228

data/.gitignore

@@ -0,0 +1,3 @@
+/pkg
+/spec/examples.txt
+Gemfile.lock

data/.rspec

@@ -0,0 +1 @@
+--require spec_helper

data/Gemfile

@@ -0,0 +1,3 @@
+source "https://rubygems.org"
+
+gemspec

data/Rakefile

@@ -0,0 +1 @@
+require 'bundler/gem_tasks'

data/aws_utils.gemspec

@@ -0,0 +1,18 @@
+require_relative 'lib/infopark/aws_utils/version'
+
+Gem::Specification.new do |s|
+  s.name = 'infopark-aws_utils'
+  s.version = Infopark::AwsUtils::VERSION
+  s.summary = 'A utility lib to ease the use of the AWS SDK'
+  s.description = s.summary
+  s.authors = ['Tilo Prütz']
+  s.email = 'tilo@infopark.de'
+  s.files = `git ls-files -z`.split("\0")
+  s.license = 'UNLICENSED'
+
+  s.add_dependency "aws-sdk", ">=2.10"
+
+  s.add_development_dependency "bundler"
+  s.add_development_dependency "rake"
+  s.add_development_dependency "rspec"
+end

data/lib/infopark/aws_utils/env.rb

@@ -0,0 +1,95 @@
+require 'aws-sdk'
+
+require 'infopark/aws_utils'
+
+module Infopark
+  module AwsUtils
+
+class Env
+  def initialize(profile_name = nil)
+    @credentials = Aws::SharedCredentials.new(profile_name: profile_name)
+    @clients = Hash.new do |clients, mod|
+      mod.const_get(:Client).new(credentials: @credentials, region: 'eu-west-1')
+    end
+  end
+
+  def profile_name
+    @credentials.profile_name
+  end
+
+  def ecs
+    @clients[Aws::ECS]
+  end
+
+  def ecr
+    @clients[Aws::ECR]
+  end
+
+  def ec2
+    @clients[Aws::EC2]
+  end
+
+  def as
+    @clients[Aws::AutoScaling]
+  end
+
+  def alb
+    @clients[Aws::ElasticLoadBalancingV2]
+  end
+
+  def aas
+    @clients[Aws::ApplicationAutoScaling]
+  end
+
+  def cw
+    @clients[Aws::CloudWatch]
+  end
+
+  def cwl
+    @clients[Aws::CloudWatchLogs]
+  end
+
+  def sts
+    @clients[Aws::STS]
+  end
+
+
+  def account_type
+    return "dev" if dev_account?
+    return "prod" if account?(PROD_ACCOUNT_ID)
+    raise "Could not determine account type."
+  end
+
+  def dev_account?
+    account?(DEV_ACCOUNT_ID)
+  end
+
+  def latest_base_image
+    available_images = AwsUtils.gather_all(ec2, :describe_images,
+        owners: [AWS_AMI_OWNER],
+        filters: [
+          {name: "root-device-type", values: ["instance-store"]},
+          {name: "image-type", values: ["machine"]},
+          {name: "virtualization-type", values: ["hvm"]},
+        ]).reject {|image| image.name.include?(".rc-") || image.name.include?("-minimal-") }
+    available_images.sort_by(&:creation_date).last
+  end
+
+  def find_image_by_id(id)
+    AwsUtils.gather_all(ec2, :describe_images, image_ids: [id]).first
+  end
+
+  def find_image_by_name(name)
+    AwsUtils.gather_all(ec2, :describe_images, owners: [DEV_ACCOUNT_ID],
+        filters: [{name: "name", values: [name]}]).first
+  end
+
+  private
+
+  def account?(account_id)
+    sts.get_caller_identity.account == account_id
+  end
+end
+
+  end
+end

data/lib/infopark/aws_utils/version.rb

@@ -0,0 +1,5 @@
+module Infopark
+  module AwsUtils
+    VERSION = "0.1.0"
+  end
+end

data/lib/infopark/aws_utils.rb

@@ -0,0 +1,57 @@
+require 'infopark/aws_utils/env'
+require 'json'
+
+module Infopark
+  module AwsUtils
+    AWS_AMI_OWNER = "137112412989"
+
+    class << self
+      def gather_all(client, method, **options)
+        @gather_cache ||= {}
+        cache_key = [client, method, options]
+        return @gather_cache[cache_key] if @gather_cache[cache_key]
+
+        result = []
+        loop do
+          response = client.send(method, **options)
+          key = (response.members - [:next_token, :failures]).first
+          if response.members.include?(:failures) && !response.failures.empty?
+            raise "Failed gathering all #{method}: #{response.failures}"
+          end
+          result += response[key]
+          unless options[:next_token] = response.members.include?(:next_token) && response.next_token
+            break
+          end
+        end
+        @gather_cache[cache_key] = result
+      end
+
+      def wait_for(progress, client, waiter, delay: 2, max_attempts: 60, **waiter_params)
+        client.wait_until(waiter, waiter_params) do |w|
+          w.delay = delay
+          w.max_attempts = max_attempts
+          w.before_wait { progress.increment }
+        end
+      end
+
+      protected
+
+      def local_config
+        @local_config ||= (
+          path = "#{Dir.home}/.config/infopark/aws_utils.json"
+          if File.exists?(path)
+            JSON.parse(File.read(path))
+          else
+            {}
+          end
+        )
+      end
+    end
+
+    DEV_ACCOUNT_ID = ENV['INFOPARK_AWS_DEV_ACCOUNT_ID'] || self.local_config['dev_account_id']
+    PROD_ACCOUNT_ID = ENV['INFOPARK_AWS_PROD_ACCOUNT_ID'] || self.local_config['prod_account_id']
+
+    puts "WARN: The Infopark AWS development account ID is not configured." unless DEV_ACCOUNT_ID
+    puts "WARN: The Infopark AWS production account ID is not configured." unless PROD_ACCOUNT_ID
+  end
+end

data/spec/env_spec.rb

@@ -0,0 +1,38 @@
+RSpec.describe Infopark::AwsUtils::Env do
+  subject(:env) { Infopark::AwsUtils::Env.new }
+
+  describe "#account_type" do
+    let(:account_id) { nil }
+    let(:sts) do
+      instance_double(Aws::STS::Client,
+        get_caller_identity: double(:caller_identity, account: account_id)
+      )
+    end
+
+    before { allow(Aws::STS::Client).to receive(:new).and_return(sts) }
+
+    subject(:account_type) { env.account_type }
+
+    context "for profile in development account" do
+      let(:account_id) { "012615398682" }
+
+      it { is_expected.to eq("dev") }
+    end
+
+    context "for profile in production account" do
+      let(:account_id) { "115379056088" }
+
+      it { is_expected.to eq("prod") }
+    end
+
+    context "for profile in some other account" do
+      let(:account_id) { "137112412989" }
+
+      it "fails" do
+        expect {
+          account_type
+        }.to raise_error("Could not determine account type.")
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,31 @@
+require 'infopark/aws_utils'
+
+RSpec.configure do |config|
+  config.expect_with :rspec do |expectations|
+    expectations.include_chain_clauses_in_custom_matcher_descriptions = true
+  end
+
+  config.mock_with :rspec do |mocks|
+    mocks.verify_partial_doubles = true
+  end
+
+  config.shared_context_metadata_behavior = :apply_to_host_groups
+
+  config.filter_run_when_matching :focus
+
+  config.example_status_persistence_file_path = "spec/examples.txt"
+
+  config.disable_monkey_patching!
+
+  config.warnings = true
+
+  if config.files_to_run.one?
+    config.default_formatter = "doc"
+  end
+
+  config.profile_examples = 10
+
+  config.order = :random
+
+  Kernel.srand config.seed
+end

metadata

@@ -0,0 +1,109 @@
+--- !ruby/object:Gem::Specification
+name: infopark-aws_utils
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Tilo Prütz
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-06-30 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.10'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.10'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A utility lib to ease the use of the AWS SDK
+email: tilo@infopark.de
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- Gemfile
+- Rakefile
+- aws_utils.gemspec
+- lib/infopark/aws_utils.rb
+- lib/infopark/aws_utils/env.rb
+- lib/infopark/aws_utils/version.rb
+- spec/env_spec.rb
+- spec/spec_helper.rb
+homepage: 
+licenses:
+- UNLICENSED
+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.4.5.1
+signing_key: 
+specification_version: 4
+summary: A utility lib to ease the use of the AWS SDK
+test_files: []