#
# 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
#

require 'src/website'
require 'cgi'

class Mentalzero < Website

  VALID_MENTALZERO_URL_REGEX = /^(http:\/\/)?(www.)?mentalzero\.com\/[a-z0-9\-_\.]+\.html$/i

  def self.owns_url?(url)
    return url =~ VALID_MENTALZERO_URL_REGEX
  end


  def get_video_url()
    page_data = get_page_data(@url)
    filepath = parse_video_url(page_data)

    return CGI::unescape(filepath)
  end


  def get_video_filename()
    #
    # TODO: Merge this with the algorithm in the Xvideos class. Only
    # the last lines differ. Maybe we could have a collection of
    # filename-getting algorithms.
    #
    # Override the default since a nice human-readable name is right
    # at the end of @url.
    file_and_params = @url.split('/').pop()

    filename = file_and_params
    # Unless it contains URL parameters. We don't want those.
    if file_and_params.include?('?')
      # There must be some parameters. Strip them off.
      param_start_idx = file_and_params.index('?')
      filename = file_and_params[0...(param_start_idx)]
    end

    # Use the Website method to get the real extension of the file
    # (the file /name/ we get back will be useless, but we can ignore
    # it).
    extension = super().split('.').pop()
    filename.sub!('.html', '.' + extension)

    return filename
  end

  protected;

  def parse_video_url(page_data)
    video_url_regex = /&file=(.+?)&site/i
    matches = video_url_regex.match(page_data)

    if matches.nil? or (matches.length < 2)
      raise StandardError.new("Couldn't parse the 'file' object parameter.")
    else
      return matches[1]
    end
  end

end