Authenticate to get existing index.yaml

Author: Steven Barnes

pkg/releaser/releaser.go

@@ -54,6 +54,7 @@ type GitHub interface {
 
 type HttpClient interface {
 	Get(url string) (*http.Response, error)
+	GetWithToken(url string, token string) (*http.Response, error)
 }
 
 type Git interface {
@@ -77,6 +78,13 @@ func (c *DefaultHttpClient) Get(url string) (resp *http.Response, err error) {
 	return http.Get(url)
 }
 
+func (c *DefaultHttpClient) GetWithToken(url string, token string) (resp *http.Response, err error) {
+	req, _ := http.NewRequest("GET", url, nil)
+	req.Header.Add("Authorization", fmt.Sprintf("token %s", token))
+	client := &http.Client{}
+	return client.Do(req)
+}
+
 type Releaser struct {
 	config     *config.Options
 	github     GitHub
@@ -108,10 +116,19 @@ func (r *Releaser) UpdateIndexFile() (bool, error) {
 	}
 
 	var indexFile *repo.IndexFile
+	var resp *http.Response
+	var err error
 
-	resp, err := r.httpClient.Get(fmt.Sprintf("%s/index.yaml", r.config.ChartsRepo))
-	if err != nil {
-		return false, err
+	if r.config.Token != "" {
+		resp, err = r.httpClient.GetWithToken(fmt.Sprintf("%s/index.yaml", r.config.ChartsRepo), r.config.Token)
+		if err != nil {
+			return false, err
+		}
+	} else {
+		resp, err = r.httpClient.Get(fmt.Sprintf("%s/index.yaml", r.config.ChartsRepo))
+		if err != nil {
+			return false, err
+		}
 	}
 
 	defer resp.Body.Close()

pkg/releaser/releaser_test.go

@@ -57,6 +57,16 @@ func (m *MockClient) Get(url string) (*http.Response, error) {
 	}
 }
 
+func (m *MockClient) GetWithToken(url string, token string) (*http.Response, error) {
+	if m.statusCode == http.StatusOK {
+		file, _ := os.Open(m.file)
+		reader := bufio.NewReader(file)
+		return &http.Response{StatusCode: http.StatusOK, Body: ioutil.NopCloser(reader)}, nil
+	} else {
+		return &http.Response{StatusCode: http.StatusNotFound, Body: ioutil.NopCloser(nil)}, nil
+	}
+}
+
 func (f *FakeGitHub) CreateRelease(ctx context.Context, input *github.Release) error {
 	f.Called(ctx, input)
 	f.release = input