Update scripts

scripts/models/country.ts

@@ -1,20 +1,58 @@
-type CountryProps = {
+import { Collection, Dictionary } from '@freearhey/core'
+import { Region, Language } from '.'
+
+type CountryData = {
   code: string
   name: string
-  languages: string[]
+  lang: string
   flag: string
 }
 
 export class Country {
   code: string
   name: string
-  languages: string[]
   flag: string
+  languageCode: string
+  language?: Language
+  subdivisions?: Collection
+  regions?: Collection
 
-  constructor({ code, name, languages, flag }: CountryProps) {
-    this.code = code
-    this.name = name
-    this.languages = languages
-    this.flag = flag
+  constructor(data: CountryData) {
+    this.code = data.code
+    this.name = data.name
+    this.flag = data.flag
+    this.languageCode = data.lang
+  }
+
+  withSubdivisions(subdivisionsGroupedByCountryCode: Dictionary): this {
+    this.subdivisions = subdivisionsGroupedByCountryCode.get(this.code) || new Collection()
+
+    return this
+  }
+
+  withRegions(regions: Collection): this {
+    this.regions = regions.filter(
+      (region: Region) => region.code !== 'INT' && region.includesCountryCode(this.code)
+    )
+
+    return this
+  }
+
+  withLanguage(languagesGroupedByCode: Dictionary): this {
+    this.language = languagesGroupedByCode.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()
   }
 }