2017-06-30 18:14:22 -04:00
|
|
|
require "fileutils"
|
|
|
|
require "pathname"
|
2017-07-08 18:11:27 -04:00
|
|
|
require "digest/md5"
|
2017-07-06 09:41:22 -04:00
|
|
|
require "active_support/core_ext/numeric/bytes"
|
2017-06-30 18:14:22 -04:00
|
|
|
|
2017-07-24 16:36:30 -04:00
|
|
|
# Wraps a local disk path as a Active Storage service. See `ActiveStorage::Service` for the generic API
|
|
|
|
# documentation that applies to all services.
|
2017-07-06 06:22:44 -04:00
|
|
|
class ActiveStorage::Service::DiskService < ActiveStorage::Service
|
2017-06-30 18:04:19 -04:00
|
|
|
attr_reader :root
|
|
|
|
|
2017-07-03 10:02:05 -04:00
|
|
|
def initialize(root:)
|
2017-06-30 18:04:19 -04:00
|
|
|
@root = root
|
|
|
|
end
|
|
|
|
|
2017-07-06 10:01:11 -04:00
|
|
|
def upload(key, io, checksum: nil)
|
2017-07-09 11:04:28 -04:00
|
|
|
instrument :upload, key, checksum: checksum do
|
|
|
|
IO.copy_stream(io, make_path_for(key))
|
|
|
|
ensure_integrity_of(key, checksum) if checksum
|
|
|
|
end
|
2017-06-30 18:04:19 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def download(key)
|
|
|
|
if block_given?
|
2017-07-09 11:04:28 -04:00
|
|
|
instrument :streaming_download, key do
|
2017-07-13 18:09:56 -04:00
|
|
|
File.open(path_for(key), "rb") do |file|
|
2017-07-10 16:17:48 -04:00
|
|
|
while data = file.read(64.kilobytes)
|
2017-07-09 11:04:28 -04:00
|
|
|
yield data
|
|
|
|
end
|
2017-06-30 18:04:19 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
else
|
2017-07-09 11:04:28 -04:00
|
|
|
instrument :download, key do
|
|
|
|
File.binread path_for(key)
|
|
|
|
end
|
2017-06-30 18:04:19 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def delete(key)
|
2017-07-09 11:04:28 -04:00
|
|
|
instrument :delete, key do
|
|
|
|
begin
|
|
|
|
File.delete path_for(key)
|
|
|
|
rescue Errno::ENOENT
|
|
|
|
# Ignore files already deleted
|
|
|
|
end
|
|
|
|
end
|
2017-07-01 06:05:58 -04:00
|
|
|
end
|
|
|
|
|
2017-07-03 15:08:36 -04:00
|
|
|
def exist?(key)
|
2017-07-09 11:04:28 -04:00
|
|
|
instrument :exist, key do |payload|
|
|
|
|
answer = File.exist? path_for(key)
|
|
|
|
payload[:exist] = answer
|
|
|
|
answer
|
|
|
|
end
|
2017-06-30 18:04:19 -04:00
|
|
|
end
|
|
|
|
|
2017-07-23 18:50:31 -04:00
|
|
|
def url(key, expires_in:, disposition:, filename:, content_type:)
|
2017-07-09 11:04:28 -04:00
|
|
|
instrument :url, key do |payload|
|
2017-07-23 16:51:01 -04:00
|
|
|
verified_key_with_expiration = ActiveStorage.verifier.generate(key, expires_in: expires_in, purpose: :blob_key)
|
2017-07-03 15:06:09 -04:00
|
|
|
|
2017-07-10 16:17:48 -04:00
|
|
|
generated_url =
|
2017-07-25 21:03:48 -04:00
|
|
|
if defined?(Rails.application)
|
2017-07-27 16:15:38 -04:00
|
|
|
Rails.application.routes.url_helpers.rails_disk_service_path \
|
2017-07-23 18:50:31 -04:00
|
|
|
verified_key_with_expiration,
|
|
|
|
disposition: disposition, filename: filename, content_type: content_type
|
2017-07-09 11:04:28 -04:00
|
|
|
else
|
2017-07-23 18:50:31 -04:00
|
|
|
"/rails/active_storage/disk/#{verified_key_with_expiration}/#{filename}?disposition=#{disposition}&content_type=#{content_type}"
|
2017-07-09 11:04:28 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
payload[:url] = generated_url
|
2017-07-10 16:17:48 -04:00
|
|
|
|
2017-07-09 11:04:28 -04:00
|
|
|
generated_url
|
2017-07-03 14:14:28 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2017-07-25 21:03:48 -04:00
|
|
|
def url_for_direct_upload(key, expires_in:, content_type:, content_length:, checksum:)
|
|
|
|
instrument :url, key do |payload|
|
|
|
|
verified_token_with_expiration = ActiveStorage.verifier.generate(
|
|
|
|
{
|
|
|
|
key: key,
|
|
|
|
content_type: content_type,
|
|
|
|
content_length: content_length,
|
|
|
|
checksum: checksum
|
|
|
|
},
|
2017-08-03 17:32:59 -04:00
|
|
|
{
|
|
|
|
expires_in: expires_in,
|
|
|
|
purpose: :blob_token
|
|
|
|
}
|
2017-07-25 21:03:48 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
generated_url =
|
|
|
|
if defined?(Rails.application)
|
2017-07-27 16:15:38 -04:00
|
|
|
Rails.application.routes.url_helpers.update_rails_disk_service_path verified_token_with_expiration
|
2017-07-25 21:03:48 -04:00
|
|
|
else
|
|
|
|
"/rails/active_storage/disk/#{verified_token_with_expiration}"
|
|
|
|
end
|
|
|
|
|
|
|
|
payload[:url] = generated_url
|
|
|
|
|
|
|
|
generated_url
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2017-07-30 11:00:55 -04:00
|
|
|
def headers_for_direct_upload(key, content_type:, **)
|
|
|
|
{ "Content-Type" => content_type }
|
|
|
|
end
|
|
|
|
|
2017-06-30 18:04:19 -04:00
|
|
|
private
|
|
|
|
def path_for(key)
|
2017-07-01 06:05:58 -04:00
|
|
|
File.join root, folder_for(key), key
|
2017-06-30 18:04:19 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def folder_for(key)
|
2017-07-01 06:05:58 -04:00
|
|
|
[ key[0..1], key[2..3] ].join("/")
|
2017-06-30 18:04:19 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def make_path_for(key)
|
|
|
|
path_for(key).tap { |path| FileUtils.mkdir_p File.dirname(path) }
|
|
|
|
end
|
2017-07-06 10:01:11 -04:00
|
|
|
|
|
|
|
def ensure_integrity_of(key, checksum)
|
|
|
|
unless Digest::MD5.file(path_for(key)).base64digest == checksum
|
2017-07-25 21:03:48 -04:00
|
|
|
delete key
|
2017-07-06 10:01:11 -04:00
|
|
|
raise ActiveStorage::IntegrityError
|
|
|
|
end
|
|
|
|
end
|
2017-06-30 18:04:19 -04:00
|
|
|
end
|