epg/scripts/commands/update-guides.js

const { db, logger, file, api } = require('../core')
const grabber = require('epg-grabber')
const _ = require('lodash')

const LOGS_DIR = process.env.LOGS_DIR || 'scripts/logs'
const PUBLIC_DIR = process.env.PUBLIC_DIR || '.gh-pages'
const GUIDES_PATH = `${LOGS_DIR}/guides.log`
const ERRORS_PATH = `${LOGS_DIR}/errors.log`

async function main() {
  await setUp()
  await generateGuides()
}

main()

async function generateGuides() {
  logger.info(`Generating guides/...`)

  const grouped = groupByGroup(await loadQueue())

  logger.info('Loading "database/programs.db"...')
  await db.programs.load()
  await api.channels.load()

  for (const key in grouped) {
    const filepath = `${PUBLIC_DIR}/guides/${key}.epg.xml`
    let items = grouped[key]

    const errors = []
    for (const item of items) {
      if (item.error) {
        const error = {
          xmltv_id: item.channel.xmltv_id,
          site: item.channel.site,
          site_id: item.channel.site_id,
          lang: item.channel.lang,
          date: item.date,
          error: item.error
        }
        errors.push(error)
        await logError(error)
      }
    }

    const programs = await loadProgramsForItems(items)

    logger.info(`Creating "${filepath}"...`)
    const channels = items
      .map(item => {
        const channel = api.channels.find({ id: item.channel.xmltv_id })
        if (!channel) return null

        return {
          id: channel.id,
          display_name: channel.name,
          url: item.channel.site,
          icon: channel.logo
        }
      })
      .filter(i => i)
    const output = grabber.convertToXMLTV({ channels, programs })
    await file.create(filepath, output)

    await logGuide({
      group: key,
      count: items.length,
      status: errors.length > 0 ? 1 : 0
    })
  }

  logger.info(`Done`)
}

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 loadProgramsForItems(items = []) {
  const qids = items.map(i => i._id)

  return await db.programs.find({ _qid: { $in: qids } }).sort({ channel: 1, start: 1 })
}

async function setUp() {
  logger.info(`Creating '${GUIDES_PATH}'...`)
  await file.create(GUIDES_PATH)
  await file.create(ERRORS_PATH)
}

async function logGuide(data) {
  await file.append(GUIDES_PATH, JSON.stringify(data) + '\r\n')
}

async function logError(data) {
  await file.append(ERRORS_PATH, JSON.stringify(data) + '\r\n')
}