prima-twig 0.13.1 → 0.13.37

data/bin/twig-update-ami

@@ -0,0 +1,197 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require_relative '../lib/prima_twig.rb'
+require_relative '../lib/prima_aws_client.rb'
+require 'launchy'
+require 'json'
+require 'aws-sdk-s3'
+
+class TwigUpdateAmi
+  include Command
+  include PrimaAwsClient
+  def initialize
+    output 'Controllo se ci sono aggiornamenti da fare...'
+    exec "gem update prima-twig && twig update-ami #{ARGV.join ' '}" unless `gem outdated`.lines.grep(/^prima-twig \(.*\)/).empty?
+    @s3 = Aws::S3::Client.new
+    @s3_bucket = 'prima-deploy'
+    @templates_base_url = "https://s3-eu-west-1.amazonaws.com"
+  end
+
+  def execute!(args)
+      update_amis(args[0], args[1], args[2], args[3], args[4])
+  end
+
+  private
+
+  def update_amis(ami_template, ami_id, ami_name, ami_description, env)
+    output "updating instance definition #{ami_template}".light_green
+    Dir.chdir 'ami'
+    update_instance_name(ami_id, ami_name, ami_description, ami_template)
+    output 'running packer update (this could take some time)'.light_green
+    new_ami_id = update_packer(ami_template, env)
+    # new_ami_id = 'ami-026890988d91ee8c6'
+    Dir.chdir '..'
+    stop_if(new_ami_id.to_s.empty?, 'Failed to generate AMI!'.red)
+    output "new ami id: #{new_ami_id}"
+
+    output 'searching for ami to update...'
+    ami_mappings = JSON.parse(@s3.get_object(bucket: @s3_bucket, key: "ami/ami-mappings.json")["body"].read())
+    old_amis = update_ami_mappings(ami_mappings, ami_template, env, new_ami_id)
+    stop_if(old_amis.empty?, "No ami to update! No #{ami_template} in env #{env}, exiting".yellow)
+
+    output "retrieving stacks that uses old ami ids: #{old_amis}"
+    exports = list_exports()
+    stacks = get_stacks_from_exports(exports, old_amis)
+    stop_if(stacks.empty?, "No stack to update found! This means that ami-mapping file is not in sync, please check manually")
+
+    stacks.each do |stack|
+      output "processing stack #{stack}"
+      if stack.include?('qa')
+        output "skipping stack #{stack} because is a qa"
+        next
+      else
+        stack_tags = tags_to_hashes(get_stack_tags(stack))
+        stack_tags['TemplateVersion'] = stack_tags['TemplateVersion'].to_i + 1
+
+        if stack.include?('batch')
+          stack_parameters = update_stack_parameters(get_stack_parameters(stack),
+            [
+              { parameter_key: 'AMIID', parameter_value: new_ami_id },
+              { parameter_key: 'TemplateVersion', parameter_value: stack_tags['TemplateVersion'].to_s }
+            ]
+          )
+          if stack.include?('offsite-backups')
+            stack_template = File.read("./cloudformation/stacks/batch/compute-environment-offsite-backups.yml")
+          else
+            stack_template = File.read("./cloudformation/stacks/batch/compute-environment.yml")
+          end
+        else
+          stack_parameters = update_stack_parameters(get_stack_parameters(stack),
+            [
+              { parameter_key: 'AMIID', parameter_value: new_ami_id },
+              { parameter_key: 'DesiredCapacity', parameter_value: get_desired_capacity(stack).to_s },
+              { parameter_key: 'TemplateVersion', parameter_value: stack_tags['TemplateVersion'].to_s }
+            ]
+          )
+          stack_template = File.read("./cloudformation/stacks/asg/#{stack.to_s.split("/")[1]}.yml")
+        end
+        update_stack(stack, stack_template, stack_parameters, hashes_to_tags(stack_tags))
+      end
+    end
+
+    stacks.each do |stack|
+      if stack.include?('qa')
+        next
+      end
+      wait_for_stack_ready(stack, ['CREATE_FAILED', 'ROLLBACK_IN_PROGRESS', 'ROLLBACK_FAILED', 'DELETE_IN_PROGRESS', 'DELETE_FAILED', 'DELETE_COMPLETE', 'UPDATE_ROLLBACK_FAILED', 'UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS', 'UPDATE_ROLLBACK_COMPLETE', 'ROLLBACK_COMPLETE'])
+    end
+
+    output 'writing new ami mapping'
+    File.open("ami/ami-mappings.json", 'w+') do |f|
+      mapping_file = JSON.pretty_generate(ami_mappings)
+      f.write(mapping_file)
+      @s3.put_object(bucket: @s3_bucket, key: "ami/ami-mappings.json", body: mapping_file)
+    end
+
+    output 'Update finished! ( ͡° ͜ʖ ͡°)'
+  end
+
+  def get_stacks_from_exports(exports, old_amis)
+    stacks = []
+    old_amis.each do |old_ami|
+      exports.each do |export|
+        if export.value.eql?(old_ami)
+          stacks.insert(0,export.exporting_stack_id)
+        end
+      end
+    end
+    stacks
+  end
+
+  def update_ami_mappings(mappings, ami_template, env, new_ami_id)
+    old_values = []
+    mappings.each do |item|
+       if item['ami_template'].eql?(ami_template) and item['env'].eql?(env)
+         old_values.insert(0,item['ami_id'])
+         item['ami_id'] = new_ami_id
+       end
+     end
+    old_values.uniq
+  end
+
+  def update_stack_parameters(stack_parameters, new_parameters)
+    new_parameters.each do |new_param|
+      stack_parameters.reject{ |k| k["parameter_key"] == new_param["parameter_key"] }
+      stack_parameters.push(new_param)
+    end
+    stack_parameters
+  end
+
+  def update_instance_name(ami_id, ami_name, ami_description, ecs_json_path)
+    ecs_json = JSON.parse File.read(ecs_json_path)
+
+    ecs_json['builders'][0]['source_ami'] = ami_id
+    ecs_json['builders'][0]['ami_name'] = ami_name
+    ecs_json['builders'][0]['ami_description'] = ami_description
+
+    File.open ecs_json_path, 'w' do |f|
+      f.write(JSON.pretty_generate(ecs_json))
+    end
+  end
+
+  def get_desired_capacity(stack_name)
+    stack_outputs = get_stack_outputs(stack_name)
+    stack_outputs.each do |out|
+      if out.export_name.include?('EC2Fleet') or out.export_name.include?('AutoScalingGroup')
+        return get_autoscaling_capacity(out.output_value)
+      end
+    end
+  end
+
+  def update_packer(json_filename, env)
+    execute_command "AWS_MAX_ATTEMPTS=90 AWS_POLL_DELAY_SECONDS=60 packer build -var datadog_apikey=`biscuit get -f ../configs/secrets/common.yml common_production_apikey_datadog` -var github_token=`biscuit get -f ../configs/secrets/common.yml common_private_repo_github_token` -var drone_key=\"`biscuit get -f ../configs/secrets/common.yml drone_license_key`\" -var env=#{env} -machine-readable ./#{json_filename} | tee build.log"
+    `grep 'artifact,0,id' build.log | cut -d, -f6 | cut -d: -f2`.sub(/\n/, '')
+  end
+
+  def help_content
+    <<-HELP
+
+    twig-update-ami
+    ===========
+
+    Updates ami version and applies it to stacks on cloudformation
+
+    Synopsis
+    --------
+
+      twig-update-ami
+
+    Description
+    -----------
+
+      from artemide main folder run
+      `twig-update-ami ${AMI_TEMPLATE} ${AMI_ID} ${AMI_NAME} ${AMI_DESCRIPTION} ${ENV}`
+
+    Subcommand for Twig: <http://rondevera.github.io/twig/>
+    Author: Eugenio Laghi <https://github.com/eugeniolaghi>
+
+    HELP
+  end
+
+  class ::Hash
+      def deep_merge(second)
+          merger = proc { |key, v1, v2| Hash === v1 && Hash === v2 ? v1.merge(v2, &merger) : Array === v1 && Array === v2 ? v1 | v2 : [:undefined, nil, :nil].include?(v2) ? v1 : v2 }
+          self.merge(second.to_h, &merger)
+      end
+  end
+
+  args = ARGV.dup
+
+  if args.include?('--help')
+    puts help_content
+    exit
+  end
+
+  TwigUpdateAmi.new.execute!(args)
+end

