Pass page content to logo function

src/index.js

@@ -26,7 +26,6 @@ async function main() {
 
   const queue = []
   channels.forEach(channel => {
-    channel.logo = config.logo ? config.logo({ channel }) : null
     dates.forEach(date => {
       queue.push({ date, channel })
     })
@@ -39,24 +38,11 @@ async function main() {
     const progs = await client
       .get(url)
       .then(response => {
-        const parserOptions = Object.assign({}, item, config, {
+        item.channel.logo = config.logo
-          content: response.data
+          ? config.logo({ channel: item.channel, content: response.data })
-        })
+          : null
-        const programs = config
-          .parser(parserOptions)
-          .filter(i => i)
-          .map(p => {
-            p.channel = item.channel.xmltv_id
-            return p
-          })
 
-        console.log(
+        return utils.parsePrograms({ response, item, config })
-          `  ${config.site} - ${item.channel.xmltv_id} - ${item.date.format('MMM D, YYYY')} (${
-            programs.length
-          } programs)`
-        )
-
-        return programs
       })
       .then(utils.sleep(config.delay))
       .catch(err => {

src/utils.js

@@ -137,6 +137,28 @@ utils.convertToXMLTV = function ({ config, channels, programs }) {
   return output
 }
 
+utils.parsePrograms = function ({ response, item, config }) {
+  const options = Object.assign({}, item, config, {
+    content: response.data
+  })
+
+  const programs = config
+    .parser(options)
+    .filter(i => i)
+    .map(p => {
+      p.channel = item.channel.xmltv_id
+      return p
+    })
+
+  console.log(
+    `  ${config.site} - ${item.channel.xmltv_id} - ${item.date.format('MMM D, YYYY')} (${
+      programs.length
+    } programs)`
+  )
+
+  return programs
+}
+
 utils.writeToFile = function (filename, data) {
   const dir = path.resolve(path.dirname(filename))
   if (!fs.existsSync(dir)) {