ufo 6.1.3 → 6.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 881d674019c976e4c8cae9030f69bd2125e59c7af970a1fa2005dcbfb77a4887
-  data.tar.gz: 5c3e8f73e200390396c99aa292f949f5c0a7b0e457bab47e2b88dbc583adc0fa
+  metadata.gz: 5b86133b126853170107907bce7c02f4620387d8b11c4a5e6e0f03176fda1349
+  data.tar.gz: 3c543c7b436ec7214043d5f5cf45a798af2ce054aed970fa5de991848f6a4664
 SHA512:
-  metadata.gz: 7eaca7eaa3999ce52dd4087dd2bd7862156f34c59443fcf06140f50356beb50237c4dea35d2dc68a1171576e17529c18144e5c97bc6aa4da4ea1edfa3b9411ef
-  data.tar.gz: 820645713455edf1d3e7fd242c68c21f32267f6eb031c67dd9236578d93c56e0a616750170f202ab27f32c86a0093a673fa5e7321c800b3d82121f38f156b29f
+  metadata.gz: f4658807edc6f36b07cf8048f9f7f9c75ed90f3a03cd481ea1b763d4332bdf887a17d9199988021fa3866d73e7d0b0f43b1d03ef031915fdf5e0d3ad00a5904c
+  data.tar.gz: d52d10df7faf76256f336abfc89a9ccd1afb999973d52e7cbf660481c17eb54ae8333c09e600de12455020c707fd1821a32beee0a893c72750337c1aecbcfaaa

data/.cody/acceptance/bin/build.sh

