X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=bin%2Fwhatever-dl;h=a40af28fab676f14b3ac36084c3bbdb7ca7cb731;hb=HEAD;hp=c20b8bd7922e761c0bf553109381425f30830bee;hpb=168ef3b2ccf5b97d561a3c542a18e8e7587de291;p=dead%2Fwhatever-dl.git

diff --git a/bin/whatever-dl b/bin/whatever-dl
index c20b8bd..a40af28 100755
--- a/bin/whatever-dl
+++ b/bin/whatever-dl
@@ -1,4 +1,4 @@
-#!/usr/bin/ruby -w
+#!/usr/bin/ruby -wKU
 #
 # whatever-dl, a script to download online (web-based) videos.
 #
@@ -19,48 +19,120 @@
 # http://www.fsf.org/licensing/licenses/gpl.html
 #
 
-# All of the website classes are located in one
-# directory, so we can 'require' them automatically.
-Dir.glob('src/websites/*.rb').each do |r|
-  require r
+# This should load everything we need for us.
+require 'whatever-dl'
+
+# And getoptlong to check for our one option, --continue.
+require 'getoptlong'
+
+log = Logger.new(STDOUT)
+log.level = Logger::WARN
+
+def usage()
+  puts <<EOF
+
+Usage: whatever-dl [options] <url>
+
+Options:
+  -c, --continue        Continue downloading a previously-attempted file.
+
+EOF
+
 end
 
 
-# Only actually do something if this script was called
-# directly (i.e. not from the tests).
-if (__FILE__ == $0) then
-  if (ARGV.length < 1) then
-    # If the user didn't give us a URL, yell
-    # at him or her.
-    puts 'Usage: whatever-dl <url>'
-    Kernel.exit(1)
-  end
+# Default options.
+options = { :continue => false }
 
-  # Check the URL against each website's class.
-  # The class will know whether or not the URL
-  # "belongs" to its website.
+# Parse the command-line options into the options hash.
+opts = GetoptLong.new(["--continue", "-c", GetoptLong::NO_ARGUMENT],
+                      ["--help", "-h", GetoptLong::NO_ARGUMENT])
 
-  site = nil
-  
-  Website.subclasses.each do |w|
-    if w.owns_url?(ARGV[0])
-      site = w.new()
-      break
-    end
+opts.each do |opt, arg|
+  case opt
+  when '--help'
+    usage()
+    Kernel.exit(ExitCodes::EXIT_SUCCESS)
+  when '--continue'
+    options[:continue] = true
   end
+end
 
-  if site.nil?
-    puts 'Invalid URL.'
-    exit(1)
-  end
+cfg = Configuration.new()
+
+# Warn about nonsensical options.
+if options[:continue] and not cfg.download_method == 'wget'
+  log.warn("The --continue flag does nothing unless download_method is wget.")
+end
+
+# Note that GetoptLong steals its arguments from ARGV, so we don't need
+# to take optional arguments into account when figuring out whether or not
+# we were passed a URL.
+if (ARGV.length < 1) then
+  # If the user didn't give us a URL, yell
+  # at him or her.
+  usage()
+  Kernel.exit(ExitCodes::EXIT_NO_URL)
+end
+
+
+# Factory method.
+site = Website.create(ARGV[0])
+
+if site.nil?
+  log.error('Invalid URL.')
+  exit(ExitCodes::EXIT_INVALID_URL)
+end
   
-  video_url = site.get_video_url(ARGV[0])
+video_url = site.get_video_url()
 
-  if video_url.nil?
-    puts 'Error retrieving video URL.'
-    exit(2)
-  end
+if video_url.nil?
+  msg  = 'Error retrieving video URL: '
+  msg += "Site not supported, and the generic parser couldn't find any videos."
+  log.error(msg)
+  exit(ExitCodes::EXIT_COULDNT_GET_VIDEO_URL)
+end
+
+# The Downloader class is a factory; it should decide
+# which subclass we get.
+puts "download_method: #{cfg.download_method}"
+downloader = Downloader.create(cfg.download_method)
+
+# Attempt to download the file, and rescue and report
+# any (predictable) exceptions. The wget downloader will
+# naturally not report any of these, since it will die in
+# its own process.
+begin
+  downloader.download(video_url,
+                      site.get_video_filename(),
+                      site.headers(),
+                      continue=options[:continue])
+
+rescue Errno::ECONNREFUSED => e
+  msg =  'The connection to the server (to download the video file) '
+  msg += 'was refused. Check your connection, and try again later.'
+  log.error(msg)
+  Kernel.exit(ExitCodes::EXIT_CONNECTION_REFUSED)
+
+rescue Errno::EACCES => e
+  msg =  'Access denied. Check that you have write permission '
+  msg += "to the output file/directory. Details: #{e.message}."
+  log.error(msg)
+  Kernel.exit(ExitCodes::EXIT_ACCESS_DENIED)
+
+rescue OpenURI::HTTPError => e
+  msg =  'An HTTP error occurred while downloading '
+  msg += " the video file: #{e.message}."
+  log.error(msg)
+  Kernel.exit(ExitCodes::EXIT_HTTP_ERROR)
+
+rescue IOError => e
+  log.error("Input/Output Error: #{e.message}")
+  Kernel.exit(ExitCodes::EXIT_IO_ERROR)
 
-  # *classy*
-  Kernel.exec("wget \"#{video_url}\"")
 end
+
+# Write an empty line at the end for aesthetic reasons.
+puts ''
+
+Kernel.exit(ExitCodes::EXIT_SUCCESS)