Update scripts

scripts/models/feed.ts

@@ -1,5 +1,5 @@
-import { Collection, Dictionary } from '@freearhey/core'
 import { Country, Language, Region, Channel, Subdivision } from './index'
+import { Collection, Dictionary } from '@freearhey/core'
 import type { FeedData } from '../types/feed'
 
 export class Feed {
@@ -106,7 +106,7 @@ export class Feed {
     regionsKeyByCode: Dictionary,
     subdivisionsKeyByCode: Dictionary
   ): this {
-    let broadcastCountries = new Collection()
+    const broadcastCountries = new Collection()
 
     if (this.isInternational()) {
       this.broadcastCountries = broadcastCountries

scripts/models/stream.ts

@@ -2,6 +2,7 @@ import { Feed, Channel, Category, Region, Subdivision, Country, Language } from
 import { URL, Collection, Dictionary } from '@freearhey/core'
 import type { StreamData } from '../types/stream'
 import parser from 'iptv-playlist-parser'
+import { IssueData } from '../core'
 
 export class Stream {
   name?: string
@@ -39,6 +40,24 @@ export class Stream {
     this.label = data.label || undefined
   }
 
+  update(issueData: IssueData): this {
+    const data = {
+      label: issueData.getString('label'),
+      quality: issueData.getString('quality'),
+      httpUserAgent: issueData.getString('httpUserAgent'),
+      httpReferrer: issueData.getString('httpReferrer'),
+      newStreamUrl: issueData.getString('newStreamUrl')
+    }
+
+    if (data.label !== undefined) this.label = data.label
+    if (data.quality !== undefined) this.setQuality(data.quality)
+    if (data.httpUserAgent !== undefined) this.userAgent = data.httpUserAgent
+    if (data.httpReferrer !== undefined) this.referrer = data.httpReferrer
+    if (data.newStreamUrl !== undefined) this.url = data.newStreamUrl
+
+    return this
+  }
+
   fromPlaylistItem(data: parser.PlaylistItem): this {
     if (!data.name) throw new Error('"name" property is required')
     if (!data.url) throw new Error('"url" property is required')
@@ -98,12 +117,6 @@ export class Stream {
     return this
   }
 
-  setLabel(label: string): this {
-    this.label = label
-
-    return this
-  }
-
   setQuality(quality: string): this {
     const { verticalResolution, isInterlaced } = parseQuality(quality)
 
@@ -113,18 +126,6 @@ export class Stream {
     return this
   }
 
-  setUserAgent(userAgent: string): this {
-    this.userAgent = userAgent
-
-    return this
-  }
-
-  setReferrer(referrer: string): this {
-    this.referrer = referrer
-
-    return this
-  }
-
   getLine(): number {
     return this.line || -1
   }
@@ -387,7 +388,7 @@ function parseQuality(quality: string | null): {
   isInterlaced: boolean | null
 } {
   if (!quality) return { verticalResolution: null, isInterlaced: null }
-  let [, verticalResolutionString] = quality.match(/^(\d+)/) || [null, undefined]
+  const [, verticalResolutionString] = quality.match(/^(\d+)/) || [null, undefined]
   const isInterlaced = /i$/i.test(quality)
   let verticalResolution = 0
   if (verticalResolutionString) verticalResolution = parseInt(verticalResolutionString)