iptv/scripts/generate.js

191 lines
5.5 KiB
JavaScript
Raw Normal View History

2021-02-10 14:39:40 +01:00
const db = require('./db')
2021-01-30 00:26:23 +01:00
const utils = require('./utils')
2019-11-02 11:26:21 +01:00
const ROOT_DIR = './.gh-pages'
2021-02-10 14:39:40 +01:00
db.load()
2019-11-01 22:27:05 +01:00
function main() {
2019-11-03 18:25:33 +01:00
createRootDirectory()
2019-11-13 15:27:59 +01:00
createNoJekyllFile()
2019-11-02 11:50:56 +01:00
generateIndex()
2021-01-27 16:54:08 +01:00
generateSFWIndex()
2021-01-30 00:26:23 +01:00
generateChannelsJson()
2019-11-01 22:27:05 +01:00
generateCountryIndex()
generateLanguageIndex()
generateCategoryIndex()
2019-11-02 11:50:56 +01:00
generateCountries()
2019-11-01 22:27:05 +01:00
generateLanguages()
2021-01-30 00:26:23 +01:00
generateCategories()
finish()
2019-11-01 22:27:05 +01:00
}
2019-11-03 18:25:33 +01:00
function createRootDirectory() {
2021-02-10 14:39:40 +01:00
console.log('Creating .gh-pages folder...')
2021-01-30 00:26:23 +01:00
utils.createDir(ROOT_DIR)
2019-11-02 11:26:21 +01:00
}
2019-11-13 15:27:59 +01:00
function createNoJekyllFile() {
2021-02-10 15:03:47 +01:00
console.log('Creating .nojekyll file...')
2021-01-30 00:26:23 +01:00
utils.createFile(`${ROOT_DIR}/.nojekyll`)
2019-11-13 15:27:59 +01:00
}
2019-11-02 11:50:56 +01:00
function generateIndex() {
2021-01-30 00:26:23 +01:00
console.log('Generating index.m3u...')
2019-11-02 11:50:56 +01:00
const filename = `${ROOT_DIR}/index.m3u`
2021-01-30 00:26:23 +01:00
utils.createFile(filename, '#EXTM3U\n')
2019-11-02 11:50:56 +01:00
2021-02-10 14:39:40 +01:00
const channels = utils.sortBy(db.channels.all(), ['name', 'url'])
2020-04-11 03:34:50 +02:00
for (let channel of channels) {
2021-01-30 00:26:23 +01:00
utils.appendToFile(filename, channel.toString())
2019-11-02 11:50:56 +01:00
}
}
2021-01-27 16:54:08 +01:00
function generateSFWIndex() {
2021-01-30 00:26:23 +01:00
console.log('Generating index.sfw.m3u...')
2021-01-27 16:54:08 +01:00
const filename = `${ROOT_DIR}/index.sfw.m3u`
2021-01-30 00:26:23 +01:00
utils.createFile(filename, '#EXTM3U\n')
2021-01-27 16:54:08 +01:00
2021-02-10 14:39:40 +01:00
const sorted = utils.sortBy(db.channels.all(), ['name', 'url'])
2021-01-30 00:26:23 +01:00
const channels = utils.filterNSFW(sorted)
2021-01-27 16:54:08 +01:00
for (let channel of channels) {
2021-01-30 00:26:23 +01:00
utils.appendToFile(filename, channel.toString())
2021-01-27 16:54:08 +01:00
}
}
2021-01-30 00:26:23 +01:00
function generateChannelsJson() {
console.log('Generating channels.json...')
2020-05-04 15:23:36 +02:00
const filename = `${ROOT_DIR}/channels.json`
2021-02-10 14:39:40 +01:00
const channels = db.channels
.sortBy(['name', 'url'])
.all()
.map(c => c.toJSON())
2021-01-30 00:26:23 +01:00
utils.createFile(filename, JSON.stringify(channels))
2020-05-04 15:23:36 +02:00
}
2019-11-01 22:08:27 +01:00
function generateCountryIndex() {
2021-01-30 00:26:23 +01:00
console.log('Generating index.country.m3u...')
2019-11-02 11:26:21 +01:00
const filename = `${ROOT_DIR}/index.country.m3u`
2021-01-30 00:26:23 +01:00
utils.createFile(filename, '#EXTM3U\n')
2019-11-01 22:08:27 +01:00
2021-02-10 14:39:40 +01:00
const channels = db.channels.sortBy(['name', 'url']).forCountry({ code: null })
for (const channel of channels) {
const category = channel.category
channel.category = ''
utils.appendToFile(filename, channel.toString())
channel.category = category
}
const countries = db.countries.sortBy(['name']).all()
for (const country of countries) {
const channels = db.channels.sortBy(['name', 'url']).forCountry(country)
for (const channel of channels) {
2021-02-10 09:57:56 +01:00
const category = channel.category
2021-02-10 14:39:40 +01:00
channel.category = country.name
2021-02-10 09:57:56 +01:00
utils.appendToFile(filename, channel.toString())
channel.category = category
}
}
2019-11-01 22:08:27 +01:00
}
2019-11-01 22:08:27 +01:00
function generateLanguageIndex() {
2021-01-30 00:26:23 +01:00
console.log('Generating index.language.m3u...')
2019-11-02 11:26:21 +01:00
const filename = `${ROOT_DIR}/index.language.m3u`
2021-01-30 00:26:23 +01:00
utils.createFile(filename, '#EXTM3U\n')
2019-11-01 22:08:27 +01:00
2021-02-10 15:03:47 +01:00
const channels = db.channels.sortBy(['name', 'url']).forLanguage({ code: null })
2021-02-10 14:39:40 +01:00
for (const channel of channels) {
2020-05-04 15:23:36 +02:00
const category = channel.category
2021-02-10 15:03:47 +01:00
channel.category = ''
2021-01-30 00:26:23 +01:00
utils.appendToFile(filename, channel.toString())
2020-05-04 15:23:36 +02:00
channel.category = category
}
2021-02-10 15:03:47 +01:00
const languages = db.languages.sortBy(['name']).all()
for (const language of languages) {
const channels = db.channels.sortBy(['name', 'url']).forLanguage(language)
for (const channel of channels) {
const category = channel.category
channel.category = language.name
utils.appendToFile(filename, channel.toString())
channel.category = category
}
}
2019-11-01 22:08:27 +01:00
}
function generateCategoryIndex() {
2021-01-30 00:26:23 +01:00
console.log('Generating index.category.m3u...')
const filename = `${ROOT_DIR}/index.category.m3u`
2021-01-30 00:26:23 +01:00
utils.createFile(filename, '#EXTM3U\n')
2019-11-01 22:08:27 +01:00
2021-02-10 14:39:40 +01:00
const channels = db.channels.sortBy(['category', 'name', 'url']).all()
for (const channel of channels) {
2021-01-30 00:26:23 +01:00
utils.appendToFile(filename, channel.toString())
}
}
2021-02-10 15:03:47 +01:00
function generateCategories() {
console.log('Generating a playlist for each category...')
const outputDir = `${ROOT_DIR}/categories`
2021-01-30 00:26:23 +01:00
utils.createDir(outputDir)
2019-11-02 11:50:56 +01:00
2021-02-10 15:03:47 +01:00
for (const category of db.categories.all()) {
const filename = `${outputDir}/${category.id}.m3u`
2021-01-30 00:26:23 +01:00
utils.createFile(filename, '#EXTM3U\n')
2019-11-02 13:43:41 +01:00
2021-02-10 15:03:47 +01:00
const channels = db.channels.sortBy(['name', 'url']).forCategory(category)
2021-01-30 04:18:21 +01:00
for (const channel of channels) {
2021-01-30 00:26:23 +01:00
utils.appendToFile(filename, channel.toString())
2019-11-02 11:50:56 +01:00
}
}
2021-02-10 15:03:47 +01:00
const other = `${outputDir}/other.m3u`
const channels = db.channels.sortBy(['name', 'url']).forCategory({ id: null })
utils.createFile(other, '#EXTM3U\n')
for (const channel of channels) {
utils.appendToFile(other, channel.toString())
}
2019-11-02 11:50:56 +01:00
}
2021-02-10 15:03:47 +01:00
function generateCountries() {
console.log('Generating a playlist for each country...')
const outputDir = `${ROOT_DIR}/countries`
2021-01-30 00:26:23 +01:00
utils.createDir(outputDir)
2019-11-02 11:26:21 +01:00
2021-02-10 15:03:47 +01:00
for (const country of db.countries.all()) {
const filename = `${outputDir}/${country.code}.m3u`
2021-01-30 00:26:23 +01:00
utils.createFile(filename, '#EXTM3U\n')
2019-11-02 13:43:41 +01:00
2021-02-10 15:03:47 +01:00
const channels = db.channels.sortBy(['name', 'url']).forCountry(country)
2021-01-30 04:18:21 +01:00
for (const channel of channels) {
2021-01-30 00:26:23 +01:00
utils.appendToFile(filename, channel.toString())
2019-11-01 22:21:08 +01:00
}
}
}
2021-02-10 15:03:47 +01:00
function generateLanguages() {
console.log('Generating a playlist for each language...')
const outputDir = `${ROOT_DIR}/languages`
2021-01-30 00:26:23 +01:00
utils.createDir(outputDir)
2019-11-02 11:50:56 +01:00
2021-02-10 15:03:47 +01:00
for (const language of db.languages.all()) {
const filename = `${outputDir}/${language.code}.m3u`
2021-01-30 00:26:23 +01:00
utils.createFile(filename, '#EXTM3U\n')
2019-11-02 13:43:41 +01:00
2021-02-10 15:03:47 +01:00
const channels = db.channels.sortBy(['name', 'url']).forLanguage(language)
2021-01-30 04:18:21 +01:00
for (const channel of channels) {
2021-01-30 00:26:23 +01:00
utils.appendToFile(filename, channel.toString())
2019-11-01 22:21:08 +01:00
}
}
}
2021-01-30 00:26:23 +01:00
function finish() {
console.log(
2021-02-10 14:39:40 +01:00
`Countries: ${db.countries.count()}. Languages: ${db.languages.count()}. Categories: ${db.categories.count()}. Channels: ${db.channels.count()}.`
2021-01-30 00:26:23 +01:00
)
2021-02-10 14:39:40 +01:00
console.log('Done.')
2021-01-30 00:26:23 +01:00
}
2019-11-01 22:08:27 +01:00
main()