summaryrefslogtreecommitdiff
path: root/go/internal/gitlabnet/client.go
diff options
context:
space:
mode:
Diffstat (limited to 'go/internal/gitlabnet/client.go')
-rw-r--r--go/internal/gitlabnet/client.go72
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")
}