Update scripts

scripts/models/playlist.ts

@@ -1,26 +1,65 @@
 import { Collection } from '@freearhey/core'
+import * as sdk from '@iptv-org/sdk'
 import { Stream } from '../models'
 
 type PlaylistOptions = {
-  public: boolean
+  public?: boolean
 }
 
 export class Playlist {
   streams: Collection<Stream>
-  options: {
-    public: boolean
-  }
+  public: boolean
 
   constructor(streams: Collection<Stream>, options?: PlaylistOptions) {
     this.streams = streams
-    this.options = options || { public: false }
+    this.public = options?.public === true
+  }
+
+  getGuides(): Collection<sdk.Models.Guide> {
+    const guides = new Collection<sdk.Models.Guide>()
+
+    this.streams.forEach(stream => {
+      guides.concat(stream.getGuides())
+    })
+
+    return guides
+  }
+
+  getGuideUrls(): Collection<string> {
+    function byFormat(source: sdk.Types.GuideSource) {
+      if (source.format === 'GZIP') return 2
+      if (source.format === 'XML') return 1
+      return 0
+    }
+
+    const urls = new Collection<string>()
+
+    this.getGuides().forEach((guide: sdk.Models.Guide) => {
+      const sources = new Collection(guide.sources)
+
+      const source = sources
+        .filter((source: sdk.Types.GuideSource) => ['GZIP', 'XML'].includes(source.format))
+        .sortBy([byFormat], ['desc'])
+        .first()
+
+      if (source) urls.add(source.url)
+    })
+
+    return urls.uniq()
   }
 
   toString() {
-    let output = '#EXTM3U\r\n'
+    let output = '#EXTM3U'
+
+    const guideUrls = this.getGuideUrls()
+    if (guideUrls.count()) {
+      output += ` x-tvg-url="${guideUrls.join(',')}"`
+    }
+
+    output += '\r\n'
 
     this.streams.forEach((stream: Stream) => {
-      output += stream.toString(this.options) + '\r\n'
+      output += stream.toString({ public: this.public }) + '\r\n'
     })
 
     return output