Extended to support https and ftp

lawrencewoodman · Apr 11, 2013 · a26c94c · a26c94c
1 parent 60fc463
commit a26c94c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/_plugins/domain_name.filter.rb b/_plugins/domain_name.filter.rb
@@ -4,7 +4,7 @@ module DomainNameFilter
 
   # Return the url's domain name
   def domain_name(url)
-    return url.sub(%r{(http://){0,1}([^/]*)(/.*$){0,1}}i, '\\2')
+    return url.sub(%r{([a-z]+?://)?([^/]*)(/.*$)?}i, '\\2')
   end
 
 end

diff --git a/test/test_domain_name.rb b/test/test_domain_name.rb
@@ -15,7 +15,10 @@ def test_domain_name
             'http://example.com/reviews/index.php',
             'example.com',
             'example.com/reviews/',
-            'example.com/reviews/index.php']
+            'example.com/reviews/index.php',
+            'https://example.com',
+            'https://example.com/lvl/page.html',
+            'ftp://example.com']
 
     urls.each do |url|
       @context['var'] = url