Refactor

Kikobeats · Sep 16, 2018 · f5e23de · f5e23de
1 parent 8700bb6
commit f5e23de
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/index.js b/src/index.js
@@ -32,6 +32,8 @@ const TAGS = {
   ]
 }
 
+const REGEX_HTTP_PROTOCOL = /https?/i
+
 const reduceSelector = (collection, fn, acc = []) => {
   collection.each(function () {
     acc = fn(acc, this)
@@ -41,9 +43,17 @@ const reduceSelector = (collection, fn, acc = []) => {
 
 const includes = (collection, fn) => findIndex(collection, fn) !== -1
 
+const isHttpUrl = url => {
+  try {
+    return REGEX_HTTP_PROTOCOL.test(new URL(url).protocol)
+  } catch (err) {
+    return false
+  }
+}
+
 const getLink = ({ url, el, attribute }) => {
   const attr = get(el, `attribs.${attribute}`, '')
-  if (isEmpty(attr) || attr.startsWith('mailto:') || attr.startsWith('callto:')) { return null }
+  if (isEmpty(attr) || !isHttpUrl(attr)) return null
 
   try {
     const normalizedUrl = normalizeUrl(url, attr)