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

Update create.js

Browse source
This commit is contained in:
Aleksandr Statciuk 2023-04-30 05:48:11 +03:00
parent ccf635673a
commit bee42ff2d5
1 changed files with 5 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

27
scripts/commands/database/create.js
View file

@ -1,4 +1,4 @@
const { db, file, parser, store, logger, api } = require('../../core') const { db, file, parser, store, logger } = require('../../core')
const { program } = require('commander') const { program } = require('commander')
const _ = require('lodash') const _ = require('lodash')
@ -8,21 +8,7 @@ const options = program
.opts() .opts()
async function main() { async function main() {
logger.info('starting...')
await saveToDatabase(await findStreams())
}
main()
async function findStreams() {
logger.info(`loading channels...`)
await api.channels.load()
const channels = _.keyBy(await api.channels.all(), 'id')
logger.info(`looking for streams...`) logger.info(`looking for streams...`)
await db.streams.load()
const streams = [] const streams = []
const files = await file.list(`${options.inputDir}/**/*.m3u`) const files = await file.list(`${options.inputDir}/**/*.m3u`)
for (const filepath of files) { for (const filepath of files) {
@ -31,9 +17,8 @@ async function findStreams() {
item.filepath = filepath item.filepath = filepath
const stream = store.create() const stream = store.create()
const channel = channels[item.tvg.id]
stream.set('channel', { channel: channel ? channel.id : null }) stream.set('channel', { channel: item.tvg.id })
stream.set('title', { title: item.name }) stream.set('title', { title: item.name })
stream.set('filepath', { filepath: item.filepath }) stream.set('filepath', { filepath: item.filepath })
stream.set('url', { url: item.url }) stream.set('url', { url: item.url })
@ -45,13 +30,11 @@ async function findStreams() {
} }
logger.info(`found ${streams.length} streams`) logger.info(`found ${streams.length} streams`)
return streams
}
async function saveToDatabase(streams = []) {
logger.info('saving to the database...') logger.info('saving to the database...')
await db.streams.load()
await db.streams.reset() await db.streams.reset()
const data = streams.map(stream => stream.data()) const data = streams.map(stream => stream.data())
await db.streams.insert(data) await db.streams.insert(data)
} }
main()