util.go

package dbl

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

type ratelimitResponse struct {
	RetryAfter int `json:"retry-after"`
}

func (c *Client) readBody(res *http.Response) ([]byte, error) {
	defer res.Body.Close()

	if res.StatusCode == 401 {
		return nil, ErrUnauthorizedRequest
	}

	if res.StatusCode != 200 {
		return nil, ErrRequestFailed
	}

	body, err := ioutil.ReadAll(res.Body)

	if err != nil {
		return nil, err
	}

	if res.StatusCode == 429 {
		rr := &ratelimitResponse{}

		err = json.Unmarshal(body, rr)

		if err != nil {
			return nil, err
		}

		c.Lock()
		c.RetryAfter = rr.RetryAfter
		c.Unlock()

		return nil, ErrRemoteRatelimit
	}

	return body, nil
}

func (c *Client) createRequest(method, endpoint string, body io.Reader) (*http.Request, error) {
	req, err := http.NewRequest(method, BaseURL+endpoint, body)

	if err != nil {
		return nil, err
	}

	req.Header.Set("Authorization", c.token)

	return req, nil
}