epg/scripts/commands/api/generate.ts

import { Logger, Storage, Collection } from '@freearhey/core'
import { ChannelsParser } from '../../core'
import path from 'path'
import { SITES_DIR, API_DIR } from '../../constants'
import epgGrabber from 'epg-grabber'

type OutputItem = {
  channel: string | null
  feed: string | null
  site: string
  site_id: string
  site_name: string
  lang: string
}

async function main() {
  const logger = new Logger()

  logger.start('staring...')

  logger.info('loading channels...')
  const sitesStorage = new Storage(SITES_DIR)
  const parser = new ChannelsParser({ storage: sitesStorage })

  let files: string[] = []
  files = await sitesStorage.list('**/*.channels.xml')

  let parsedChannels = new Collection()
  for (const filepath of files) {
    parsedChannels = parsedChannels.concat(await parser.parse(filepath))
  }

  logger.info(`  found ${parsedChannels.count()} channel(s)`)

  const output = parsedChannels.map((channel: epgGrabber.Channel): OutputItem => {
    const xmltv_id = channel.xmltv_id || ''
    const [channelId, feedId] = xmltv_id.split('@')

    return {
      channel: channelId || null,
      feed: feedId || null,
      site: channel.site || '',
      site_id: channel.site_id || '',
      site_name: channel.name,
      lang: channel.lang || ''
    }
  })

  const apiStorage = new Storage(API_DIR)
  const outputFilename = 'guides.json'
  await apiStorage.save('guides.json', output.toJSON())

  logger.info(`saved to "${path.join(API_DIR, outputFilename)}"`)
}

main()