Merge pull request #162 from fluxcd/feature/github-enterprise

Add support for GitHub enterprise commit status

internal/notifier/github.go

@@ -20,6 +20,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"net/url"
 	"strings"
 	"time"
 
@@ -39,7 +40,12 @@ func NewGitHub(addr string, token string) (*GitHub, error) {
 		return nil, errors.New("github token cannot be empty")
 	}
 
-	_, id, err := parseGitAddress(addr)
+	host, id, err := parseGitAddress(addr)
+	if err != nil {
+		return nil, err
+	}
+
+	baseUrl, err := url.Parse(host)
 	if err != nil {
 		return nil, err
 	}
@@ -49,10 +55,15 @@ func NewGitHub(addr string, token string) (*GitHub, error) {
 		return nil, fmt.Errorf("invalid repository id %q", id)
 	}
 
-	ctx := context.Background()
 	ts := oauth2.StaticTokenSource(&oauth2.Token{AccessToken: token})
-	tc := oauth2.NewClient(ctx, ts)
+	tc := oauth2.NewClient(context.Background(), ts)
 	client := github.NewClient(tc)
+	if baseUrl.Host != "github.com" {
+		client, err = github.NewEnterpriseClient(host, host, tc)
+		if err != nil {
+			return nil, fmt.Errorf("could not create enterprise GitHub client: %v", err)
+		}
+	}
 
 	return &GitHub{
 		Owner:  comp[0],

internal/notifier/github_test.go

@@ -28,6 +28,15 @@ func TestNewGitHubBasic(t *testing.T) {
 	assert.Nil(t, err)
 	assert.Equal(t, g.Owner, "foo")
 	assert.Equal(t, g.Repo, "bar")
+	assert.Equal(t, g.Client.BaseURL.Host, "api.github.com")
+}
+
+func TestNewEmterpriseGitHubBasic(t *testing.T) {
+	g, err := NewGitHub("https://foobar.com/foo/bar", "foobar")
+	assert.Nil(t, err)
+	assert.Equal(t, g.Owner, "foo")
+	assert.Equal(t, g.Repo, "bar")
+	assert.Equal(t, g.Client.BaseURL.Host, "foobar.com")
 }
 
 func TestNewGitHubInvalidUrl(t *testing.T) {