This commit is contained in:
Aleksandr Statciuk 2022-02-07 02:27:15 +03:00
parent bad3eddf9d
commit 86e0c2f7da
31 changed files with 611 additions and 698 deletions

View file

@ -9,274 +9,17 @@ async function main() {
await generator.generate('countries', streams)
await generator.generate('languages', streams)
await generator.generate('regions', streams)
await generator.generate('index_m3u', streams)
// await generateRegions()
// await generateIndex()
// await generateIndexNSFW()
// await generateIndexCategory()
// await generateIndexCountry()
// await generateIndexLanguage()
// await generateIndexRegion()
// await generateChannelsJson()
// await saveLogs()
}
main()
// async function generateRegions() {
// logger.info(`Generating regions/...`)
// for (const region of regions) {
// const { count } = await generator.generate(
// `${PUBLIC_PATH}/regions/${region.code.toLowerCase()}.m3u`,
// {
// regions: { $elemMatch: region }
// }
// )
// await log('regions', {
// name: region.name,
// code: region.code,
// count
// })
// }
// const { count: undefinedCount } = await generator.generate(
// `${PUBLIC_PATH}/regions/undefined.m3u`,
// { regions: { $size: 0 } },
// {
// saveEmpty: true,
// onLoad: function (items) {
// return items.map(item => {
// item.group_title = 'Undefined'
// return item
// })
// }
// }
// )
// await log('regions', {
// name: 'Undefined',
// code: 'UNDEFINED',
// count: undefinedCount
// })
// }
async function generateIndexNSFW() {
logger.info(`Generating index.nsfw.m3u...`)
await generator.generate(
`${PUBLIC_PATH}/index.nsfw.m3u`,
{},
{
includeNSFW: true,
onLoad: function (items) {
return items.map(item => {
if (!item.categories || !item.categories.length) {
item.group_title = 'Other'
}
return item
})
},
sortBy: item => {
if (item.group_title === 'Other') return '_'
return item.group_title || ''
}
}
)
}
async function generateIndex() {
logger.info(`Generating index.m3u...`)
await generator.generate(
`${PUBLIC_PATH}/index.m3u`,
{},
{
onLoad: function (items) {
return items.map(item => {
if (!item.categories || !item.categories.length) {
item.group_title = 'Other'
}
return item
})
},
sortBy: item => {
if (item.group_title === 'Other') return '_'
return item.group_title || ''
}
}
)
}
async function generateIndexCategory() {
logger.info(`Generating index.category.m3u...`)
await generator.generate(
`${PUBLIC_PATH}/index.category.m3u`,
{},
{
onLoad: function (items) {
let results = items
.filter(item => !item.categories || !item.categories.length)
.map(item => {
const newItem = _.cloneDeep(item)
newItem.group_title = 'Other'
return newItem
})
for (const category of _.sortBy(Object.values(categories), ['name'])) {
let filtered = items
.filter(item => {
return (
Array.isArray(item.categories) &&
item.categories.map(c => c.slug).includes(category.slug)
)
})
.map(item => {
const newItem = _.cloneDeep(item)
newItem.group_title = category.name
return newItem
})
results = results.concat(filtered)
}
return results
},
sortBy: item => {
if (item.group_title === 'Other') return '_'
return item.group_title
}
}
)
}
async function generateIndexCountry() {
logger.info(`Generating index.country.m3u...`)
await generator.generate(
`${PUBLIC_PATH}/index.country.m3u`,
{},
{
onLoad: function (items) {
let results = items
.filter(item => !item.countries || !item.countries.length)
.map(item => {
const newItem = _.cloneDeep(item)
newItem.group_title = 'Undefined'
newItem.categories = []
return newItem
})
for (const country of _.sortBy(Object.values(countries), ['name'])) {
let filtered = items
.filter(item => {
return (
Array.isArray(item.countries) &&
item.countries.map(c => c.code).includes(country.code)
)
})
.map(item => {
const newItem = _.cloneDeep(item)
newItem.group_title = country.name
return newItem
})
results = results.concat(filtered)
}
return results
},
sortBy: item => {
if (item.group_title === 'Undefined') return '_'
return item.group_title
}
}
)
}
async function generateIndexLanguage() {
logger.info(`Generating index.language.m3u...`)
await generator.generate(
`${PUBLIC_PATH}/index.language.m3u`,
{},
{
onLoad: function (items) {
let results = items
.filter(item => !item.languages || !item.languages.length)
.map(item => {
const newItem = _.cloneDeep(item)
newItem.group_title = 'Undefined'
newItem.categories = []
return newItem
})
for (const language of languages) {
let filtered = items
.filter(item => {
return (
Array.isArray(item.languages) &&
item.languages.map(c => c.code).includes(language.code)
)
})
.map(item => {
const newItem = _.cloneDeep(item)
newItem.group_title = language.name
return newItem
})
results = results.concat(filtered)
}
return results
},
sortBy: item => {
if (item.group_title === 'Undefined') return '_'
return item.group_title
}
}
)
}
async function generateIndexRegion() {
logger.info(`Generating index.region.m3u...`)
await generator.generate(
`${PUBLIC_PATH}/index.region.m3u`,
{},
{
onLoad: function (items) {
let results = items
.filter(item => !item.regions.length)
.map(item => {
const newItem = _.cloneDeep(item)
newItem.group_title = 'Undefined'
newItem.categories = []
return newItem
})
for (const region of regions) {
let filtered = items
.filter(item => {
return item.regions.map(c => c.code).includes(region.code)
})
.map(item => {
const newItem = _.cloneDeep(item)
newItem.group_title = region.name
return newItem
})
results = results.concat(filtered)
}
return results
},
sortBy: item => {
if (item.group_title === 'Undefined') return '_'
return item.group_title
}
}
)
}
async function loadStreams() {
await api.channels.load()
let channels = await api.channels.all()
@ -318,3 +61,192 @@ async function loadStreams() {
return stream
})
}
// async function generateIndexNSFW() {
// logger.info(`Generating index.nsfw.m3u...`)
// await generator.generate(
// `${PUBLIC_PATH}/index.nsfw.m3u`,
// {},
// {
// includeNSFW: true,
// onLoad: function (items) {
// return items.map(item => {
// if (!item.categories || !item.categories.length) {
// item.group_title = 'Other'
// }
// return item
// })
// },
// sortBy: item => {
// if (item.group_title === 'Other') return '_'
// return item.group_title || ''
// }
// }
// )
// }
// async function generateIndexCategory() {
// logger.info(`Generating index.category.m3u...`)
// await generator.generate(
// `${PUBLIC_PATH}/index.category.m3u`,
// {},
// {
// onLoad: function (items) {
// let results = items
// .filter(item => !item.categories || !item.categories.length)
// .map(item => {
// const newItem = _.cloneDeep(item)
// newItem.group_title = 'Other'
// return newItem
// })
// for (const category of _.sortBy(Object.values(categories), ['name'])) {
// let filtered = items
// .filter(item => {
// return (
// Array.isArray(item.categories) &&
// item.categories.map(c => c.slug).includes(category.slug)
// )
// })
// .map(item => {
// const newItem = _.cloneDeep(item)
// newItem.group_title = category.name
// return newItem
// })
// results = results.concat(filtered)
// }
// return results
// },
// sortBy: item => {
// if (item.group_title === 'Other') return '_'
// return item.group_title
// }
// }
// )
// }
// async function generateIndexCountry() {
// logger.info(`Generating index.country.m3u...`)
// await generator.generate(
// `${PUBLIC_PATH}/index.country.m3u`,
// {},
// {
// onLoad: function (items) {
// let results = items
// .filter(item => !item.countries || !item.countries.length)
// .map(item => {
// const newItem = _.cloneDeep(item)
// newItem.group_title = 'Undefined'
// newItem.categories = []
// return newItem
// })
// for (const country of _.sortBy(Object.values(countries), ['name'])) {
// let filtered = items
// .filter(item => {
// return (
// Array.isArray(item.countries) &&
// item.countries.map(c => c.code).includes(country.code)
// )
// })
// .map(item => {
// const newItem = _.cloneDeep(item)
// newItem.group_title = country.name
// return newItem
// })
// results = results.concat(filtered)
// }
// return results
// },
// sortBy: item => {
// if (item.group_title === 'Undefined') return '_'
// return item.group_title
// }
// }
// )
// }
// async function generateIndexLanguage() {
// logger.info(`Generating index.language.m3u...`)
// await generator.generate(
// `${PUBLIC_PATH}/index.language.m3u`,
// {},
// {
// onLoad: function (items) {
// let results = items
// .filter(item => !item.languages || !item.languages.length)
// .map(item => {
// const newItem = _.cloneDeep(item)
// newItem.group_title = 'Undefined'
// newItem.categories = []
// return newItem
// })
// for (const language of languages) {
// let filtered = items
// .filter(item => {
// return (
// Array.isArray(item.languages) &&
// item.languages.map(c => c.code).includes(language.code)
// )
// })
// .map(item => {
// const newItem = _.cloneDeep(item)
// newItem.group_title = language.name
// return newItem
// })
// results = results.concat(filtered)
// }
// return results
// },
// sortBy: item => {
// if (item.group_title === 'Undefined') return '_'
// return item.group_title
// }
// }
// )
// }
// async function generateIndexRegion() {
// logger.info(`Generating index.region.m3u...`)
// await generator.generate(
// `${PUBLIC_PATH}/index.region.m3u`,
// {},
// {
// onLoad: function (items) {
// let results = items
// .filter(item => !item.regions.length)
// .map(item => {
// const newItem = _.cloneDeep(item)
// newItem.group_title = 'Undefined'
// newItem.categories = []
// return newItem
// })
// for (const region of regions) {
// let filtered = items
// .filter(item => {
// return item.regions.map(c => c.code).includes(region.code)
// })
// .map(item => {
// const newItem = _.cloneDeep(item)
// newItem.group_title = region.name
// return newItem
// })
// results = results.concat(filtered)
// }
// return results
// },
// sortBy: item => {
// if (item.group_title === 'Undefined') return '_'
// return item.group_title
// }
// }
// )
// }