Merge pull request #273 from tildeio/protocol-for-url

Add hook for extracting protocol from URLs
tildeio · Jan 31, 2015 · db589d9 · db589d9
2 parents 513c917 + acb274d
commit db589d9
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/packages/morph/lib/dom-helper.js b/packages/morph/lib/dom-helper.js
@@ -403,4 +403,16 @@ prototype.parseHTML = function(html, contextualElement) {
   }
 };
 
+var parsingNode;
+
+// Used to determine whether a URL needs to be sanitized.
+prototype.protocolForURL = function(url) {
+  if (!parsingNode) {
+    parsingNode = this.document.createElement('a');
+  }
+
+  parsingNode.href = url;
+  return parsingNode.protocol;
+};
+
 export default DOMHelper;
diff --git a/packages/morph/tests/dom-helper-test.js b/packages/morph/tests/dom-helper-test.js
@@ -372,6 +372,16 @@ test('#parseHTML of number', function(){
   equalHTML(nodes, '5');
 });
 
+test('#protocolForURL', function() {
+  var protocol = dom.protocolForURL("http://www.emberjs.com");
+  equal(protocol, "http:");
+
+  // Inherit protocol from document if unparseable
+  protocol = dom.protocolForURL("   javascript:lulzhacked()");
+  /*jshint scripturl:true*/
+  equal(protocol, "javascript:");
+});
+
 test('#cloneNode shallow', function(){
   var divElement = document.createElement('div');