Class: FileFM::Downloaders::Swift

Inherits:
Object
  • Object
show all
Defined in:
lib/filefm/downloaders/swift.rb

Class Method Summary collapse

Class Method Details

.download(link, opts = {}) ⇒ Object 



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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
 
# File 'lib/filefm/downloaders/swift.rb', line 4

def self.download(link, opts = {})
  require 'fog'
  require 'fog/rackspace/storage'
  
  uri = URI.parse(link)
  container = uri.path.split("/")[1]
  object = uri.path.split("/")[2..-1].join("/")

  if (not opts[:username] or not opts[:password])
    raise "Invalid Credentials"
  end
  secure = opts[:secure] == true 
  scheme = secure ? "https" : "http"
  username = opts[:username]
  password = opts[:password]

  #puts "#{scheme}://#{uri.host}/auth/v1.0"
  conn = Fog::Storage.new({
    :provider => 'Rackspace',
    :rackspace_username => username,
    :rackspace_api_key => password,
    :rackspace_auth_url => "#{scheme}://#{uri.host}:#{uri.port}/auth/v1.0"
                  })


  out = RestClient.get "#{scheme}://#{uri.host}:#{uri.port}/auth/v1.0", 'X-Storage-User' => username, 'X-Storage-Pass' => password
  storage_url = out.headers[:x_storage_url]
  auth_token = out.headers[:x_auth_token]
  raise "Error authenticating" unless out.code == 200

  o = {
    :location => nil,
    :size => nil,
    :filename => nil
  }.merge(opts)
  
  @link = storage_url + "#{uri.path}"
  @size = o[:size]
  @location = o[:location] ||= ""
  @filename = o[:filename]
  @progress = 0
  
  headers = {
              "User-Agent" => "FileFM #{VERSION}",
              "X-Auth-Token" => auth_token
             }

  uri = URI.parse storage_url + uri.path
  http = Net::HTTP.new(uri.host, uri.port)
  http.use_ssl = true if uri.scheme == "https"
  http.open_timeout = 3 # seconds
  http.read_timeout = 3 # seconds
  request = Net::HTTP::Get.new(uri.request_uri)

  request.initialize_http_header headers

  container = conn.directories.get container
  raise "Container not found" if container.nil?

  object = container.files.get object
  @size = object.content_length
  
  puts "unknown file size for #{@filename} but downloading..." if @size.nil?
  #puts @link.to_s
  
  if opts[:output]
    dest_file = opts[:output]
  else
    dest_file = @location + (@filename ||= File.basename(uri.path))
  end
  response = http.request(request) do |response|
    if opts[:progressbar]
      bar = ProgressBar.new("Progress", @size.to_i) unless @size.nil?
      bar.format_arguments=[:title, :percentage, :bar, :stat_for_file_transfer] unless @size.nil?
    end
          
    File.open(dest_file, "wb") do |file|
      response.read_body do |segment|
        if opts[:progressbar]
          @progress += segment.length
          bar.set(@progress) unless @size.nil?
        end
        file.write(segment)
      end
    end
  end

  FileUtils.rm dest_file unless response.is_a? Net::HTTPOK
  raise "Error downlading file: #{response.class.to_s}" unless response.is_a? Net::HTTPOK
end