Made the output filename the responsibility of the website subclass.

[dead/whatever-dl.git] / bin / whatever-dl

diff --git a/bin/whatever-dl b/bin/whatever-dl
index c20b8bd7922e761c0bf553109381425f30830bee..a59e61b2d56a0c92ef5187315f1192456aa4e6fe 100755 (executable)
--- a/bin/whatever-dl
+++ b/bin/whatever-dl
@@ -19,6 +19,11 @@
 # http://www.fsf.org/licensing/licenses/gpl.html
 #
 
+# We require the UriUtilities class to handle
+# the download of the video URL.
+require 'src/uri_utilities'
+
+
 # All of the website classes are located in one
 # directory, so we can 'require' them automatically.
 Dir.glob('src/websites/*.rb').each do |r|
@@ -26,6 +31,16 @@ Dir.glob('src/websites/*.rb').each do |r|
 end
 
 
+EXIT_SUCCESS = 0
+EXIT_NO_URL = 1
+EXIT_INVALID_URL = 2
+EXIT_COULDNT_GET_VIDEO_URL = 3
+EXIT_OUTPUT_FILE_ALREADY_EXISTS = 4
+EXIT_ERROR_READING_FROM_VIDEO_URL = 5
+EXIT_CONNECTION_REFUSED = 6
+EXIT_HTTP_ERROR = 7
+EXIT_ACCESS_DENIED = 8
+
 # Only actually do something if this script was called
 # directly (i.e. not from the tests).
 if (__FILE__ == $0) then
@@ -33,34 +48,52 @@ if (__FILE__ == $0) then
     # If the user didn't give us a URL, yell
     # at him or her.
     puts 'Usage: whatever-dl <url>'
-    Kernel.exit(1)
+    Kernel.exit(EXIT_NO_URL)
   end
 
-  # Check the URL against each website's class.
-  # The class will know whether or not the URL
-  # "belongs" to its website.
-
-  site = nil
-  
-  Website.subclasses.each do |w|
-    if w.owns_url?(ARGV[0])
-      site = w.new()
-      break
-    end
-  end
+  # Factory method.
+  site = Website.create(ARGV[0])
 
   if site.nil?
     puts 'Invalid URL.'
-    exit(1)
+    exit(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)
+    exit(EXIT_COULDNT_GET_VIDEO_URL)
   end
 
-  # *classy*
-  Kernel.exec("wget \"#{video_url}\"")
+  video_uri = URI.parse(video_url)
+  uu = UriUtilities.new()
+  
+  if File.exists?(site.get_video_filename())
+    puts "Error: output file already exists. Please remove #{site.get_video_filename()}, and try again."
+    Kernel.exit(EXIT_OUTPUT_FILE_ALREADY_EXISTS)
+  end
+
+
+  # Attempt to download the file, and rescue and report
+  # any (predictable) exceptions.
+  begin
+    puts "Fetching #{video_url}"
+    puts "Saving as #{site.get_video_filename()}."
+    puts ""
+    uu.download_with_progress_bar(video_uri, site.get_video_filename())
+  rescue Errno::ECONNREFUSED => e
+    puts 'The connection to the server (to download the video file) was refused. Check your connection, and try again later.'
+    Kernel.exit(EXIT_CONNECTION_REFUSED)
+  rescue Errno::EACCES => e
+    puts "Access denied. Check that you have write permission to the output file/directory. Details: #{e.message}."
+  rescue OpenURI::HTTPError => e
+    puts "An HTTP error occurred while downloading the video file: #{e.message}."
+    Kernel.exit(EXIT_HTTP_ERROR)
+  end
+
+  # Write an empty line at the end for aesthetic reasons.
+  puts ""
+  
+  Kernel.exit(EXIT_SUCCESS)
 end