Contents

require_relative "includes"
require_relative "logging"

class AwsAssumeRole::Runner
    include AwsAssumeRole::Logging
    extend Dry::Initializer

    param :command, Dry::Types["coercible.array"].member(Dry::Types["strict.string"])
    option :exit_on_error, Dry::Types["strict.bool"], default: proc { true }
    option :expected_exit_code, Dry::Types["strict.int"], default: proc { 0 }
    option :environment, Dry::Types["strict.hash"], default: proc { {} }
    option :credentials, optional: true

    def initialize(params, options = {})
        super(params, options)
        command_to_exec = command.join(" ")
        process_credentials unless credentials.blank?
        system environment, command_to_exec
        exit_status = $CHILD_STATUS.exitstatus
        process_error(exit_status) if exit_status != expected_exit_code
    end

    private

    def process_credentials
        cred_env = {
            "AWS_ACCESS_KEY_ID" => credentials.credentials.access_key_id,
            "AWS_SECRET_ACCESS_KEY" => credentials.credentials.secret_access_key,
            "AWS_SESSION_TOKEN" => credentials.credentials.session_token,
        }
        @environment = environment.merge cred_env
    end

    def process_error(exit_status)
        logger.error "#{command} failed with #{exit_status}"
        exit exit_status if exit_on_error
        raise "#{command} failed with #{exit_status}"
    end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
aws_assume_role-0.1.1	lib/aws_assume_role/runner.rb
aws_assume_role-0.1.0	lib/aws_assume_role/runner.rb