elbas 0.27.0 → 3.0.0

data/lib/elbas/aws/taggable.rb

@@ -0,0 +1,18 @@
+module Elbas
+  module AWS
+    module Taggable
+      def tag(key, value)
+        @tags ||= {}
+
+        Elbas::Retryable.times(3).delay(5) do
+          aws_counterpart.create_tags tags: [{ key: key, value: value }]
+          @tags[key] = value
+        end
+      end
+
+      def tags
+        @tags || {}
+      end
+    end
+  end
+end

data/lib/elbas/capistrano.rb

@@ -1,26 +1,28 @@
-require 'aws-sdk-v1'
 require 'capistrano/dsl'
 
 load File.expand_path("../tasks/elbas.rake", __FILE__)
 
 def autoscale(groupname, *args)
   include Capistrano::DSL
-  include Elbas::AWS::AutoScaling
+  include Elbas::Logger
 
-  autoscale_group   = autoscaling.groups[groupname]
-  running_instances = autoscale_group.ec2_instances.filter('instance-state-name', 'running')
+  set :aws_autoscale_group_name, groupname
 
-  set :aws_autoscale_group, groupname
+  asg = Elbas::AWS::AutoscaleGroup.new groupname
+  instances = asg.instances.running
 
-  running_instances.each do |instance|
-    hostname = instance.dns_name || instance.private_ip_address
-    p "ELBAS: Adding server: #{hostname}"
-    server(hostname, *args)
+  instances.each do |instance|
+    info "Adding server: #{instance.hostname}"
+    server instance.hostname, *args
   end
 
-  if running_instances.count > 0
-    after('deploy', 'elbas:scale')
+  if instances.any?
+    after 'deploy', 'elbas:deploy'
   else
-    p "ELBAS: AMI could not be created because no running instances were found. Is your autoscale group name correct?"
+    error <<~MESSAGE
+      Could not create AMI because no running instances were found in the specified
+      AutoScale group. Ensure that the AutoScale group name is correct and that
+      there is at least one running instance attached to it.
+    MESSAGE
   end
 end

data/lib/elbas/errors/no_launch_template.rb

@@ -0,0 +1,6 @@
+module Elbas
+  module Errors
+    class NoLaunchTemplate < StandardError
+    end
+  end
+end

data/lib/elbas/logger.rb

@@ -1,7 +1,15 @@
+require 'capistrano/doctor/output_helpers'
+
 module Elbas
   module Logger
+    include Capistrano::Doctor::OutputHelpers
+
     def info(message)
-      p "ELBAS: #{message}"
+      $stdout.puts [prefix, message, "\n"].join
+    end
+
+    def prefix
+      @prefix ||= color.colorize("\n[ELBAS] ", :cyan)
     end
   end
-end
+end

data/lib/elbas/retryable.rb

@@ -1,16 +1,41 @@
 module Elbas
-  module Retryable
-    def with_retry(max: fetch(:elbas_retry_max, 3), delay: fetch(:elbas_retry_delay, 5))
-      tries ||= 0
-      tries += 1
-      yield
+  class Retryable
+    include Elbas::Logger
+
+    def initialize
+      @max = 0
+      @delay = 0
+    end
+
+    def times(max, &block)
+      @max = max
+      run block if block_given?
+    end
+
+    def delay(seconds, &block)
+      @delay = seconds
+      run block if block_given?
+    end
+
+    def run(proc)
+      attempts ||= 0
+      attempts += 1
+      proc.call
     rescue => e
-      p "Rescued #{e.message}"
-      if tries < max
-        p "Retrying in #{delay} seconds..."
-        sleep delay
+      info "Rescued error in retryable action: #{e.message}"
+      if attempts < @max
+        info "Retrying in #{@delay} seconds..."
+        sleep @delay
         retry
       end
     end
+
+    def self.times(max, &block)
+      new.tap { |r| r.times max }
+    end
+
+    def self.delay(seconds, &block)
+      new.tap { |r| r.delay seconds }
+    end
   end
 end

data/lib/elbas/tasks/elbas.rake

@@ -1,17 +1,26 @@
 require 'elbas'
+include Elbas::Logger
 
 namespace :elbas do
