iptv/test/index.js

105 lines
1.9 KiB
JavaScript
Raw Normal View History

2019-07-20 09:33:16 +02:00
const util = require('../helpers/util')
2018-12-22 20:07:47 +01:00
const axios = require('axios')
2019-08-08 13:11:11 +02:00
const https = require('https')
2018-11-16 00:41:15 +01:00
2019-08-08 15:06:49 +02:00
const verbose = false
2018-12-22 20:07:47 +01:00
const errorLog = 'error.log'
2019-07-20 09:33:16 +02:00
const config = {
timeout: 60000,
delay: 200
}
2018-11-16 00:41:15 +01:00
2019-07-20 09:33:16 +02:00
let stats = {
tests: 0,
channels: 0,
failures: 0
}
2018-11-16 00:41:15 +01:00
const instance = axios.create({
2019-08-08 13:11:11 +02:00
timeout: config.timeout,
httpsAgent: new https.Agent({
rejectUnauthorized: false
})
})
instance.defaults.headers.common["User-Agent"] = "VLC/2.2.4 LibVLC/2.2.4"
2018-12-22 20:07:47 +01:00
2019-07-20 09:33:16 +02:00
async function test() {
2018-11-16 00:41:15 +01:00
2019-07-20 09:33:16 +02:00
stats.tests++
2018-11-16 00:41:15 +01:00
2019-08-07 22:59:44 +02:00
const playlist = util.parsePlaylist('index.m3u')
const countries = playlist.items
2018-11-16 00:41:15 +01:00
2018-12-22 20:07:47 +01:00
for(let country of countries) {
2018-11-16 00:41:15 +01:00
2019-08-07 22:59:44 +02:00
if (skipPlaylist(country.url)) {
continue;
}
2019-08-07 22:59:44 +02:00
console.log(`Checking '${country.url}'...`)
const playlist = util.parsePlaylist(country.url)
2018-12-22 20:07:47 +01:00
2019-08-07 22:59:44 +02:00
for(let item of playlist.items) {
2018-12-22 20:07:47 +01:00
2019-08-08 13:11:11 +02:00
if(item.url.indexOf('rtmp://') > -1) continue
await new Promise(resolve => {
2019-07-20 09:33:16 +02:00
setTimeout(resolve, config.delay)
})
2019-07-20 09:33:16 +02:00
stats.channels++
2018-11-16 00:41:15 +01:00
2018-11-16 04:46:05 +01:00
try {
2018-12-22 20:07:47 +01:00
if(verbose) {
console.log(`Checking '${item.url}'...`)
}
await instance.get(item.url)
2018-12-22 20:07:47 +01:00
continue
} catch (err) {
stats.failures++
2019-08-08 13:11:11 +02:00
writeToLog(country.url, err.message, item.url)
2018-12-22 20:07:47 +01:00
2018-11-16 00:41:15 +01:00
}
2018-12-22 20:07:47 +01:00
2018-11-16 00:41:15 +01:00
}
2018-12-22 20:07:47 +01:00
}
2018-11-16 00:41:15 +01:00
2019-07-20 09:33:16 +02:00
if(stats.failures === 0) {
2018-11-16 00:41:15 +01:00
2019-07-20 09:33:16 +02:00
console.log(`OK (${stats.tests} tests, ${stats.channels} channels)`)
2018-12-22 20:07:47 +01:00
} else {
2018-11-16 00:41:15 +01:00
2019-07-20 09:33:16 +02:00
console.log(`FAILURES! (${stats.tests} tests, ${stats.channels} channels, ${stats.failures} failures)`)
2018-11-16 00:41:15 +01:00
2018-12-22 20:07:47 +01:00
}
2018-11-16 00:41:15 +01:00
}
2018-12-26 20:41:17 +01:00
console.log('Test is running...')
2019-07-20 09:33:16 +02:00
test()
2019-08-07 22:59:44 +02:00
2019-08-08 02:32:27 +02:00
function writeToLog(country, msg, url) {
2019-08-07 22:59:44 +02:00
var now = new Date()
2019-08-08 02:32:27 +02:00
var line = `${country}: ${msg} '${url}'`
2019-08-07 22:59:44 +02:00
util.appendToFile(errorLog, now.toISOString() + ' ' + line + '\n')
console.log(`Error: ${msg} '${url}'`)
}
function skipPlaylist(filename) {
let test_country = process.env.npm_config_country
if (test_country && filename !== 'channels/' + test_country + '.m3u') {
return true;
}
return false;
}