Update scripts

scripts/core/queue.ts

@@ -0,0 +1,94 @@
+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
+  }
+}