Move all of the 'src' code under the more-standard 'lib'.

[dead/whatever-dl.git] / lib / whatever-dl / websites / redtube.rb

diff --git a/lib/whatever-dl/websites/redtube.rb b/lib/whatever-dl/websites/redtube.rb
new file mode 100644 (file)
index 0000000..78480c9
--- /dev/null
+++ b/lib/whatever-dl/websites/redtube.rb
@@ -0,0 +1,77 @@
+#
+# 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 Redtube < Website
+
+  VALID_REDTUBE_URL_REGEX = /^(http:\/\/)?(www\.)?redtube\.com\/(\d+)$/
+
+  def self.owns_url?(url)
+    return url =~ VALID_REDTUBE_URL_REGEX
+  end
+
+
+  def get_video_url()
+    page_data = self.get_page_data(@url)
+    begin
+      # We prefer to parse the HTML5 version because it can come in a
+      # nicer container format.
+      return parse_html5_src(page_data)
+    rescue StandardError
+      return self.parse_hashlink(page_data)
+    end
+  end
+
+
+  protected;
+
+  def parse_video_id()
+    return /\d+/.match(@url)[0]
+  end
+
+
+  def parse_html5_src(page_data)
+    html5_src_regex = /\"<source src='([^']+?)'/
+
+    matches = html5_src_regex.match(page_data)
+
+    if matches.nil? or matches.length < 2
+      raise StandardError.new("Could not parse the HTML5 source src.")
+    else
+      return matches[1]
+    end
+
+  end
+
+
+  def parse_hashlink(page_data)
+    hashlink_regex = /hashlink=([^&]+)&/
+
+    matches = hashlink_regex.match(page_data)
+
+    if matches.nil? or matches.length < 2
+      raise StandardError.new("Could not parse the 'hashlink' Flash variable.")
+    end
+
+    # The hashlink variable /is/ the video URL, but it's encoded.
+    return CGI::unescape(matches[1])
+  end
+
+end