epg/scripts/core/queue.ts

import { Storage, Collection, DateTime, Logger, Dictionary } from '@freearhey/core'
import { ChannelsParser, ConfigLoader, ApiChannel } from './'
import { SITES_DIR, DATA_DIR, CURR_DATE } from '../constants'
import { Channel, SiteConfig } from 'epg-grabber'
import path from 'path'
import { GrabOptions } from '../commands/epg/grab'

export type QueueItem = {
  channel: Channel
  date: string
  config: SiteConfig
  error: string | null
}

type QueueProps = {
  logger: Logger
  options: GrabOptions
  parsedChannels: Collection
}

export class Queue {
  configLoader: ConfigLoader
  logger: Logger
  sitesStorage: Storage
  dataStorage: Storage
  parser: ChannelsParser
  parsedChannels: Collection
  options: GrabOptions
  date: DateTime
  _items: QueueItem[] = []

  constructor({ parsedChannels, logger, options }: QueueProps) {
    this.parsedChannels = parsedChannels
    this.logger = logger
    this.sitesStorage = new Storage()
    this.dataStorage = new Storage(DATA_DIR)
    this.parser = new ChannelsParser({ storage: new Storage() })
    this.date = new DateTime(CURR_DATE)
    this.options = options
    this.configLoader = new ConfigLoader()
  }

  async create() {
    const channelsContent = await this.dataStorage.json('channels.json')
    const channels = new Collection(channelsContent).map(data => new ApiChannel(data))

    const queue = new Dictionary()

    for (const channel of this.parsedChannels.all()) {
      if (!channel.site || !channel.xmltv_id) continue
      if (this.options.lang && channel.lang !== this.options.lang) continue

      const configPath = path.resolve(SITES_DIR, `${channel.site}/${channel.site}.config.js`)
      const config: SiteConfig = await this.configLoader.load(configPath)

      const found: ApiChannel = channels.first(
        (_channel: ApiChannel) => _channel.id === channel.xmltv_id
      )
      if (found) {
        channel.logo = found.logo
      }

      const days = this.options.days || config.days || 1
      const dates = Array.from({ length: days }, (_, day) => this.date.add(day, 'd'))
      dates.forEach((date: DateTime) => {
        const dateString = date.toJSON()
        const key = `${channel.site}:${channel.lang}:${channel.xmltv_id}:${dateString}`

        if (queue.missing(key)) {
          queue.set(key, {
            channel,
            date: dateString,
            config,
            error: null
          })
        }
      })
    }

    this._items = Object.values(queue.data())
  }

  size(): number {
    return this._items.length
  }

  items(): QueueItem[] {
    return this._items
  }

  isEmpty(): boolean {
    return this._items.length === 0
  }
}