prima-twig 0.31.1 → 0.31.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 101de521a31bf054b2642be15b80adafb4fb998e
-  data.tar.gz: d851a7a0c8e8aedb8807ad93896126bca5b4dd7d
+  metadata.gz: c6f7a49c077da9462b2283d2e4cd246c3d0af86e
+  data.tar.gz: 7ad946b20bc7c0b742ae6e99efa021686b49f047
 SHA512:
-  metadata.gz: 6757dfb7c78f1df2703d9ea9d35fa08684533d0934536b9ac2227297d32081d28df49a03388e8ddfd846086c5e09cf3879d06c4537b3484f6b8f0385a88f21b2
-  data.tar.gz: 938a531398ea60012ede9d4a41e6065ce03bbe053d6858007aaeafe5af661b2d2d8a2e8514219414222cc667da8436c4b8ef35ea09f37a06aa1ecfdcf3cfdca6
+  metadata.gz: e53e291ecc59c2916c5935a46c9260f17f7fc10fdac268dac0376b1fd4cd320d7864533cb2ef94ae315784696ce9d26b883bed7eec5a02d5f4c6de73ea7246f1
+  data.tar.gz: b960e4d770f769191b3a4d8b9f5baf47db8c051929c2f387f4745e7993cff1385125fcc30ed01f80dc90a463f6546eb93c32806d91d15fca3949ec4451036d55

data/bin/twig-update-ami

@@ -0,0 +1,131 @@
+#!/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'
+
+#
+class TwigUpdateAmi
+  include Command
+  include PrimaAwsClient
+  def initialize
+    @s3 = Aws::S3::Client.new
+    @s3_bucket = 'prima-deploy'
+    @instances = JSON.parse File.read('../twig-binaries/cloudformation.json')
+  end
+
+  def execute!(args)
+    update_amis args[0], args[1], args[2]
+  end
+
+  private
+
+  def update_amis(ami_id, ami_name, ami_description)
+    @instances['instances'].each do |instance|
+      output 'updating instance definition'.light_green
+      Dir.chdir 'ami'
+      update_instance_name(ami_id, ami_name, ami_description, instance['json'])
+      output 'running packer update (this could take some time)'.light_green
+      new_ami_id = update_packer instance['json']
+      Dir.chdir '..'
+      output 'new ami id: ' + new_ami_id
+      instance['stacks'].each do |stack|
+        output 'updating ' + stack['yaml_filename'] + ' and copying onto s3'
+        update_yml_files(new_ami_id, stack['yaml_filename'])
+        copy_yml_files_to_s3(stack['yaml_filename'], stack['s3_key'])
+        output 'updating stack on cloudformation'
+        update_stack_url(stack['stack_name'], stack['template_url'], stack['parameters']) if stack['stack_name']
+      end
+    end
+  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 update_packer(json_filename)
+    `packer build -machine-readable ./#{json_filename} | tee build.log`
+    `grep 'artifact,0,id' build.log | cut -d, -f6 | cut -d: -f2`.sub(/\n/, '')
+  end
+
+  def update_yml_files(ami_id, yaml_filename)
+    yaml_file = File.read yaml_filename
+
+    yaml_file.sub!(/ami-[0-9a-z]{8}/, ami_id)
+
+    File.open yaml_filename, 'w' do |f|
+      f.write yaml_file
+    end
+
+    if yaml_filename.include? 'spotfleet' do
+      old_handle = (File.read yaml_filename)[/InstanceReadyWaitHandleUpdate[0-9]+/]
+      handle_version = old_handle.sub('InstanceReadyWaitHandleUpdate', '').to_i + 1
+
+      old_condition = (File.read yaml_filename)[/InstanceReadyWaitConditionUpdate[0-9]+/]
+      condition_version = old_condition.sub('InstanceReadyWaitConditionUpdate', '').to_i + 1
+
+      file_content = File.read yaml_filename
+      file_content.gsub!(old_handle, 'InstanceReadyWaitHandleUpdate' + handle_version.to_s)
+      file_content.gsub!(old_condition, 'InstanceReadyWaitConditionUpdate' + condition_version.to_s)
+
+      File.open yaml_filename, 'w' do |f|
+        f.write file_content
+      end
+    end
+    end
+  end
+
+  def copy_yml_files_to_s3(source, s3_key)
+    body = File.read source
+    @s3.put_object(
+      body: body,
+      bucket: @s3_bucket,
+      key: s3_key
+    )
+  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-binaries/bin/twig-update-ami ${AMI_ID} ${AMI_NAME} ${AMI_DESCRIPTION}`
+
+    Subcommand for Twig: <http://rondevera.github.io/twig/>
+    Author: Eugenio Laghi <https://github.com/eugeniolaghi>
+
+    HELP
+  end
+
+  args = ARGV.dup
+
+  if args.include?('--help')
+    puts help_content
+    exit
+  end
+
+  TwigUpdateAmi.new.execute!(args)
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: prima-twig
 version: !ruby/object:Gem::Version
-  version: 0.31.1
+  version: 0.31.2
 platform: ruby
 authors:
 - Matteo Giachino
@@ -179,6 +179,7 @@ executables:
 - twig-open-pr
 - twig-pick-issue
 - twig-review
+- twig-update-ami
 extensions: []
 extra_rdoc_files: []
 files:
@@ -189,6 +190,7 @@ files:
 - bin/twig-open-pr
 - bin/twig-pick-issue
 - bin/twig-review
+- bin/twig-update-ami
 - lib/command.rb
 - lib/prima_aws_client.rb
 - lib/prima_twig.rb