activestorage/lib/active_storage/downloading.rb


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

# frozen_string_literal: true

require "tmpdir"

module ActiveStorage
  module Downloading
    private
      # Opens a new tempfile in #tempdir and copies blob data into it. Yields the tempfile.
      def download_blob_to_tempfile #:doc:
        open_tempfile_for_blob do |file|
          download_blob_to file
          yield file
        end
      end

      def open_tempfile_for_blob
        tempfile = Tempfile.open([ "ActiveStorage", blob.filename.extension_with_delimiter ], tempdir)

        begin
          yield tempfile
        ensure
          tempfile.close!
        end
      end

      # Efficiently downloads blob data into the given file.
      def download_blob_to(file) #:doc:
        file.binmode
        blob.download { |chunk| file.write(chunk) }
        file.rewind
      end

      # Returns the directory in which tempfiles should be opened. Defaults to +Dir.tmpdir+.
      def tempdir #:doc:
        Dir.tmpdir
      end
  end
end