Create update_legacy.js

scripts/commands/guides/update_legacy.js

@@ -0,0 +1,84 @@
+const { db, logger, file, zip } = require('../../core')
+const { generateXMLTV, Program, Channel } = require('epg-grabber')
+const _ = require('lodash')
+
+const PUBLIC_DIR = process.env.PUBLIC_DIR || '.gh-pages'
+const CURR_DATE = process.env.CURR_DATE || new Date()
+
+async function main() {
+  logger.info(`Generating guides/...`)
+
+  logger.info('Loading "database/programs.db"...')
+  await db.programs.load()
+
+  let total = 0
+  const grouped = groupByGroup(await loadQueue())
+  for (const key in grouped) {
+    let channels = {}
+    let programs = []
+    for (const item of grouped[key]) {
+      if (item.error) continue
+
+      const itemPrograms = await loadProgramsForItem(item)
+      programs = programs.concat(itemPrograms)
+
+      if (channels[item.channel.id]) continue
+      channels[item.channel.id] = new Channel(item.channel)
+    }
+    programs = _.sortBy(programs, ['channel', 'start'])
+    programs = programs.map(p => new Program(p, channels[p.channel]))
+    programs = _.uniqBy(programs, p => p.channel + p.start)
+    total += programs.length
+    channels = Object.values(channels)
+    channels = _.sortBy(channels, 'id')
+
+    const xmlFilepath = `${PUBLIC_DIR}/guides/${key}.epg.xml`
+    const gzFilepath = `${PUBLIC_DIR}/guides/${key}.epg.xml.gz`
+    const jsonFilepath = `${PUBLIC_DIR}/guides/${key}.epg.json`
+    logger.info(`Creating "${xmlFilepath}"...`)
+    const xmltv = generateXMLTV({ channels, programs, date: CURR_DATE })
+    await file.create(xmlFilepath, xmltv)
+    logger.info(`Creating "${gzFilepath}"...`)
+    const compressed = await zip.compress(xmltv)
+    await file.create(gzFilepath, compressed)
+    logger.info(`Creating "${jsonFilepath}"...`)
+    await file.create(jsonFilepath, JSON.stringify({ channels, programs }))
+  }
+
+  if (!total) {
+    logger.error('\nError: No programs found')
+    process.exit(1)
+  } else {
+    logger.info(`Done`)
+  }
+}
+
+main()
+
+function groupByGroup(items = []) {
+  const groups = {}
+
+  items.forEach(item => {
+    item.groups.forEach(key => {
+      if (!groups[key]) {
+        groups[key] = []
+      }
+
+      groups[key].push(item)
+    })
+  })
+
+  return groups
+}
+
+async function loadQueue() {
+  logger.info('Loading queue...')
+
+  await db.queue.load()
+
+  return await db.queue.find({}).sort({ xmltv_id: 1 })
+}
+
+async function loadProgramsForItem(item) {
+  return await db.programs.find({ _qid: item._id }).sort({ channel: 1, start: 1 })
+}