Class: PKGWizard::StreamingDownloader

Inherits:
Object
  • Object
show all
Defined in:
lib/pkg-wizard/streaming_downloader.rb

Overview

StreamingDownloader code based on HTTPDownloader code from www.vagrantup.com

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.match?(uri) ⇒ Boolean

Returns:

  • (Boolean) 


12
13
14
15
16
 
# File 'lib/pkg-wizard/streaming_downloader.rb', line 12

def self.match?(uri)
  # URI.parse barfs on '<drive letter>:\\files \on\ windows'
  extracted = URI.extract(uri).first
  extracted && extracted.include?(uri)
end

Instance Method Details

#download!(source_url, destination_file) ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
 
# File 'lib/pkg-wizard/streaming_downloader.rb', line 28

def download!(source_url, destination_file)
proxy_uri = URI.parse(ENV["http_proxy"] || "")
uri = URI.parse(source_url)
http = Net::HTTP.new(uri.host, uri.port, proxy_uri.host, proxy_uri.port, proxy_uri.user, proxy_uri.password)

if uri.scheme == "https"
  http.use_ssl = true
  http.verify_mode = OpenSSL::SSL::VERIFY_NONE
end

http.start do |h|
  h.request_get(uri.request_uri) do |response|
    total = response.content_length
    progress = 0
    segment_count = 0

    response.read_body do |segment|
      # Report the progress out
      progress += segment.length
      segment_count += 1

      # Progress reporting is limited to every 25 segments just so
      # we're not constantly updating
      if segment_count % 25 == 0
        report_progress(progress, total)
        segment_count = 0
      end


      # Store the segment
      destination_file.write(segment)
    end
  end
end
puts
rescue SocketError
  raise Errors::DownloaderHTTPSocketError.new
end

#report_progress(progress, total, show_parts = true) ⇒ Object 



18
19
20
21
22
23
24
25
26
 
# File 'lib/pkg-wizard/streaming_downloader.rb', line 18

def report_progress(progress, total, show_parts=true)
  line_reset = "\r\e[0K" 
  percent = (progress.to_f / total.to_f) * 100
  line = "Progress: #{percent.to_i}%"
  line << " (#{progress} / #{total})" if show_parts
  line = "#{line_reset}#{line}"
  $stdout.sync = true
  $stdout.print line
end