country.go

package gocountries

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
)

const baseURL = "https://restcountries.eu/rest/v2/%s"

// Country model
type Country struct {
	Name           string
	Capital        string
	AltSpellings   []string
	Relevance      string
	Region         string
	Subregion      string
	Translations   map[string]string
	Population     int32
	LatLng         []float32
	Demonym        string
	Area           float32
	Gini           float32
	Timezones      []string
	Borders        []string
	NativeName     string
	CallingCodes   []string
	TopLevelDomain []string
	Alpha2Code     string
	Alpha3Code     string
	Currencies     []map[string]string
	Languages      []map[string]string
	NumericCode    string
	Flag           string
	RegionalBlocs  []RegionalBloc
	Cioc           string
}

// RegionalBloc model
type RegionalBloc struct {
	Acronym       string
	Name          string
	OtherAcronyms []string
	OtherNames    []string
}

func doRestcountriesCall(apiSuffix string) ([]byte, error) {
	url := fmt.Sprintf(baseURL, apiSuffix)
	res, err := http.Get(url)
	if err != nil {
		return []byte{}, err
	}
	defer res.Body.Close()
	if res.StatusCode != http.StatusOK {
		e := fmt.Errorf("Unexpected API status code %s", res.Status)
		return []byte{}, e
	}
	body, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return []byte{}, err
	}
	return body, nil
}

func doCountriesByName(name string, isFullName bool) ([]Country, error) {
	apiSuffix := "name/%s"

	if isFullName {
		apiSuffix += "?fullText=true"
	}

	resData, err := doRestcountriesCall(fmt.Sprintf(apiSuffix, name))

	if err != nil {
		return nil, err
	}
	var c []Country
	err = json.Unmarshal(resData, &c)
	if err != nil {
		return nil, err
	}
	return c, nil
}

// CountriesByName searches for countries by their name. It can be the native name or partial name
func CountriesByName(name string) ([]Country, error) {
	return doCountriesByName(name, false)
}

// CountriesByFullName searches for countries by their full name
func CountriesByFullName(name string) ([]Country, error) {
	return doCountriesByName(name, true)
}

// CountriesByCapital searches for countries with capital city matching 'name'
func CountriesByCapital(name string) ([]Country, error) {
	resData, err := doRestcountriesCall(fmt.Sprintf("capital/%s", name))

	if err != nil {
		return nil, err
	}
	var c []Country
	err = json.Unmarshal(resData, &c)
	if err != nil {
		return nil, err
	}
	return c, nil
}

// AllCountries retrieves every country
func AllCountries() ([]Country, error) {
	resData, err := doRestcountriesCall(fmt.Sprintf("all"))

	if err != nil {
		return nil, err
	}
	var c []Country
	err = json.Unmarshal(resData, &c)
	if err != nil {
		return nil, err
	}
	return c, nil
}