iptv/scripts/format.js

const categories = require('./helpers/categories')
const parser = require('./helpers/parser')
const utils = require('./helpers/utils')
const file = require('./helpers/file')
const log = require('./helpers/log')

async function main() {
  log.start()

  log.print(`Parsing 'index.m3u'...`)
  let playlists = parser.parseIndex().filter(i => i.url !== 'channels/unsorted.m3u')
  for (const playlist of playlists) {
    log.print(`\nProcessing '${playlist.url}'...`)
    await parser
      .parsePlaylist(playlist.url)
      .then(removeWrongCategories)
      .then(addMissingData)
      .then(sortChannels)
      .then(playlist => {
        if (file.read(playlist.url) !== playlist.toString()) {
          log.print('updated')
          playlist.updated = true
        }

        playlist.save()
      })
  }

  log.print('\n')
  log.finish()
}

async function removeWrongCategories(playlist) {
  for (const channel of playlist.channels) {
    if (!channel.category) continue
    const valid = categories.find(c => c.name === channel.category)
    if (!valid) {
      channel.category = ''
    }
  }

  return playlist
}

async function addMissingData(playlist) {
  for (const channel of playlist.channels) {
    const code = file.getBasename(playlist.url)
    // tvg-name
    if (!channel.tvg.name && code !== 'unsorted' && channel.name) {
      channel.tvg.name = channel.name.replace(/\"/gi, '')
    }
    // tvg-id
    if (!channel.tvg.id && code !== 'unsorted' && channel.tvg.name) {
      const id = utils.name2id(channel.tvg.name)
      channel.tvg.id = id ? `${id}.${code}` : ''
    }
    // country
    if (!channel.countries.length) {
      const name = utils.code2name(code)
      channel.countries = name ? [{ code, name }] : []
      channel.tvg.country = channel.countries.map(c => c.code.toUpperCase()).join(';')
    }
  }

  return playlist
}

async function sortChannels(playlist) {
  const channels = [...playlist.channels]
  utils.sortBy(channels, ['name', 'url'])
  playlist.channels = channels

  return playlist
}

main()