iptv-org-mirror/epg
1
0
Fork 0
mirror of https://github.com/iptv-org/epg.git synced 2025-05-10 00:50:09 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update update-api.js

Browse source
This commit is contained in:
Aleksandr Statciuk 2022-01-21 20:10:45 +03:00
parent 158197fb22
commit 122b666e11
2 changed files with 49 additions and 52 deletions
Download patch file Download diff file Expand all files Collapse all files

93
scripts/commands/update-api.js
View file

@ -5,31 +5,45 @@ const DB_DIR = process.env.DB_DIR || 'scripts/database'
const API_DIR = process.env.API_DIR || '.gh-pages/api' const API_DIR = process.env.API_DIR || '.gh-pages/api'
async function main() { async function main() {
await generateChannelsJson() await saveToChannelsJson(await loadChannels())
await generateProgramsJson() await saveToProgramsJson(await loadPrograms())
logger.info(`Done`)
} }
main() main()
async function generateChannelsJson() { async function loadChannels() {
logger.info('Generating channels.json...') logger.info('Loading channels from database...')
const channels = await loadChannels() await db.channels.load()
const channelsPath = `${API_DIR}/channels.json` const channels = await db.channels.find({}).sort({ xmltv_id: 1 })
logger.info(`Saving to "${channelsPath}"...`)
await file.create(channelsPath, JSON.stringify(channels))
}
async function generateProgramsJson() { const output = {}
logger.info('Generating programs.json...') for (const channel of channels) {
if (!output[channel.xmltv_id]) {
output[channel.xmltv_id] = {
id: channel.xmltv_id,
name: [],
logo: channel.logo || null,
country: channel.country,
guides: []
}
} else {
output[channel.xmltv_id].logo = output[channel.xmltv_id].logo || channel.logo
}
const programs = await loadPrograms() output[channel.xmltv_id].name.push(channel.name)
output[channel.xmltv_id].name = _.uniq(output[channel.xmltv_id].name)
channel.groups.forEach(group => {
if (channel.programCount) {
output[channel.xmltv_id].guides.push(
`https://iptv-org.github.io/epg/guides/${group}.epg.xml`
)
}
})
}
const programsPath = `${API_DIR}/programs.json` return Object.values(output)
logger.info(`Saving to "${programsPath}"...`)
await file.create(programsPath, JSON.stringify(programs))
} }
async function loadPrograms() { async function loadPrograms() {
@ -37,9 +51,9 @@ async function loadPrograms() {
await db.programs.load() await db.programs.load()
let items = await db.programs.find({}) let programs = await db.programs.find({})
items = items.map(item => { programs = programs.map(item => {
const categories = Array.isArray(item.category) ? item.category : [item.category] const categories = Array.isArray(item.category) ? item.category : [item.category]
return { return {
@ -58,36 +72,19 @@ async function loadPrograms() {
}) })
logger.info('Sort programs...') logger.info('Sort programs...')
return _.sortBy(items, ['channel', 'site', 'start']) programs = _.sortBy(programs, ['channel', 'site', 'start'])
return programs
} }
async function loadChannels() { async function saveToChannelsJson(channels = []) {
logger.info('Loading channels from database...') const channelsPath = `${API_DIR}/channels.json`
logger.info(`Saving to "${channelsPath}"...`)
await db.channels.load() await file.create(channelsPath, JSON.stringify(channels))
}
const items = await db.channels.find({}).sort({ xmltv_id: 1 })
async function saveToProgramsJson(programs = []) {
const output = {} const programsPath = `${API_DIR}/programs.json`
for (const item of items) { logger.info(`Saving to "${programsPath}"...`)
if (!output[item.xmltv_id]) { await file.create(programsPath, JSON.stringify(programs))
output[item.xmltv_id] = {
id: item.xmltv_id,
name: [],
logo: item.logo || null,
country: item.country,
guides: []
}
} else {
output[item.xmltv_id].logo = output[item.xmltv_id].logo || item.logo
}
output[item.xmltv_id].name.push(item.name)
output[item.xmltv_id].name = _.uniq(output[item.xmltv_id].name)
output[item.xmltv_id].guides.push(
`https://iptv-org.github.io/epg/guides/${item.gid}/${item.site}.epg.xml`
)
}
return Object.values(output)
} }

8
tests/__data__/input/database/channels.db
View file

@ -1,4 +1,4 @@
{"lang":"en","xmltv_id":"BravoEast.us","site_id":"237","logo":"https://www.directv.com/images/logos/channels/dark/large/579.png","name":"Bravo East","site":"directv.com","channelsPath":"sites/directv.com/directv.com_us.channels.xml","configPath":"sites/directv.com/directv.com.config.js","groups":["us"],"cluster_id":84,"country":"US","_id":"00AluKCrCnfgrl8W"} {"lang":"en","xmltv_id":"BravoEast.us","site_id":"237","logo":"https://www.directv.com/images/logos/channels/dark/large/579.png","name":"Bravo East","site":"directv.com","channelsPath":"sites/directv.com/directv.com_us.channels.xml","configPath":"sites/directv.com/directv.com.config.js","groups":["us/directv.com"],"cluster_id":84,"country":"US","programCount":1,"_id":"00AluKCrCnfgrl8W"}
{"lang":"fr","country":"US","xmltv_id":"CNNInternationalEurope.us","site_id":"53","logo":null,"name":"CNN International Europe","site":"chaines-tv.orange.fr","channelsPath":"sites/chaines-tv.orange.fr/chaines-tv.orange.fr_fr.channels.xml","configPath":"tests/__data__/input/sites/example.com.config.js","groups":["fr"],"cluster_id":1,"_id":"0Wefq0oMR3feCcuY"} {"lang":"fr","country":"US","xmltv_id":"CNNInternationalEurope.us","site_id":"53","logo":null,"name":"CNN International Europe","site":"chaines-tv.orange.fr","channelsPath":"sites/chaines-tv.orange.fr/chaines-tv.orange.fr_fr.channels.xml","configPath":"tests/__data__/input/sites/example.com.config.js","groups":["fr/chaines-tv.orange.fr"],"cluster_id":1,"programCount":1,"_id":"0Wefq0oMR3feCcuY"}
{"lang":"ru","country":"US","xmltv_id":"CNNInternationalEurope.us","site_id":"140","logo":"https://www.magticom.ge/images/channels/MjAxOC8wOS8xMC9lZmJhNWU5Yy0yMmNiLTRkMTAtOWY5Ny01ODM0MzY0ZTg0MmEuanBn.jpg","name":"CNN Int","site":"magticom.ge","channelsPath":"sites/magticom.ge/magticom.ge_ge.channels.xml","configPath":"tests/__data__/input/sites/example.com.config.js","groups":["ge"],"cluster_id":1,"_id":"1XzrxNkSF2AQNBrT"} {"lang":"ru","country":"US","xmltv_id":"CNNInternationalEurope.us","site_id":"140","logo":"https://www.magticom.ge/images/channels/MjAxOC8wOS8xMC9lZmJhNWU5Yy0yMmNiLTRkMTAtOWY5Ny01ODM0MzY0ZTg0MmEuanBn.jpg","name":"CNN Int","site":"magticom.ge","channelsPath":"sites/magticom.ge/magticom.ge_ge.channels.xml","configPath":"tests/__data__/input/sites/example.com.config.js","groups":["ge/magticom.ge"],"cluster_id":1,"programCount":1,"_id":"1XzrxNkSF2AQNBrT"}
{"lang":"en","country":"ZA","xmltv_id":"MNetMovies2.za","site_id":"404a052b-3dea-4cac-a19c-de9a7d6f191d#MAP","logo":"https://rndcdn.dstv.com/dstvcms/2020/08/31/M-Net_Movies_2_Logo_4-3_lightbackground_xlrg.png","name":"M-Net Movies 2","site":"dstv.com","channelsPath":"sites/dstv.com/dstv.com_zw.channels.xml","configPath":"sites/dstv.com/dstv.com.config.js","groups":["zw"],"cluster_id":120,"_id":"1lnhXpN7g0ER5XwN"} {"lang":"en","country":"ZA","xmltv_id":"MNetMovies2.za","site_id":"404a052b-3dea-4cac-a19c-de9a7d6f191d#MAP","logo":"https://rndcdn.dstv.com/dstvcms/2020/08/31/M-Net_Movies_2_Logo_4-3_lightbackground_xlrg.png","name":"M-Net Movies 2","site":"dstv.com","channelsPath":"sites/dstv.com/dstv.com_zw.channels.xml","configPath":"sites/dstv.com/dstv.com.config.js","groups":["zw/dstv.com"],"cluster_id":120,"programCount":1,"_id":"1lnhXpN7g0ER5XwN"}