lono 4.1.0 → 4.2.0

data/lib/lono/cfn/current.rb

@@ -0,0 +1,95 @@
+require "yaml"
+
+class Lono::Cfn
+  class Current
+    def initialize(options={})
+      Lono::ProjectChecker.check_lono_project
+      @options = options
+      @file = ".lono/current"
+      @path = "#{Lono.root}/#{@file}"
+    end
+
+    def run
+      @options[:rm] ? rm : set
+    end
+
+    def rm
+      FileUtils.rm_f(@path)
+      puts "Current settings have been removed. Removed #{@file}"
+    end
+
+    def set
+      if @options.empty?
+        show
+      else
+        d = data # assign data to d to create local variable for merge to work
+        d = d.merge(@options).delete_if do |k,v|
+          v&.empty? || v == ['']
+        end
+        text = YAML.dump(d)
+        FileUtils.mkdir_p(File.dirname(@path))
+        IO.write(@path, text)
+        puts "Current settings saved in .lono/current"
+        show
+      end
+    end
+
+    def show
+      if data.empty?
+        puts <<-EOL
+There are no current settings.  To set a current stack run:
+
+    lono cfn current --name mystack
+    lono cfn current -h # for more examples
+EOL
+        return
+      end
+
+      data.each do |key, value|
+        puts "Current #{key}: #{value}"
+      end
+    end
+
+    def data
+      YAML.load(IO.read(@path)) rescue {}
+    end
+
+    def suffix
+      current = data["suffix"]
+      return current unless current&.empty?
+    end
+
+    # reads suffix, returns nil if not set
+    def self.suffix
+      Current.new.suffix
+    end
+
+    def name
+      current = data["name"]
+      return current unless current&.empty?
+    end
+
+    # reads name, returns nil if not set
+    def self.name
+      Current.new.name
+    end
+
+    # reads name, will exit if current name not set
+    def self.name!(name=:current)
+      return name if name != :current
+
+      name = Current.name
+      return name if name
+
+      # TODO: don't think it is possible to get here...
+      puts "ERROR: stack name must be specified.".colorize(:red)
+      puts <<-EOL
+Example:
+    lono cfn #{ARGV[1]} STACK
+You can also set a current stack name to be remembered with:
+    lono cfn current --name STACK
+EOL
+      exit 1
+    end
+  end
+end

data/lib/lono/cfn/delete.rb

@@ -3,7 +3,7 @@ class Lono::Cfn::Delete
   include Lono::Cfn::Util
 
   def initialize(stack_name, options={})
-    @stack_name = stack_name
+    @stack_name = switch_current(stack_name)
     @options = options
   end
 
@@ -21,5 +21,19 @@ class Lono::Cfn::Delete
         puts "#{@stack_name.inspect} stack does not exist".colorize(:red)
       end
     end
+
+    return unless @options[:wait]
+    start_time = Time.now
+    status.wait
+    took = Time.now - start_time
+    puts "Time took for stack deletion: #{status.pretty_time(took).green}."
+  end
+
+  def status
+    @status ||= Lono::Cfn::Status.new(@stack_name)
+  end
+
+  def switch_current(stack_name)
+    Lono::Cfn::Current.name!(stack_name)
   end
 end

data/lib/lono/cfn/preview.rb

@@ -22,14 +22,13 @@ class Lono::Cfn::Preview < Lono::Cfn::Base
     end
     exit_unless_updatable!(stack_status(@stack_name))
 
-    template_body = IO.read(@template_path)
     params = {
       change_set_name: change_set_name,
       stack_name: @stack_name,
-      template_body: template_body,
       parameters: params,
       capabilities: capabilities, # ["CAPABILITY_IAM", "CAPABILITY_NAMED_IAM"]
     }
+    set_template_body!(params)
     show_parameters(params, "cfn.create_change_set")
     begin
       cfn.create_change_set(params)
@@ -67,7 +66,10 @@ class Lono::Cfn::Preview < Lono::Cfn::Base
     case change_set.status
     when "CREATE_COMPLETE"
       puts "CloudFormation preview for '#{@stack_name}' stack update. Changes:"
-      change_set.changes.each do |change|
+      changes = change_set.changes.sort_by do |change|
+        change["resource_change"]["action"]
+      end
+      changes.each do |change|
         display_change(change)
       end
     when "FAILED"

data/lib/lono/cfn/status.rb

