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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
package http
import (
"fmt"
"io"
"net/http"
"strings"
"gopkg.in/src-d/go-git.v2/clients/common"
"gopkg.in/src-d/go-git.v2/core"
"gopkg.in/src-d/go-git.v2/formats/pktline"
)
type GitUploadPackService struct {
Client *http.Client
endpoint common.Endpoint
auth HTTPAuthMethod
}
func NewGitUploadPackService() *GitUploadPackService {
return &GitUploadPackService{
Client: http.DefaultClient,
}
}
func (s *GitUploadPackService) Connect(url common.Endpoint) error {
s.endpoint = url
return nil
}
func (s *GitUploadPackService) ConnectWithAuth(url common.Endpoint, auth common.AuthMethod) error {
httpAuth, ok := auth.(HTTPAuthMethod)
if !ok {
return InvalidAuthMethodErr
}
s.endpoint = url
s.auth = httpAuth
return nil
}
func (s *GitUploadPackService) Info() (*common.GitUploadPackInfo, error) {
url := fmt.Sprintf("%s/info/refs?service=%s", s.endpoint, common.GitUploadPackServiceName)
res, err := s.doRequest("GET", url, nil)
if err != nil {
return nil, err
}
defer res.Body.Close()
i := common.NewGitUploadPackInfo()
return i, i.Decode(pktline.NewDecoder(res.Body))
}
func (s *GitUploadPackService) Fetch(r *common.GitUploadPackRequest) (io.ReadCloser, error) {
url := fmt.Sprintf("%s/%s", s.endpoint, common.GitUploadPackServiceName)
res, err := s.doRequest("POST", url, r.Reader())
if err != nil {
return nil, err
}
h := make([]byte, 8)
if _, err := res.Body.Read(h); err != nil {
return nil, core.NewUnexpectedError(err)
}
return res.Body, nil
}
func (s *GitUploadPackService) doRequest(method, url string, content *strings.Reader) (*http.Response, error) {
var body io.Reader
if content != nil {
body = content
}
req, err := http.NewRequest(method, url, body)
if err != nil {
return nil, core.NewPermanentError(err)
}
s.applyHeadersToRequest(req, content)
s.applyAuthToRequest(req)
res, err := s.Client.Do(req)
if err != nil {
return nil, core.NewUnexpectedError(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()))
}
}
func (s *GitUploadPackService) applyAuthToRequest(req *http.Request) {
if s.auth == nil {
return
}
s.auth.setAuth(req)
}
|
