diff --git a/scripts/update-readme.js b/scripts/update-readme.js
index fdea7885d9..268b35a37e 100644
--- a/scripts/update-readme.js
+++ b/scripts/update-readme.js
@@ -1,99 +1,120 @@
-const helper = require('./helper')
+const utils = require('./utils')
+const parser = require('./parser')
-let output = {
+const list = {
countries: [],
languages: [],
categories: []
}
function main() {
- console.log(`Parsing index...`)
parseIndex()
- console.log(`Generating countries table...`)
generateCountriesTable()
- console.log(`Generating languages table...`)
generateLanguagesTable()
- console.log(`Generating categories table...`)
generateCategoriesTable()
- console.log(`Generating README.md...`)
generateReadme()
- console.log(`Done.`)
+ finish()
}
function parseIndex() {
- const root = helper.parsePlaylist('index.m3u')
+ console.log(`Parsing index...`)
+ const items = parser.parseIndex()
- let countries = {}
- let languages = {}
- let categories = {}
+ const countries = {}
+ const languages = {}
+ const categories = {}
- for (let categoryCode in helper.supportedCategories) {
- categories[categoryCode] = {
- category: helper.supportedCategories[categoryCode],
- channels: 0,
- playlist: `https://iptv-org.github.io/iptv/categories/${categoryCode}.m3u
`
- }
+ countries['undefined'] = {
+ country: 'Undefined',
+ channels: 0,
+ playlist: `https://iptv-org.github.io/iptv/countries/undefined.m3u
`,
+ epg: '',
+ name: 'Undefined'
}
- for (let rootItem of root.items) {
- const playlist = helper.parsePlaylist(rootItem.url)
- const countryName = rootItem.name
- const countryCode = helper.getBasename(rootItem.url).toLowerCase()
- const countryEpg = playlist.header.attrs['x-tvg-url']
- ? `${playlist.header.attrs['x-tvg-url']}
`
- : ''
+ languages['undefined'] = {
+ language: 'Undefined',
+ channels: 0,
+ playlist: `https://iptv-org.github.io/iptv/languages/undefined.m3u
`
+ }
- for (let item of playlist.items) {
+ for (const category of utils.supportedCategories) {
+ categories[category.id] = {
+ category: category.name,
+ channels: 0,
+ playlist: `https://iptv-org.github.io/iptv/categories/${category.id}.m3u
`
+ }
+ }
+ categories['other'] = {
+ category: 'Other',
+ channels: 0,
+ playlist: `https://iptv-org.github.io/iptv/categories/other.m3u
`
+ }
+
+ for (const item of items) {
+ const playlist = parser.parsePlaylist(item.url)
+ for (let channel of playlist.channels) {
// countries
- if (countries[countryCode]) {
- countries[countryCode].channels++
+ if (!channel.countries.length) {
+ countries['undefined'].channels++
} else {
- let flag = helper.code2flag(countryCode)
-
- countries[countryCode] = {
- country: flag + ' ' + countryName,
- channels: 1,
- playlist: `https://iptv-org.github.io/iptv/countries/${countryCode}.m3u
`,
- epg: countryEpg
+ for (let country of channel.countries) {
+ if (countries[country.code]) {
+ countries[country.code].channels++
+ } else {
+ let flag = utils.code2flag(country.code)
+ countries[country.code] = {
+ country: flag + ' ' + country.name,
+ channels: 1,
+ playlist: `https://iptv-org.github.io/iptv/countries/${country.code}.m3u
`,
+ epg: playlist.header.attrs['x-tvg-url']
+ ? `${playlist.header.attrs['x-tvg-url']}
`
+ : '',
+ name: country.name
+ }
+ }
}
}
// languages
- const languageNames = item.tvg.language || 'Undefined'
- for (let languageName of languageNames.split(';')) {
- let languageCode = 'undefined'
- if (languageName !== 'Undefined') {
- languageCode = helper.getISO6391Code(languageName)
- if (!languageCode) continue
- }
-
- if (languages[languageCode]) {
- languages[languageCode].channels++
- } else {
- languages[languageCode] = {
- language: languageName,
- channels: 1,
- playlist: `https://iptv-org.github.io/iptv/languages/${languageCode}.m3u
`
+ if (!channel.languages.length) {
+ languages['undefined'].channels++
+ } else {
+ for (let language of channel.languages) {
+ if (languages[language.code]) {
+ languages[language.code].channels++
+ } else {
+ languages[language.code] = {
+ language: language.name,
+ channels: 1,
+ playlist: `https://iptv-org.github.io/iptv/languages/${language.code}.m3u
`
+ }
}
}
}
// categories
- const categoryName = helper.filterGroup(item.group.title) || 'Other'
- const categoryCode = categoryName.toLowerCase()
- if (categories[categoryCode]) {
- categories[categoryCode].channels++
+ const categoryId = channel.category.toLowerCase()
+ if (!categoryId) {
+ categories['other'].channels++
+ } else if (categories[categoryId]) {
+ categories[categoryId].channels++
}
}
}
- output.countries = Object.values(countries)
- output.languages = Object.values(languages)
- output.categories = Object.values(categories)
+ list.countries = Object.values(countries)
+ list.languages = Object.values(languages)
+ list.categories = Object.values(categories)
}
function generateCountriesTable() {
- const table = helper.generateTable(output.countries, {
+ console.log(`Generating countries table...`)
+ list.countries = utils.sortBy(list.countries, ['name'])
+ list.countries.forEach(function (i) {
+ delete i.name
+ })
+ const table = utils.generateTable(list.countries, {
columns: [
{ name: 'Country', align: 'left' },
{ name: 'Channels', align: 'right' },
@@ -102,27 +123,13 @@ function generateCountriesTable() {
]
})
- helper.createFile('./.readme/_countries.md', table)
+ utils.createFile('./.readme/_countries.md', table)
}
function generateLanguagesTable() {
- output.languages.sort((a, b) => {
- if (a.language === 'Undefined') {
- return 1
- }
- if (b.language === 'Undefined') {
- return -1
- }
- if (a.language < b.language) {
- return -1
- }
- if (a.language > b.language) {
- return 1
- }
- return 0
- })
-
- const table = helper.generateTable(output.languages, {
+ console.log(`Generating languages table...`)
+ list.languages = utils.sortBy(list.languages, ['language'])
+ const table = utils.generateTable(list.languages, {
columns: [
{ name: 'Language', align: 'left' },
{ name: 'Channels', align: 'right' },
@@ -130,27 +137,13 @@ function generateLanguagesTable() {
]
})
- helper.createFile('./.readme/_languages.md', table)
+ utils.createFile('./.readme/_languages.md', table)
}
function generateCategoriesTable() {
- output.categories.sort((a, b) => {
- if (a.category === 'Other') {
- return 1
- }
- if (b.category === 'Other') {
- return -1
- }
- if (a.category < b.category) {
- return -1
- }
- if (a.category > b.category) {
- return 1
- }
- return 0
- })
-
- const table = helper.generateTable(output.categories, {
+ console.log(`Generating categories table...`)
+ list.categories = utils.sortBy(list.categories, ['category'])
+ const table = utils.generateTable(list.categories, {
columns: [
{ name: 'Category', align: 'left' },
{ name: 'Channels', align: 'right' },
@@ -158,11 +151,16 @@ function generateCategoriesTable() {
]
})
- helper.createFile('./.readme/_categories.md', table)
+ utils.createFile('./.readme/_categories.md', table)
}
function generateReadme() {
- helper.compileMarkdown('../.readme/config.json')
+ console.log(`Generating README.md...`)
+ utils.compileMarkdown('../.readme/config.json')
+}
+
+function finish() {
+ console.log(`Done.`)
}
main()