@@ -0,0 +1,212 @@
+class Lono::Cfn
+  class Status
+    include AwsService
+
+    attr_reader :events
+    def initialize(stack_name, options={})
+      @stack_name = switch_current(stack_name)
+      @options = options
+      reset
+    end
+
+    # used for the lono cfn status command
+    def run
+      unless stack_exists?(@stack_name)
+        puts "The stack #{@stack_name.colorize(:green)} does not exist."
+        return
+      end
+
+      resp = cfn.describe_stacks(stack_name: @stack_name)
+      stack = resp.stacks.first
+
+      puts "The current status for the stack #{@stack_name.colorize(:green)} is #{stack.stack_status.colorize(:green)}"
+      if stack.stack_status =~ /_IN_PROGRESS$/
+        puts "Stack events (tailing):"
+        # tail all events until done
+        @hide_time_took = true
+        wait
+      else
+        puts "Stack events:"
+        # show the last events that was user initiated
+        refresh_events
+        show_events(true)
+      end
+    end
+
+    def switch_current(stack_name)
+      Lono::Cfn::Current.name!(stack_name)
+    end
+
+    def reset
+      @events = [] # constantly replaced with recent events
+      @last_shown_event_id = nil
+      @stack_deletion_completed = nil
+    end
+
+    # check for /(_COMPLETE|_FAILED)$/ status
+    def wait
+      puts "Waiting for stack to complete"
+      start_time = Time.now
+
+      refresh_events
+      until completed || @stack_deletion_completed
+        show_events
+      end
+      show_events(true) # show the final event
+
+      if @stack_deletion_completed
+        puts "Stack #{@stack_name} deleted."
+        return
+      end
+
+      if last_event_status =~ /_FAILED/
+        puts "Stack failed: #{last_event_status}".colorize(:red)
+        puts "Stack reason #{@events[0]["resource_status_reason"]}".colorize(:red)
+      elsif last_event_status =~ /_ROLLBACK_/
+        puts "Stack rolled back: #{last_event_status}".colorize(:red)
+      else # success
+        puts "Stack success status: #{last_event_status}".colorize(:green)
+      end
+
+      # Never gets here when deleting a stack because the describe stack returns nothing
+      # once the stack is deleted. Gets here for stack create and update though.
+      return if @hide_time_took # set in run
+      took = Time.now - start_time
+      puts "Time took for stack deployment: #{pretty_time(took).green}."
+    end
+
+    def completed
+      last_event_status =~ /(_COMPLETE|_FAILED)$/ &&
+      @events[0]["resource_type"] == "AWS::CloudFormation::Stack"
+    end
+
+    def last_event_status
+      @events[0]["resource_status"]
+    end
+
+    # Only shows new events
+    def show_events(final=false)
+      if @last_shown_event_id.nil?
+        i = find_index(:start)
+        print_events(i)
+      else
+        i = find_index(:last_shown)
+        # puts "last_shown index #{i}"
+        print_events(i-1) unless i == 0
+      end
+
+      return if final
+      sleep 5 unless ENV['TEST']
+      refresh_events
+    end
+
+    def print_events(i)
+      @events[0..i].reverse.each do |e|
+        print_event(e)
+      end
+      @last_shown_event_id = @events[0]["event_id"]
+      # puts "@last_shown_event_id #{@last_shown_event_id.inspect}"
+    end
+
+    def print_event(e)
+      message = [
+        event_time(e["timestamp"]),
+        e["resource_status"],
+        e["resource_type"],
+        e["logical_resource_id"],
+        e["resource_status_reason"]
+      ].join(" ")
+      message = message.colorize(:red) if e["resource_status"] =~ /_FAILED/
+      puts message
+    end
+
+    # https://stackoverflow.com/questions/18000432/rails-12-hour-am-pm-range-for-a-day
+    def event_time(timestamp)
+      Time.parse(timestamp.to_s).localtime.strftime("%I:%M:%S%p")
+    end
+
+    # refreshes the loaded events in memory
+    def refresh_events
+      resp = cfn.describe_stack_events(stack_name: @stack_name)
+      @events = resp["stack_events"]
+    rescue Aws::CloudFormation::Errors::ValidationError => e
+      if e.message =~ /Stack .* does not exis/
+        @stack_deletion_completed = true
+      else
+        raise
+      end
+    end
+
+    def find_index(name)
+      send("#{name}_index")
+    end
+
+    def start_index
+      @events.find_index do |event|
+        event["resource_type"] == "AWS::CloudFormation::Stack" &&
+        event["resource_status_reason"] == "User Initiated"
+      end
+    end
+
+    def last_shown_index
+      @events.find_index do |event|
+        event["event_id"] == @last_shown_event_id
+      end
+    end
+
+    def success?
+      resource_status = @events[0]["resource_status"]
+      %w[CREATE_COMPLETE UPDATE_COMPLETE].include?(resource_status)
+    end
+
+    def update_rollback?
+      @events[0]["resource_status"] == "UPDATE_ROLLBACK_COMPLETE"
+    end
+
+    def find_update_failed_event
+      i = @events.find_index do |event|
+        event["resource_type"] == "AWS::CloudFormation::Stack" &&
+        event["resource_status_reason"] == "User Initiated"
+      end
+
+      @events[0..i].reverse.find do |e|
+        e["resource_status"] == "UPDATE_FAILED"
+      end
+    end
+
+    def rollback_error_message
+      return unless update_rollback?
+
+      event = find_update_failed_event
+      return unless event
+
+      reason = event["resource_status_reason"]
+      messages_map.each do |pattern, message|
+        if reason =~ pattern
+          return message
+        end
+      end
+
+      reason # default message is original reason if not found in messages map
+    end
+
+    def messages_map
+      {
+        /CloudFormation cannot update a stack when a custom-named resource requires replacing/ => "A workaround is to run ufo again with STATIC_NAME=0 and to switch to dynamic names for resources. Then run ufo again with STATIC_NAME=1 to get back to statically name resources. Note, there are caveats with the workaround.",
+        /cannot be associated with more than one load balancer/ => "There's was an issue updating the stack. Target groups can only be associated with one load balancer at a time. The workaround for this is to use UFO_FORCE_TARGET_GROUP=1 and run the command again. This will force the recreation of the target group resource.",
+        /SetSubnets is not supported for load balancers of type/ => "Changing subnets for Network Load Balancers is currently not supported. You can try workarouding this with UFO_FORCE_ELB=1 and run the command again. This will force the recreation of the elb resource."
+      }
+    end
+
+    # http://stackoverflow.com/questions/4175733/convert-duration-to-hoursminutesseconds-or-similar-in-rails-3-or-ruby
+    def pretty_time(total_seconds)
+      minutes = (total_seconds / 60) % 60
+      seconds = total_seconds % 60
+      if total_seconds < 60
+        "#{seconds.to_i}s"
+      else
+        "#{minutes.to_i}m #{seconds.to_i}s"
+      end
+    end
+  end
+end

