Sha256: 13311d0dbad62629e8265116d811fdaa1854fac6cfb72b5f90c48b1395d6af66
Contents?: true
Size: 1.83 KB
Versions: 13
Compression:
Stored size: 1.83 KB
Contents
require "json"
# Can test:
#
# ufo ship app-web --cluster prod --noop
module Ufo
class Ecr::Cleaner
include Util
include AwsService
def initialize(docker_image_name, options={})
# docker_image_name does not containg the tag
# Example: 123456789.dkr.ecr.us-east-1.amazonaws.com/image
@docker_image_name = docker_image_name
@options = options
@keep = options[:ecr_keep] || settings["ecr_keep"]
@tag_prefix = options[:tag_prefix] || "ufo"
end
def cleanup
return false unless ecr_image?
return false unless @keep
update_auth_token
image_tags = fetch_image_tags
delete_tags = image_tags[@keep..-1] # ordered by most recent images first
delete_images(delete_tags)
end
def fetch_image_tags
ecr.list_images(repository_name: repo_name).
image_ids.
map { |image_id| image_id.image_tag }.
select { |image_tag| image_tag =~ Regexp.new("^#{@tag_prefix}-") }.
sort.reverse
end
def delete_images(tags)
return if tags.nil? || tags.empty?
unless @options[:mute]
puts "Keeping #{@keep} most recent ECR images."
puts "Deleting these ECR images:"
tag_list = tags.map { |t| " #{repo_name}:#{t}" }
puts tag_list
end
image_ids = tags.map { |tag| {image_tag: tag} }
ecr.batch_delete_image(
repository_name: repo_name,
image_ids: image_ids) unless @options[:noop]
end
def update_auth_token
repo_domain = "#{@docker_image_name.split('/').first}"
auth = Ecr::Auth.new(repo_domain)
auth.update
end
def repo_name
# @docker_image_name example: 123456789.dkr.ecr.us-east-1.amazonaws.com/image
@docker_image_name.split('/').last
end
def ecr_image?
@docker_image_name =~ /\.amazonaws\.com/
end
end
end
Version data entries
13 entries across 13 versions & 1 rubygems