pipedream 0.1.0 → 0.4.0

data/vendor/cfn-status/lib/cfn/status.rb

@@ -0,0 +1,220 @@
+require "cfn/status/version"
+
+module Cfn
+  autoload :AwsService, "cfn/aws_service"
+
+  class Status
+    class Error < StandardError; end
+
+    include AwsService
+
+    attr_reader :events
+    def initialize(stack_name, options={})
+      @stack_name = stack_name
+      @options = options
+      reset
+    end
+
+    # used for the bolt cfn status command
+    def run
+      unless stack_exists?(@stack_name)
+        puts "The stack #{@stack_name.color(:green)} does not exist."
+        return true
+      end
+
+      resp = cfn.describe_stacks(stack_name: @stack_name)
+      stack = resp.stacks.first
+
+      puts "The current status for the stack #{@stack_name.color(:green)} is #{stack.stack_status.color(: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
+      success?
+    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}".color(:red)
+        puts "Stack reason #{@events[0]["resource_status_reason"]}".color(:red)
+      elsif last_event_status =~ /_ROLLBACK_/
+        puts "Stack rolled back: #{last_event_status}".color(:red)
+      else # success
+        puts "Stack success status: #{last_event_status}".color(: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).color(:green)}."
+      success?
+    end
+
+    def completed
+      last_event_status =~ /(_COMPLETE|_FAILED)$/ &&
+      @events[0]["logical_resource_id"] == @stack_name &&
+      @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, final)
+        print_events(i)
+      else
+        i = find_index(:last_shown, final)
+        # 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.color(: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, final=false)
+      send("#{name}_index", final)
+    end
+
+    def start_index(final=false)
+      index = @events.find_index do |event|
+        event["resource_type"] == "AWS::CloudFormation::Stack" &&
+        event["resource_status_reason"] == "User Initiated"
+      end
+      # Instead of paginating until until we find the first "User Initiated" "AWS::CloudFormation::Stack" event
+      # we'll use the max.
+      index ? index : @events.size - 1
+    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/vendor/cfn-status/spec/cfn/status_spec.rb

@@ -0,0 +1,81 @@
+RSpec.describe Cfn::Status do
+  it "has a version number" do
+    expect(Cfn::Status::VERSION).not_to be nil
+  end
+
+  let(:status) do
+    status = Cfn::Status.new("test-stack")
+    allow(status).to receive(:cfn).and_return(cfn)
+    status
+  end
+  let(:cfn) do
+    service = double("service").as_null_object
+    allow(service).to receive(:describe_stacks).and_return(stack_status)
+    allow(service).to receive(:describe_stack_events).and_return(stack_events)
+    service
+  end
+
+  context "in progress" do
+    let(:stack_status) { "UPDATE_IN_PROGRESS" }
+    let(:stack_events) { JSON.load(IO.read("spec/fixtures/cfn/stack-events-in-progress.json")) }
+    it "lists events since user initiated event" do
+      status.refresh_events
+      i = status.find_index(:start)
+      expect(i).to eq 15
+      # uncomment to view and debug
+      # status.show_events
+      # puts "****"
+      # status.show_events # show not show anything
+    end
+
+    it "lists events since last shown event" do
+      # first display
+      status.refresh_events
+      i = status.find_index(:start)
+      expect(i).to eq 15
+
+      # move the last event back in time 4 events, so should print 3 events
+      status.instance_variable_set(:@last_shown_event_id, "TargetGroup-ec634c43-b887-4bde-a525-7c69782865a6")
+
+      captured_events = []
+      allow(status).to receive(:print_event) do |e|
+        captured_events << "#{e["resource_type"]} #{e["resource_status"]}"
+      end
+      status.show_events
+      expect(captured_events).to eq([
+        "AWS::ElasticLoadBalancingV2::LoadBalancer DELETE_IN_PROGRESS",
+        "AWS::ElasticLoadBalancingV2::LoadBalancer DELETE_COMPLETE",
+        "AWS::EC2::SecurityGroup DELETE_IN_PROGRESS",
+      ])
+    end
+  end
+
+  context "complete" do
+    let(:stack_status) { "UPDATE_COMPLETE" }
+    let(:stack_events) { JSON.load(IO.read("spec/fixtures/cfn/stack-events-complete.json")) }
+    it "lists events all the way to completion" do
+      status.refresh_events
+      i = status.find_index(:start)
+      expect(i).to eq 17
+      # uncomment to view and debug
+      # status.show_events
+    end
+  end
+
+  context "update_rollback" do
+    let(:stack_status) { "UPDATE_ROLLBACK_COMPLETE" }
+    let(:stack_events) { JSON.load(IO.read("spec/fixtures/cfn/stack-events-update-rollback-complete.json")) }
+    it "lists events all the way to update rollback complete" do
+      status.refresh_events
+      expect(status.success?).to be false
+      expect(status.update_rollback?).to be true
+      expect(status.rollback_error_message).to include("STATIC_NAME")
+
+      # i = status.find_index(:start)
+      # expect(i).to eq 17
+      # uncomment to view and debug
+      # status.show_events
+    end
+  end
+end
+