iptv/helpers/generate.js

76 lines
1.5 KiB
JavaScript

const util = require('./util')
const urlParser = require('url')
const outputFile = 'index.full.m3u'
const debug = false
let cache = {}
let stats = {
total: 0,
duplicates: 0
}
function init() {
let countries = util.parsePlaylist('index.m3u')
if(debug) {
countries = countries.slice(0, 1)
}
util.createFile(outputFile)
for(let country of countries) {
const countryName = util.getTitle(country.title)
const countryCode = util.getBasename(country.file).toUpperCase()
const playlist = util.parsePlaylist(country.file)
for(let item of playlist) {
const channel = util.parseChannelData(item)
const groupTitle = [ countryName, channel.group ].filter(i => i).join(';')
const info = `-1 tvg-id="${channel.id}" tvg-name="${channel.name}" tvg-logo="${channel.logo}" group-title="${groupTitle}",${channel.title}`
const file = channel.file
if(checkCache(file)) {
stats.duplicates++
} else {
util.writeToFile(outputFile, info, file)
addToCache(file)
}
stats.total++
}
}
}
init()
console.log(`Total: ${stats.total}. Duplicates: ${stats.duplicates}. Unique: ${stats.total - stats.duplicates}`)
function addToCache(url) {
let id = getUrlPath(url)
cache[id] = true
}
function checkCache(url) {
let id = getUrlPath(url)
return cache.hasOwnProperty(id)
}
function getUrlPath(u) {
let parsed = urlParser.parse(u)
let searchQuery = parsed.search || ''
return parsed.host + parsed.pathname + searchQuery
}