]> gitweb.michael.orlitzky.com - dead/whatever-dl.git/blob - bin/whatever-dl
Modified the Youtube class to accept in.youtube.com URLs.
[dead/whatever-dl.git] / bin / whatever-dl
1 #!/usr/bin/ruby -w
2 #
3 # whatever-dl, a script to download online (web-based) videos.
4 #
5 # Copyright Michael Orlitzky
6 #
7 # http://michael.orlitzky.com/
8 #
9 # This program is free software: you can redistribute it and/or modify
10 # it under the terms of the GNU General Public License as published by
11 # the Free Software Foundation, either version 3 of the License, or
12 # (at your option) any later version.
13 #
14 # This program is distributed in the hope that it will be useful,
15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 # GNU General Public License for more details.
18 #
19 # http://www.fsf.org/licensing/licenses/gpl.html
20 #
21
22 # We need Pathname to get the real filesystem path
23 # of this script (and not, for example, the path of
24 # a symlink which points to it.
25 require 'pathname'
26
27 # This bit of magic adds the parent directory (the
28 # project root) to the list of ruby load paths.
29 # Thus, our require statements will work regardless of
30 # how or from where the script was run.
31 executable = Pathname.new(__FILE__).realpath.to_s
32 $: << File.dirname(executable) + '/../'
33
34 # We require the UriUtilities class to handle
35 # the download of the video URL.
36 require 'src/uri_utilities'
37
38
39 # The Dir.glob that's coming up doesn't use the
40 # Ruby library path so we need to tell it where to
41 # look explicitly.
42 websites_pattern = File.dirname(executable) + '/../src/websites/*.rb'
43
44 # All of the website classes are located in one
45 # directory, so we can 'require' them automatically.
46 Dir.glob(websites_pattern).each do |r|
47 require r
48 end
49
50
51 EXIT_SUCCESS = 0
52 EXIT_NO_URL = 1
53 EXIT_INVALID_URL = 2
54 EXIT_COULDNT_GET_VIDEO_URL = 3
55 EXIT_OUTPUT_FILE_ALREADY_EXISTS = 4
56 EXIT_ERROR_READING_FROM_VIDEO_URL = 5
57 EXIT_CONNECTION_REFUSED = 6
58 EXIT_HTTP_ERROR = 7
59 EXIT_ACCESS_DENIED = 8
60
61 # Only actually do something if this script was called
62 # directly (i.e. not from the tests).
63 if (__FILE__ == $0) then
64 if (ARGV.length < 1) then
65 # If the user didn't give us a URL, yell
66 # at him or her.
67 puts 'Usage: whatever-dl <url>'
68 Kernel.exit(EXIT_NO_URL)
69 end
70
71 # Factory method.
72 site = Website.create(ARGV[0])
73
74 if site.nil?
75 puts 'Invalid URL.'
76 exit(EXIT_INVALID_URL)
77 end
78
79 video_url = site.get_video_url()
80
81 if video_url.nil?
82 puts 'Error retrieving video URL.'
83 exit(EXIT_COULDNT_GET_VIDEO_URL)
84 end
85
86 video_uri = URI.parse(video_url)
87 uu = UriUtilities.new()
88
89 if File.exists?(site.get_video_filename())
90 puts "Error: output file already exists. Please remove #{site.get_video_filename()}, and try again."
91 Kernel.exit(EXIT_OUTPUT_FILE_ALREADY_EXISTS)
92 end
93
94
95 # Attempt to download the file, and rescue and report
96 # any (predictable) exceptions.
97 begin
98 puts "Fetching #{video_url}"
99 puts "Saving as #{site.get_video_filename()}."
100 puts ""
101 uu.download_with_progress_bar(video_uri, site.get_video_filename())
102 rescue Errno::ECONNREFUSED => e
103 puts 'The connection to the server (to download the video file) was refused. Check your connection, and try again later.'
104 Kernel.exit(EXIT_CONNECTION_REFUSED)
105 rescue Errno::EACCES => e
106 puts "Access denied. Check that you have write permission to the output file/directory. Details: #{e.message}."
107 rescue OpenURI::HTTPError => e
108 puts "An HTTP error occurred while downloading the video file: #{e.message}."
109 Kernel.exit(EXIT_HTTP_ERROR)
110 end
111
112 # Write an empty line at the end for aesthetic reasons.
113 puts ""
114
115 Kernel.exit(EXIT_SUCCESS)
116 end