Update scripts

scripts/commands/channels/parse.ts

@@ -1,8 +1,9 @@
-import { Logger, File, Collection, Storage } from '@freearhey/core'
-import { ChannelsParser, XML } from '../../core'
-import { Channel } from 'epg-grabber'
-import { Command } from 'commander'
+import { Logger, File, Storage } from '@freearhey/core'
+import { ChannelsParser } from '../../core'
+import { ChannelList } from '../../models'
 import { pathToFileURL } from 'node:url'
+import epgGrabber from 'epg-grabber'
+import { Command } from 'commander'
 
 const program = new Command()
 program
@@ -21,17 +22,25 @@ type ParseOptions = {
 const options: ParseOptions = program.opts()
 
 async function main() {
+  function isPromise(promise: object[] | Promise<object[]>) {
+    return (
+      !!promise &&
+      typeof promise === 'object' &&
+      typeof (promise as Promise<object[]>).then === 'function'
+    )
+  }
+
   const storage = new Storage()
-  const parser = new ChannelsParser({ storage })
   const logger = new Logger()
+  const parser = new ChannelsParser({ storage })
   const file = new File(options.config)
   const dir = file.dirname()
   const config = (await import(pathToFileURL(options.config).toString())).default
   const outputFilepath = options.output || `${dir}/${config.site}.channels.xml`
 
-  let channels = new Collection()
+  let channelList = new ChannelList({ channels: [] })
   if (await storage.exists(outputFilepath)) {
-    channels = await parser.parse(outputFilepath)
+    channelList = await parser.parse(outputFilepath)
   }
 
   const args: {
@@ -49,45 +58,31 @@ async function main() {
   if (isPromise(parsedChannels)) {
     parsedChannels = await parsedChannels
   }
-  parsedChannels = parsedChannels.map((channel: Channel) => {
+  parsedChannels = parsedChannels.map((channel: epgGrabber.Channel) => {
     channel.site = config.site
 
     return channel
   })
 
-  let output = new Collection()
-  parsedChannels.forEach((channel: Channel) => {
-    const found: Channel | undefined = channels.first(
-      (_channel: Channel) => _channel.site_id == channel.site_id
-    )
+  const newChannelList = new ChannelList({ channels: [] })
+  parsedChannels.forEach((channel: epgGrabber.Channel) => {
+    if (!channel.site_id) return
+
+    const found: epgGrabber.Channel | undefined = channelList.get(channel.site_id)
 
     if (found) {
       channel.xmltv_id = found.xmltv_id
       channel.lang = found.lang
     }
 
-    output.add(channel)
+    newChannelList.add(channel)
   })
 
-  output = output.orderBy([
-    (channel: Channel) => channel.lang || '_',
-    (channel: Channel) => (channel.xmltv_id ? channel.xmltv_id.toLowerCase() : '0'),
-    (channel: Channel) => channel.site_id
-  ])
+  newChannelList.sort()
 
-  const xml = new XML(output)
-
-  await storage.save(outputFilepath, xml.toString())
+  await storage.save(outputFilepath, newChannelList.toString())
 
   logger.info(`File '${outputFilepath}' successfully saved`)
 }
 
 main()
-
-function isPromise(promise: object[] | Promise<object[]>) {
-  return (
-    !!promise &&
-    typeof promise === 'object' &&
-    typeof (promise as Promise<object[]>).then === 'function'
-  )
-}