# http://www.fsf.org/licensing/licenses/gpl.html
#
+# We need Pathname to get the real filesystem path
+# of this script (and not, for example, the path of
+# a symlink which points to it.
+require 'pathname'
+
+# This bit of magic adds the parent directory (the
+# project root) to the list of ruby load paths.
+# Thus, our require statements will work regardless of
+# how or from where the script was run.
+executable = Pathname.new(__FILE__).realpath.to_s
+$: << File.dirname(executable) + '/../'
+
# We require the UriUtilities class to handle
# the download of the video URL.
require 'src/uri_utilities'
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.'
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
# Attempt to download the file, and rescue and report
# any (predictable) exceptions.
begin
- uu.download_with_progress_bar(video_uri, outfile_name)
+ 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
+ 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