govid.go

package govid

import (
	"encoding/json"

	"github.com/valyala/fasthttp"
)

// The different endpoints of the API
const (
	AllDataEndpoint      = "https://coronavirus-19-api.herokuapp.com/all"
	AllCountriesEndpoint = "https://coronavirus-19-api.herokuapp.com/countries"
	CountryDataEndpoint  = "https://coronavirus-19-api.herokuapp.com/countries/"
)

// Gets data about the global cases, recoveries and deaths
// Returns:
//   (*GlobalData, error)
func GetGlobalData() (*GlobalData, error) {
	// Create the request
	req := fasthttp.AcquireRequest()
	req.SetRequestURI(AllDataEndpoint)

	resp := fasthttp.AcquireResponse()

	// Release the request and response
	defer fasthttp.ReleaseResponse(resp)
	defer fasthttp.ReleaseRequest(req)

	// Do the request
	err := fasthttp.Do(req, resp)

	if err != nil {
		return nil, err
	}

	var body []byte

	body = resp.Body()

	// Deserialize the data
	var data GlobalData
	err = json.Unmarshal(body, &data)

	if err != nil {
		return nil, err
	}

	return &data, nil
}

// Gets data about a given country.
// Params:
//  country string
// Returns:
//  (*CountryData, error)
func GetCountryData(country string) (*CountryData, error) {
	// Create the request
	req := fasthttp.AcquireRequest()
	req.SetRequestURI(CountryDataEndpoint + country)

	resp := fasthttp.AcquireResponse()

	// Release the request and response
	defer fasthttp.ReleaseResponse(resp)
	defer fasthttp.ReleaseRequest(req)

	// Do the request
	err := fasthttp.Do(req, resp)

	if err != nil {
		return nil, err
	}

	var body []byte

	body = resp.Body()

	// Deserialize the data
	var data CountryData
	err = json.Unmarshal(body, &data)

	if err != nil {
		return nil, err
	}

	return &data, nil
}

// Gets a list which contains all the data about every available country
// Returns:
//  ([]CountryData, error)
func GetAllCountriesData() ([]CountryData, error) {
	// Create the request
	req := fasthttp.AcquireRequest()
	req.SetRequestURI(AllCountriesEndpoint)

	resp := fasthttp.AcquireResponse()

	// Release the request and response
	defer fasthttp.ReleaseResponse(resp)
	defer fasthttp.ReleaseRequest(req)

	// Do the request
	err := fasthttp.Do(req, resp)

	if err != nil {
		return nil, err
	}

	var body []byte

	body = resp.Body()

	// Deserialize the data
	var data []CountryData
	err = json.Unmarshal(body, &data)

	if err != nil {
		return nil, err
	}

	return data, nil
}

// The same as GetAllCountriesData, but allows you to explicitly specify a limit on many elements
// to get.
// Params:
//    limit int
// Returns:
//    ([]CountryData, error)
func GetCountriesDataByLimit(limit int) ([]CountryData, error) {
	arr, err := GetAllCountriesData()

	if err != nil {
		return nil, err
	}

	return arr[0:limit], nil
}

// Similar to GetCountriesDataByLimit, but lets you specify a start and end limit for more control.
// Params:
//    endLimit int
//    startLimit int
// Returns:
//   ([]CountryData, error)
func GetCountriesDataByEndLimit(endLimit int, startLimit int) ([]CountryData, error) {
	arr, err := GetAllCountriesData()

	if err != nil {
		return nil, err
	}

	return arr[startLimit:endLimit], nil
}