client.go

package geonet

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

const defaultVersion = "2"
const baseURI = "https://api.geonet.org.nz"

var sfmt = fmt.Sprintf

// Client represents a client of the Geonet API
type Client struct {
	*http.Client
	Version   string
	UserAgent string
}

// NewClient creates a new Geonet API client
func NewClient() *Client {
	return &Client{&http.Client{}, defaultVersion, "Geonet Golang API Client penguinpower/go-geonet"}
}

// Request builds a request object to be used with a net/http client from
// the given method and path
func (c *Client) Request(method, path string) (*http.Request, error) {
	method = strings.ToUpper(method)
	url := fmt.Sprintf("%s/%s", baseURI, strings.TrimLeft(path, "/"))
	req, err := http.NewRequest(method, url, nil)
	req.Header.Set("User-Agent", c.UserAgent)
	req.Header.Set("Accept", "application/vnd.geo+json;version="+c.Version)
	return req, err
}

// Do will execute the given request
func (c *Client) Do(req *http.Request) (*http.Response, error) {
	log.Printf("requesting %s", req.URL)
	return c.Client.Do(req)
}

// Get will perform a GET operation on the given path and unmarshal the
// results into the given pointer interface
func (c *Client) Get(path string, v interface{}) error {
	req, err := c.Request("GET", path)
	if err != nil {
		return err
	}

	res, err := c.Do(req)
	if err != nil {
		return err
	}

	data, err := ioutil.ReadAll(res.Body)
	// log.Println(string(data))
	if err != nil {
		return err
	}

	return json.Unmarshal(data, v)
}