client.go

package octokit

import (
	"github.com/lostisland/go-sawyer"
	"github.com/lostisland/go-sawyer/hypermedia"
	"net/http"
	"net/url"
)

func NewClient(authMethod AuthMethod) *Client {
	return NewClientWith(gitHubAPIURL, nil, authMethod)
}

func NewClientWith(baseURL string, httpClient *http.Client, authMethod AuthMethod) *Client {
	client, _ := sawyer.NewFromString(baseURL, httpClient)
	return &Client{sawyerClient: client, UserAgent: userAgent, AuthMethod: authMethod}
}

type Client struct {
	UserAgent    string
	AuthMethod   AuthMethod
	sawyerClient *sawyer.Client
	rootRels     hypermedia.Relations
}

func (c *Client) NewRequest(urlStr string) (req *Request, err error) {
	sawyerReq, err := c.sawyerClient.NewRequest(urlStr)
	if err != nil {
		return
	}

	sawyerReq.Header.Add("Accept", defaultMediaType)
	sawyerReq.Header.Add("User-Agent", c.UserAgent)
	if c.AuthMethod != nil {
		sawyerReq.Header.Add("Authorization", c.AuthMethod.String())
	}

	if basicAuth, ok := c.AuthMethod.(BasicAuth); ok && basicAuth.OneTimePassword != "" {
		sawyerReq.Header.Add("X-GitHub-OTP", basicAuth.OneTimePassword)
	}

	req = &Request{sawyerReq: sawyerReq}
	return
}

func (c *Client) head(url *url.URL, output interface{}) (result *Result) {
	req, err := c.NewRequest(url.String())
	if err != nil {
		result = newResult(nil, err)
		return
	}

	resp, err := req.Head(output)
	result = newResult(resp, err)

	return
}

func (c *Client) get(url *url.URL, output interface{}) (result *Result) {
	req, err := c.NewRequest(url.String())
	if err != nil {
		result = newResult(nil, err)
		return
	}

	resp, err := req.Get(output)
	result = newResult(resp, err)

	return
}

func (c *Client) post(url *url.URL, input interface{}, output interface{}) (result *Result) {
	req, err := c.NewRequest(url.String())
	if err != nil {
		result = newResult(nil, err)
		return
	}

	resp, err := req.Post(input, output)
	result = newResult(resp, err)

	return
}

func (c *Client) put(url *url.URL, input interface{}, output interface{}) (result *Result) {
	req, err := c.NewRequest(url.String())
	if err != nil {
		result = newResult(nil, err)
		return
	}

	resp, err := req.Put(input, output)
	result = newResult(resp, err)

	return
}

func (c *Client) delete(url *url.URL, output interface{}) (result *Result) {
	req, err := c.NewRequest(url.String())
	if err != nil {
		result = newResult(nil, err)
		return
	}

	resp, err := req.Delete(output)
	result = newResult(resp, err)

	return
}