mirror of https://github.com/iptv-org/iptv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
2.6 KiB
TypeScript
96 lines
2.6 KiB
TypeScript
import { Collection, Dictionary } from '@freearhey/core'
|
|
import { Region, Language, Subdivision } from '.'
|
|
import type { CountryData, CountrySerializedData } from '../types/country'
|
|
import { SubdivisionSerializedData } from '../types/subdivision'
|
|
import { RegionSerializedData } from '../types/region'
|
|
|
|
export class Country {
|
|
code: string
|
|
name: string
|
|
flag: string
|
|
languageCode: string
|
|
language?: Language
|
|
subdivisions?: Collection
|
|
regions?: Collection
|
|
cities?: Collection
|
|
|
|
constructor(data?: CountryData) {
|
|
if (!data) return
|
|
|
|
this.code = data.code
|
|
this.name = data.name
|
|
this.flag = data.flag
|
|
this.languageCode = data.lang
|
|
}
|
|
|
|
withSubdivisions(subdivisionsGroupedByCountryCode: Dictionary): this {
|
|
this.subdivisions = new Collection(subdivisionsGroupedByCountryCode.get(this.code))
|
|
|
|
return this
|
|
}
|
|
|
|
withRegions(regions: Collection): this {
|
|
this.regions = regions.filter((region: Region) => region.includesCountryCode(this.code))
|
|
|
|
return this
|
|
}
|
|
|
|
withCities(citiesGroupedByCountryCode: Dictionary): this {
|
|
this.cities = new Collection(citiesGroupedByCountryCode.get(this.code))
|
|
|
|
return this
|
|
}
|
|
|
|
withLanguage(languagesKeyByCode: Dictionary): this {
|
|
this.language = languagesKeyByCode.get(this.languageCode)
|
|
|
|
return this
|
|
}
|
|
|
|
getLanguage(): Language | undefined {
|
|
return this.language
|
|
}
|
|
|
|
getRegions(): Collection {
|
|
return this.regions || new Collection()
|
|
}
|
|
|
|
getSubdivisions(): Collection {
|
|
return this.subdivisions || new Collection()
|
|
}
|
|
|
|
getCities(): Collection {
|
|
return this.cities || new Collection()
|
|
}
|
|
|
|
serialize(): CountrySerializedData {
|
|
return {
|
|
code: this.code,
|
|
name: this.name,
|
|
flag: this.flag,
|
|
languageCode: this.languageCode,
|
|
language: this.language ? this.language.serialize() : null,
|
|
subdivisions: this.subdivisions
|
|
? this.subdivisions.map((subdivision: Subdivision) => subdivision.serialize()).all()
|
|
: [],
|
|
regions: this.regions ? this.regions.map((region: Region) => region.serialize()).all() : []
|
|
}
|
|
}
|
|
|
|
deserialize(data: CountrySerializedData): this {
|
|
this.code = data.code
|
|
this.name = data.name
|
|
this.flag = data.flag
|
|
this.languageCode = data.languageCode
|
|
this.language = data.language ? new Language().deserialize(data.language) : undefined
|
|
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
|
|
}
|
|
}
|