mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
c624df326a
Yaroslav agreed to hand over the gem name ❤️
53 lines
1.1 KiB
Ruby
53 lines
1.1 KiB
Ruby
require "google/cloud/storage"
|
|
require "active_support/core_ext/object/to_query"
|
|
|
|
class ActiveStorage::Site::GCSSite < ActiveStorage::Site
|
|
attr_reader :client, :bucket
|
|
|
|
def initialize(project:, keyfile:, bucket:)
|
|
@client = Google::Cloud::Storage.new(project: project, keyfile: keyfile)
|
|
@bucket = @client.bucket(bucket)
|
|
end
|
|
|
|
def upload(key, io)
|
|
bucket.create_file(io, key)
|
|
end
|
|
|
|
def download(key)
|
|
io = file_for(key).download
|
|
io.rewind
|
|
io.read
|
|
end
|
|
|
|
def delete(key)
|
|
file_for(key).try(:delete)
|
|
end
|
|
|
|
def exist?(key)
|
|
file_for(key).present?
|
|
end
|
|
|
|
|
|
def url(key, expires_in:, disposition:, filename:)
|
|
file_for(key).signed_url(expires: expires_in) + "&" +
|
|
{ "response-content-disposition" => "#{disposition}; filename=\"#{filename}\"" }.to_query
|
|
end
|
|
|
|
def byte_size(key)
|
|
file_for(key).size
|
|
end
|
|
|
|
def checksum(key)
|
|
convert_to_hex base64: file_for(key).md5
|
|
end
|
|
|
|
|
|
private
|
|
def file_for(key)
|
|
bucket.file(key)
|
|
end
|
|
|
|
def convert_to_hex(base64:)
|
|
base64.unpack("m0").first.unpack("H*").first
|
|
end
|
|
end
|