Create QueueCreator

scripts/core/index.ts

@@ -9,3 +9,4 @@ export * from './guideManager'
 export * from './guide'
 export * from './apiChannel'
 export * from './apiClient'
+export * from './queueCreator'

scripts/core/queue.ts

@@ -1,9 +1,5 @@
-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'
+import { Dictionary } from '@freearhey/core'
+import { SiteConfig, Channel } from 'epg-grabber'
 
 export type QueueItem = {
   channel: Channel
@@ -12,83 +8,38 @@ export type QueueItem = {
   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[] = []
+  _data: Dictionary
 
-  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()
+  constructor() {
+    this._data = new Dictionary()
   }
 
-  async create() {
-    const channelsContent = await this.dataStorage.json('channels.json')
-    const channels = new Collection(channelsContent).map(data => new ApiChannel(data))
+  missing(key: string): boolean {
+    return this._data.missing(key)
+  }
 
-    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())
+  add(
+    key: string,
+    { channel, config, date }: { channel: Channel; date: string | null; config: SiteConfig }
+  ) {
+    this._data.set(key, {
+      channel,
+      date,
+      config,
+      error: null
+    })
   }
 
   size(): number {
-    return this._items.length
+    return Object.values(this._data.data()).length
   }
 
   items(): QueueItem[] {
-    return this._items
+    return Object.values(this._data.data()) as QueueItem[]
   }
 
   isEmpty(): boolean {
-    return this._items.length === 0
+    return this.size() === 0
   }
 }

scripts/core/queueCreator.ts

@@ -0,0 +1,71 @@
+import { Storage, Collection, DateTime, Logger } from '@freearhey/core'
+import { ChannelsParser, ConfigLoader, ApiChannel, Queue } from './'
+import { SITES_DIR, DATA_DIR, CURR_DATE } from '../constants'
+import { SiteConfig } from 'epg-grabber'
+import path from 'path'
+import { GrabOptions } from '../commands/epg/grab'
+
+type QueueCreatorProps = {
+  logger: Logger
+  options: GrabOptions
+  parsedChannels: Collection
+}
+
+export class QueueCreator {
+  configLoader: ConfigLoader
+  logger: Logger
+  sitesStorage: Storage
+  dataStorage: Storage
+  parser: ChannelsParser
+  parsedChannels: Collection
+  options: GrabOptions
+  date: DateTime
+
+  constructor({ parsedChannels, logger, options }: QueueCreatorProps) {
+    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(): Promise<Queue> {
+    const channelsContent = await this.dataStorage.json('channels.json')
+    const channels = new Collection(channelsContent).map(data => new ApiChannel(data))
+
+    const queue = new Queue()
+    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.add(key, {
+            channel,
+            date: dateString,
+            config
+          })
+        }
+      })
+    }
+
+    return queue
+  }
+}