iptv/scripts/generate.js

231 lines
6.3 KiB
JavaScript

const utils = require('./utils')
const parser = require('./parser')
const ROOT_DIR = './.gh-pages'
let list = {
all: [],
countries: {},
languages: {},
categories: {}
}
function main() {
parseIndex()
createRootDirectory()
createNoJekyllFile()
generateIndex()
generateSFWIndex()
generateChannelsJson()
generateCountryIndex()
generateLanguageIndex()
generateCategoryIndex()
generateCountries()
generateLanguages()
generateCategories()
finish()
}
function createRootDirectory() {
console.log('Creating root directory...')
utils.createDir(ROOT_DIR)
}
function createNoJekyllFile() {
console.log('Creating .nojekyll...')
utils.createFile(`${ROOT_DIR}/.nojekyll`)
}
function parseIndex() {
console.log(`Parsing index...`)
const items = parser.parseIndex()
for (const category of utils.supportedCategories) {
list.categories[category.id] = []
}
list.categories['other'] = []
for (let item of items) {
const playlist = parser.parsePlaylist(item.url)
for (let channel of playlist.channels) {
// all
list.all.push(channel)
// country
if (!channel.countries.length) {
const countryCode = 'undefined'
if (!list.countries[countryCode]) {
list.countries[countryCode] = []
}
list.countries[countryCode].push(channel)
} else {
for (let country of channel.countries) {
const countryCode = country.code || 'undefined'
if (!list.countries[countryCode]) {
list.countries[countryCode] = []
}
list.countries[countryCode].push(channel)
}
}
// language
if (!channel.languages.length) {
const languageCode = 'undefined'
if (!list.languages[languageCode]) {
list.languages[languageCode] = []
}
list.languages[languageCode].push(channel)
} else {
for (let language of channel.languages) {
const languageCode = language.code || 'undefined'
if (!list.languages[languageCode]) {
list.languages[languageCode] = []
}
list.languages[languageCode].push(channel)
}
}
// category
const categoryId = channel.category.toLowerCase()
if (!list.categories[categoryId]) {
list.categories['other'].push(channel)
} else {
list.categories[categoryId].push(channel)
}
}
}
}
function generateIndex() {
console.log('Generating index.m3u...')
const filename = `${ROOT_DIR}/index.m3u`
utils.createFile(filename, '#EXTM3U\n')
const channels = utils.sortBy(list.all, ['name', 'url'])
for (let channel of channels) {
utils.appendToFile(filename, channel.toString())
}
}
function generateSFWIndex() {
console.log('Generating index.sfw.m3u...')
const filename = `${ROOT_DIR}/index.sfw.m3u`
utils.createFile(filename, '#EXTM3U\n')
const sorted = utils.sortBy(list.all, ['name', 'url'])
const channels = utils.filterNSFW(sorted)
for (let channel of channels) {
utils.appendToFile(filename, channel.toString())
}
}
function generateChannelsJson() {
console.log('Generating channels.json...')
const filename = `${ROOT_DIR}/channels.json`
const sorted = utils.sortBy(list.all, ['name', 'url'])
const channels = sorted.map(c => c.toJSON())
utils.createFile(filename, JSON.stringify(channels))
}
function generateCountryIndex() {
console.log('Generating index.country.m3u...')
const filename = `${ROOT_DIR}/index.country.m3u`
utils.createFile(filename, '#EXTM3U\n')
const channels = utils.sortBy(list.all, ['tvgCountry', 'name', 'url'])
for (let channel of channels) {
const category = channel.category
channel.category = channel.tvgCountry
utils.appendToFile(filename, channel.toString())
channel.category = category
}
}
function generateLanguageIndex() {
console.log('Generating index.language.m3u...')
const filename = `${ROOT_DIR}/index.language.m3u`
utils.createFile(filename, '#EXTM3U\n')
const channels = utils.sortBy(list.all, ['tvgLanguage', 'name', 'url'])
for (let channel of channels) {
const category = channel.category
channel.category = channel.tvgLanguage
utils.appendToFile(filename, channel.toString())
channel.category = category
}
}
function generateCategoryIndex() {
console.log('Generating index.category.m3u...')
const filename = `${ROOT_DIR}/index.category.m3u`
utils.createFile(filename, '#EXTM3U\n')
const channels = utils.sortBy(list.all, ['category', 'name', 'url'])
for (let channel of channels) {
utils.appendToFile(filename, channel.toString())
}
}
function generateCountries() {
console.log('Generating /countries...')
const outputDir = `${ROOT_DIR}/countries`
utils.createDir(outputDir)
for (const countryId in list.countries) {
const filename = `${outputDir}/${countryId}.m3u`
utils.createFile(filename, '#EXTM3U\n')
let channels = Object.values(list.countries[countryId])
channels = utils.sortBy(channels, ['name', 'url'])
for (const channel of channels) {
utils.appendToFile(filename, channel.toString())
}
}
}
function generateLanguages() {
console.log('Generating /languages...')
const outputDir = `${ROOT_DIR}/languages`
utils.createDir(outputDir)
for (const languageId in list.languages) {
const filename = `${outputDir}/${languageId}.m3u`
utils.createFile(filename, '#EXTM3U\n')
let channels = Object.values(list.languages[languageId])
channels = utils.sortBy(channels, ['name', 'url'])
for (const channel of channels) {
utils.appendToFile(filename, channel.toString())
}
}
}
function generateCategories() {
console.log('Generating /categories...')
const outputDir = `${ROOT_DIR}/categories`
utils.createDir(outputDir)
for (const category of utils.supportedCategories) {
const filename = `${outputDir}/${category.id}.m3u`
utils.createFile(filename, '#EXTM3U\n')
let channels = Object.values(list.categories[category.id])
channels = utils.sortBy(channels, ['name', 'url'])
for (const channel of channels) {
utils.appendToFile(filename, channel.toString())
}
}
}
function finish() {
console.log('Done.\n')
console.log(
`Countries: ${Object.values(list.countries).length}. Languages: ${
Object.values(list.languages).length
}. Categories: ${Object.values(list.categories).length}. Channels: ${list.all.length}.`
)
}
main()