Update index.js

src/index.js

@@ -1,74 +1,44 @@
-#! /usr/bin/env node
-
-const { Command } = require('commander')
-const program = new Command()
+const axios = require('axios').default
 const utils = require('./utils')
-const { name, version, description } = require('../package.json')
 
-program
-  .name(name)
-  .version(version, '-v, --version')
-  .description(description)
-  .option('-c, --config <config>', 'Path to [site].config.js file')
-  .option('-o, --output <output>', 'Path to output file', 'guide.xml')
-  .option('--channels <channels>', 'Path to channels.xml file')
-  .option('--lang <lang>', 'Set default language for all programs', 'en')
-  .option('--days <days>', 'Number of days for which to grab the program', 1)
-  .option('--delay <delay>', 'Delay between requests (in mileseconds)', 3000)
-  .option('--debug', 'Enable debug mode', false)
-  .parse(process.argv)
+const Result = {
+  events: {
+    data: () => {},
+    error: () => {},
+    done: () => {}
+  },
+  on(type, cb) {
+    this.events[type] = cb
+  }
+}
 
-const options = program.opts()
-const config = utils.loadConfig(options)
-
-async function main() {
-  console.log('\r\nStarting...')
-
-  const channels = utils.parseChannels(config.channels)
-  const utcDate = utils.getUTCDate()
-  const dates = Array.from({ length: config.days }, (_, i) => utcDate.add(i, 'd'))
-
-  const queue = []
-  channels.forEach(channel => {
+module.exports = {
+  grab: async function (channel, config, cb) {
+    cb(Result)
+    const utcDate = utils.getUTCDate()
+    const dates = Array.from({ length: config.days }, (_, i) => utcDate.add(i, 'd'))
+    const queue = []
     dates.forEach(date => {
       queue.push({ date, channel })
     })
-  })
 
-  let programs = []
-  console.log('Parsing:')
-  for (let item of queue) {
-    if (options.debug) console.time('    Response Time')
-    await utils
-      .buildRequest(item, config)
-      .then(utils.fetchData)
-      .then(async response => {
-        if (options.debug) console.timeEnd('    Response Time')
-        if (options.debug) console.time('    Parsing Time')
-        const results = await utils.parseResponse(item, response, config)
-        if (options.debug) console.timeEnd('    Parsing Time')
-        programs = programs.concat(results)
-      })
-      .then(utils.sleep(config.delay))
-      .catch(err => {
-        console.log(
-          `  ${config.site} - ${item.channel.xmltv_id} - ${item.date.format(
-            'MMM D, YYYY'
-          )} (0 programs)`
-        )
-        console.log(`    Error: ${err.message}`)
-        if (options.debug) {
-          console.timeEnd('    Response Time')
-          console.timeEnd('    Parsing Time')
-        }
-      })
+    let programs = []
+    for (let item of queue) {
+      await utils
+        .buildRequest(item, config)
+        .then(request => utils.fetchData(request))
+        .then(response => utils.parseResponse(item, response, config))
+        .then(results => {
+          item.programs = results
+          Result.events['data'](item)
+          programs = programs.concat(results)
+        })
+        .catch(err => {
+          Result.events['error'](err)
+        })
+        .finally(utils.sleep(config.delay))
+    }
+
+    Result.events['done'](programs)
   }
-
-  const xml = utils.convertToXMLTV({ config, channels, programs })
-  utils.writeToFile(config.output, xml)
-
-  console.log(`File '${config.output}' successfully saved`)
-  console.log('Finish')
 }
-
-main()