Last active
November 6, 2022 12:52
-
-
Save dinatih/dbfdfd4e84faac4037448a06c9fdc016 to your computer and use it in GitHub Desktop.
ActiveStorageAclOption https://github.com/rails/rails/issues/31419#issuecomment-370900013
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# config/initializers/activestorage.rb | |
Rails.application.config.to_prepare do | |
# Provides the class-level DSL for declaring that an Active Record model has attached blobs. | |
ActiveStorage::Attached::Macros.module_eval do | |
def has_one_attached(name, dependent: :purge_later, acl: :private) | |
class_eval <<-CODE, __FILE__, __LINE__ + 1 | |
def #{name} | |
@active_storage_attached_#{name} ||= ActiveStorage::Attached::One.new("#{name}", self, dependent: #{dependent == :purge_later ? ":purge_later" : "false"}, acl: "#{acl}") | |
end | |
def #{name}=(attachable) | |
#{name}.attach(attachable) | |
end | |
CODE | |
has_one :"#{name}_attachment", -> { where(name: name) }, class_name: "ActiveStorage::Attachment", as: :record, inverse_of: :record, dependent: false | |
has_one :"#{name}_blob", through: :"#{name}_attachment", class_name: "ActiveStorage::Blob", source: :blob | |
scope :"with_attached_#{name}", -> { includes("#{name}_attachment": :blob) } | |
if dependent == :purge_later | |
after_destroy_commit { public_send(name).purge_later } | |
else | |
before_destroy { public_send(name).detach } | |
end | |
end | |
def has_many_attached(name, dependent: :purge_later, acl: :private) | |
class_eval <<-CODE, __FILE__, __LINE__ + 1 | |
def #{name} | |
@active_storage_attached_#{name} ||= ActiveStorage::Attached::Many.new("#{name}", self, dependent: #{dependent == :purge_later ? ":purge_later" : "false"}, acl: "#{acl}") | |
end | |
def #{name}=(attachables) | |
#{name}.attach(attachables) | |
end | |
CODE | |
has_many :"#{name}_attachments", -> { where(name: name) }, as: :record, class_name: "ActiveStorage::Attachment", inverse_of: :record, dependent: false do | |
def purge | |
each(&:purge) | |
reset | |
end | |
def purge_later | |
each(&:purge_later) | |
reset | |
end | |
end | |
has_many :"#{name}_blobs", through: :"#{name}_attachments", class_name: "ActiveStorage::Blob", source: :blob | |
scope :"with_attached_#{name}", -> { includes("#{name}_attachments": :blob) } | |
if dependent == :purge_later | |
after_destroy_commit { public_send(name).purge_later } | |
else | |
before_destroy { public_send(name).detach } | |
end | |
end | |
end | |
ActiveStorage::Blob.class_eval do | |
def service_url(expires_in: service.url_expires_in, disposition: :inline, filename: nil, **options) | |
filename = ActiveStorage::Filename.wrap(filename || self.filename) | |
expires_in = false if metadata[:acl] == 'public' | |
service.url key, expires_in: expires_in, filename: filename, content_type: content_type, | |
disposition: forcibly_serve_as_binary? ? :attachment : disposition, **options | |
end | |
def upload(io) | |
self.checksum = compute_checksum_in_chunks(io) | |
self.content_type = extract_content_type(io) | |
self.byte_size = io.size | |
self.identified = true | |
service.upload(key, io, checksum: checksum, acl: metadata[:acl]) | |
end | |
end | |
ActiveStorage::Attached.class_eval do | |
attr_reader :name, :record, :dependent, :acl | |
def initialize(name, record, dependent:, acl: 'private') | |
@name, @record, @dependent, @acl = name, record, dependent, acl | |
end | |
private | |
def create_blob_from(attachable) | |
case attachable | |
when ActiveStorage::Blob | |
attachable | |
when ActionDispatch::Http::UploadedFile, Rack::Test::UploadedFile | |
ActiveStorage::Blob.create_after_upload! \ | |
io: attachable.open, | |
filename: attachable.original_filename, | |
content_type: attachable.content_type, | |
metadata: { acl: acl } | |
when Hash | |
ActiveStorage::Blob.create_after_upload!({ metadata: { acl: acl } }.deep_merge(attachable)) | |
when String | |
ActiveStorage::Blob.find_signed(attachable) | |
else | |
nil | |
end | |
end | |
end | |
if defined?(ActiveStorage::Service) | |
ActiveStorage::Service.class_eval do | |
def upload(key, io, checksum: nil, acl: 'private') | |
raise NotImplementedError | |
end | |
end | |
end | |
if defined?(ActiveStorage::Service::DiskService) | |
ActiveStorage::Service::DiskService.class_eval do | |
def upload(key, io, checksum: nil, acl: 'private') | |
instrument :upload, key: key, checksum: checksum do | |
IO.copy_stream(io, make_path_for(key)) | |
ensure_integrity_of(key, checksum) if checksum | |
end | |
end | |
end | |
end | |
if defined?(ActiveStorage::Service::S3Service) | |
# from activestorage/lib/active_storage/service/s3_service.rb | |
ActiveStorage::Service::S3Service.class_eval do | |
def upload(key, io, checksum: nil, acl: 'private') | |
instrument :upload, key: key, checksum: checksum, acl: acl do | |
begin | |
object_for(key).put(upload_options.merge(body: io, content_md5: checksum, | |
acl: acl == 'public' ? 'public-read' : 'private')) | |
rescue Aws::S3::Errors::BadDigest | |
raise ActiveStorage::IntegrityError | |
end | |
end | |
end | |
def url(key, expires_in:, filename:, disposition:, content_type:) | |
instrument :url, key: key, expires_in: expires_in do |payload| | |
generated_url = if expires_in == false | |
object_for(key).public_url | |
else | |
object_for(key).presigned_url :get, | |
expires_in: expires_in.to_i, | |
response_content_disposition: content_disposition_with( | |
type: disposition, filename: filename | |
), | |
response_content_type: content_type | |
end | |
payload[:url] = generated_url | |
generated_url | |
end | |
end | |
end | |
end | |
end |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
When attaching a file generated via a tempfile, I had an issue (Rails 6.0) with build_after_unfurling method
"ArgumentError: wrong number of arguments (given 1, expected 0; required keywords: io, filename)"
Here is the corrected code :
Full code :