mirror of
https://github.com/iptv-org/iptv.git
synced 2025-05-11 17:40:03 -04:00
105 lines
3.1 KiB
TypeScript
105 lines
3.1 KiB
TypeScript
import { PlaylistParser, DataProcessor, DataLoader } from '../../core'
|
|
import type { DataProcessorData } from '../../types/dataProcessor'
|
|
import { DATA_DIR, LOGS_DIR, STREAMS_DIR } from '../../constants'
|
|
import type { DataLoaderData } from '../../types/dataLoader'
|
|
import { Logger, Storage, File } from '@freearhey/core'
|
|
import { Stream } from '../../models'
|
|
import { uniqueId } from 'lodash'
|
|
import {
|
|
IndexCategoryGenerator,
|
|
IndexLanguageGenerator,
|
|
IndexCountryGenerator,
|
|
IndexRegionGenerator,
|
|
CategoriesGenerator,
|
|
CountriesGenerator,
|
|
LanguagesGenerator,
|
|
RegionsGenerator,
|
|
IndexGenerator
|
|
} from '../../generators'
|
|
|
|
async function main() {
|
|
const logger = new Logger()
|
|
const logFile = new File('generators.log')
|
|
|
|
logger.info('loading data from api...')
|
|
const processor = new DataProcessor()
|
|
const dataStorage = new Storage(DATA_DIR)
|
|
const loader = new DataLoader({ storage: dataStorage })
|
|
const data: DataLoaderData = await loader.load()
|
|
const {
|
|
feedsGroupedByChannelId,
|
|
channelsKeyById,
|
|
categories,
|
|
countries,
|
|
regions
|
|
}: DataProcessorData = processor.process(data)
|
|
|
|
logger.info('loading streams...')
|
|
const streamsStorage = new Storage(STREAMS_DIR)
|
|
const parser = new PlaylistParser({
|
|
storage: streamsStorage,
|
|
feedsGroupedByChannelId,
|
|
channelsKeyById
|
|
})
|
|
const files = await streamsStorage.list('**/*.m3u')
|
|
let streams = await parser.parse(files)
|
|
const totalStreams = streams.count()
|
|
streams = streams.uniqBy((stream: Stream) =>
|
|
stream.hasId() ? stream.getChannelId() + stream.getFeedId() : uniqueId()
|
|
)
|
|
logger.info(`found ${totalStreams} streams (including ${streams.count()} unique)`)
|
|
|
|
logger.info('sorting streams...')
|
|
streams = streams.orderBy(
|
|
[
|
|
(stream: Stream) => stream.getId(),
|
|
(stream: Stream) => stream.getVerticalResolution(),
|
|
(stream: Stream) => stream.getLabel()
|
|
],
|
|
['asc', 'asc', 'desc']
|
|
)
|
|
|
|
logger.info('generating categories/...')
|
|
await new CategoriesGenerator({ categories, streams, logFile }).generate()
|
|
|
|
logger.info('generating countries/...')
|
|
await new CountriesGenerator({
|
|
countries,
|
|
streams,
|
|
logFile
|
|
}).generate()
|
|
|
|
logger.info('generating languages/...')
|
|
await new LanguagesGenerator({ streams, logFile }).generate()
|
|
|
|
logger.info('generating regions/...')
|
|
await new RegionsGenerator({
|
|
streams,
|
|
regions,
|
|
logFile
|
|
}).generate()
|
|
|
|
logger.info('generating index.m3u...')
|
|
await new IndexGenerator({ streams, logFile }).generate()
|
|
|
|
logger.info('generating index.category.m3u...')
|
|
await new IndexCategoryGenerator({ streams, logFile }).generate()
|
|
|
|
logger.info('generating index.country.m3u...')
|
|
await new IndexCountryGenerator({
|
|
streams,
|
|
logFile
|
|
}).generate()
|
|
|
|
logger.info('generating index.language.m3u...')
|
|
await new IndexLanguageGenerator({ streams, logFile }).generate()
|
|
|
|
logger.info('generating index.region.m3u...')
|
|
await new IndexRegionGenerator({ streams, regions, logFile }).generate()
|
|
|
|
logger.info('saving generators.log...')
|
|
const logStorage = new Storage(LOGS_DIR)
|
|
logStorage.saveFile(logFile)
|
|
}
|
|
|
|
main()
|