Add a provider field fixes #77

mozilla · Oct 13, 2016 · 6e73102 · 6e73102
1 parent ee886dc
commit 6e73102
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/parser.js b/parser.js
@@ -138,6 +138,18 @@ function getMetadata(doc, url, rules) {
     metadata.url = url;
   }
 
+  metadata.provider = '';
+  if(url) {
+    const parsedUrl = urlparse.parse(url);
+    metadata.provider = parsedUrl
+      .hostname
+      .replace('www.', '')
+      .replace('co.', '')
+      .split('.')
+      .slice(0, -1)
+      .join(' ');
+  }
+
   if(url && !metadata.icon_url) {
     metadata.icon_url = makeUrlAbsolute(url, '/favicon.ico');
   }

diff --git a/tests/getMetadata.test.js b/tests/getMetadata.test.js
@@ -60,6 +60,14 @@ describe('Get Metadata Tests', function() {
     assert.equal(metadata.image_url, sampleImageHTTP, `Unable to find ${sampleImageHTTP} in ${relativeHtml}`);
   });
 
+  it('adds a provider when URL passed in', () => {
+    const sampleProvider = 'example';
+    const doc = stringToDom(sampleHtml);
+    const metadata = getMetadata(doc, sampleUrl);
+
+    assert.equal(metadata.provider, sampleProvider, `Unable to find ${sampleProvider} in ${sampleUrl}`);
+  });
+
   it('uses default favicon when no favicon is found', () => {
     const noIconHtml = `
       <html>