Contents

module Jets
  class AwsInfo
    extend Memoist
    include AwsServices

    def region
      return 'us-east-1' if test?

      return ENV['JETS_AWS_REGION'] if ENV['JETS_AWS_REGION'] # highest precedence
      return ENV['AWS_REGION'] if ENV['AWS_REGION']

      region = nil

      # First try to get it from the ~/.aws/config
      region = `aws configure get region 2>&1`.strip rescue nil
      exit_code = $?.exitstatus
      if exit_code != 0
        exception_message = region.split("\n").grep(/botocore\.exceptions/).first
        if exception_message
          puts "WARN: #{exception_message}".colorize(:yellow)
        else
          # show full message as warning
          puts region.colorize(:yellow)
        end
        puts "You can also get rid of this message by setting AWS_REGION or configuring ~/.aws/config with the region"
        region = nil
      end
      region = nil if region == ''
      return region if region

      # Second try the metadata endpoint, should be available on AWS Lambda environment
      # https://stackoverflow.com/questions/4249488/find-region-from-within-an-ec2-instance
      begin
        az = `curl -s --max-time 3 --connect-timeout 5 http://169.254.169.254/latest/meta-data/placement/availability-zone`
        region = az.strip.chop # remove last char
        region = nil if region == ''
      rescue
      end
      return region if region

      'us-east-1' # default if all else fails
    end
    memoize :region

    # aws sts get-caller-identity
    def account
      return '123456789' if test?
      # ensure region set, required for sts.get_caller_identity.account to work
      ENV['AWS_REGION'] ||= region
      sts.get_caller_identity.account
    end
    memoize :account

    def test?
      ENV['TEST'] || ENV['CIRCLECI']
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
jets-0.10.4	lib/jets/aws_info.rb