iptv-org.github.io/src/models/broadcastArea.ts

import type { BroadcastAreaData, BroadcastAreaSerializedData } from '~/types/broadcastArea'
import type { SubdivisionSerializedData } from '~/types/subdivision'
import type { CountrySerializedData } from '~/types/country'
import type { RegionSerializedData } from '~/types/region'
import { type Dictionary, Collection } from '@freearhey/core/browser'
import { Region, Country, Subdivision } from './'

export class BroadcastArea {
  code: string
  name?: string
  countries?: Collection
  subdivisions?: Collection
  regions?: Collection

  constructor(data?: BroadcastAreaData) {
    if (!data) return

    this.code = data.code
  }

  withName(
    countriesKeyByCode: Dictionary,
    subdivisionsKeyByCode: Dictionary,
    regionsKeyByCode: Dictionary
  ): this {
    const [type, code] = this.code.split('/')

    switch (type) {
      case 's': {
        const subdivision: Subdivision = subdivisionsKeyByCode.get(code)
        if (subdivision) this.name = subdivision.name
        break
      }
      case 'c': {
        const country: Country = countriesKeyByCode.get(code)
        if (country) this.name = country.name
        break
      }
      case 'r': {
        const region: Region = regionsKeyByCode.get(code)
        if (region) this.name = region.name
        break
      }
    }

    return this
  }

  withLocations(
    countriesKeyByCode: Dictionary,
    subdivisionsKeyByCode: Dictionary,
    regionsKeyByCode: Dictionary,
    regions: Collection
  ): this {
    const [type, code] = this.code.split('/')

    let _countries = new Collection()
    let _regions = new Collection()
    let _subdivisions = new Collection()

    regions = regions.filter((region: Region) => region.code !== 'INT')

    switch (type) {
      case 's': {
        const subdivision: Subdivision = subdivisionsKeyByCode.get(code)
        if (!subdivision) break
        _subdivisions.add(subdivision)
        const country: Country = countriesKeyByCode.get(subdivision.countryCode)
        if (!country) break
        _countries.add(country)
        const countryRegions = regions.filter((region: Region) =>
          region.countryCodes.includes(country.code)
        )
        countryRegions.forEach((region: Region) => {
          _regions.add(region)
        })
        break
      }
      case 'c': {
        const country = countriesKeyByCode.get(code)
        if (!country) break
        _countries.add(country)
        const countryRegions = regions.filter((region: Region) =>
          region.countryCodes.includes(country.code)
        )
        countryRegions.forEach((region: Region) => {
          _regions.add(region)
        })
        break
      }
      case 'r': {
        const region: Region = regionsKeyByCode.get(code)
        if (!region) break
        _regions.add(region)
        break
      }
    }

    this.countries = _countries.uniqBy((country: Country) => country.code)
    this.regions = _regions.uniqBy((region: Region) => region.code)
    this.subdivisions = _subdivisions.uniqBy((subdivision: Subdivision) => subdivision.code)

    return this
  }

  getName(): string {
    return this.name || ''
  }

  getCountries(): Collection {
    if (!this.countries) return new Collection()

    return this.countries
  }

  getRegions(): Collection {
    if (!this.regions) return new Collection()

    return this.regions
  }

  getSubdivisions(): Collection {
    if (!this.subdivisions) return new Collection()

    return this.subdivisions
  }

  getLocationCodes(): Collection {
    let locationCodes = new Collection()

    this.getCountries().forEach((country: Country) => {
      locationCodes.add(country.code)
    })

    this.getRegions().forEach((region: Region) => {
      locationCodes.add(region.code)
    })

    this.getSubdivisions().forEach((subdivision: Subdivision) => {
      locationCodes.add(subdivision.code)
    })

    return locationCodes
  }

  getLocationNames(): Collection {
    let locationNames = new Collection()

    this.getCountries().forEach((country: Country) => {
      locationNames.add(country.name)
    })

    this.getRegions().forEach((region: Region) => {
      locationNames.add(region.name)
    })

    this.getSubdivisions().forEach((subdivision: Subdivision) => {
      locationNames.add(subdivision.name)
    })

    return locationNames
  }

  serialize(): BroadcastAreaSerializedData {
    return {
      code: this.code,
      name: this.getName(),
      countries: this.getCountries()
        .map((country: Country) => country.serialize())
        .all(),
      subdivisions: this.getSubdivisions()
        .map((subdivision: Subdivision) => subdivision.serialize())
        .all(),
      regions: this.getRegions()
        .map((region: Region) => region.serialize())
        .all()
    }
  }

  deserialize(data: BroadcastAreaSerializedData): this {
    this.code = data.code
    this.name = data.name
    this.countries = new Collection(data.countries).map((data: CountrySerializedData) =>
      new Country().deserialize(data)
    )
    this.subdivisions = new Collection(data.subdivisions).map((data: SubdivisionSerializedData) =>
      new Subdivision().deserialize(data)
    )
    this.regions = new Collection(data.regions).map((data: RegionSerializedData) =>
      new Region().deserialize(data)
    )

    return this
  }
}