public
/
epg-grabber
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update utils.js

This commit is contained in:
Aleksandr Statciuk 2022-05-06 13:41:08 +03:00
parent 06c1644449
commit 04b935fb19
1 changed files with 75 additions and 64 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

129
src/utils.js
View File

@ -150,7 +150,9 @@ utils.escapeString = function (string, defaultValue = '') {
} }
utils.convertToXMLTV = function ({ channels, programs }) { utils.convertToXMLTV = function ({ channels, programs }) {
let output = `<?xml version="1.0" encoding="UTF-8" ?><tv date="${dayjs.utc().format('YYYYMMDD')}">\r\n` let output = `<?xml version="1.0" encoding="UTF-8" ?><tv date="${dayjs
.utc()
.format('YYYYMMDD')}">\r\n`
for (let channel of channels) { for (let channel of channels) {
const id = utils.escapeString(channel['xmltv_id']) const id = utils.escapeString(channel['xmltv_id'])
const displayName = utils.escapeString(channel.name) const displayName = utils.escapeString(channel.name)
@ -179,15 +181,26 @@ utils.convertToXMLTV = function ({ channels, programs }) {
const xmltv_ns = createXMLTVNS(program.season, program.episode) const xmltv_ns = createXMLTVNS(program.season, program.episode)
const onscreen = createOnScreen(program.season, program.episode) const onscreen = createOnScreen(program.season, program.episode)
const date = program.date || '' const date = program.date || ''
const credits = createCredits({'director':program.director,'actor':program.actor, 'writer':program.writer,'adapter':program.adapter,'producer':program.producer, 'composer':program.composer, 'editor':program.editor, 'presenter':program.presenter, 'commentator':program.commentator, 'guest':program.guest}) const credits = createCredits({
director: program.director,
actor: program.actor,
writer: program.writer,
adapter: program.adapter,
producer: program.producer,
composer: program.composer,
editor: program.editor,
presenter: program.presenter,
commentator: program.commentator,
guest: program.guest
})
const icon = utils.escapeString(program.icon) const icon = utils.escapeString(program.icon)
const sub_title = program.sub_title || '' const sub_title = program.sub_title || ''
const url = program.url? createURL(program.url, channel) : '' const url = program.url ? createURL(program.url, channel) : ''
if (start && stop && title) { if (start && stop && title) {
output += `<programme start="${start}" stop="${stop}" channel="${channel}"><title lang="${lang}">${title}</title>` output += `<programme start="${start}" stop="${stop}" channel="${channel}"><title lang="${lang}">${title}</title>`
if (sub_title){ if (sub_title) {
output += `<sub-title>${sub_title}</sub-title>` output += `<sub-title>${sub_title}</sub-title>`
} }
@ -252,17 +265,17 @@ utils.convertToXMLTV = function ({ channels, programs }) {
function createURL(urlObj, channel = '') { function createURL(urlObj, channel = '') {
const urls = Array.isArray(urlObj) ? urlObj : [urlObj] const urls = Array.isArray(urlObj) ? urlObj : [urlObj]
let output = '' let output = ''
for (let url of urls){ for (let url of urls) {
if (typeof url === 'string' || url instanceof String){ if (typeof url === 'string' || url instanceof String) {
url = {value: url} url = { value: url }
} }
let attr = url.system ? ` system="${url.system}"` : '' let attr = url.system ? ` system="${url.system}"` : ''
if(url.value.includes('http')){ if (url.value.includes('http')) {
output += `<url${attr}>${url.value}</url>` output += `<url${attr}>${url.value}</url>`
} else if(channel){ } else if (channel) {
let chan = channels.find((c) => c.xmltv_id.localeCompare(channel) === 0) let chan = channels.find(c => c.xmltv_id.localeCompare(channel) === 0)
if (chan && chan.site){ if (chan && chan.site) {
output += `<url${attr}>https://${chan.site}${url.value}</url>` output += `<url${attr}>https://${chan.site}${url.value}</url>`
} }
} }
@ -271,41 +284,41 @@ utils.convertToXMLTV = function ({ channels, programs }) {
return output return output
} }
function createImage(imgObj, channel='') { function createImage(imgObj, channel = '') {
const imgs = Array.isArray(imgObj) ? imgObj : [imgObj] const imgs = Array.isArray(imgObj) ? imgObj : [imgObj]
let output = '' let output = ''
for (let img of imgs){ for (let img of imgs) {
if (typeof img === 'string' || img instanceof String){ if (typeof img === 'string' || img instanceof String) {
img = {value: img} img = { value: img }
} }
const imageTypes = ['poster', 'backdrop', 'still', 'person', 'character']; const imageTypes = ['poster', 'backdrop', 'still', 'person', 'character']
const imageSizes = ['1', '2', '3']; const imageSizes = ['1', '2', '3']
const imageOrients = ['P', 'L']; const imageOrients = ['P', 'L']
let attr = '' let attr = ''
if(img.type && imageTypes.some(el => img.type.includes(el))){ if (img.type && imageTypes.some(el => img.type.includes(el))) {
attr += ` type="${img.type}"` attr += ` type="${img.type}"`
} }
if(img.size && imageSizes.some(el => img.size.includes(el))){ if (img.size && imageSizes.some(el => img.size.includes(el))) {
attr += ` size="${img.size}"` attr += ` size="${img.size}"`
} }
if(img.orient && imageOrients.some(el => img.orient.includes(el))){ if (img.orient && imageOrients.some(el => img.orient.includes(el))) {
attr += ` orient="${img.orient}"` attr += ` orient="${img.orient}"`
} }
if(img.system){ if (img.system) {
attr += ` system="${img.system}"` attr += ` system="${img.system}"`
} }
if(img.value.includes('http')){ if (img.value.includes('http')) {
output += `<image${attr}>${img.value}</image>` output += `<image${attr}>${img.value}</image>`
} else if(channel){ } else if (channel) {
let chan = channels.find((c) => c.xmltv_id.localeCompare(channel) === 0) let chan = channels.find(c => c.xmltv_id.localeCompare(channel) === 0)
if (chan && chan.site){ if (chan && chan.site) {
output += `<image${attr}>https://${chan.site}${img.value}</image>` output += `<image${attr}>https://${chan.site}${img.value}</image>`
} }
} }
@ -315,40 +328,38 @@ utils.convertToXMLTV = function ({ channels, programs }) {
} }
function createCredits(obj) { function createCredits(obj) {
let cast = Object.entries(obj) let cast = Object.entries(obj)
.filter((x) => x[1]) .filter(x => x[1])
.map(([name, value]) => ({name, value})) .map(([name, value]) => ({ name, value }))
let output = '' let output = ''
for(let type of cast){ for (let type of cast) {
const r = Array.isArray(type.value) ? type.value: [type.value] const r = Array.isArray(type.value) ? type.value : [type.value]
for(let person of r){ for (let person of r) {
if (typeof person === 'string' || person instanceof String){ if (typeof person === 'string' || person instanceof String) {
person = {value: person} person = { value: person }
}
let attr = ''
if (type.name.localeCompare('actor') === 0 && type.value.role) {
attr += ` role="${type.value.role}"`
}
if (type.name.localeCompare('actor') === 0 && type.value.guest) {
attr += ` guest="${type.value.guest}"`
}
output += `<${type.name}${attr}>${person.value}`
if (person.url) {
output += createURL(person.url)
}
if (person.image) {
output += createImage(person.image)
}
output += `</${type.name}>`
} }
let attr = ''
if(type.name.localeCompare('actor') === 0 && type.value.role){
attr += ` role="${type.value.role}"`
}
if(type.name.localeCompare('actor') === 0 && type.value.guest){
attr += ` guest="${type.value.guest}"`
}
output += `<${type.name}${attr}>${person.value}`
if(person.url){
output += createURL(person.url)
}
if (person.image){
output += createImage(person.image)
}
output += `</${type.name}>`
} }
} return output
return output
} }
return output return output
} }
@ -468,8 +479,8 @@ utils.parsePrograms = async function (data, config) {
category: program.category || null, category: program.category || null,
season: program.season || null, season: program.season || null,
episode: program.episode || null, episode: program.episode || null,
sub_title : program.sub_title || null, sub_title: program.sub_title || null,
url : program.url || null, url: program.url || null,
icon: program.icon || null, icon: program.icon || null,
channel: channel.xmltv_id, channel: channel.xmltv_id,
lang: program.lang || channel.lang || config.lang || 'en', lang: program.lang || channel.lang || config.lang || 'en',