-  task :scale do
-    set :aws_access_key_id,     fetch(:aws_access_key_id,     ENV['AWS_ACCESS_KEY_ID'])
-    set :aws_secret_access_key, fetch(:aws_secret_access_key, ENV['AWS_SECRET_ACCESS_KEY'])
+  task :deploy do
+    asg = Elbas::AWS::AutoscaleGroup.new fetch(:aws_autoscale_group_name)
 
-    Elbas::AMI.create do |ami|
-      p "ELBAS: Created AMI: #{ami.aws_counterpart.id}"
-      Elbas::LaunchConfiguration.create(ami) do |lc|
-        p "ELBAS: Created Launch Configuration: #{lc.aws_counterpart.name}"
-        lc.attach_to_autoscale_group!
-      end
+    info "Creating AMI from a running instance..."
+    ami = Elbas::AWS::AMI.create asg.instances.running.sample
+    ami.tag 'ELBAS-Deploy-group', asg.name
+    ami.tag 'ELBAS-Deploy-id', env.timestamp.to_i.to_s
+    info  "Created AMI: #{ami.id}"
+
+    info "Updating launch template with the new AMI..."
+    launch_template = asg.launch_template.update ami
+    info "Updated launch template, new default version = #{launch_template.version}"
+
+    info "Cleaning up old AMIs..."
+    ami.ancestors.each do |ancestor|
+      info "Deleting old AMI: #{ancestor.id}"
+      ancestor.delete
     end
 
+    info "Deployment complete!"
   end
-end
+end

data/lib/elbas/version.rb

@@ -1,3 +1,3 @@
 module Elbas
-  VERSION = '0.27.0'.freeze
+  VERSION = '3.0.0'.freeze
 end

data/spec/aws/ami_spec.rb

@@ -0,0 +1,140 @@
+describe Elbas::AWS::AMI do
+  subject { Elbas::AWS::AMI.new 'test' }
+
+  describe '#initialize' do
+    it 'sets the id' do
+      expect(subject.id).to eq 'test'
+    end
+
+    it 'has an aws-sdk counterpart' do
+      expect(subject.aws_counterpart).to be_a_kind_of ::Aws::EC2::Image
+      expect(subject.aws_counterpart.id).to eq 'test'
+    end
+
+    context 'with snapshots' do
+      subject do
+        Elbas::AWS::AMI.new 'test', [
+          double(:bdm, ebs: double(:ebs, snapshot_id: 'snap-1'))
+        ]
+      end
+
+      it 'sets snapshots to Snapshot objects' do
+        expect(subject.snapshots.size).to eq 1
+        expect(subject.snapshots[0]).to be_a_kind_of Elbas::AWS::Snapshot
+      end
+
+      it 'sets the ID on the Snapshots' do
+        expect(subject.snapshots[0].id).to eq 'snap-1'
+      end
+    end
+  end
+
+  describe '#deploy_id' do
+    it 'returns the ELBAS-Deploy-id tag, if set' do
+      webmock :post, /ec2/ => 201, with: Hash[body: /Action=CreateTags/]
+      subject.tag 'ELBAS-Deploy-id', 'test'
+      expect(subject.deploy_id).to eq 'test'
+    end
+
+    it 'returns nil if the tag was never set' do
+      expect(subject.deploy_id).to be_nil
+    end
+  end
+
+  describe '#deploy_group' do
+    it 'returns the ELBAS-Deploy-group tag, if set' do
+      webmock :post, /ec2/ => 201, with: Hash[body: /Action=CreateTags/]
+      subject.tag 'ELBAS-Deploy-group', 'test'
+      expect(subject.deploy_group).to eq 'test'
+    end
+
+    it 'returns nil if the tag was never set' do
+      expect(subject.deploy_group).to be_nil
+    end
+  end
+
+  describe '#ancestors' do
+    before do
+      webmock :post, /ec2/ => 201, with: Hash[body: /Action=CreateTags/]
+      subject.tag 'ELBAS-Deploy-group', 'test'
+      subject.tag 'ELBAS-Deploy-id', 'test'
+
+      webmock :post, %r{ec2.(.*).amazonaws.com\/\z} => 'DescribeImages.200.xml',
+        with: Hash[body: /Action=DescribeImages/]
+    end
+
+    it 'includes AMIs from the same deploy group, different deploy ID' do
+      expect {
+        subject.tag 'ELBAS-Deploy-id', 'not-test'
+      }.to change {
+        subject.ancestors.size
+      }.by 1
+    end
+  end
+
+  describe '#delete' do
+    before do
+      webmock :post, /ec2/ => 201, with: Hash[body: /Action=DeregisterImage/]
+    end
+
+    it 'calls the deregister AMI API' do
+      subject.delete
+      expect(WebMock)
+        .to have_requested(:post, /ec2/)
+        .with body: /Action=DeregisterImage&ImageId=test/
+    end
+
+    context 'with snapshots' do
+      subject do
+        Elbas::AWS::AMI.new 'test', [
+          double(:bdm, ebs: double(:ebs, snapshot_id: 'snap-1'))
+        ]
+      end
+
+      it 'deletes the AMIs snapshots too' do
+        webmock :post, /ec2/ => 201, with: Hash[body: /Action=DeleteSnapshot/]
+
+        subject.delete
+        expect(WebMock)
+          .to have_requested(:post, /ec2/)
+          .with body: /Action=DeleteSnapshot&SnapshotId=snap-1/
+      end
+    end
+  end
+
+  describe '.create' do
+    subject { described_class }
+    let(:instance) { Elbas::AWS::Instance.new 'i-1234567890', nil, nil }
+
+    before do
+      webmock :post, %r{ec2.(.*).amazonaws.com\/\z} => 'CreateImage.200.xml',
+        with: Hash[body: /Action=CreateImage/]
+    end
+
+    it 'calls the API with the instance given' do
+      subject.create instance
+      expect(WebMock)
+        .to have_requested(:post, /ec2/)
+        .with body: /Action=CreateImage&InstanceId=i-1234567890&Name=ELBAS-ami-(\d+)/
+    end
+
+    it 'uses the no_reboot option by default' do
+      subject.create instance
+      expect(WebMock)
+        .to have_requested(:post, /ec2/)
+        .with(body: /NoReboot=true/)
+    end
+
+    it 'uses no_reboot as false, if given' do
+      subject.create instance, no_reboot: false
+      expect(WebMock)
+        .to have_requested(:post, /ec2/)
+        .with(body: /NoReboot=false/)
+    end
+
+    it 'returns the an AMI object with the new id' do
+      ami = subject.create instance
+      expect(ami.id).to eq 'ami-4fa54026' # from stub
+    end
+  end
+end

