Correct rule installation in AWS

This commit is contained in:
Sam 2017-10-09 10:26:58 +11:00
parent 70bb2aa426
commit a8b4255bae
2 changed files with 62 additions and 39 deletions

View File

@ -44,30 +44,70 @@ class S3Helper
rescue Aws::S3::Errors::NoSuchKey rescue Aws::S3::Errors::NoSuchKey
end end
def update_lifecycle(id, days, prefix: nil) # make sure we have a cors config for assets
# otherwise we will have no fonts
def ensure_cors!
rule = nil
begin
rule = s3_resource.client.get_bucket_cors(
bucket: @s3_bucket_name
).cors_rules&.first
rescue Aws::S3::Errors::NoSuchCORSConfiguration
# no rule
end
unless rule
puts "installing CORS rule"
s3_resource.client.put_bucket_cors(
bucket: @s3_bucket_name,
cors_configuration: {
cors_rules: [{
allowed_headers: ["Authorization"],
allowed_methods: ["GET", "HEAD"],
allowed_origins: ["*"],
max_age_seconds: 3000
}]
}
)
end
end
def update_lifecycle(id, days, prefix: nil, tag: nil)
filter = {}
if prefix
filter[:prefix] = prefix
elsif tag
filter[:tag] = tag
end
# cf. http://docs.aws.amazon.com/AmazonS3/latest/dev/object-lifecycle-mgmt.html # cf. http://docs.aws.amazon.com/AmazonS3/latest/dev/object-lifecycle-mgmt.html
rule = { rule = {
id: id, id: id,
status: "Enabled", status: "Enabled",
expiration: { days: days } expiration: { days: days },
filter: filter
} }
if prefix rules = []
rule[:prefix] = prefix
end
begin
rules = s3_resource.client.get_bucket_lifecycle_configuration(bucket: @s3_bucket_name).rules rules = s3_resource.client.get_bucket_lifecycle_configuration(bucket: @s3_bucket_name).rules
rescue Aws::S3::Errors::NoSuchLifecycleConfiguration
# skip trying to merge
end
rules.delete_if do |r| rules.delete_if do |r|
r.id == id r.id == id
end end
rules.map! { |r| r.to_h }
rules << rule rules << rule
s3_resource.client.put_bucket_lifecycle(bucket: @s3_bucket_name, s3_resource.client.put_bucket_lifecycle_configuration(
bucket: @s3_bucket_name,
lifecycle_configuration: { lifecycle_configuration: {
rules: rules rules: rules
}) })

View File

@ -71,28 +71,8 @@ def assets
results results
end end
def in_manifest def asset_paths
found = [] Set.new(assets.map { |_, asset_path| asset_path })
assets.each do |_, path|
fullpath = (Rails.root + "public/assets/#{path}").to_s
asset_path = "assets/#{path}"
found << asset_path
if File.exist?(fullpath + '.br')
found << brotli_s3_path(asset_path)
end
if File.exist?(fullpath + '.gz')
found << gzip_s3_path(asset_path)
end
if File.exist?(fullpath + '.map')
found << asset_path + '.map'
end
end
Set.new(found)
end end
def ensure_s3_configured! def ensure_s3_configured!
@ -104,6 +84,7 @@ end
task 's3:upload_assets' => :environment do task 's3:upload_assets' => :environment do
ensure_s3_configured! ensure_s3_configured!
helper.ensure_cors!
assets.each do |asset| assets.each do |asset|
upload(*asset) upload(*asset)
@ -112,25 +93,27 @@ end
task 's3:expire_missing_assets' => :environment do task 's3:expire_missing_assets' => :environment do
ensure_s3_configured! ensure_s3_configured!
keep = in_manifest
count = 0 count = 0
keep = 0
in_manifest = asset_paths
puts "Ensuring AWS assets are tagged correctly for removal" puts "Ensuring AWS assets are tagged correctly for removal"
helper.list.each do |f| helper.list('assets/').each do |f|
if keep.include?(f.key) if !in_manifest.include?(f.key)
helper.tag_file(f.key, old: true) helper.tag_file(f.key, old: true)
count += 1 count += 1
else else
# ensure we do not delete this by mistake # ensure we do not delete this by mistake
helper.tag_file(f.key, {}) helper.tag_file(f.key, {})
keep += 1
end end
end end
puts "#{count} assets were flagged for removal in 10 days" puts "#{count} assets were flagged for removal in 10 days (#{keep} assets will be retained)"
puts "Ensuring AWS rule exists for purging old assets" puts "Ensuring AWS rule exists for purging old assets"
#helper.update_lifecycle("delete_old_assets", 10, prefix: 'old=true') helper.update_lifecycle("delete_old_assets", 10, tag: { key: 'old', value: 'true' })
puts "Waiting on https://github.com/aws/aws-sdk-ruby/issues/1623"
end end