mirror of
https://github.com/iptv-org/iptv
synced 2026-03-31 00:01:59 -04:00
133 lines
3.7 KiB
TypeScript
133 lines
3.7 KiB
TypeScript
import { Collection, Logger } from '@freearhey/core'
|
|
import { OptionValues, program } from 'commander'
|
|
import { Stream, Playlist } from '../../models'
|
|
import { Storage } from '@freearhey/storage-js'
|
|
import { STREAMS_DIR } from '../../constants'
|
|
import { PlaylistParser } from '../../core'
|
|
import { getStreamInfo } from '../../utils'
|
|
import cliProgress from 'cli-progress'
|
|
import { loadData } from '../../api'
|
|
import { eachLimit } from 'async'
|
|
import path from 'node:path'
|
|
import os from 'node:os'
|
|
|
|
program
|
|
.argument('[filepath...]', 'Path to file to format')
|
|
.option(
|
|
'-p, --parallel <number>',
|
|
'Batch size of streams to test concurrently',
|
|
(value: string) => parseInt(value),
|
|
os.cpus().length
|
|
)
|
|
.option('-x, --proxy <url>', 'Use the specified proxy')
|
|
.option(
|
|
'-t, --timeout <number>',
|
|
'The number of milliseconds before the request will be aborted',
|
|
(value: string) => parseInt(value),
|
|
1000
|
|
)
|
|
.parse(process.argv)
|
|
|
|
const options: OptionValues = program.opts()
|
|
|
|
async function main() {
|
|
const logger = new Logger()
|
|
|
|
logger.info('loading data from api...')
|
|
await loadData()
|
|
|
|
logger.info('loading streams...')
|
|
const streamsStorage = new Storage(STREAMS_DIR)
|
|
const parser = new PlaylistParser({
|
|
storage: streamsStorage
|
|
})
|
|
let files = program.args.length ? program.args : await streamsStorage.list('**/*.m3u')
|
|
files = files.map((filepath: string) => path.basename(filepath))
|
|
let streams = await parser.parse(files)
|
|
|
|
logger.info(`found ${streams.count()} streams`)
|
|
|
|
logger.info('normalizing links...')
|
|
streams = streams.map(stream => {
|
|
stream.normalizeURL()
|
|
return stream
|
|
})
|
|
|
|
logger.info('removing duplicates...')
|
|
streams = streams.uniqBy(stream => stream.url)
|
|
|
|
logger.info('removing wrong id...')
|
|
streams = streams.map((stream: Stream) => {
|
|
const channel = stream.getChannel()
|
|
if (channel) return stream
|
|
|
|
stream.tvgId = ''
|
|
stream.channel = ''
|
|
stream.feed = ''
|
|
|
|
return stream
|
|
})
|
|
|
|
logger.info('adding the missing feed id...')
|
|
streams = streams.map((stream: Stream) => {
|
|
const feed = stream.getFeed()
|
|
if (feed) {
|
|
stream.feed = feed.id
|
|
stream.tvgId = stream.getId()
|
|
}
|
|
|
|
return stream
|
|
})
|
|
|
|
logger.info('adding the missing quality...')
|
|
const progressBar = new cliProgress.SingleBar({
|
|
clearOnComplete: true,
|
|
format: `[{bar}] {percentage}% | {value}/{total}`
|
|
})
|
|
progressBar.start(streams.count(), 0)
|
|
await eachLimit(streams.all(), options.parallel, async (stream: Stream) => {
|
|
progressBar.increment()
|
|
if (stream.quality) return
|
|
|
|
const streamInfo = await getStreamInfo(stream.url, {
|
|
httpUserAgent: stream.user_agent,
|
|
httpReferrer: stream.referrer,
|
|
timeout: options.timeout,
|
|
proxy: options.proxy
|
|
})
|
|
|
|
if (streamInfo) {
|
|
const height = streamInfo?.resolution?.height
|
|
|
|
if (height) {
|
|
stream.quality = `${height}p`
|
|
}
|
|
}
|
|
})
|
|
progressBar.stop()
|
|
|
|
logger.info('sorting links...')
|
|
streams = streams.sortBy(
|
|
[
|
|
(stream: Stream) => stream.title,
|
|
(stream: Stream) => stream.getVerticalResolution(),
|
|
(stream: Stream) => stream.label,
|
|
(stream: Stream) => stream.url
|
|
],
|
|
['asc', 'desc', 'asc', 'asc']
|
|
)
|
|
|
|
logger.info('saving...')
|
|
const groupedStreams = streams.groupBy((stream: Stream) => stream.getFilepath())
|
|
for (const filepath of groupedStreams.keys()) {
|
|
const streams = new Collection(groupedStreams.get(filepath))
|
|
|
|
if (streams.isEmpty()) return
|
|
|
|
const playlist = new Playlist(streams, { public: false })
|
|
await streamsStorage.save(filepath, playlist.toString())
|
|
}
|
|
}
|
|
|
|
main()
|