Class: Sprout::RemoteFileLoader

Inherits:
Object
  • Object
show all
Defined in:
lib/sprout/remote_file_loader.rb

Overview

:nodoc:

Instance Method Summary collapse

Instance Method Details

#fetch(uri) ⇒ Object 



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
66
67
 
# File 'lib/sprout/remote_file_loader.rb', line 34

def fetch(uri)
  uri = URI.parse(uri)
  progress = nil
  response = nil
  name = uri.path.split("/").pop
  
  begin
    open(uri.to_s, :content_length_proc => lambda {|t|
      if t && t > 0
        progress = ProgressBar.new(name, t)
        progress.file_transfer_mode
        progress.set(0)
      else
        progress = ProgressBar.new(name, 0)
        progress.file_transfer_mode
        progress.set(0)
      end
    },
    :progress_proc => lambda {|s|
      progress.set s if progress
    }) do |f|
      response = f.read
      progress.finish
    end
  rescue SocketError => sock_err
    raise RemoteFileLoaderError.new("[ERROR] #{sock_err.to_s}")
  rescue OpenURI::HTTPError => http_err
    raise RemoteFileLoaderError.new("[ERROR] Failed to load file from: '#{uri.to_s}'\n[REMOTE ERROR] #{http_err.io.read.strip}")
  rescue Errno::ECONNREFUSED => econ_err
    raise Errno::ECONNREFUSED.new("[ERROR] Connection refused at: '#{uri.to_s}'")
  end

  return response
end

#get_remote_file(uri, force = false, md5 = nil) ⇒ Object 



8
9
10
11
12
13
 
# File 'lib/sprout/remote_file_loader.rb', line 8

def get_remote_file(uri, force=false, md5=nil)
  response = fetch(uri.to_s)
  if(force || response_is_valid?(response, md5))
    return response
  end
end

#response_is_valid?(response, expected_md5sum = nil) ⇒ Boolean

Returns:

  • (Boolean) 


15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
 
# File 'lib/sprout/remote_file_loader.rb', line 15

def response_is_valid?(response, expected_md5sum=nil)
  if(expected_md5sum)
    md5 = Digest::MD5.new
    md5 << response
    
    if(expected_md5sum != md5.hexdigest)
      puts "The MD5 Sum of the downloaded file (#{md5.hexdigest}) does not match what was expected (#{expected_md5sum})."
      puts "Would you like to install anyway? [Yn]"
      response = $stdin.gets.chomp!
      if(response.downcase == 'y')
        return true
      else
        raise RemoteFileLoaderError.new('MD5 Checksum failed')
      end
    end
  end
  return true
end