]> gitweb.michael.orlitzky.com - dead/whatever-dl.git/blob - lib/whatever-dl/websites/mentalzero.rb
Move all of the 'src' code under the more-standard 'lib'.
[dead/whatever-dl.git] / lib / whatever-dl / websites / mentalzero.rb
1 #
2 # Copyright Michael Orlitzky
3 #
4 # http://michael.orlitzky.com/
5 #
6 # This program is free software: you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation, either version 3 of the License, or
9 # (at your option) any later version.
10 #
11 # This program is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 # GNU General Public License for more details.
15 #
16 # http://www.fsf.org/licensing/licenses/gpl.html
17 #
18
19 require 'src/website'
20 require 'cgi'
21
22 class Mentalzero < Website
23
24 VALID_MENTALZERO_URL_REGEX = /^(http:\/\/)?(www.)?mentalzero\.com\/[a-z0-9\-_\.]+\.html$/i
25
26 def self.owns_url?(url)
27 return url =~ VALID_MENTALZERO_URL_REGEX
28 end
29
30
31 def get_video_url()
32 page_data = get_page_data(@url)
33 filepath = parse_video_url(page_data)
34
35 return CGI::unescape(filepath)
36 end
37
38
39 def get_video_filename()
40 # Override the default since a nice human-readable name is right
41 # at the end of @url.
42 file_and_params = @url.split('/').pop()
43
44 filename = file_and_params
45 # Unless it contains URL parameters. We don't want those.
46 if file_and_params.include?('?')
47 # There must be some parameters. Strip them off.
48 param_start_idx = file_and_params.index('?')
49 filename = file_and_params[0...(param_start_idx)]
50 end
51
52 # Use the Website method to get the real extension of the file
53 # (the file /name/ we get back will be useless, but we can ignore
54 # it).
55 extension = super().split('.').pop()
56 filename.sub!('.html', '.' + extension)
57
58 return filename
59 end
60
61 protected;
62
63 def parse_video_url(page_data)
64 video_url_regex = /&file=(.+?)&site/i
65 matches = video_url_regex.match(page_data)
66
67 if matches.nil? or (matches.length < 2)
68 raise StandardError.new("Couldn't parse the 'file' object parameter.")
69 else
70 return matches[1]
71 end
72 end
73
74 end