Create octokit.Client

octokit/octokit.go

@@ -0,0 +1,83 @@
+package octokit
+
+import (
+	"bytes"
+	"encoding/base64"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"net/http"
+)
+
+const (
+	GitHubHost    string = "github.com"
+	GitHubApiUrl  string = "https://" + GitHubApiHost
+	GitHubApiHost string = "api.github.com"
+	OAuthAppUrl   string = "http://owenou.com/gh"
+)
+
+type Client struct {
+	httpClient *http.Client
+	Login      string
+	Password   string
+	Token      string
+}
+
+func (c *Client) get(path string, extraHeaders map[string]string) ([]byte, error) {
+	return c.request("GET", path, extraHeaders, nil)
+}
+
+func (c *Client) post(path string, extraHeaders map[string]string, content *bytes.Buffer) ([]byte, error) {
+	return c.request("POST", path, extraHeaders, content)
+}
+
+func (c *Client) request(method, path string, extraHeaders map[string]string, content io.Reader) ([]byte, error) {
+	url := fmt.Sprintf("%s/%s", GitHubApiUrl, path)
+	request, err := http.NewRequest(method, url, content)
+	if err != nil {
+		return nil, err
+	}
+
+	c.setDefaultHeaders(request)
+
+	if extraHeaders != nil {
+		for h, v := range extraHeaders {
+			request.Header.Set(h, v)
+		}
+	}
+
+	response, err := c.httpClient.Do(request)
+	if err != nil {
+		return nil, err
+	}
+
+	body, err := ioutil.ReadAll(response.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	return body, nil
+}
+
+func (c *Client) setDefaultHeaders(request *http.Request) {
+	request.Header.Set("Accept", "application/vnd.github.beta+json")
+	if c.Login != "" && c.Password != "" {
+		request.Header.Set("Authorization", fmt.Sprintf("Basic %s", hashAuth(c.Login, c.Password)))
+	}
+	if c.Token != "" {
+		request.Header.Set("Authorization", fmt.Sprintf("token %s", c.Token))
+	}
+}
+
+func hashAuth(u, p string) string {
+	var a = fmt.Sprintf("%s:%s", u, p)
+	return base64.StdEncoding.EncodeToString([]byte(a))
+}
+
+func NewClientWithPassword(login, password string) *Client {
+	return &Client{&http.Client{}, login, password, ""}
+}
+
+func NewClient() *Client {
+	return &Client{&http.Client{}, "", "", ""}
+}

octokit/octokit_test.go

@@ -0,0 +1,14 @@
+package octokit
+
+import (
+	"github.com/bmizerany/assert"
+	"testing"
+)
+
+func TestGet(t *testing.T) {
+	c := NewClient()
+	body, err := c.get("repos/jingweno/gh/commits", nil)
+
+	assert.Equal(t, nil, err)
+	assert.T(t, len(body) > 0)
+}