1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
# frozen_string_literal: true
module ActiveStorage
class Downloader # :nodoc:
attr_reader :service
def initialize(service)
@service = service
end
def open(key, checksum: nil, verify: true, name: "ActiveStorage-", tmpdir: nil)
open_tempfile(name, tmpdir) do |file|
download key, file
verify_integrity_of(file, checksum: checksum) if verify
yield file
end
end
private
def open_tempfile(name, tmpdir = nil)
file = Tempfile.open(name, tmpdir)
begin
yield file
ensure
file.close!
end
end
def download(key, file)
file.binmode
service.download(key) { |chunk| file.write(chunk) }
file.flush
file.rewind
end
def verify_integrity_of(file, checksum:)
unless OpenSSL::Digest::MD5.file(file).base64digest == checksum
raise ActiveStorage::IntegrityError
end
end
end
end
|
