hanayo/vendor/github.com/pariz/gountries/query.go

143 lines
2.9 KiB
Go
Raw Normal View History

2019-02-23 13:29:15 +00:00
package gountries
import (
"strings"
)
// Query holds a reference to the QueryHolder struct
var queryInited = false
var queryInstance *Query
// Query contains queries for countries, cities, etc.
type Query struct {
Countries map[string]Country
Subdivisions map[string][]SubDivision
NameToAlpha2 map[string]string
Alpha3ToAlpha2 map[string]string
}
// FindCountryByName fincs a country by given name
func (q *Query) FindCountryByName(name string) (result Country, err error) {
lowerName := strings.ToLower(name)
alpha2, exists := q.NameToAlpha2[lowerName]
if !exists {
return Country{}, makeError("Could not find country with name", name)
}
return q.Countries[alpha2], nil
}
// FindCountryByAlpha fincs a country by given code
func (q *Query) FindCountryByAlpha(code string) (result Country, err error) {
codeU := strings.ToUpper(code)
switch {
case len(code) == 2:
country, exists := q.Countries[codeU]
if !exists {
return Country{}, makeError("Could not find country with code %s", code)
}
return country, nil
case len(code) == 3:
alpha2, exists := q.Alpha3ToAlpha2[codeU]
if !exists {
return Country{}, makeError("Could not find country with code", code)
}
return q.Countries[alpha2], nil
default:
return Country{}, makeError("Invalid code format", code)
}
}
// FindAllCountries returns a list of all countries
func (q *Query) FindAllCountries() (countries map[string]Country) {
return q.Countries
}
// FindCountries finds a Country based on the given struct data
func (q Query) FindCountries(c Country) (countries []Country) {
for _, country := range q.Countries {
// Name
//
if c.Name.Common != "" && strings.ToLower(c.Name.Common) == strings.ToLower(country.Name.Common) {
continue
}
// Alpha
//
if c.Alpha2 != "" && c.Alpha2 != country.Alpha2 {
continue
}
if c.Alpha3 != "" && c.Alpha3 != country.Alpha3 {
continue
}
// Geo
//
if c.Geo.Continent != "" && strings.ToLower(c.Geo.Continent) != strings.ToLower(country.Geo.Continent) {
continue
}
if c.Geo.Region != "" && strings.ToLower(c.Geo.Region) != strings.ToLower(country.Geo.Region) {
continue
}
if c.Geo.SubRegion != "" && strings.ToLower(c.Geo.SubRegion) != strings.ToLower(country.Geo.SubRegion) {
continue
}
// Misc
//
if c.InternationalPrefix != "" && strings.ToLower(c.InternationalPrefix) != strings.ToLower(country.InternationalPrefix) {
continue
}
// Bordering countries
//
allMatch := false
if len(c.BorderingCountries()) > 0 {
for _, c1 := range c.BorderingCountries() {
match := false
for _, c2 := range country.BorderingCountries() {
match = c1.Alpha2 == c2.Alpha2
if match == true {
break
}
}
if match == true {
allMatch = true
} else {
allMatch = false
break
}
}
if allMatch == false {
continue
}
}
// Append if all matches
//
countries = append(countries, country)
}
return
}