Swiftgram/submodules/CountrySelectionUI/Sources/CountryList.swift

import Foundation
import AppBundle
import TelegramStringFormatting

public func emojiFlagForISOCountryCode(_ countryCode: String) -> String {
    if countryCode.count != 2 {
        return ""
    }
    
    if countryCode == "FT" {
        return "🏴‍☠️"
    } else if countryCode == "XX" {
        return "🏳️"
    } else if countryCode == "XG" {
        return "🛰️"
    } else if countryCode == "XV" {
        return "🌍"
    } else if countryCode == "TS" {
        return "🏳️"
    }
    
    if ["YL"].contains(countryCode) {
        return ""
    }
    
    return flagEmoji(countryCode: countryCode)
}

private func loadCountriesInfo() -> [(Int, String, String)] {
    guard let filePath = getAppBundle().path(forResource: "PhoneCountries", ofType: "txt") else {
        return []
    }
    guard let stringData = try? Data(contentsOf: URL(fileURLWithPath: filePath)) else {
        return []
    }
    guard let data = String(data: stringData, encoding: .utf8) else {
        return []
    }
    
    let delimiter = ";"
    let endOfLine1 = "\r\n"
    let endOfLine2 = "\n"

    var array: [(Int, String, String)] = []
    
    var currentLocation = data.startIndex
    while true {
        guard let codeRange = data.range(of: delimiter, options: [], range: currentLocation ..< data.endIndex, locale: nil) else {
            break
        }
        
        guard let countryCode = Int(data[currentLocation ..< codeRange.lowerBound]) else {
            break
        }
        
        guard let idRange = data.range(of: delimiter, options: [], range: codeRange.upperBound ..< data.endIndex) else {
            break
        }
        
        let countryId = String(data[codeRange.upperBound ..< idRange.lowerBound])
        
        guard let patternRange = data.range(of: delimiter, options: [], range: idRange.upperBound ..< data.endIndex) else {
            break
        }
                        
        let countryName: String
        let nameRange1 = data.range(of: endOfLine1, options: [], range: patternRange.upperBound ..< data.endIndex)
        let nameRange2 = data.range(of: endOfLine2, options: [], range: patternRange.upperBound ..< data.endIndex)
        var nameRange: Range<String.Index>?
        if let nameRange1 = nameRange1, let nameRange2 = nameRange2 {
            if nameRange1.lowerBound < nameRange2.lowerBound {
                nameRange = nameRange1
            } else {
                nameRange = nameRange2
            }
        } else {
            nameRange = nameRange1 ?? nameRange2
        }
        if let nameRange = nameRange {
            countryName = String(data[patternRange.upperBound ..< nameRange.lowerBound])
            currentLocation = nameRange.upperBound
        } else {
            countryName = String(data[patternRange.upperBound ..< data.index(data.endIndex, offsetBy: -1)])
        }
        
        array.append((countryCode, countryId, countryName))
        
        if nameRange == nil {
            break
        }
    }
    return array
}

let phoneCountriesInfo = loadCountriesInfo()

public let countryCodeToIdAndName: [Int: (String, String)] = {
    var dict: [Int: (String, String)] = [:]
    for (code, id, name) in phoneCountriesInfo {
        if dict[code] == nil {
            dict[code] = (id, name)
        }
    }
    return dict
}()

public struct CountryCodeAndId: Hashable {
    let code: Int
    let id: String
    
    public init(code: Int, id: String) {
        self.code = code
        self.id = id
    }
}

public let countryCodeAndIdToName: [CountryCodeAndId: String] = {
    var dict: [CountryCodeAndId: String] = [:]
    for (code, id, name) in phoneCountriesInfo {
        dict[CountryCodeAndId(code: code, id: id)] = name
    }
    return dict
}()