iptv/scripts/commands/report/create.ts

import { Logger, Storage, Collection, Dictionary } from '@freearhey/core'
import { DATA_DIR, STREAMS_DIR } from '../../constants'
import { IssueLoader, PlaylistParser } from '../../core'
import { Blocked, Channel, Issue, Stream } from '../../models'

async function main() {
  const logger = new Logger()
  const loader = new IssueLoader()

  const storage = new Storage(DATA_DIR)

  logger.info('loading channels from api...')
  const channelsContent = await storage.json('channels.json')
  const groupedChannels = new Collection(channelsContent)
    .map(data => new Channel(data))
    .groupBy((channel: Channel) => channel.id)

  logger.info('loading blocklist from api...')
  const blocklistContent = await storage.json('blocklist.json')
  const groupedBlocklist = new Collection(blocklistContent)
    .map(data => new Blocked(data))
    .groupBy((blocked: Blocked) => blocked.channel)

  logger.info('loading streams...')
  const streamsStorage = new Storage(STREAMS_DIR)
  const parser = new PlaylistParser({ storage: streamsStorage })
  const files = await streamsStorage.list('**/*.m3u')
  const streams = await parser.parse(files)
  const groupedStreams = streams.groupBy((stream: Stream) => stream.url)

  logger.info('creating report...')
  let report = new Collection()

  logger.info('checking streams:add requests...')
  const addRequests = await loader.load({ labels: ['streams:add'] })
  const buffer = new Dictionary()
  addRequests.forEach((issue: Issue) => {
    const channelId = issue.data.get('channel_id') || undefined
    const streamUrl = issue.data.get('stream_url') || undefined

    const result = new Dictionary({
      issueNumber: issue.number,
      type: 'streams:add',
      channelId,
      status: undefined
    })

    if (!channelId) result.set('status', 'missing_id')
    else if (!streamUrl) result.set('status', 'missing_link')
    else if (groupedBlocklist.has(channelId)) result.set('status', 'blocked')
    else if (groupedChannels.missing(channelId)) result.set('status', 'invalid_id')
    else if (groupedStreams.has(streamUrl)) result.set('status', 'fullfilled')
    else if (buffer.has(streamUrl)) result.set('status', 'duplicate')
    else result.set('status', 'pending')

    buffer.set(streamUrl, true)

    report.add(result.data())
  })

  logger.info('checking streams:edit requests...')
  const editRequests = await loader.load({ labels: ['streams:edit'] })
  editRequests.forEach((issue: Issue) => {
    const channelId = issue.data.get('channel_id') || undefined
    const streamUrl = issue.data.get('stream_url') || undefined

    const result = new Dictionary({
      issueNumber: issue.number,
      type: 'streams:edit',
      channelId,
      status: undefined
    })

    if (!streamUrl) result.set('status', 'missing_link')
    else if (groupedStreams.missing(streamUrl)) result.set('status', 'invalid_link')
    else if (channelId && groupedChannels.missing(channelId)) result.set('status', 'invalid_id')
    else result.set('status', 'pending')

    report.add(result.data())
  })

  logger.info('checking broken streams reports...')
  const brokenStreamReports = await loader.load({ labels: ['broken stream'] })
  brokenStreamReports.forEach((issue: Issue) => {
    const streamUrl = issue.data.get('stream_url') || undefined

    const result = new Dictionary({
      issueNumber: issue.number,
      type: 'broken stream',
      channelId: undefined,
      status: undefined
    })

    if (!streamUrl) result.set('status', 'missing_link')
    else if (groupedStreams.missing(streamUrl)) result.set('status', 'invalid_link')
    else result.set('status', 'pending')

    report.add(result.data())
  })

  report = report.orderBy(item => item.issueNumber)

  console.table(report.all())
}

main()