#!/usr/bin/ruby -w
#
# whatever-dl, a script to download online (web-based) videos.
#
# Copyright Michael Orlitzky
#
# http://michael.orlitzky.com/
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# 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|
  require 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
  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(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

  if site.nil?
    puts 'Invalid URL.'
    exit(EXIT_INVALID_URL)
  end
  
  video_url = site.get_video_url(ARGV[0])

  if video_url.nil?
    puts 'Error retrieving video URL.'
    exit(EXIT_COULDNT_GET_VIDEO_URL)
  end

  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."
    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)
  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

  Kernel.exit(EXIT_SUCCESS)
end