Expose tags and cheerio opts

Kikobeats · Jun 21, 2018 · 9c27283 · 9c27283
1 parent cf1af6f
commit 9c27283
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/src/index.js b/src/index.js
@@ -5,7 +5,7 @@ const { getUrl } = require('@metascraper/helpers')
 const cheerio = require('cheerio')
 const matcher = require('matcher')
 
-const LINKS_ATTRIBUTES = {
+const TAGS = {
   background: ['body'],
   cite: ['blockquote', 'del', 'ins', 'q'],
   data: ['object'],
@@ -67,11 +67,16 @@ const getLinksByAttribute = ({ selector, attribute, url, whitelist }) => {
   )
 }
 
-module.exports = ({ html = '', url = '', whitelist = false } = {}) => {
-  const $ = cheerio.load(html)
+module.exports = ({
+  html = '',
+  url = '',
+  whitelist = false,
+  cheerioOpts = {}
+} = {}) => {
+  const $ = cheerio.load(html, cheerioOpts)
 
   return reduce(
-    LINKS_ATTRIBUTES,
+    TAGS,
     (acc, htmlTags, attribute) => {
       const links = getLinksByAttribute({
         selector: $(htmlTags.join(',')),
@@ -84,3 +89,5 @@ module.exports = ({ html = '', url = '', whitelist = false } = {}) => {
     []
   )
 }
+
+module.exports.TAGS = TAGS