mirror of
https://github.com/codeninjasllc/discourse.git
synced 2024-11-27 17:46:05 -05:00
bb0c2813ac
FIX: keep the "uploading..." indicator until the server replies via the MessageBus FIX: text was disapearing when uploading an avatar PERF: always use a region for S3 (defaults to 'us-east-1') FEATURE: ApplyCDN middleware when using S3 FIX: use the same pattern to store files on S3 and locally PERF: keep a local cache of uploads when generating thumbnails FEATURE: migrate_to_s3 rake task
77 lines
2.1 KiB
Ruby
77 lines
2.1 KiB
Ruby
require "aws-sdk"
|
|
|
|
class S3Helper
|
|
|
|
def initialize(s3_bucket, tombstone_prefix=nil)
|
|
raise Discourse::InvalidParameters.new("s3_bucket") if s3_bucket.blank?
|
|
|
|
@s3_bucket = s3_bucket
|
|
@tombstone_prefix = tombstone_prefix
|
|
|
|
check_missing_site_settings
|
|
end
|
|
|
|
def upload(file, unique_filename, options={})
|
|
obj = s3_bucket.object(unique_filename)
|
|
obj.upload_file(file, options)
|
|
end
|
|
|
|
def remove(unique_filename, copy_to_tombstone=false)
|
|
bucket = s3_bucket
|
|
# copy the file in tombstone
|
|
if copy_to_tombstone && @tombstone_prefix.present?
|
|
bucket.object(@tombstone_prefix + unique_filename).copy_from(copy_source: "#{@s3_bucket}/#{unique_filename}")
|
|
end
|
|
# delete the file
|
|
bucket.object(unique_filename).delete
|
|
rescue Aws::S3::Errors::NoSuchKey
|
|
end
|
|
|
|
def update_tombstone_lifecycle(grace_period)
|
|
return if @tombstone_prefix.blank?
|
|
|
|
# cf. http://docs.aws.amazon.com/AmazonS3/latest/dev/object-lifecycle-mgmt.html
|
|
s3_resource.client.put_bucket_lifecycle({
|
|
bucket: @s3_bucket,
|
|
lifecycle_configuration: {
|
|
rules: [
|
|
{
|
|
id: "purge-tombstone",
|
|
status: "Enabled",
|
|
expiration: { days: grace_period },
|
|
prefix: @tombstone_prefix
|
|
}
|
|
]
|
|
}
|
|
})
|
|
end
|
|
|
|
private
|
|
|
|
def s3_resource
|
|
opts = {}
|
|
|
|
opts = {
|
|
access_key_id: SiteSetting.s3_access_key_id,
|
|
secret_access_key: SiteSetting.s3_secret_access_key
|
|
} unless SiteSetting.s3_use_iam_profile
|
|
|
|
opts[:region] = SiteSetting.s3_region unless SiteSetting.s3_region.blank?
|
|
|
|
Aws::S3::Resource.new(opts)
|
|
end
|
|
|
|
def s3_bucket
|
|
bucket = s3_resource.bucket(@s3_bucket)
|
|
bucket.create unless bucket.exists?
|
|
bucket
|
|
end
|
|
|
|
def check_missing_site_settings
|
|
unless SiteSetting.s3_use_iam_profile
|
|
raise Discourse::SiteSettingMissing.new("s3_access_key_id") if SiteSetting.s3_access_key_id.blank?
|
|
raise Discourse::SiteSettingMissing.new("s3_secret_access_key") if SiteSetting.s3_secret_access_key.blank?
|
|
end
|
|
end
|
|
|
|
end
|