Update scripts

2025-12-17 02:47:33 -05:00 · 2025-10-08 21:25:22 +03:00
parent 25fa704e14
commit ad2c83e333
73 changed files with 3215 additions and 4784 deletions
--- a/scripts/generators/countriesGenerator.ts
+++ b/scripts/generators/countriesGenerator.ts
@@ -1,68 +1,80 @@
-import { Country, Stream, Playlist } from '../models'
-import { Collection, Storage, File } from '@freearhey/core'
-import { PUBLIC_DIR, EOL } from '../constants'
-import { Generator } from './generator'
-
-type CountriesGeneratorProps = {
-  streams: Collection
-  countries: Collection
-  logFile: File
-}
-
-export class CountriesGenerator implements Generator {
-  streams: Collection
-  countries: Collection
-  storage: Storage
-  logFile: File
-
-  constructor({ streams, countries, logFile }: CountriesGeneratorProps) {
-    this.streams = streams.clone()
-    this.countries = countries
-    this.storage = new Storage(PUBLIC_DIR)
-    this.logFile = logFile
-  }
-
-  async generate(): Promise<void> {
-    const streams = this.streams
-      .orderBy((stream: Stream) => stream.getTitle())
-      .filter((stream: Stream) => stream.isSFW())
-
-    this.countries.forEach(async (country: Country) => {
-      const countryStreams = streams.filter((stream: Stream) =>
-        stream.isBroadcastInCountry(country)
-      )
-      if (countryStreams.isEmpty()) return
-
-      const playlist = new Playlist(countryStreams, { public: true })
-      const filepath = `countries/${country.code.toLowerCase()}.m3u`
-      await this.storage.save(filepath, playlist.toString())
-      this.logFile.append(
-        JSON.stringify({ type: 'country', filepath, count: playlist.streams.count() }) + EOL
-      )
-    })
-
-    const internationalStreams = streams.filter((stream: Stream) => stream.isInternational())
-    const internationalPlaylist = new Playlist(internationalStreams, { public: true })
-    const internationalFilepath = 'countries/int.m3u'
-    await this.storage.save(internationalFilepath, internationalPlaylist.toString())
-    this.logFile.append(
-      JSON.stringify({
-        type: 'country',
-        filepath: internationalFilepath,
-        count: internationalPlaylist.streams.count()
-      }) + EOL
-    )
-
-    const undefinedStreams = streams.filter((stream: Stream) => !stream.hasBroadcastArea())
-    const undefinedPlaylist = new Playlist(undefinedStreams, { public: true })
-    const undefinedFilepath = 'countries/undefined.m3u'
-    await this.storage.save(undefinedFilepath, undefinedPlaylist.toString())
-    this.logFile.append(
-      JSON.stringify({
-        type: 'country',
-        filepath: undefinedFilepath,
-        count: undefinedPlaylist.streams.count()
-      }) + EOL
-    )
-  }
-}
+import { Storage, File } from '@freearhey/storage-js'
+import { PUBLIC_DIR, EOL } from '../constants'
+import { Stream, Playlist } from '../models'
+import { Collection } from '@freearhey/core'
+import { Generator } from './generator'
+import * as sdk from '@iptv-org/sdk'
+
+type CountriesGeneratorProps = {
+  streams: Collection<Stream>
+  countries: Collection<sdk.Models.Country>
+  logFile: File
+}
+
+export class CountriesGenerator implements Generator {
+  streams: Collection<Stream>
+  countries: Collection<sdk.Models.Country>
+  storage: Storage
+  logFile: File
+
+  constructor({ streams, countries, logFile }: CountriesGeneratorProps) {
+    this.streams = streams.clone()
+    this.countries = countries
+    this.storage = new Storage(PUBLIC_DIR)
+    this.logFile = logFile
+  }
+
+  async generate(): Promise<void> {
+    const streams = this.streams
+      .sortBy((stream: Stream) => stream.title)
+      .filter((stream: Stream) => stream.isSFW())
+
+    const streamsGroupedByCountryCode = {}
+    streams.forEach((stream: Stream) => {
+      stream.getBroadcastCountries().forEach((country: sdk.Models.Country) => {
+        if (streamsGroupedByCountryCode[country.code]) {
+          streamsGroupedByCountryCode[country.code].add(stream)
+        } else {
+          streamsGroupedByCountryCode[country.code] = new Collection<Stream>([stream])
+        }
+      })
+    })
+
+    for (const countryCode in streamsGroupedByCountryCode) {
+      const countryStreams = streamsGroupedByCountryCode[countryCode]
+
+      const playlist = new Playlist(countryStreams, { public: true })
+      const filepath = `countries/${countryCode.toLowerCase()}.m3u`
+      await this.storage.save(filepath, playlist.toString())
+      this.logFile.append(
+        JSON.stringify({ type: 'country', filepath, count: playlist.streams.count() }) + EOL
+      )
+    }
+
+    const internationalStreams = streams.filter((stream: Stream) => stream.isInternational())
+    const internationalPlaylist = new Playlist(internationalStreams, { public: true })
+    const internationalFilepath = 'countries/int.m3u'
+    await this.storage.save(internationalFilepath, internationalPlaylist.toString())
+    this.logFile.append(
+      JSON.stringify({
+        type: 'country',
+        filepath: internationalFilepath,
+        count: internationalPlaylist.streams.count()
+      }) + EOL
+    )
+
+    const undefinedStreams = streams.filter((stream: Stream) =>
+      stream.getBroadcastAreaCodes().isEmpty()
+    )
+    const undefinedPlaylist = new Playlist(undefinedStreams, { public: true })
+    const undefinedFilepath = 'countries/undefined.m3u'
+    await this.storage.save(undefinedFilepath, undefinedPlaylist.toString())
+    this.logFile.append(
+      JSON.stringify({
+        type: 'country',
+        filepath: undefinedFilepath,
+        count: undefinedPlaylist.streams.count()
+      }) + EOL
+    )
+  }
+}