@@ -22,7 +22,7 @@ REPO=$(aws ecr describe-repositories --repository-name test/demo | jq -r '.repos
 
 # DockerHub
 # toomanyrequests: You have reached your pull rate limit. You may increase the limit by authenticating
-docker login --username $DOCKER_USER --password $DOCKER_PASS
+echo "$DOCKER_PASS" | docker login --username $DOCKER_USER --password-stdin
 TOKEN=$(curl -s --user "$DOCKER_USER:$DOCKER_PASS" "https://auth.docker.io/token?service=registry.docker.io&scope=repository:ratelimitpreview/test:pull" | jq -r .token)
 echo "Current rate limit:"
 curl -s --head -H "Authorization: Bearer $TOKEN" https://registry-1.docker.io/v2/ratelimitpreview/test/manifests/latest

data/CHANGELOG.md

@@ -3,6 +3,15 @@
 All notable changes to this project will be documented in this file.
 This project *tries* to adhere to [Semantic Versioning](http://semver.org/), even before v1.0.
 
+## [6.2.0] - 2022-03-16
+- [#152](https://github.com/tongueroo/ufo/pull/152) ufo docker base: s3 storage support
+
+## [6.1.5] - 2022-03-16
+- [#151](https://github.com/tongueroo/ufo/pull/151) fix symlink when .git ext is in the repo
+
+## [6.1.4] - 2022-03-16
+- [#150](https://github.com/tongueroo/ufo/pull/150) ufo central: fix edge case when trying report broken symlink and logger not available
+
 ## [6.1.3] - 2022-03-14
 - [#149](https://github.com/tongueroo/ufo/pull/149) stack_output helper improve message when stack not found
 

data/lib/ufo/aws_services/concerns.rb

@@ -0,0 +1,55 @@
+require "cfn_status"
+
+module Ufo::AwsServices
+  module Concerns
+    extend Memoist
+
+    def find_stack(stack_name)
+      resp = cfn.describe_stacks(stack_name: stack_name)
+      resp.stacks.first
+    rescue Aws::CloudFormation::Errors::ValidationError => e
+      # example: Stack with id demo-web does not exist
+      if e.message =~ /Stack with/ && e.message =~ /does not exist/
+        nil
+      else
+        raise
+      end
+    end
+
+    def stack_resources(stack_name)
+      resp = cfn.describe_stack_resources(stack_name: stack_name)
+      resp.stack_resources
+    rescue Aws::CloudFormation::Errors::ValidationError => e
+      e.message.include?("does not exist") ? return : raise
+    end
+
+    def task_definition_arns(family, max_items=10)
+      resp = ecs.list_task_definitions(
+        family_prefix: family,
+        sort: "DESC",
+      )
+      arns = resp.task_definition_arns
+      arns = arns.select do |arn|
+        task_definition = arn.split('/').last.split(':').first
+        task_definition == family
+      end
+      arns[0..max_items]
+    end
+
+    def status
+      CfnStatus.new(@stack_name) # NOTE: @stack_name must be set in the including Class
+    end
+    memoize :status
+
+    def find_stack_resources(stack_name)
+      resp = cfn.describe_stack_resources(stack_name: stack_name)
+      resp.stack_resources
+    rescue Aws::CloudFormation::Errors::ValidationError => e
+      if e.message.include?("does not exist")
+        nil
+      else
+        raise
+      end
+    end
+  end
+end

data/lib/ufo/aws_services.rb

@@ -6,15 +6,16 @@ require "aws-sdk-ec2"
 require "aws-sdk-ecr"
 require "aws-sdk-ecs"
 require "aws-sdk-elasticloadbalancingv2"
+require "aws-sdk-s3"
 require "aws-sdk-ssm"
 require "aws-sdk-wafv2"
 
 require "aws_mfa_secure/ext/aws" # add MFA support
-require "cfn_status"
 
 module Ufo
   module AwsServices
     extend Memoist
+    include Concerns
 
     def acm
       Aws::ACM::Client.new(aws_options)
@@ -26,10 +27,10 @@ module Ufo
     end
     memoize :applicationautoscaling
 
-    def cloudformation
+    def cfn
       Aws::CloudFormation::Client.new(aws_options)
     end
-    memoize :cloudformation
+    memoize :cfn
 
     def cloudwatchlogs
       Aws::CloudWatchLogs::Client.new(aws_options)
@@ -56,6 +57,11 @@ module Ufo
     end
     memoize :elb
 
+    def s3
+      Aws::S3::Client.new(aws_options)
+    end
+    memoize :s3
+
     # ssm is a helper method
     def ssm_client
       Aws::SSM::Client.new(aws_options)
@@ -98,42 +104,5 @@ module Ufo
       ) if ENV['UFO_DEBUG_AWS_SDK']
       options
     end
-
-    def find_stack(stack_name)
-      resp = cloudformation.describe_stacks(stack_name: stack_name)
-      resp.stacks.first
-    rescue Aws::CloudFormation::Errors::ValidationError => e
-      # example: Stack with id demo-web does not exist
-      if e.message =~ /Stack with/ && e.message =~ /does not exist/
-        nil
-      else
-        raise
-      end
-    end
-
-    def stack_resources(stack_name)
-      resp = cloudformation.describe_stack_resources(stack_name: stack_name)
-      resp.stack_resources
-    rescue Aws::CloudFormation::Errors::ValidationError => e
-      e.message.include?("does not exist") ? return : raise
-    end
-
-    def task_definition_arns(family, max_items=10)
-      resp = ecs.list_task_definitions(
-        family_prefix: family,
-        sort: "DESC",
-      )
-      arns = resp.task_definition_arns
-      arns = arns.select do |arn|
-        task_definition = arn.split('/').last.split(':').first
-        task_definition == family
-      end
-      arns[0..max_items]
-    end
-
-    def status
-      CfnStatus.new(@stack_name) # NOTE: @stack_name must be set in the including Class
-    end
-    memoize :status
   end
 end

data/lib/ufo/cfn/stack/status.rb

@@ -122,7 +122,7 @@ class Ufo::Cfn::Stack
 
     # refreshes the loaded events in memory
     def refresh_events
-      resp = cloudformation.describe_stack_events(stack_name: @stack_name)
+      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/

data/lib/ufo/cfn/stack.rb

@@ -31,7 +31,7 @@ module Ufo::Cfn
       @stack = find_stack(@stack_name)
       if @stack && rollback_complete?(@stack)
         logger.info "Existing stack in ROLLBACK_COMPLETE state. Deleting stack before continuing."
-        cloudformation.delete_stack(stack_name: @stack_name)
+        cfn.delete_stack(stack_name: @stack_name)
         status.wait
         status.reset
         @stack = nil # at this point stack has been deleted
@@ -53,7 +53,7 @@ module Ufo::Cfn
 
     def perform(action)
       logger.info "#{action[0..-2].capitalize}ing stack #{@stack_name.color(:green)}"
-      cloudformation.send("#{action}_stack", stack_options) # Example: cloudformation.send("update_stack", stack_options)
+      cfn.send("#{action}_stack", stack_options) # Example: cfn.send("update_stack", stack_options)
     rescue Aws::CloudFormation::Errors::ValidationError => e
       handle_stack_error(e)
     end
@@ -152,10 +152,10 @@ module Ufo::Cfn
       end
 
       if stack.stack_status == "CREATE_IN_PROGRESS"
-        cloudformation.delete_stack(stack_name: @stack_name)
+        cfn.delete_stack(stack_name: @stack_name)
         logger.info "Canceling stack creation"
       elsif stack.stack_status == "UPDATE_IN_PROGRESS"
-        cloudformation.cancel_update_stack(stack_name: @stack_name)
+        cfn.cancel_update_stack(stack_name: @stack_name)
         logger.info "Canceling stack update"
       else
         logger.info "The stack is not in a state to that is cancelable: #{stack.stack_status}"

data/lib/ufo/cli/central/base.rb

@@ -1,5 +1,6 @@
 class Ufo::CLI::Central
   class Base
+    include Ufo::Utils::Logging
     include Ufo::Utils::Pretty
     include Ufo::Utils::Sure
 

data/lib/ufo/cli/central/update.rb

@@ -30,6 +30,7 @@ class Ufo::CLI::Central
 
       # FileUtils.ln_s(target, link, options)
       # ~/.ufo/central/repo -> .ufo
+      src.sub!(/\.git$/,'')
       FileUtils.ln_sf(src, ".ufo", verbose: false) # force in case of existing broken symlink
       FileUtils.rm_rf(".ufo.bak")
 

data/lib/ufo/cli/destroy.rb

@@ -17,7 +17,7 @@ class Ufo::CLI
         return
       end
 
-      cloudformation.delete_stack(stack_name: @stack_name)
+      cfn.delete_stack(stack_name: @stack_name)
       puts "Deleting stack #{@stack_name.color(:green)}"
 
       return unless @options[:wait]

data/lib/ufo/config.rb

@@ -114,7 +114,10 @@ module Ufo
       config.ship.docker.quiet = false # only affects ufo ship docker commands output
 
       config.state = ActiveSupport::OrderedOptions.new
+      config.state.bucket = nil # Set to use existing bucket. When not set ufo creates a managed s3 bucket
+      config.state.managed = true # false will disable creation of managed bucket entirely
       config.state.reminder = true
+      config.state.storage = "s3" # s3 or file
 
       config.waf = ActiveSupport::OrderedOptions.new
       config.waf.web_acl_arn = nil

data/lib/ufo/docker/builder.rb

@@ -162,7 +162,7 @@ module Ufo::Docker
 
     def update_dockerfile
       updater = if File.exist?("#{Ufo.root}/Dockerfile.erb") # dont use @dockerfile on purpose
-        State.new(docker_image, @options)
+        State.new(@options.merge(base_image: docker_image))
       else
         Dockerfile.new(docker_image, @options)
       end

data/lib/ufo/docker/compiler.rb

@@ -9,9 +9,9 @@ module Ufo::Docker
       return unless File.exist?(@erb_file)
 
       puts "Compiled #{File.basename(@erb_file).color(:green)} to #{File.basename(@dockerfile).color(:green)}"
-      path = "#{Ufo.root}/.ufo/state/data.yml"
-      vars = YAML.load_file(path)[Ufo.env] if File.exist?(path)
-      vars ||= {}
+
+      state = State.new
+      vars = state.read
       result = RenderMePretty.result(@erb_file, vars)
       comment =<<~EOL.chop # remove the trailing newline
         # IMPORTANT: This file was generated from #{File.basename(@erb_file)} as a part of running:

data/lib/ufo/docker/state/base.rb

@@ -0,0 +1,14 @@
+class Ufo::Docker::State
+  class Base
+    include Ufo::Utils::Logging
+    include Ufo::Utils::Pretty
+
+    def initialize(options={})
+      @options = options
+      # base_image only passed in with: ufo docker base
+      # State#update uses it.
+      # State#read wont have access to it and gets it from stored state
+      @base_image = options[:base_image]
+    end
+  end
+end

data/lib/ufo/docker/state/bucket.rb

@@ -0,0 +1,2 @@
+class Ufo::Docker::State
+  class S3 < Base

data/lib/ufo/docker/state/file.rb

@@ -0,0 +1,52 @@
+class Ufo::Docker::State
+  class File < Base
+    def read
+      current_data
+    end
+
+    def update
+      data = current_data
+      data["base_image"] = @base_image
+
+      pretty_path = state_path.sub("#{Ufo.root}/", "")
+      FileUtils.mkdir_p(::File.dirname(state_path))
+      IO.write(state_path, YAML.dump(data))
+
+      logger.info "The #{pretty_path} base_image has been updated with the latest base image:".color(:green)
+      logger.info "    #{@base_image}".color(:green)
+      reminder_message
+    end
+
+    def current_data
+      ::File.exist?(state_path) ? YAML.load_file(state_path) : {}
+    end
+
+    def state_path
+      "#{Ufo.root}/.ufo/state/#{Ufo.app}/#{Ufo.env}/data.yml"
+    end
+
+    def reminder_message
+      return unless Ufo.config.state.reminder
+      repo = ENV['UFO_CENTRAL_REPO']
+      return unless repo
+      logger.info "It looks like you're using a central deployer pattern".color(:yellow)
+      logger.info <<~EOL
+        Remember to commit the state file:
+
+            state file: #{pretty_path(state_path)}
+            repo:       #{repo}
+
+      EOL
+
+      logger.info <<~EOL
+        You can disable these reminder messages with:
+
+        .ufo/config.rb
+
+            Ufo.configure do |config|
+              config.state.reminder = false
+            end
+      EOL
+    end
+  end
+end

data/lib/ufo/docker/state/s3.rb

@@ -0,0 +1,80 @@
+class Ufo::Docker::State
+  class S3 < Base
+    extend Memoist
+    include Ufo::AwsServices
+
+    def read
+      current_data
+    end
+
+    def update
+      data = current_data
+      data["base_image"] = @base_image
+
+      # write data to s3
+      body = YAML.dump(data)
+      s3.put_object(
+        body: body,
+        bucket: s3_bucket,
+        key: s3_key,
+      )
+      logger.info "Updated base image in s3://#{s3_bucket}/#{s3_key}"
+      logger.info "    #{@base_image}".color(:green)
+    end
+
+    # TODO: edge cases: no bucket, no permission
+    def current_data
+      resp = s3.get_object(bucket: s3_bucket, key: s3_key)
+      YAML.load(resp.body)
+    rescue Aws::S3::Errors::NoSuchKey
+      logger.debug "WARN: s3 key does not exist: #{s3_key}"
+      {}
+    rescue Aws::S3::Errors::NoSuchBucket
+      logger.error "ERROR: S3 bucket does not exist to store state: #{s3_bucket}".color(:red)
+      logger.error <<~EOL
+          Please double check the config.
+
+          See: http://ufoships.com/docs/config/state/
+
+      EOL
+      exit 1
+    end
+
+    def s3_key
+      "ufo/state/#{app}/#{Ufo.env}/data.yml"
+    end
+
+    # ufo docker base is called before Ufo.config is loaded. This ensures it is loaded
+    def app
+      Ufo.config
+      Ufo.app
+    end
+
+    def s3_bucket
+      state = Ufo.config.state
+      if state.bucket
+        state.bucket
+      elsif state.managed
+        ensure_s3_bucket_exist
+        Ufo::S3::Bucket.name
+      else
+        logger.error "ERROR: No s3 bucket to store state".color(:red)
+        logger.error <<~EOL
+          UFO needs a bucket to store the built docker base image.
+
+          Configure an existing bucket or enable UFO to create a bucket.
+
+          See: http://ufoships.com/docs/config/state/
+        EOL
+        exit 1
+      end
+    end
+
+    def ensure_s3_bucket_exist
+      bucket = Ufo::S3::Bucket.new
+      return if bucket.exist?
+      bucket.deploy
+    end
+    memoize :ensure_s3_bucket_exist
+  end
+end

data/lib/ufo/docker/state.rb

@@ -1,63 +1,29 @@
 module Ufo::Docker
   class State
-    include Ufo::Utils::Logging
-    include Ufo::Utils::Pretty
+    extend Memoist
 
-    def initialize(docker_image, options={})
-      @docker_image, @options = docker_image, options
+    def initialize(options={})
+      @options = options
     end
 
     def update
-      data = current_data
-      data[Ufo.env] ||= {}
-      data[Ufo.env]["base_image"] = @docker_image
-      pretty_path = state_path.sub("#{Ufo.root}/", "")
-      FileUtils.mkdir_p(File.dirname(state_path))
-      IO.write(state_path, YAML.dump(data))
-      logger.info "The #{pretty_path} base_image has been updated with the latest base image:".color(:green)
-      logger.info "  #{@docker_image}".color(:green)
-      reminder_message
+      storage.update
     end
 
-    def current_data
-      File.exist?(state_path) ? YAML.load_file(state_path) : {}
+    def read
+      storage.read
     end
 
-    def state_path
-      path = "#{Ufo.root}/.ufo/state"
-      if ENV['UFO_APP'] # env var activates app path
-        path = "#{Ufo.root}/.ufo/state/#{Ufo.app}"
-      end
-      "#{path}/data.yml"
-    end
-
-    def reminder_message
-      return unless Ufo.config.state.reminder
-      repo = ENV['UFO_CENTRAL_REPO']
-      return unless repo
-      logger.info "It looks like you're using a central deployer pattern".color(:yellow)
-      logger.info <<~EOL
-        Remember to commit the state file:
-
-            state file: #{pretty_path(state_path)}
-            repo:       #{repo}
-
-      EOL
-
-      unless ENV['UFO_APP']
-        logger.info "WARN: It also doesnt look like UFO_ENV is set".color(:yellow)
-        logger.info "UFO_ENV should be set when you're using ufo in a central manner"
-      end
-
-      logger.info <<~EOL
-        You can disable these reminder messages with:
-
-        .ufo/config.rb
-
-            Ufo.configure do |config|
-              config.state.reminder = false
-            end
-      EOL
+  private
+    # Examples:
+    #   File.new(@docker_image, @options)
+    #   S3.new(@docker_image, @options)
+    def storage
+      storage = Ufo.config.state.storage
+      class_name = "Ufo::Docker::State::#{storage.camelize}"
+      klass = class_name.constantize
+      klass.new(@options)
     end
+    memoize :storage
   end
 end

data/lib/ufo/info.rb

@@ -73,7 +73,7 @@ module Ufo
     end
 
     def stack_resources
-      resp = cloudformation.describe_stack_resources(stack_name: @stack_name)
+      resp = cfn.describe_stack_resources(stack_name: @stack_name)
       resp.stack_resources
     end
     memoize :stack_resources

data/lib/ufo/s3/aws_setup.rb

@@ -0,0 +1,17 @@
+module Ufo::S3
+  class AwsSetup
+    include Ufo::AwsServices
+    include Ufo::Utils::Logging
+
+    def check!
+      s3.config.region
+    rescue Aws::Errors::MissingRegionError => e
+      logger.info "ERROR: #{e.class}: #{e.message}".color(:red)
+      logger.info <<~EOL
+        Unable to detect the AWS_REGION to make AWS API calls. This is might be because the AWS access
+        has not been set up yet. Please either your ~/.aws files.
+      EOL
+      exit 1
+    end
+  end
+end

data/lib/ufo/s3/bucket.rb

@@ -0,0 +1,174 @@
+module Ufo::S3
+  class Bucket
+    extend Memoist
+    extend Ufo::AwsServices
+    include Ufo::AwsServices
+    include Ufo::Utils::Logging
+
+    STACK_NAME = ENV['UFO_STACK_NAME'] || "ufo"
+    def initialize(options={})
+      @options = options
+    end
+
+    def deploy
+      stack = find_stack
+      if rollback.complete?
+        logger.info "Existing '#{STACK_NAME}' stack in ROLLBACK_COMPLETE state. Deleting stack before continuing."
+        disable_termination_protection
+        cfn.delete_stack(stack_name: STACK_NAME)
+        status.wait
+        stack = nil
+      end
+
+      if stack
+        update
+      else
+        create
+      end
+    end
+
+    def exist?
+      !!bucket_name
+    end
+
+    def bucket_name
+      self.class.name
+    end
+
+    def show
+      if bucket_name
+        logger.info "UFO bucket name: #{bucket_name}"
+      else
+        logger.info "UFO bucket does not exist yet."
+      end
+    end
+
+    # Launches a cloudformation to create an s3 bucket
+    def create
+      logger.info "Creating #{STACK_NAME} stack for s3 bucket to store state"
+      cfn.create_stack(
+        stack_name: STACK_NAME,
+        template_body: template_body,
+        enable_termination_protection: true,
+      )
+      success = status.wait
+      unless success
+        logger.info "ERROR: Unable to create UFO stack with managed s3 bucket".color(:red)
+        exit 1
+      end
+    end
+
+    def update
+      logger.info "Updating #{STACK_NAME} stack with the s3 bucket"
+      cfn.update_stack(stack_name: STACK_NAME, template_body: template_body)
+    rescue Aws::CloudFormation::Errors::ValidationError => e
+      raise unless e.message.include?("No updates are to be performed")
+    end
+
+    def delete
+      are_you_sure?
+
+      logger.info "Deleting #{STACK_NAME} stack with the s3 bucket"
+      disable_termination_protection
+      empty_bucket!
+      cfn.delete_stack(stack_name: STACK_NAME)
+    end
+
+    def disable_termination_protection
+      cfn.update_termination_protection(
+        stack_name: STACK_NAME,
+        enable_termination_protection: false,
+      )
+    end
+
+    def find_stack
+      resp = cfn.describe_stacks(stack_name: STACK_NAME)
+      resp.stacks.first
+    rescue Aws::CloudFormation::Errors::ValidationError
+      nil
+    end
+
+    def status
+      CfnStatus.new(STACK_NAME)
+    end
+
+  private
+
+    def empty_bucket!
+      return unless bucket_name # in case of UFO stack ROLLBACK_COMPLETE from failed bucket creation
+
+      resp = s3.list_objects(bucket: bucket_name)
+      if resp.contents.size > 0
+        # IE: objects = [{key: "objectkey1"}, {key: "objectkey2"}]
+        objects = resp.contents.map { |item| {key: item.key} }
+        s3.delete_objects(
+          bucket: bucket_name,
+          delete: {
+            objects: objects,
+            quiet: false,
+          }
+        )
+        empty_bucket! # keep deleting objects until bucket is empty
+      end
+    end
+
+
+    def are_you_sure?
+      return true if @options[:yes]
+
+      if bucket_name.nil?
+        logger.info "The UFO stack and s3 bucket does not exist."
+        exit
+      end
+
+      logger.info "Are you yes you want the UFO bucket #{bucket_name.color(:green)} to be emptied and deleted? (y/N)"
+      yes = $stdin.gets.strip
+      confirmed = yes =~ /^Y/i
+      unless confirmed
+        logger.info "Phew that was close."
+        exit
+      end
+    end
+
+    def template_body
+      <<~YAML
+        Description: UFO managed s3 bucket
+        Resources:
+          Bucket:
+            Type: AWS::S3::Bucket
+            Properties:
+              BucketEncryption:
+                 ServerSideEncryptionConfiguration:
+                 - ServerSideEncryptionByDefault:
+                    SSEAlgorithm: AES256
+              Tags:
+                - Key: Name
+                  Value: UFO
+        Outputs:
+          Bucket:
+            Value:
+              Ref: Bucket
+      YAML
+    end
+
+    def rollback
+      Rollback.new(STACK_NAME)
+    end
+
+    class << self
+      @@name = nil
+      def name
+        return @@name if @@name # only memoize once bucket has been created
+
+        AwsSetup.new.check!
+
+        stack = new.find_stack
+        return unless stack
+
+        stack_resources = find_stack_resources(STACK_NAME)
+        bucket = stack_resources.find { |r| r.logical_resource_id == "Bucket" }
+        @@name = bucket.physical_resource_id # actual bucket name
+      end
+    end
+  end
+end

data/lib/ufo/s3/rollback.rb

@@ -0,0 +1,52 @@
+module Ufo::S3
+  class Rollback
+    extend Memoist
+    include Ufo::AwsServices
+    include Ufo::Utils::Logging
+
+    def initialize(stack)
+      @stack = stack
+    end
+
+    def delete_stack
+      return unless complete?
+      logger.info "Existing stack in ROLLBACK_COMPLETE state. Deleting stack before continuing."
+      cfn.delete_stack(stack_name: @stack)
+      status.wait
+      status.reset
+      true
+    end
+
+    def continue_update
+      continue_update?
+      begin
+        cfn.continue_update_rollback(stack_name: @stack)
+      rescue Aws::CloudFormation::Errors::ValidationError => e
+        logger.info "ERROR: Continue update: #{e.message}".color(:red)
+        quit 1
+      end
+    end
+
+    def continue_update?
+      logger.info <<~EOL
+        The stack is in the UPDATE_ROLLBACK_FAILED state. More info here: https://amzn.to/2IiEjc5
+        Would you like to try to continue the update rollback? (y/N)
+      EOL
+
+      yes = @options[:yes] ? "y" : $stdin.gets
+      unless yes =~ /^y/
+        logger.info "Exiting without continuing the update rollback."
+        quit 0
+      end
+    end
+
+    def complete?
+      stack&.stack_status == 'ROLLBACK_COMPLETE'
+    end
+
+    def stack
+      find_stack(@stack)
+    end
+    memoize :stack
+  end
+end

data/lib/ufo/version.rb

@@ -1,3 +1,3 @@
 module Ufo
-  VERSION = "6.1.3"
+  VERSION = "6.2.0"
 end

data/ufo.gemspec

@@ -28,6 +28,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency "aws-sdk-ecr"
   spec.add_dependency "aws-sdk-ecs"
   spec.add_dependency "aws-sdk-elasticloadbalancingv2"
+  spec.add_dependency "aws-sdk-s3"
   spec.add_dependency "aws-sdk-ssm"
   spec.add_dependency "aws-sdk-wafv2"
   spec.add_dependency "aws_data"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ufo
 version: !ruby/object:Gem::Version
-  version: 6.1.3
+  version: 6.2.0
 platform: ruby
 authors:
 - Tung Nguyen
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-03-14 00:00:00.000000000 Z
+date: 2022-03-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-logs
@@ -150,6 +150,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-s3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: aws-sdk-ssm
   requirement: !ruby/object:Gem::Requirement
@@ -472,6 +486,7 @@ files:
 - lib/ufo.rb
 - lib/ufo/autoloader.rb
 - lib/ufo/aws_services.rb
+- lib/ufo/aws_services/concerns.rb
 - lib/ufo/booter.rb
 - lib/ufo/cfn/base.rb
 - lib/ufo/cfn/deploy.rb
@@ -581,6 +596,10 @@ files:
 - lib/ufo/docker/dockerfile.rb
 - lib/ufo/docker/pusher.rb
 - lib/ufo/docker/state.rb
+- lib/ufo/docker/state/base.rb
+- lib/ufo/docker/state/bucket.rb
+- lib/ufo/docker/state/file.rb
+- lib/ufo/docker/state/s3.rb
 - lib/ufo/ecr/auth.rb
 - lib/ufo/ecr/cleaner.rb
 - lib/ufo/ecs/service.rb
@@ -598,6 +617,9 @@ files:
 - lib/ufo/logger/formatter.rb
 - lib/ufo/names.rb
 - lib/ufo/param.rb
+- lib/ufo/s3/aws_setup.rb
+- lib/ufo/s3/bucket.rb
+- lib/ufo/s3/rollback.rb
 - lib/ufo/task_definition.rb
 - lib/ufo/task_definition/builder.rb
 - lib/ufo/task_definition/context.rb