iptv/scripts/test.js

const { program } = require('commander')
const utils = require('./utils')
const parser = require('./parser')
const axios = require('axios')
const https = require('https')
const ProgressBar = require('progress')

program
  .usage('[OPTIONS]...')
  .option('-d, --debug', 'Debug mode')
  .option('-c, --country <country>', 'Comma-separated list of country codes', '')
  .option('-e, --exclude <exclude>', 'Comma-separated list of country codes to be excluded', '')
  .option('--delay <delay>', 'Delay between parser requests', 1000)
  .option('--timeout <timeout>', 'Set timeout for each request', 5000)
  .parse(process.argv)

const config = program.opts()

const instance = axios.create({
  timeout: config.timeout,
  maxContentLength: 20000,
  httpsAgent: new https.Agent({
    rejectUnauthorized: false
  }),
  validateStatus: function (status) {
    return (status >= 200 && status < 300) || status === 403
  }
})

let stats = {
  playlists: 0,
  channels: 0,
  failures: 0
}

async function test() {
  let items = parser.parseIndex()
  items = utils.filterPlaylists(items, config.country, config.exclude)

  for (const item of items) {
    const playlist = parser.parsePlaylist(item.url)
    const bar = new ProgressBar(`Processing '${item.url}'...:current/:total`, {
      total: playlist.channels.length
    })

    stats.playlists++

    for (let channel of playlist.channels) {
      bar.tick()
      stats.channels++

      if (channel.url.startsWith('rtmp://')) continue

      await instance
        .get(channel.url)
        .then(utils.sleep(config.delay))
        .catch(error => {
          if (error.response) {
            stats.failures++
            utils.writeToLog(item.url, error.message, channel.url)
          }
        })
    }
  }

  if (stats.failures === 0) {
    console.log(`\nOK (${stats.playlists} playlists, ${stats.channels} channels)`)
  } else {
    console.log(
      `\nFAILURES! (${stats.playlists} playlists, ${stats.channels} channels, ${stats.failures} failures)
      \n\nCheck the "error.log" file to see which links failed.`
    )

    process.exit(1)
  }
}

console.log('Test is running...\n')

test()