src/wget_downloader.rb

   1 #
   2 # Copyright Michael Orlitzky
   3 #
   4 # http://michael.orlitzky.com/
   5 #
   6 # This program is free software: you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation, either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14 # GNU General Public License for more details.
  15 #
  16 # http://www.fsf.org/licensing/licenses/gpl.html
  17 #
  18
  19 class WgetDownloader < Downloader
  20
  21   def download(url, outfile, headers = {}, continue = false)
  22     if (continue == false and File.exists?(outfile))
  23       raise IOError.new("Output file already exists. Please remove #{outfile}, and try again. If this is a partially-downloaded file, you can use the --continue flag to pick up where it left off.")
  24     end
  25
  26     options = ''
  27
  28     if continue == true
  29       options += '--continue '
  30     end
  31
  32     headers.each_key do |key|
  33       options += "--header '#{key}: #{headers[key]}' "
  34     end
  35
  36     # This one's easy.
  37     cmd = "wget #{options} -O \"#{outfile}\" \"#{url}\""
  38     puts "\nExecuting external command: #{cmd}\n\n"
  39     Kernel.exec(cmd)
  40   end
  41
  42 end