Update scripts

scripts/models/feed.ts

@@ -1,4 +1,4 @@
-import { Country, Language, Region, Channel, Subdivision } from './index'
+import { Country, Language, Region, Channel, Subdivision, BroadcastArea } from './index'
 import { Collection, Dictionary } from '@freearhey/core'
 import type { FeedData } from '../types/feed'
 
@@ -9,12 +9,7 @@ export class Feed {
   name: string
   isMain: boolean
   broadcastAreaCodes: Collection
-  broadcastCountryCodes: Collection
-  broadcastCountries?: Collection
-  broadcastRegionCodes: Collection
-  broadcastRegions?: Collection
-  broadcastSubdivisionCodes: Collection
-  broadcastSubdivisions?: Collection
+  broadcastArea?: BroadcastArea
   languageCodes: Collection
   languages?: Collection
   timezoneIds: Collection
@@ -32,25 +27,6 @@ export class Feed {
     this.languageCodes = new Collection(data.languages)
     this.timezoneIds = new Collection(data.timezones)
     this.videoFormat = data.video_format
-    this.broadcastCountryCodes = new Collection()
-    this.broadcastRegionCodes = new Collection()
-    this.broadcastSubdivisionCodes = new Collection()
-
-    this.broadcastAreaCodes.forEach((areaCode: string) => {
-      const [type, code] = areaCode.split('/')
-
-      switch (type) {
-        case 'c':
-          this.broadcastCountryCodes.add(code)
-          break
-        case 'r':
-          this.broadcastRegionCodes.add(code)
-          break
-        case 's':
-          this.broadcastSubdivisionCodes.add(code)
-          break
-      }
-    })
   }
 
   withChannel(channelsKeyById: Dictionary): this {
@@ -93,76 +69,36 @@ export class Feed {
     return this
   }
 
-  withBroadcastSubdivisions(subdivisionsKeyByCode: Dictionary): this {
-    this.broadcastSubdivisions = this.broadcastSubdivisionCodes.map((code: string) =>
-      subdivisionsKeyByCode.get(code)
-    )
-
-    return this
-  }
-
-  withBroadcastCountries(
+  withBroadcastArea(
+    citiesKeyByCode: Dictionary,
+    subdivisionsKeyByCode: Dictionary,
     countriesKeyByCode: Dictionary,
-    regionsKeyByCode: Dictionary,
-    subdivisionsKeyByCode: Dictionary
+    regionsKeyByCode: Dictionary
   ): this {
-    const broadcastCountries = new Collection()
-
-    if (this.isInternational()) {
-      this.broadcastCountries = broadcastCountries
-      return this
-    }
-
-    this.broadcastCountryCodes.forEach((code: string) => {
-      broadcastCountries.add(countriesKeyByCode.get(code))
-    })
-
-    this.broadcastRegionCodes.forEach((code: string) => {
-      const region: Region = regionsKeyByCode.get(code)
-      if (region) {
-        region.countryCodes.forEach((countryCode: string) => {
-          broadcastCountries.add(countriesKeyByCode.get(countryCode))
-        })
-      }
-    })
-
-    this.broadcastSubdivisionCodes.forEach((code: string) => {
-      const subdivision: Subdivision = subdivisionsKeyByCode.get(code)
-      if (subdivision) {
-        broadcastCountries.add(countriesKeyByCode.get(subdivision.countryCode))
-      }
-    })
-
-    this.broadcastCountries = broadcastCountries.uniq().filter(Boolean)
-
-    return this
-  }
-
-  withBroadcastRegions(regions: Collection): this {
-    if (!this.broadcastCountries) return this
-    const countriesCodes = this.broadcastCountries.map((country: Country) => country.code)
-
-    this.broadcastRegions = regions.filter((region: Region) => {
-      if (region.code === 'INT') return false
-      const intersected = region.countryCodes.intersects(countriesCodes)
-      return intersected.notEmpty()
-    })
+    this.broadcastArea = new BroadcastArea(this.broadcastAreaCodes).withLocations(
+      citiesKeyByCode,
+      subdivisionsKeyByCode,
+      countriesKeyByCode,
+      regionsKeyByCode
+    )
 
     return this
   }
 
   hasBroadcastArea(): boolean {
-    return (
-      this.isInternational() || (!!this.broadcastCountries && this.broadcastCountries.notEmpty())
-    )
+    return !!this.broadcastArea
   }
 
   getBroadcastCountries(): Collection {
-    return this.broadcastCountries || new Collection()
+    if (!this.broadcastArea) return new Collection()
+
+    return this.broadcastArea.getCountries()
   }
 
   getBroadcastRegions(): Collection {
-    return this.broadcastRegions || new Collection()
+    if (!this.broadcastArea) return new Collection()
+
+    return this.broadcastArea.getRegions()
   }
 
   getTimezones(): Collection {
@@ -184,35 +120,22 @@ export class Feed {
     )
   }
 
-  isInternational(): boolean {
-    return this.broadcastAreaCodes.includes('r/INT')
-  }
-
   isBroadcastInSubdivision(subdivision: Subdivision): boolean {
-    if (this.isInternational()) return false
-    if (this.broadcastSubdivisionCodes.includes(subdivision.code)) return true
-    if (
-      this.broadcastSubdivisionCodes.isEmpty() &&
-      subdivision.country &&
-      this.isBroadcastInCountry(subdivision.country)
-    )
-      return true
+    if (!this.broadcastArea) return false
 
-    return false
+    return this.broadcastArea.includesSubdivision(subdivision)
   }
 
   isBroadcastInCountry(country: Country): boolean {
-    if (this.isInternational()) return false
+    if (!this.broadcastArea) return false
 
-    return this.getBroadcastCountries().includes(
-      (_country: Country) => _country.code === country.code
-    )
+    return this.broadcastArea?.includesCountry(country)
   }
 
   isBroadcastInRegion(region: Region): boolean {
-    if (this.isInternational()) return false
+    if (!this.broadcastArea) return false
 
-    return this.getBroadcastRegions().includes((_region: Region) => _region.code === region.code)
+    return this.broadcastArea?.includesRegion(region)
   }
 
   getGuides(): Collection {