Move scripts to subfolders

scripts/commands/cluster/load.js

@@ -0,0 +1,73 @@
+const _ = require('lodash')
+const grabber = require('epg-grabber')
+const { program } = require('commander')
+const { db, logger, timer, file, parser } = require('../../core')
+
+const options = program
+  .requiredOption('-c, --cluster-id <cluster-id>', 'The ID of cluster to load', parser.parseNumber)
+  .option('--delay <delay>', 'Delay between requests (in mileseconds)', parser.parseNumber)
+  .option(
+    '-t, --timeout <timeout>',
+    'Set a timeout for each request (in mileseconds)',
+    parser.parseNumber
+  )
+  .option('--debug', 'Enable debug mode', false)
+  .parse(process.argv)
+  .opts()
+
+const LOGS_DIR = process.env.LOGS_DIR || 'scripts/logs'
+const CLUSTER_PATH = `${LOGS_DIR}/cluster/load/cluster_${options.clusterId}.log`
+
+async function main() {
+  logger.info('Starting...')
+  timer.start()
+
+  logger.info(`Loading cluster: ${options.clusterId}`)
+  logger.info(`Creating '${CLUSTER_PATH}'...`)
+  await file.create(CLUSTER_PATH)
+  await db.queue.load()
+  let items = await db.queue.find({ cluster_id: options.clusterId })
+  items = _.orderBy(items, [i => i.channel.xmltv_id.toLowerCase(), 'date'])
+  const total = items.length
+
+  logger.info('Loading...')
+  const results = {}
+  let i = 1
+  for (const item of items) {
+    let config = require(file.resolve(item.configPath))
+
+    config = _.merge(config, {
+      debug: options.debug,
+      delay: options.delay,
+      request: {
+        timeout: options.timeout
+      }
+    })
+
+    await grabber.grab(item.channel, item.date, config, async (data, err) => {
+      logger.info(
+        `[${i}/${total}] ${item.channel.site} - ${item.channel.xmltv_id} - ${data.date.format(
+          'MMM D, YYYY'
+        )} (${data.programs.length} programs)`
+      )
+
+      if (err) logger.error(err.message)
+
+      const result = {
+        _qid: item._id,
+        programs: data.programs,
+        error: err ? err.message : null
+      }
+
+      await file.append(CLUSTER_PATH, JSON.stringify(result) + '\n')
+
+      if (i < total) i++
+    })
+  }
+
+  db.queue.compact()
+
+  logger.info(`Done in ${timer.format('HH[h] mm[m] ss[s]')}`)
+}
+
+main()