This commit is contained in:
freearhey 2019-11-02 00:08:27 +03:00
parent 043ac262eb
commit 5412372c05
1 changed files with 168 additions and 100 deletions

View File

@ -1,124 +1,192 @@
const util = require('./util')
const ISO6391 = require('iso-639-1')
const debug = false
const types = ['full', 'country', 'content', 'language']
const categories = util.supportedCategories.map(c => c.toLowerCase())
let stats = {
countries: 0,
channels: 0
let list = {
all: [],
countries: {},
languages: {},
categories: {}
}
let languageBuffer = {
undefined: []
}
function parseIndex() {
const root = util.parsePlaylist('index.m3u')
let categoryBuffer = {}
categories.push('other')
categories.forEach(category => {
categoryBuffer[category] = []
})
let countries = {}
let languages = {}
let categories = {}
function main() {
console.log(`Parsing 'index.m3u'...`)
const playlist = util.parsePlaylist('index.m3u')
let countries = playlist.items
if(debug) {
console.log('Debug mode is turn on')
countries = countries.slice(0, 1)
}
for(let type of types) {
const filename = `index.${type}.m3u`
console.log(`Creating '${filename}'...`)
util.createFile(filename, '#EXTM3U\n')
}
for(let category of categories) {
const filename = `categories/${category}.m3u`
console.log(`Creating '${filename}'...`)
util.createFile(filename, '#EXTM3U\n')
const categoryName = util.supportedCategories.find(c => c.toLowerCase() === category) || 'Other'
}
for(let country of countries) {
console.log(`Parsing '${country.url}'...`)
const playlist = util.parsePlaylist(country.url)
const c = {
name: country.name,
code: util.getBasename(country.url).toUpperCase()
}
for(let rootItem of root.items) {
const playlist = util.parsePlaylist(rootItem.url)
const countryCode = util.getBasename(rootItem.url).toLowerCase()
const countryName = rootItem.name
for(let item of playlist.items) {
const channel = util.createChannel(item)
channel.countryCode = countryCode
channel.countryName = countryName
let channel = util.createChannel(item)
// all
list.all.push(channel)
let category = channel.group.toLowerCase()
if(categoryBuffer[category]) {
categoryBuffer[category].push(channel)
} else {
categoryBuffer['other'].push(channel)
// country
if(!countries[countryCode]) {
countries[countryCode] = []
}
countries[countryCode].push(channel)
let languageCode = ISO6391.getCode(channel.language)
if(languageCode) {
if(!languageBuffer[languageCode]) {
languageBuffer[languageCode] = []
}
languageBuffer[languageCode].push(channel)
} else {
languageBuffer['undefined'].push(channel)
// language
const languageCode = ISO6391.getCode(channel.language) || 'undefined'
if(!languages[languageCode]) {
languages[languageCode] = []
}
languages[languageCode].push(channel)
let group = channel.group
for(const type of types) {
if(type === 'full') {
channel.group = [ c.name, channel.group ].filter(i => i).join(';')
} else if(type === 'country') {
channel.group = c.name
} else if(type === 'language') {
channel.group = channel.language
} else {
channel.group = group
}
util.appendToFile(`index.${type}.m3u`, channel.toString())
// category
const categoryCode = channel.group.toLowerCase() || 'other'
if(!categories[categoryCode]) {
categories[categoryCode] = []
}
stats.channels++
categories[categoryCode].push(channel)
}
stats.countries++
}
util.clearCache()
for(const languageCode in languageBuffer) {
const filename = `languages/${languageCode}.m3u`
util.createFile(filename, '#EXTM3U\n')
list.countries = countries
list.languages = languages
list.categories = categories
}
function generateCountryIndex() {
const filename = `index.country.m3u`
util.createFile(filename, '#EXTM3U\n')
for(let channel of list.all) {
const group = channel.group
channel.group = channel.countryName
util.appendToFile(filename, channel.toString())
channel.group = group
}
}
function generateLanguageIndex() {
const filename = `index.language.m3u`
util.createFile(filename, '#EXTM3U\n')
const channels = list.all.sort((a, b) => {
if(a.language < b.language) { return -1 }
if(a.language > b.language) { return 1 }
return 0
})
for(let channel of channels) {
const group = channel.group
channel.group = channel.language
util.appendToFile(filename, channel.toString())
channel.group = group
}
}
function generateContentIndex() {
const filename = `index.content.m3u`
util.createFile(filename, '#EXTM3U\n')
const channels = list.all.sort((a, b) => {
if(a.group < b.group) { return -1 }
if(a.group > b.group) { return 1 }
return 0
})
for(let channel of channels) {
util.appendToFile(filename, channel.toString())
}
}
function generateFullIndex() {
const filename = `index.full.m3u`
util.createFile(filename, '#EXTM3U\n')
const channels = list.all.sort((a, b) => {
if(a.countryName < b.countryName) { return -1 }
if(a.countryName > b.countryName) { return 1 }
if(a.group < b.group) { return -1 }
if(a.group > b.group) { return 1 }
return 0
})
for(let channel of channels) {
const group = channel.group
channel.group = [ channel.countryName, channel.group ].filter(i => i).join(';')
util.appendToFile(filename, channel.toString())
channel.group = group
}
}
function main() {
console.log(`Parsing index...`)
parseIndex()
console.log('Generating index.country.m3u...')
generateCountryIndex()
console.log('Generating index.language.m3u...')
generateLanguageIndex()
console.log('Generating index.content.m3u...')
generateContentIndex()
console.log('Generating index.full.m3u...')
generateFullIndex()
// console.log(output)
// let group = channel.group
// for(const type of types) {
// if(type === 'full') {
// channel.group = [ c.name, channel.group ].filter(i => i).join(';')
// } else if(type === 'country') {
// channel.group = c.name
// } else if(type === 'language') {
// channel.group = channel.language
// } else {
// channel.group = group
// }
// util.appendToFile(`index.${type}.m3u`, channel.toString())
// }
// for(let type of types) {
// const filename = `index.${type}.m3u`
// console.log(`Creating '${filename}'...`)
// util.createFile(filename, '#EXTM3U\n')
// }
// const categories = util.supportedCategories
// for(let category of categories) {
// const categoryCode = category.toLowerCase()
// util.createFile(`categories/${categoryCode}.m3u`, '#EXTM3U\n')
// }
// util.clearCache()
// for(const languageCode in languageBuffer) {
// const filename = `languages/${languageCode}.m3u`
// util.createFile(filename, '#EXTM3U\n')
let channels = util.sortByTitleAndUrl(languageBuffer[languageCode])
for(const channel of channels) {
if(!util.checkCache(channel.url)) {
util.appendToFile(filename, channel.toString())
util.addToCache(channel.url)
}
}
}
// let channels = util.sortByTitleAndUrl(languageBuffer[languageCode])
// for(const channel of channels) {
// if(!util.checkCache(channel.url)) {
// util.appendToFile(filename, channel.toString())
// util.addToCache(channel.url)
// }
// }
// }
util.clearCache()
for(const category in categoryBuffer) {
let channels = util.sortByTitleAndUrl(categoryBuffer[category])
for(const channel of channels) {
if(!util.checkCache(channel.url)) {
util.appendToFile(`categories/${category}.m3u`, channel.toString())
util.addToCache(channel.url)
}
}
}
// util.clearCache()
// for(const category in categoryBuffer) {
// let channels = util.sortByTitleAndUrl(categoryBuffer[category])
// for(const channel of channels) {
// if(!util.checkCache(channel.url)) {
// util.appendToFile(`categories/${category}.m3u`, channel.toString())
// util.addToCache(channel.url)
// }
// }
// }
//
// console.log(`Countries: ${stats.countries}. Languages: ${stats.channels}. Categories: ${stats.channels}. Channels: ${stats.channels}.`)
}
main()
console.log(`Countries: ${stats.countries}. Channels: ${stats.channels}.`)