data/spec/aws/autoscale_group_spec.rb

@@ -0,0 +1,53 @@
+describe Elbas::AWS::AutoscaleGroup do
+  subject { Elbas::AWS::AutoscaleGroup.new 'test-asg' }
+
+  before do
+    webmock :post, %r{autoscaling.(.*).amazonaws.com\/\z} => 'DescribeAutoScalingGroups.200.xml',
+      with: Hash[body: /Action=DescribeAutoScalingGroups/]
+
+    webmock :post, %r{ec2.(.*).amazonaws.com\/\z} => 'DescribeInstances.200.xml',
+      with: Hash[body: /Action=DescribeInstances/]
+  end
+
+  describe '#initialize' do
+    it 'sets the name' do
+      expect(subject.name).to eq 'test-asg'
+    end
+
+    it 'has an aws-sdk counterpart' do
+      expect(subject.aws_counterpart).to be_a_kind_of ::Aws::AutoScaling::Types::AutoScalingGroup
+      expect(subject.aws_counterpart.auto_scaling_group_name).to eq 'test-asg'
+    end
+  end
+
+  describe '#instance_ids' do
+    it 'returns every instance ID in the ASG' do
+      expect(subject.instance_ids).to eq ['i-1234567890', 'i-500']
+    end
+  end
+
+  describe '#instances' do
+    it 'returns an instance collection with all instances' do
+      instances = subject.instances
+      expect(instances.count).to eq 2
+    end
+  end
+
+  describe '#launch_template' do
+    it 'throws an error if there is no launch template' do
+      expect { subject.launch_template }.to raise_error(Elbas::Errors::NoLaunchTemplate)
+    end
+
+    it 'returns a LaunchTemplate object with the id/name/version set' do
+      allow(subject.aws_counterpart).to receive(:launch_template) do
+        double(:lt, launch_template_id: 'test-1', launch_template_name: 'Test', version: '$Latest')
+      end
+
+      expect(subject.launch_template.id).to eq 'test-1'
+      expect(subject.launch_template.name).to eq 'Test'
+      expect(subject.launch_template.version).to eq '$Latest'
+    end
+
+
+  end
+end

