prima-twig 1.1.0

data/lib/command.rb

@@ -0,0 +1,73 @@
+require 'colorize'
+require 'English'
+require 'optparse'
+
+module Command
+  # executes command and exits if status != 0
+  def exec_step(command, output = nil)
+    if output
+      puts output
+    else
+      puts 'exec > '.green + command.to_s.yellow
+    end
+    `#{command}`
+    exit($CHILD_STATUS.exitstatus) unless $CHILD_STATUS.exitstatus.zero?
+  end
+
+  def output(msg)
+    puts 'twig binaries > '.black + msg
+  end
+
+  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
+      command.empty? || exec(command)
+      puts 'there was a problem > '.red + output_msg
+      exit(1)
+    end
+  end
+
+  def symbol_message(s)
+    case s
+    when :clean
+      'hai dei file non committati...non posso continuare'
+    when :detached_head
+      "repo in stato 'head detached'"
+    when :wrong_args
+      'argomento non corretto'
+    end
+  end
+
+  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 += values
+      msg
+    end
+  end
+
+  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::ElasticLoadBalancingV2::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