iptv/scripts/commands/playlist/generate.js

const { db, generator, api, logger, file } = require('../../core')
const { orderBy } = require('natural-orderby')
const _ = require('lodash')

async function main() {
  const streams = await loadStreams()

  logger.info('generating categories/...')
  await generator.generate('categories', streams)
  logger.info('generating countries/...')
  await generator.generate('countries', streams)
  logger.info('generating languages/...')
  await generator.generate('languages', streams)
  logger.info('generating regions/...')
  await generator.generate('regions', streams)
  logger.info('generating index.category.m3u...')
  await generator.generate('index_category_m3u', streams)
  logger.info('generating index.country.m3u...')
  await generator.generate('index_country_m3u', streams)
  logger.info('generating index.language.m3u...')
  await generator.generate('index_language_m3u', streams)
  logger.info('generating index.m3u...')
  await generator.generate('index_m3u', streams)
  logger.info('generating index.nsfw.m3u...')
  await generator.generate('index_nsfw_m3u', streams)
  logger.info('generating index.region.m3u...')
  await generator.generate('index_region_m3u', streams)
}

main()

async function loadStreams() {
  await db.streams.load()
  let streams = await db.streams.find({})
  streams = _.filter(streams, stream => stream.status !== 'error')
  streams = orderBy(streams, ['channel', 'height', 'url'], ['asc', 'desc', 'asc'])
  streams = _.uniqBy(streams, stream => stream.channel || _.uniqueId())

  await api.channels.load()
  let channels = await api.channels.all()
  channels = _.keyBy(channels, 'id')

  await api.categories.load()
  let categories = await api.categories.all()
  categories = _.keyBy(categories, 'id')

  await api.languages.load()
  let languages = await api.languages.all()
  languages = _.keyBy(languages, 'code')

  await api.guides.load()
  let guides = await api.guides.all()
  guides = _.groupBy(guides, 'channel')

  streams = streams.map(stream => {
    const channel = channels[stream.channel] || null
    const filename = file.getFilename(stream.filepath)
    const [_, code] = filename.match(/^([a-z]{2})(_|$)/) || [null, null]
    const defaultBroadcastArea = code ? [`c/${code.toUpperCase()}`] : []

    stream.guides = channel && Array.isArray(guides[channel.id]) ? guides[channel.id] : []
    stream.categories = channel ? channel.categories.map(id => categories[id]) : []
    stream.languages = channel ? channel.languages.map(id => languages[id]) : []
    stream.broadcast_area = channel ? channel.broadcast_area : defaultBroadcastArea
    stream.is_nsfw = channel ? channel.is_nsfw : false
    stream.logo = channel ? channel.logo : null

    return stream
  })

  streams = orderBy(streams, ['title'], ['asc'])

  return streams
}