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