Add function for PullRequestService to download raw content of pull request

Author: huydx

.gitignore

@@ -1 +1,2 @@
 *.test
+.idea

github/github.go

@@ -52,6 +52,11 @@ const (
 	// https://developer.github.com/changes/2014-12-09-new-attributes-for-stars-api/
 	mediaTypeStarringPreview = "application/vnd.github.v3.star+json"
 
+	//https://developer.github.com/v3/media/
+	mediaTypeDiff  = "application/vnd.github.v3.diff"
+	mediaTypePatch = "application/vnd.github.v3.patch"
+	mediaTypeSha   = "application/vnd.github.v3.sha"
+
 	// https://developer.github.com/changes/2015-11-11-protected-branches-api/
 	mediaTypeProtectedBranchesPreview = "application/vnd.github.loki-preview+json"
 
@@ -154,6 +159,17 @@ type UploadOptions struct {
 	Name string `url:"name,omitempty"`
 }
 
+type MediaRawType int
+
+const (
+	Diff MediaRawType = 1 + iota
+	Patch
+)
+
+type MediaRawTypeOption struct {
+	Type MediaRawType
+}
+
 // addOptions adds the parameters in opt as URL query parameters to s.  opt
 // must be a struct whose fields may contain "url" tags.
 func addOptions(s string, opt interface{}) (string, error) {

github/pulls.go

@@ -8,6 +8,7 @@ package github
 import (
 	"fmt"
 	"time"
+	"bytes"
 )
 
 // PullRequestsService handles communication with the pull request related
@@ -134,6 +135,29 @@ func (s *PullRequestsService) Get(owner string, repo string, number int) (*PullR
 	return pull, resp, err
 }
 
+// Get pull request raw format (diff or patch)
+func (s *PullRequestsService) GetRaw(owner string, repo string, number int, opt *MediaRawTypeOption) (string, *Response, error) {
+	u := fmt.Sprintf("repos/%v/%v/pulls/%d", owner, repo, number)
+	req, err := s.client.NewRequest("GET", u, nil)
+	if err != nil {
+		return "", nil, err
+	}
+
+	if opt.Type == Diff {
+		req.Header.Set("Accept", mediaTypeDiff)
+	}  else if opt.Type == Patch {
+		req.Header.Set("Accept", mediaTypePatch)
+	}
+
+	ret := new(bytes.Buffer)
+	resp, err := s.client.Do(req, ret)
+	if err != nil {
+		return "", resp, err
+	}
+
+	return string(ret.Bytes()), resp, err
+}
+
 // NewPullRequest represents a new pull request to be created.
 type NewPullRequest struct {
 	Title *string `json:"title,omitempty"`

github/pulls_test.go

@@ -67,6 +67,50 @@ func TestPullRequestsService_Get(t *testing.T) {
 	}
 }
 
+func TestPullRequestService_GetRawPatch(t *testing.T) {
+	setup()
+	defer teardown()
+	var rawStr = "@@patch content"
+
+	mux.HandleFunc("/repos/o/r/pulls/1", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "GET")
+		testHeader(t, r, "Accept", mediaTypePatch)
+		fmt.Fprint(w, rawStr)
+	})
+
+	ret, _, err := client.PullRequests.GetRaw("o", "r", 1, &MediaRawTypeOption{Patch})
+	if err != nil {
+		t.Errorf("PullRequests.GetRaw return error: %v", err)
+	}
+
+	if ret != rawStr {
+		t.Errorf("PullRequests.GetRaw returned %s want %s", ret, rawStr)
+	}
+}
+
+
+func TestPullRequestService_GetRawDiff(t *testing.T) {
+	setup()
+	defer teardown()
+	var rawStr = "@@diff content"
+
+	mux.HandleFunc("/repos/o/r/pulls/1", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "GET")
+		testHeader(t, r, "Accept", mediaTypeDiff)
+		fmt.Fprint(w, rawStr)
+	})
+
+	ret, _, err := client.PullRequests.GetRaw("o", "r", 1, &MediaRawTypeOption{Diff})
+	if err != nil {
+		t.Errorf("PullRequests.GetRaw return error: %v", err)
+	}
+
+	if ret != rawStr {
+		t.Errorf("PullRequests.GetRaw returned %s want %s", ret, rawStr)
+	}
+}
+
+
 func TestPullRequestsService_Get_headAndBase(t *testing.T) {
 	setup()
 	defer teardown()