Added the ability to download videos from http://www.yikers.com/.

[dead/whatever-dl.git] / src / website.rb

diff --git a/src/website.rb b/src/website.rb
index 96290dee3de99ac7d00f34e14298110be82455a0..b5a501f9db8184d403d93ee1c13a34900b10506d 100644 (file)
--- a/src/website.rb
+++ b/src/website.rb
@@ -16,6 +16,10 @@
 # http://www.fsf.org/licensing/licenses/gpl.html
 #
 
+# Necessary in a lot of subclasses; plus, we need it
+# to parse the server name out of our URL.
+require 'uri'
+
 # This class keeps track of all its subclasses
 # We use this to loop through every "website" in an
 # attempt to determine to which site a URL belongs.
@@ -37,6 +41,13 @@ class Website
     @subclasses << subclass
   end
 
+
+  def server
+    # Get the HTTP server portion of our URI
+    uri = URI.parse(@url)
+    return uri.host
+  end
+
   
   public;