aiden
/
iptv
mirror of https://github.com/iptv-org/iptv
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

wip

Browse Source
pull/621/head
freearhey
parent db008d61fb
commit fc469c0568
2 changed files with 185 additions and 161 deletions

332
scripts/format.js
Unescape Escape View File

@ -1,174 +1,184 @@
const helper = require('./helper') const helper = require('./helper')
const debug = false const config = {
const verbose = false debug: false,
const parseEpg = process.env.npm_config_epg || false country: process.env.npm_config_country,
exclude: process.env.npm_config_exclude,
let stats = { epg: process.env.npm_config_epg || false
total: 0,
updated: 0,
duplicates: 0,
unvalid: 0,
removed: 0
} }
let buffer = {}
let unsorted = {}
async function main() { // let stats = {
// total: 0,
console.log(`Parsing 'index.m3u'...`) // updated: 0,
const playlist = helper.parsePlaylist('index.m3u') // duplicates: 0,
const countries = playlist.items // removed: 0
// }
if(debug) { // let buffer = {}
console.log('Debug mode is turn on')
} let playlists = {}
const unsortedPlaylist = helper.parsePlaylist('channels/unsorted.m3u') function parseIndex() {
for(const item of unsortedPlaylist.items) { const root = helper.parsePlaylist('index.m3u')
unsorted[item.url] = helper.createChannel(item) const rootItems = helper.filterPlaylists(root.items, config.country, config.exclude)
}
for(let rootItem of rootItems) {
for(let country of countries) { const playlist = helper.parsePlaylist(rootItem.url)
playlists[rootItem.url] = {}
if (helper.skipPlaylist(country.url)) {
continue
}
if(verbose) {
console.log(`Clear cache...`)
}
helper.clearCache()
console.log(`Parsing '${country.url}'...`)
const playlist = helper.parsePlaylist(country.url)
if(verbose) {
console.log(`Creating channels list...`)
}
let channels = []
for(let item of playlist.items) { for(let item of playlist.items) {
let channel = helper.createChannel(item) playlists[rootItem.url][item.url] = helper.createChannel(item)
if(helper.checkCache(channel.url)) {
stats.duplicates++
} else if(!helper.validateUrl(channel.url)) {
stats.unvalid++
} else {
channels.push(channel)
helper.addToCache(channel.url)
}
if(unsorted[channel.url]) {
if(verbose) {
console.log(`Removed '${channel.url}' from 'channels/unsorted.m3u'...`)
}
delete unsorted[channel.url]
stats.removed++
stats.duplicates++
}
} }
}
}
const epgUrl = playlist.header.attrs['x-tvg-url'] function main() {
if(epgUrl && !buffer[epgUrl] && parseEpg) { console.log(`Parsing index...`)
try { parseIndex()
console.log(`Loading '${epgUrl}'...`) console.log(playlists)
const epg = await helper.loadEPG(epgUrl) // add missing data from epg (if exists)
console.log(`Adding '${epgUrl}' to buffer...`) // sort channels
buffer[epgUrl] = epg // save all playlists back
} catch(e) { // display stats
console.log(`Could not load '${epgUrl}'`)
}
}
if(buffer[epgUrl]) { // for(let country of countries) {
console.log('Add missing tvg-id from EPG by channel title...')
for(let channel of channels) {
for(let channelId in buffer[epgUrl].channels) {
let c = buffer[epgUrl].channels[channelId]
for(let epgName of c.name) {
if(epgName.value) {
let escaped = helper.escapeStringRegexp(epgName.value)
channelTitle = channel.title.replace(/(fhd|hd|sd|高清)$/i, '').trim()
let regexp = new RegExp(`^${escaped}$`, 'i')
if(regexp.test(channelTitle)) {
if(!channel.id) {
channel.id = c.id
continue
}
}
}
}
}
}
}
if(buffer[epgUrl]) { // if (helper.skipPlaylist(country.url)) {
console.log(`Fills in missing channel's data...`) // continue
for(let channel of channels) { // }
let channelId = channel.id
if(!channelId) continue // if(verbose) {
let c = buffer[epgUrl].channels[channelId] // console.log(`Clear cache...`)
if(!c) continue // }
let updated = false // helper.clearCache()
// console.log(`Parsing '${country.url}'...`)
// const playlist = helper.parsePlaylist(country.url)
// if(verbose) {
// console.log(`Creating channels list...`)
// }
// let channels = []
// for(let item of playlist.items) {
// let channel = helper.createChannel(item)
// if(helper.checkCache(channel.url)) {
// stats.duplicates++
// } else if(!helper.validateUrl(channel.url)) {
// stats.unvalid++
// } else {
// channels.push(channel)
// helper.addToCache(channel.url)
// }
// if(unsorted[channel.url]) {
// if(verbose) {
// console.log(`Removed '${channel.url}' from 'channels/unsorted.m3u'...`)
// }
// delete unsorted[channel.url]
// stats.removed++
// stats.duplicates++
// }
// }
// const epgUrl = playlist.header.attrs['x-tvg-url']
// if(epgUrl && !buffer[epgUrl] && parseEpg) {
// try {
// console.log(`Loading '${epgUrl}'...`)
// const epg = await helper.loadEPG(epgUrl)
// console.log(`Adding '${epgUrl}' to buffer...`)
// buffer[epgUrl] = epg
// } catch(e) {
// console.log(`Could not load '${epgUrl}'`)
// }
// }
// if(buffer[epgUrl]) {
// console.log('Add missing tvg-id from EPG by channel title...')
// for(let channel of channels) {
// for(let channelId in buffer[epgUrl].channels) {
// let c = buffer[epgUrl].channels[channelId]
// for(let epgName of c.name) {
// if(epgName.value) {
// let escaped = helper.escapeStringRegexp(epgName.value)
// channelTitle = channel.title.replace(/(fhd|hd|sd|高清)$/i, '').trim()
// let regexp = new RegExp(`^${escaped}$`, 'i')
// if(regexp.test(channelTitle)) {
// if(!channel.id) {
// channel.id = c.id
// continue
// }
// }
// }
// }
// }
// }
// }
// if(buffer[epgUrl]) {
// console.log(`Fills in missing channel's data...`)
// for(let channel of channels) {
// let channelId = channel.id
// if(!channelId) continue
// let c = buffer[epgUrl].channels[channelId]
// if(!c) continue
// let updated = false
if(!channel.name && c.name.length) { // if(!channel.name && c.name.length) {
channel.name = c.name[0].value // channel.name = c.name[0].value
updated = true // updated = true
if(verbose) { // if(verbose) {
console.log(`Added name '${c.name[0].value}' to '${channel.id}'`) // console.log(`Added name '${c.name[0].value}' to '${channel.id}'`)
} // }
} // }
if(!channel.language && c.name.length && c.name[0].lang) { // if(!channel.language && c.name.length && c.name[0].lang) {
let language = helper.getISO6391Name(c.name[0].lang) // let language = helper.getISO6391Name(c.name[0].lang)
channel.language = language // channel.language = language
updated = true // updated = true
if(verbose) { // if(verbose) {
console.log(`Added language '${language}' to '${channel.id}'`) // console.log(`Added language '${language}' to '${channel.id}'`)
} // }
} // }
if(!channel.logo && c.icon.length) { // if(!channel.logo && c.icon.length) {
const icon = c.icon[0].split('|')[0] // const icon = c.icon[0].split('|')[0]
channel.logo = icon // channel.logo = icon
updated = true // updated = true
if(verbose) { // if(verbose) {
console.log(`Added logo '${icon}' to '${channel.id}'`) // console.log(`Added logo '${icon}' to '${channel.id}'`)
} // }
} // }
if(updated) { // if(updated) {
stats.updated++ // stats.updated++
} // }
} // }
} // }
if(verbose) { // if(verbose) {
console.log(`Sorting channels...`) // console.log(`Sorting channels...`)
} // }
channels = helper.sortBy(channels, ['title', 'url']) // channels = helper.sortBy(channels, ['title', 'url'])
if(!debug) { // if(!debug) {
console.log(`Updating '${country.url}'...`) // console.log(`Updating '${country.url}'...`)
helper.createFile(country.url, playlist.getHeader()) // helper.createFile(country.url, playlist.getHeader())
channels.forEach(channel => { // channels.forEach(channel => {
helper.appendToFile(country.url, channel.toShortString()) // helper.appendToFile(country.url, channel.toShortString())
}) // })
} // }
stats.total += channels.length // stats.total += channels.length
} // }
if(!debug & stats.removed > 0) { // if(!debug & stats.removed > 0) {
console.log(`Updating 'channels/unsorted.m3u'...`) // console.log(`Updating 'channels/unsorted.m3u'...`)
helper.createFile('channels/unsorted.m3u', playlist.getHeader()) // helper.createFile('channels/unsorted.m3u', playlist.getHeader())
Object.values(unsorted).forEach(channel => { // Object.values(unsorted).forEach(channel => {
helper.appendToFile('channels/unsorted.m3u', channel.toShortString()) // helper.appendToFile('channels/unsorted.m3u', channel.toShortString())
}) // })
} // }
console.log(`Total: ${stats.total}. Duplicates: ${stats.duplicates}. Unvalid: ${stats.unvalid}. Updated: ${stats.updated}.`) // console.log(`Total: ${stats.total}. Duplicates: ${stats.duplicates}. Unvalid: ${stats.unvalid}. Updated: ${stats.updated}.`)
} }
main() main()

14
scripts/helper.js
Unescape Escape View File

@ -147,6 +147,20 @@ helper.validateUrl = function(channelUrl) {
return blacklist.indexOf(host) === -1 return blacklist.indexOf(host) === -1
} }
helper.filterPlaylists = function(arr, include = '', exclude = '') {
if(include) {
const included = include.split(',').map(filename => `channels/${filename}.m3u`)
return arr.filter(i => included.indexOf(i.url) > -1)
}
if(exclude) {
const excluded = exclude.split(',').map(filename => `channels/${filename}.m3u`)
return arr.filter(i => excluded.indexOf(i.url) === -1)
}
}
helper.skipPlaylist = function(filename) { helper.skipPlaylist = function(filename) {
let testCountry = process.env.npm_config_country let testCountry = process.env.npm_config_country
let excludeList = process.env.npm_config_exclude let excludeList = process.env.npm_config_exclude

Write Preview
Loading…
Cancel
Save