src/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     #
  41     # TODO: Merge this with the algorithm in the Xvideos class. Only
  42     # the last lines differ. Maybe we could have a collection of
  43     # filename-getting algorithms.
  44     #
  45     # Override the default since a nice human-readable name is right
  46     # at the end of @url.
  47     file_and_params = @url.split('/').pop()
  48
  49     filename = file_and_params
  50     # Unless it contains URL parameters. We don't want those.
  51     if file_and_params.include?('?')
  52       # There must be some parameters. Strip them off.
  53       param_start_idx = file_and_params.index('?')
  54       filename = file_and_params[0...(param_start_idx)]
  55     end
  56
  57     # Use the Website method to get the real extension of the file
  58     # (the file /name/ we get back will be useless, but we can ignore
  59     # it).
  60     extension = super().split('.').pop()
  61     filename.sub!('.html', '.' + extension)
  62
  63     return filename
  64   end
  65
  66   protected;
  67
  68   def parse_video_url(page_data)
  69     video_url_regex = /&file=(.+?)&site/i
  70     matches = video_url_regex.match(page_data)
  71
  72     if matches.nil? or (matches.length < 2)
  73       raise StandardError.new("Couldn't parse the 'file' object parameter.")
  74     else
  75       return matches[1]
  76     end
  77   end
  78
  79 end