diff options
Diffstat (limited to 'go/internal/gitlabnet/client.go')
-rw-r--r-- | go/internal/gitlabnet/client.go | 72 |
1 files changed, 59 insertions, 13 deletions
diff --git a/go/internal/gitlabnet/client.go b/go/internal/gitlabnet/client.go index c2453e5..c0f7f97 100644 --- a/go/internal/gitlabnet/client.go +++ b/go/internal/gitlabnet/client.go @@ -1,9 +1,11 @@ package gitlabnet import ( + "bytes" "encoding/base64" "encoding/json" "fmt" + "io" "net/http" "strings" @@ -15,22 +17,24 @@ const ( secretHeaderName = "Gitlab-Shared-Secret" ) -type GitlabClient interface { - Get(path string) (*http.Response, error) - Post(path string, data interface{}) (*http.Response, error) -} - type ErrorResponse struct { Message string `json:"message"` } -func GetClient(config *config.Config) (GitlabClient, error) { - url := config.GitlabUrl - if strings.HasPrefix(url, UnixSocketProtocol) { - return buildSocketClient(config), nil +type GitlabClient struct { + httpClient *http.Client + config *config.Config + host string +} + +func GetClient(config *config.Config) (*GitlabClient, error) { + client := config.GetHttpClient() + + if client == nil { + return nil, fmt.Errorf("Unsupported protocol") } - return nil, fmt.Errorf("Unsupported protocol") + return &GitlabClient{httpClient: client.HttpClient, config: config, host: client.Host}, nil } func normalizePath(path string) string { @@ -44,6 +48,27 @@ func normalizePath(path string) string { return path } +func newRequest(method, host, path string, data interface{}) (*http.Request, error) { + path = normalizePath(path) + + var jsonReader io.Reader + if data != nil { + jsonData, err := json.Marshal(data) + if err != nil { + return nil, err + } + + jsonReader = bytes.NewReader(jsonData) + } + + request, err := http.NewRequest(method, host+path, jsonReader) + if err != nil { + return nil, err + } + + return request, nil +} + func parseError(resp *http.Response) error { if resp.StatusCode >= 200 && resp.StatusCode <= 299 { return nil @@ -59,11 +84,32 @@ func parseError(resp *http.Response) error { } -func doRequest(client *http.Client, config *config.Config, request *http.Request) (*http.Response, error) { - encodedSecret := base64.StdEncoding.EncodeToString([]byte(config.Secret)) +func (c *GitlabClient) Get(path string) (*http.Response, error) { + return c.doRequest("GET", path, nil) +} + +func (c *GitlabClient) Post(path string, data interface{}) (*http.Response, error) { + return c.doRequest("POST", path, data) +} + +func (c *GitlabClient) doRequest(method, path string, data interface{}) (*http.Response, error) { + request, err := newRequest(method, c.host, path, data) + if err != nil { + return nil, err + } + + user, password := c.config.HttpSettings.User, c.config.HttpSettings.Password + if user != "" && password != "" { + request.SetBasicAuth(user, password) + } + + encodedSecret := base64.StdEncoding.EncodeToString([]byte(c.config.Secret)) request.Header.Set(secretHeaderName, encodedSecret) - response, err := client.Do(request) + request.Header.Add("Content-Type", "application/json") + request.Close = true + + response, err := c.httpClient.Do(request) if err != nil { return nil, fmt.Errorf("Internal API unreachable") } |