data/lib/command.rb

@@ -1,60 +1,73 @@
 require 'colorize'
+require 'English'
 require 'optparse'
 
- module Command
-  def exec_step command, output = nil
+module Command
+  # executes command and exits if status != 0
+  def exec_step(command, output = nil)
     if output
       puts output
     else
-      puts "exec > ".green + "#{command}".yellow
+      puts 'exec > '.green + command.to_s.yellow
     end
     `#{command}`
-    exit($?.exitstatus) unless 0 === $?.exitstatus
+    exit($CHILD_STATUS.exitstatus) unless $CHILD_STATUS.exitstatus.zero?
   end
 
-  def output msg
-    puts "twig binaries > ".black + msg
+  def output(msg)
+    puts 'twig binaries > '.black + msg
   end
 
-  def stop_if check, msg
+  def stop_if(check, msg, command = '')
     if check
-      output_msg = case msg
-      when Symbol
-        symbol_message msg
-      when Array
-        array_message msg
-      else
-        msg
-      end
-      puts "there was a problem > ".red + output_msg
+      output_msg =
+        case msg
+        when Symbol
+          symbol_message msg
+        when Array
+          array_message msg
+        else
+          msg
+        end
+      command.empty? || exec(command)
+      puts 'there was a problem > '.red + output_msg
       exit(1)
     end
   end
 
