Contents

# frozen_string_literal: true

require 'aws-sdk-ec2'

module Capistrano
  # Patches the Capistrano Configuration class to add our aws-ec2 features
  class Configuration
    EC2_STATE_RUNNING_CODE = 16

    def ec2_instances(running_only: true, tags: {})
      ec2_resource.instances.filter do |instance|
        # Filter out non-running instances
        next false if running_only && instance.state.code != EC2_STATE_RUNNING_CODE

        # Filter out instances not matching the given tags
        tags.all? { |k, v| i.tags.any? { |tag| tag.key == k && tag.value == v } }
      end
    end

    private

    def ec2_resource
      @ec2_resource ||= ::Aws::EC2::Resource.new(region: aws_region, credentials: aws_credentials)
    end

    def aws_region
      fetch(:aws_region, ENV['AWS_REGION'] || ENV['AWS_DEFAULT_REGION'])
    end

    def aws_credentials
      @aws_credentials ||= ::Aws::Credentials.new(aws_access_key_id, aws_secret_access_key, aws_session_token)
    end

    def aws_access_key_id
      fetch(:aws_access_key_id, ENV["#{stage.upcase}_ACCESS_KEY_ID"] || ENV['AWS_ACCESS_KEY_ID'])
    end

    def aws_secret_access_key
      fetch(:aws_secret_access_key, ENV["#{stage.upcase}_SECRET_ACCESS_KEY"] || ENV['AWS_SECRET_ACCESS_KEY'])
    end

    def aws_session_token
      fetch(:aws_session_token, ENV["#{stage.upcase}_SESSION_TOKEN"])
    end

    def stage
      fetch(:stage)
    end
  end

  module DSL
    module Env
      def_delegators :env, :ec2_instances
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
capistrano-aws-ec2-0.1.7	lib/capistrano/aws/ec2/patches.rb