aboutsummaryrefslogtreecommitdiffstats
path: root/clients/http/git_upload_pack.go
blob: 96dbdcebc674f1398e72fed8fc6187292208413f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package http

import (
	"io"
	"net/http"
	"strings"

	"gopkg.in/src-d/go-git.v2/clients/common"
	"gopkg.in/src-d/go-git.v2/pktline"
)

type GitUploadPackService struct {
	Client *http.Client

	endpoint common.Endpoint
}

func NewGitUploadPackService() *GitUploadPackService {
	return &GitUploadPackService{
		Client: http.DefaultClient,
	}
}

func (s *GitUploadPackService) Connect(url common.Endpoint) error {
	s.endpoint = url

	return nil
}

func (s *GitUploadPackService) Info() (*common.GitUploadPackInfo, error) {
	res, err := s.doRequest("GET", common.GitUploadPackServiceName, nil)
	if err != nil {
		return nil, err
	}

	defer res.Body.Close()

	dec := pktline.NewDecoder(res.Body)
	return common.NewGitUploadPackInfo(dec)
}

func (s *GitUploadPackService) doRequest(method, service string, content *strings.Reader) (*http.Response, error) {
	var body io.Reader
	if content != nil {
		body = content
	}

	req, err := http.NewRequest(method, s.endpoint.Service(service), body)
	if err != nil {
		return nil, err
	}

	s.applyHeadersToRequest(req, content)

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

	if err := NewHTTPError(res); err != nil {
		return nil, err
	}

	return res, nil
}

func (s *GitUploadPackService) applyHeadersToRequest(req *http.Request, content *strings.Reader) {
	req.Header.Add("User-Agent", "git/1.0")
	req.Header.Add("Host", "github.com")

	if content == nil {
		req.Header.Add("Accept", "*/*")
	} else {
		req.Header.Add("Accept", "application/x-git-upload-pack-result")
		req.Header.Add("Content-Type", "application/x-git-upload-pack-request")
		req.Header.Add("Content-Length", string(content.Len()))
	}
}