-  def symbol_message s
+  def symbol_message(s)
     case s
     when :clean
-      "hai dei file non committati...non posso continuare'"
+      'hai dei file non committati...non posso continuare'
     when :detached_head
       "repo in stato 'head detached'"
     when :wrong_args
-      "argomento non corretto"
+      'argomento non corretto'
     end
   end
 
-  def array_message arr
+  def array_message(arr)
     case arr[0]
     when :wrong_args
       msg = symbol_message arr[0]
       msg += "\n"
-      msg += "  valore possibile: "
-      values = "[ " + arr[1].join(" | ").yellow + " ]"
+      msg += '  valore possibile: '
+      values = '[ ' + arr[1].join(' | ').yellow + ' ]'
       msg += values
       msg
     end
   end
 
-  def stop_unless check, msg
-    stop_if (not check), msg
+  def stop_unless(check, msg, command = '')
+    stop_if !check, msg, command
+  end
+
+  # executes command and returns properly colored output
+  def execute_command(cmd)
+    output "Eseguo #{cmd}".yellow
+    res = `#{cmd}`
+    color = $CHILD_STATUS.exitstatus.zero? ? 'green' : 'red'
+    output res.send color
+    stop_if (color == 'red'), "Errore durante la build dell'artifact".red
   end
 end

data/lib/prima_aws_client.rb