data/lib/lono/cfn/update.rb

@@ -18,6 +18,10 @@ class Lono::Cfn::Update < Lono::Cfn::Base
     end
     exit_unless_updatable!(stack_status(@stack_name))
 
+    options = @options.merge(lono: false, mute_params: true, mute_using: true, keep: true)
+    # create new copy of preview when update_stack is called because of IAM retry logic
+    preview = Lono::Cfn::Preview.new(@stack_name, options)
+
     error = nil
     diff.run if @options[:diff]
     preview.run if @options[:preview]
@@ -25,7 +29,7 @@ class Lono::Cfn::Update < Lono::Cfn::Base
 
     if @options[:change_set] # defaults to this
       message << " via change set: #{preview.change_set_name}"
-      change_set_update
+      preview.execute_change_set
     else
       standard_update(params)
     end
@@ -33,14 +37,13 @@ class Lono::Cfn::Update < Lono::Cfn::Base
   end
 
   def standard_update(params)
-    template_body = IO.read(@template_path)
     params = {
       stack_name: @stack_name,
-      template_body: template_body,
       parameters: params,
       capabilities: capabilities, # ["CAPABILITY_IAM", "CAPABILITY_NAMED_IAM"]
       disable_rollback: !@options[:rollback],
     }
+    set_template_body!(params)
     show_parameters(params, "cfn.update_stack")
     begin
       cfn.update_stack(params)
@@ -50,16 +53,7 @@ class Lono::Cfn::Update < Lono::Cfn::Base
     end
   end
 
-  def preview
-    options = @options.merge(lono: false, mute_params: true, mute_using: true, keep: true)
-    @preview ||= Lono::Cfn::Preview.new(@stack_name, options)
-  end
-
   def diff
     @diff ||= Lono::Cfn::Diff.new(@stack_name, @options.merge(lono: false, mute_params: true, mute_using: true))
   end
-
-  def change_set_update
-    preview.execute_change_set
-  end
 end

data/lib/lono/cfn/util.rb

@@ -5,9 +5,9 @@ module Lono::Cfn::Util
     else
       message = case action
       when :update
-        "Are you sure you want to want to update the '#{stack_name}' stack with the changes? (y/N)"
+        "Are you sure you want to want to update the #{stack_name.colorize(:green)} stack with the changes? (y/N)"
       when :delete
-        "Are you sure you want to want to delete the '#{stack_name}' stack? (y/N)"
+        "Are you sure you want to want to delete the #{stack_name.colorize(:green)} stack? (y/N)"
       end
       puts message
       sure = $stdin.gets

data/lib/lono/cli.rb

@@ -66,12 +66,6 @@ module Lono
       Completer::Script.generate
     end
 
-    desc "upgrade4", "Upgrade from version 3 to 4."
-    long_desc Help.text("upgrade4")
-    def upgrade4
-      Upgrade4.new(options).run
-    end
-
     desc "version", "Prints version"
     def version
       puts VERSION
@@ -92,5 +86,9 @@ module Lono
     desc "script SUBCOMMAND", "script subcommands"
     long_desc Help.text(:script)
     subcommand "script", Script
+
+    desc "upgrade SUBCOMMAND", "upgrade subcommands"
+    long_desc Help.text(:upgrade)
+    subcommand "upgrade", Upgrade
   end
 end