mirror of
https://github.com/iptv-org/epg.git
synced 2025-05-09 08:30:06 -04:00
104 lines
3 KiB
TypeScript
104 lines
3 KiB
TypeScript
import { Logger, Timer, Storage, Collection } from '@freearhey/core'
|
|
import { program } from 'commander'
|
|
import { QueueCreator, Job, ChannelsParser } from '../../core'
|
|
import { Channel } from 'epg-grabber'
|
|
import path from 'path'
|
|
import { SITES_DIR } from '../../constants'
|
|
|
|
program
|
|
.option('-s, --site <name>', 'Name of the site to parse')
|
|
.option(
|
|
'-c, --channels <path>',
|
|
'Path to *.channels.xml file (required if the "--site" attribute is not specified)'
|
|
)
|
|
.option('-o, --output <path>', 'Path to output file', 'guide.xml')
|
|
.option('-l, --lang <code>', 'Filter channels by language (ISO 639-2 code)')
|
|
.option('-t, --timeout <milliseconds>', 'Override the default timeout for each request')
|
|
.option('-d, --delay <milliseconds>', 'Override the default delay between request')
|
|
.option(
|
|
'--days <days>',
|
|
'Override the number of days for which the program will be loaded (defaults to the value from the site config)',
|
|
value => parseInt(value)
|
|
)
|
|
.option(
|
|
'--maxConnections <number>',
|
|
'Limit on the number of concurrent requests',
|
|
value => parseInt(value),
|
|
1
|
|
)
|
|
.option('--gzip', 'Create a compressed version of the guide as well', false)
|
|
.parse(process.argv)
|
|
|
|
export type GrabOptions = {
|
|
site?: string
|
|
channels?: string
|
|
output: string
|
|
gzip: boolean
|
|
maxConnections: number
|
|
timeout?: string
|
|
delay?: string
|
|
lang?: string
|
|
days?: number
|
|
}
|
|
|
|
const options: GrabOptions = program.opts()
|
|
|
|
async function main() {
|
|
if (!options.site && !options.channels)
|
|
throw new Error('One of the arguments must be presented: `--site` or `--channels`')
|
|
|
|
const logger = new Logger()
|
|
|
|
logger.start('starting...')
|
|
|
|
logger.info('config:')
|
|
logger.tree(options)
|
|
|
|
logger.info('loading channels...')
|
|
const storage = new Storage()
|
|
const parser = new ChannelsParser({ storage })
|
|
|
|
let files: string[] = []
|
|
if (options.site) {
|
|
let pattern = path.join(SITES_DIR, options.site, '*.channels.xml')
|
|
pattern = pattern.replace(/\\/g, '/')
|
|
files = await storage.list(pattern)
|
|
} else if (options.channels) {
|
|
files = await storage.list(options.channels)
|
|
}
|
|
|
|
let parsedChannels = new Collection()
|
|
for (const filepath of files) {
|
|
parsedChannels = parsedChannels.concat(await parser.parse(filepath))
|
|
}
|
|
if (options.lang) {
|
|
parsedChannels = parsedChannels.filter((channel: Channel) => channel.lang === options.lang)
|
|
}
|
|
logger.info(` found ${parsedChannels.count()} channel(s)`)
|
|
|
|
logger.info(`run:`)
|
|
runJob({ logger, parsedChannels })
|
|
}
|
|
|
|
main()
|
|
|
|
async function runJob({ logger, parsedChannels }: { logger: Logger; parsedChannels: Collection }) {
|
|
const timer = new Timer()
|
|
timer.start()
|
|
|
|
const queueCreator = new QueueCreator({
|
|
parsedChannels,
|
|
logger,
|
|
options
|
|
})
|
|
const queue = await queueCreator.create()
|
|
const job = new Job({
|
|
queue,
|
|
logger,
|
|
options
|
|
})
|
|
|
|
await job.run()
|
|
|
|
logger.success(` done in ${timer.format('HH[h] mm[m] ss[s]')}`)
|
|
}
|