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 47253f13d8bd00eace1ccca43e69592de5948411..a59e61b2d56a0c92ef5187315f1192456aa4e6fe 100755 (executable)
--- a/bin/whatever-dl
+++ b/bin/whatever-dl
@@ -51,25 +51,15 @@ if (__FILE__ == $0) then
     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(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.'
@@ -79,22 +69,8 @@ if (__FILE__ == $0) then
   video_uri = URI.parse(video_url)
   uu = UriUtilities.new()
   
-
-  # Here, we start out with a default file name and
-  # extension. If UriUtilities can parse a sane filename
-  # out of the URL, we'll use that. Otherwise, we fall
-  # back to the default.
-  outfile_name = 'default.ext'
-  
-  if not uu.get_filename(video_uri).nil?
-    outfile_name = uu.get_filename(video_uri)
-  else
-    puts "We couldn't determine the video's filename. Falling back to the default, #{outfile_name}."
-  end
-
-  
-  if File.exists?(outfile_name)
-    puts "Error: output file already exists. Please remove #{outfile_name}, and try again."
+  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
 
@@ -103,7 +79,9 @@ if (__FILE__ == $0) then
   # any (predictable) exceptions.
   begin
     puts "Fetching #{video_url}"
-    uu.download_with_progress_bar(video_uri, outfile_name)
+    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)
@@ -114,5 +92,8 @@ if (__FILE__ == $0) then
     Kernel.exit(EXIT_HTTP_ERROR)
   end
 
+  # Write an empty line at the end for aesthetic reasons.
+  puts ""
+  
   Kernel.exit(EXIT_SUCCESS)
 end