@@ -0,0 +1,494 @@
+require 'aws-sdk-autoscaling'
+require 'aws-sdk-batch'
+require 'aws-sdk-cloudformation'
+require 'aws-sdk-cloudfront'
+require 'aws-sdk-ec2'
+require 'aws-sdk-ecs'
+require 'aws-sdk-elasticloadbalancingv2'
+require 'aws-sdk-s3'
+require 'colorize'
+#
+module PrimaAwsClient
+  def s3_client
+    @s3 ||= Aws::S3::Client.new
+  end
+
+  def cf_client
+    @cf ||= Aws::CloudFormation::Client.new
+  end
+
+  def asg_client
+    @asg ||= Aws::AutoScaling::Client.new
+  end
+
+  def ec2_client
+    @ec2 ||= Aws::EC2::Client.new
+  end
+
+  def alb_client
+    @alb ||= Aws::ElasticLoadBalancingV2::Client.new
+  end
+
+  def ecs_client
+    @ecs ||= Aws::ECS::Client.new
+  end
+
+  def stack_list
+    stacks = []
+    next_token = ''
+    loop do
+      print '.'.yellow; STDOUT.flush
+      options = next_token != '' ? { next_token: next_token } : {}
+      begin
+        resp = cf_client.describe_stacks(options)
+      rescue Aws::CloudFormation::Errors::Throttling => e
+        output 'Throttling, retrying in 15 seconds'.red
+        sleep 15
+        resp = cf_client.describe_stacks(options)
+      end
+      stacks += resp.stacks
+      break unless resp.next_token
+      next_token = resp.next_token
+    end
+    puts '.'.yellow; STDOUT.flush
+    stacks.keep_if { |stack| stack.stack_name.include? '-qa-' }
+    stacks
+  end
+
+  def cluster_list
+    stacks = []
+    next_token = ''
+    loop do
+      print '.'.yellow; STDOUT.flush
+      options = next_token != '' ? { next_token: next_token } : {}
+      begin
+        resp = cf_client.describe_stacks(options)
+      rescue Aws::CloudFormation::Errors::Throttling => e
+        output 'Throttling, retrying in 15 seconds'.red
+        sleep 15
+        resp = cf_client.describe_stacks(options)
+      end
+      stacks += resp.stacks
+      break unless resp.next_token
+      next_token = resp.next_token
+    end
+    puts '.'.yellow; STDOUT.flush
+    stacks.keep_if { |stack| stack.stack_name.include? 'ecs-cluster-qa-' }
+    stacks
+  end
+
+  def list_exports
+    exports = []
+    next_token = ''
+    loop do
+      print '.'.yellow; STDOUT.flush
+      options = next_token != '' ? { next_token: next_token } : {}
+      begin
+        resp = cf_client.list_exports(options)
+      rescue Aws::CloudFormation::Errors::Throttling => e
+        output 'Throttling, retrying in 15 seconds'.red
+        sleep 15
+        resp = cf_client.list_exports(options)
+      end
+      exports += resp.exports
+      break unless resp.next_token
+      next_token = resp.next_token
+    end
+    puts '.'.yellow; STDOUT.flush
+    exports
+  end
+
+  def create_stack(stack_name, stack_body, parameters = [], tags = [], role = nil)
+    cf_args = {
+      stack_name: stack_name,
+      template_body: stack_body,
+      parameters: parameters,
+      tags: tags,
+      capabilities: ['CAPABILITY_IAM'],
+      on_failure: 'ROLLBACK'
+    }
+
+    if role != nil then
+      cf_args.merge!(role_arn: role)
+    end
+
+    begin
+      cf_client.create_stack(cf_args)
+    rescue Aws::CloudFormation::Errors::Throttling, Aws::CloudFormation::Errors::LimitExcedeedException => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      create_stack(stack_name, stack_body, parameters = [], tags = [])
+    else
+      output "La creazione dello stack #{stack_name} è stata avviata".green
+    end
+  end
+
+  def update_stack(stack_name, template_body, parameters = [], tags = [], role = nil)
+    cf_args = {
+      stack_name: stack_name,
+      template_body: template_body,
+      parameters: parameters,
+      tags: tags,
+      capabilities: ['CAPABILITY_IAM']
+    }
+
+    if role != nil then
+      cf_args.merge!(role_arn: role)
+    end
+
+    begin
+      cf_client.update_stack(cf_args)
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      update_stack(stack_name, template_body, parameters = [], tags = [])
+    rescue Aws::CloudFormation::Errors::ValidationError => e
+      raise e
+    else
+      output "L'update dello stack #{stack_name} è stato avviato".green
+    end
+  end
+
+  def update_stack_url(stack_name, template_url, parameters = [], tags = [], role = nil)
+    cf_args = {
+      stack_name: stack_name,
+      template_url: template_url,
+      parameters: parameters,
+      tags: tags,
+      capabilities: ['CAPABILITY_IAM']
+    }
+
+    if role != nil then
+      cf_args.merge!(role_arn: role)
+    end
+
+    begin
+      cf_client.update_stack(cf_args)
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      update_stack_url(stack_name, template_url, parameters = [], tags = [])
+    rescue Aws::CloudFormation::Errors::ValidationError => e
+      raise e
+    else
+      output "L'update dello stack #{stack_name} è stato avviato".green
+    end
+  end
+
+  def stack_exists?(stack_name)
+    begin
+      cf_client.describe_stacks(stack_name: stack_name)
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      stack_exists?(stack_name)
+    rescue Aws::CloudFormation::Errors::ValidationError => e
+      return false if e.message.include? 'does not exist'
+      raise e
+    else
+      true
+    end
+  end
+
+  def delete_stack(stack_name)
+    begin
+      cf_client.delete_stack(stack_name: stack_name)
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      delete_stack(stack_name)
+    else
+      output "Stack #{stack_name} spenta con successo\n".green
+    end
+  end
+
+  def wait_for_stack_ready(stack_name, failed_statuses = ['CREATE_FAILED', 'ROLLBACK_IN_PROGRESS', 'ROLLBACK_FAILED', 'DELETE_IN_PROGRESS', 'DELETE_FAILED', 'DELETE_COMPLETE', 'UPDATE_ROLLBACK_FAILED', 'UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS'])
+    ready = false
+    sleep_seconds = 13
+    output "Attendo che lo stack #{stack_name} finisca di essere inizializzato...\n".yellow
+    while !ready
+      ready = true if stack_ready?(stack_name, failed_statuses)
+      seconds_elapsed = 0
+      loop do
+        break if seconds_elapsed >= sleep_seconds
+        print '.'.yellow; STDOUT.flush
+        sleep 1
+        seconds_elapsed += 1
+      end
+    end
+    output "\nStack #{stack_name} pronto!\n".green
+  end
+
+  def wait_for_stack_removal(stack_name)
+    ready = false
+    sleep_seconds = 13
+    sleep 10
+    output "Attendo che lo stack #{stack_name} finisca di essere cancellato...\n".yellow
+    while !ready
+      ready = true if stack_deleted?(stack_name)
+      seconds_elapsed = 0
+      loop do
+        break if seconds_elapsed >= sleep_seconds
+        print '.'.yellow; STDOUT.flush
+        sleep 1
+        seconds_elapsed += 1
+      end
+    end
+    output "\nStack #{stack_name} eliminato!\n".green
+  end
+
+  def get_stack_tags(name)
+    begin
+      resp = cf_client.describe_stacks(stack_name: name)
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      get_stack_tags(name)
+    else
+      resp.stacks[0].tags
+    end
+  end
+
+  def get_stack_parameters(name)
+    begin
+      resp = cf_client.describe_stacks(stack_name: name)
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      get_stack_parameters(name)
+    else
+      resp.stacks[0].parameters
+    end
+  end
+
+  def get_stack_outputs(name)
+    begin
+      resp = cf_client.describe_stacks(stack_name: name)
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      get_stack_outputs(name)
+    else
+      resp.stacks[0].outputs
+    end
+  end
+
+  def get_stack_template(name)
+    begin
+      resp = cf_client.get_template(stack_name: name)
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      get_stack_template(name)
+    else
+      resp.template_body
+    end
+
+  end
+
+  def stack_ready?(stack_name, failed_statuses = ['CREATE_FAILED', 'ROLLBACK_IN_PROGRESS', 'ROLLBACK_FAILED', 'DELETE_IN_PROGRESS', 'DELETE_FAILED', 'DELETE_COMPLETE', 'UPDATE_ROLLBACK_FAILED', 'UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS'])
+    begin
+      resp = cf_client.describe_stacks(
+        stack_name: stack_name
+      )
+      stack_status = resp.stacks[0].stack_status
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      print 'Throttling'.red; STDOUT.flush
+      return false
+    end
+    raise "The stack #{stack_name} errored out" if failed_statuses.include? stack_status
+    ['CREATE_COMPLETE', 'UPDATE_COMPLETE', 'UPDATE_ROLLBACK_COMPLETE', 'ROLLBACK_COMPLETE'].include? stack_status
+  end
+
+  def stack_deleted?(stack_name, failed_statuses = ['ROLLBACK_IN_PROGRESS', 'ROLLBACK_FAILED', 'DELETE_FAILED', 'UPDATE_ROLLBACK_FAILED', 'UPDATE_ROLLBACK_COMPLETE_CLEANUP_IN_PROGRESS'])
+    begin
+      resp = cf_client.describe_stacks(
+        stack_name: stack_name
+      )
+      stack_status = resp.stacks[0].stack_status
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      print 'Throttling'.red; STDOUT.flush
+      return false
+    rescue Aws::CloudFormation::Errors::ValidationError => e
+      print 'Stack deleted'
+      return true
+    end
+    raise "The stack #{stack_name} errored out" if failed_statuses.include? stack_status
+    ['DELETE_COMPLETE'].include? stack_status
+  end
+
+  def artifact_exists?(bucket, path)
+    resp = s3_client.list_objects(
+      bucket: bucket,
+      max_keys: 1,
+      prefix: path
+    )
+    !resp.contents.empty?
+  end
+
+  def upload_artifact(source_path, destination_path, bucket_name_override=nil)
+    output "Upload dell'artifact in corso (#{(File.size(source_path).to_f / 2**20).round(2)} MiB)\n".yellow
+    s3 = Aws::S3::Resource.new
+    s3_bucket = if !bucket_name_override.nil? then bucket_name_override else @s3_bucket end
+    puts s3_bucket
+    obj = s3.bucket(s3_bucket).object(destination_path)
+    obj.upload_file(source_path)
+
+    output "#{s3_bucket}/#{destination_path} uploadato con successo!\n".green
+  end
+
+  def wait_for_artifact(bucket, path)
+    ready = artifact_exists?(bucket, path)
+    sleep_seconds = 13
+    output "Attendo che sia pronto l'artefatto #{path}...\n".yellow
+    retries = 0
+    while !ready
+      ready = true if artifact_exists?(bucket, path)
+      seconds_elapsed = 0
+      loop do
+        break if seconds_elapsed >= sleep_seconds
+        print '.'.yellow; STDOUT.flush
+        sleep 1
+        seconds_elapsed += 1
+      end
+      retries += 1
+      if retries > 150
+        output "\n Timeout raggiunto aspettando #{path}\n".red
+        exit
+      end
+    end
+    output "\nArtefatto #{path} creato!\n".green
+  end
+
+  def list_import_stacks(export_name)
+    stacks = []
+    next_token = ''
+    loop do
+      print '.'.yellow; STDOUT.flush
+      options = next_token != '' ? { export_name: export_name, next_token: next_token } : {export_name: export_name}
+      begin
+        resp = cf_client.list_imports(options)
+      rescue Aws::CloudFormation::Errors::Throttling => e
+        output 'Throttling, retrying in 15 seconds'.red
+        sleep 15
+        resp = cf_client.list_imports(options)
+      end
+      stacks += resp.imports
+      break unless resp.next_token
+      next_token = resp.next_token
+    end
+    stacks
+  end
+
+  def describe_stack_resource(cluster_stack_name, logical_resource_id)
+    begin
+      resp = cf_client.describe_stack_resource({stack_name: cluster_stack_name, logical_resource_id: logical_resource_id})
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      resp = describe_stack_resource(cluster_stack_name, logical_resource_id)
+    end
+  end
+
+  def describe_instances(instance_ids)
+    begin
+      resp = ec2_client.describe_instances({instance_ids: instance_ids})
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      resp = describe_instances(instance_ids)
+    end
+  end
+
+  def describe_auto_scaling_groups(auto_scaling_group_names, max_records)
+    begin
+      resp = asg_client.describe_auto_scaling_groups({
+        auto_scaling_group_names: auto_scaling_group_names,
+        max_records: max_records
+      })
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      resp = describe_auto_scaling_groups(auto_scaling_group_names, max_records)
+    end
+  end
+
+  def describe_load_balancers(load_balancer_arns)
+    begin
+      resp = alb_client.describe_load_balancers({load_balancer_arns: load_balancer_arns})
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      resp = describe_load_balancers(load_balancer_arns)
+    end
+  end
+
+  def update_ecs_service(cluster, service, deployment_configuration)
+    begin
+      resp = ecs_client.update_service(
+        cluster: cluster,
+        service: service,
+        deployment_configuration: deployment_configuration
+      )
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      resp = update_ecs_service(cluster, service, deployment_configuration)
+    end
+  end
+
+  def describe_ecs_tasks(cluster, tasks)
+    begin
+      resp = ecs_client.describe_tasks({
+        cluster: cluster,
+        tasks: tasks
+      })
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      resp = describe_ecs_tasks(cluster, tasks)
+    end
+  end
+
+  def run_ecs_task(cluster, task_definition, overrides, count)
+    begin
+      resp = ecs_client.run_task({
+        cluster: cluster,
+        task_definition: task_definition,
+        overrides: overrides,
+        count: count
+      })
+    rescue Aws::CloudFormation::Errors::Throttling => e
+      output 'Throttling, retrying in 15 seconds'.red
+      sleep 15
+      resp = run_ecs_task(cluster, task_definition, overrides, count)
+    end
+  end
+
+  def get_autoscaling_capacity(asg_name)
+    resp = asg_client.describe_auto_scaling_groups(auto_scaling_group_names: [asg_name])
+    resp.auto_scaling_groups[0].desired_capacity
+  end
+
+  def get_spotfleet_capacity(fleet_arn)
+    resp = ec2_client.describe_spot_fleet_requests(spot_fleet_request_ids: [fleet_arn])
+    resp.spot_fleet_request_configs[0].spot_fleet_request_config.target_capacity
+  end
+
+  def hashes_to_tags(hashes)
+    tags = []
+    hkeys = hashes.keys
+    hkeys.each do |hkey|
+      tags.insert(0, { key: hkey, value: hashes[hkey].to_s })
+    end
+    tags
+  end
+
+  def tags_to_hashes(tags)
+    hash = Hash.new
+    tags.each do |tags_obj|
+      hash[tags_obj.key] = tags_obj.value
+    end
+    hash
+  end
+end