data/spec/aws/instance_collection_spec.rb

@@ -0,0 +1,28 @@
+describe Elbas::AWS::InstanceCollection do
+  subject { Elbas::AWS::InstanceCollection.new ['i-1234567890', 'i-500'] }
+
+  before do
+    webmock :post, %r{ec2.(.*).amazonaws.com\/\z} => 'DescribeInstances.200.xml',
+      with: Hash[body: /Action=DescribeInstances/]
+  end
+
+  describe '#instances' do
+    it 'returns Instance objects with name/hostname/state' do
+      expect(subject.instances[0].id).to eq 'i-1234567890'
+      expect(subject.instances[0].hostname).to eq 'ec2-1234567890.amazonaws.com'
+      expect(subject.instances[0].state).to eq 16
+
+      expect(subject.instances[1].id).to eq 'i-500'
+      expect(subject.instances[1].hostname).to eq 'ec2-500.amazonaws.com'
+      expect(subject.instances[1].state).to eq 32
+    end
+  end
+
+  describe '#running' do
+    it 'returns only running instances' do
+      expect(subject.instances.size).to eq 2
+      expect(subject.running.size).to eq 1
+      expect(subject.running[0].id).to eq 'i-1234567890'
+    end
+  end
+end

data/spec/aws/instance_spec.rb

@@ -0,0 +1,30 @@
+describe Elbas::AWS::Instance do
+  subject { Elbas::AWS::Instance.new 'i-1234567890', 'ec2-1234567890.amazonaws.com', 16 }
+
+  describe '#initialize' do
+    it 'sets the AWS counterpart' do
+      expect(subject.aws_counterpart).to be_a_kind_of ::Aws::EC2::Instance
+      expect(subject.aws_counterpart.id).to eq 'i-1234567890'
+    end
+  end
+
+  describe '#hostname' do
+    it 'returns the public DNS' do
+      expect(subject.hostname).to eq 'ec2-1234567890.amazonaws.com'
+    end
+  end
+
+  describe '#running?' do
+    it 'returns true if the state code is 16 ("running")' do
+      expect(subject).to receive(:state) { 16 }
+      expect(subject).to be_running
+    end
+
+    it 'returns false for every other state code' do
+      [0, 32, 48, 64, 80].each do |code|
+        expect(subject).to receive(:state) { code }
+        expect(subject).to_not be_running
+      end
+    end
+  end
+end

data/spec/aws/launch_template_spec.rb

@@ -0,0 +1,52 @@
+describe Elbas::AWS::LaunchTemplate do
+  subject { Elbas::AWS::LaunchTemplate.new 'test-lt', 'test', '1' }
+
+  before do
+    webmock :post, %r{ec2.(.*).amazonaws.com\/\z} => 'CreateLaunchTemplateVersion.200.xml',
+      with: Hash[body: /Action=CreateLaunchTemplateVersion/]
+  end
+
+  describe '#initialize' do
+    it 'sets the id' do
+      expect(subject.id).to eq 'test-lt'
+    end
+
+    it 'sets the name' do
+      expect(subject.name).to eq 'test'
+    end
+
+    it 'sets the version' do
+      expect(subject.version).to eq '1'
+    end
+  end
+
+  describe '#update' do
+    it 'hits the CreateLaunchTemplateVersion API' do
+      subject.update double(:ami, id: 'ami-123')
+      expect(WebMock)
+        .to have_requested(:post, /ec2/)
+        .with(body: %r{Action=CreateLaunchTemplateVersion})
+    end
+
+    it 'creates a new launch template from the given AMI' do
+      subject.update double(:ami, id: 'ami-123')
+      expect(WebMock)
+        .to have_requested(:post, /ec2/)
+        .with(body: %r{LaunchTemplateData.ImageId=ami-123})
+    end
+
+    it 'uses itself as the source' do
+      subject.update double(:ami, id: 'ami-123')
+      expect(WebMock)
+        .to have_requested(:post, /ec2/)
+        .with(body: %r{LaunchTemplateId=test-lt&SourceVersion=1})
+    end
+
+    it 'returns a new launch template' do
+      launch_template = subject.update double(:ami, id: 'ami-123')
+      expect(launch_template.id).to eq 'lt-1234567890'
+      expect(launch_template.name).to eq 'elbas-test'
+      expect(launch_template.version).to eq 123
+    end
+  end
+end