feat: add github enterprise support

Author: christophwitzko

cmd/semantic-release/main.go

@@ -51,6 +51,7 @@ func main() {
 	vFile := flag.Bool("vf", false, "create a .version file")
 	showVersion := flag.Bool("version", false, "outputs the semantic-release version")
 	updateFile := flag.String("update", "", "updates the version of a certain file")
+	gheHost := flag.String("ghe-host", os.Getenv("GITHUB_ENTERPRISE_HOST"), "github enterprise host")
 	flag.Parse()
 
 	if *showVersion {
@@ -72,7 +73,7 @@ func main() {
 		exitIfError(errors.New("slug missing"))
 	}
 
-	repo, err := semrel.NewRepository(context.TODO(), *slug, *token)
+	repo, err := semrel.NewRepository(context.TODO(), *gheHost, *slug, *token)
 	exitIfError(err)
 
 	logger.Println("getting default branch...")

semrel.go

@@ -56,7 +56,7 @@ type Repository struct {
 	Client *github.Client
 }
 
-func NewRepository(ctx context.Context, slug, token string) (*Repository, error) {
+func NewRepository(ctx context.Context, gheHost, slug, token string) (*Repository, error) {
 	if !strings.Contains(slug, "/") {
 		return nil, errors.New("invalid slug")
 	}
@@ -65,9 +65,17 @@ func NewRepository(ctx context.Context, slug, token string) (*Repository, error)
 	repo.Owner = splited[0]
 	repo.Repo = splited[1]
 	repo.Ctx = ctx
-	repo.Client = github.NewClient(oauth2.NewClient(ctx, oauth2.StaticTokenSource(
-		&oauth2.Token{AccessToken: token},
-	)))
+	oauthClient := oauth2.NewClient(ctx, oauth2.StaticTokenSource(&oauth2.Token{AccessToken: token}))
+	if gheHost != "" {
+		gheUrl := fmt.Sprintf("https://%s/", gheHost)
+		rClient, err := github.NewEnterpriseClient(gheUrl, gheUrl, oauthClient)
+		if err != nil {
+			return nil, err
+		}
+		repo.Client = rClient
+	} else {
+		repo.Client = github.NewClient(oauthClient)
+	}
 	return repo, nil
 }
 

semrel_test.go

@@ -14,14 +14,18 @@ import (
 )
 
 func TestNewRepository(t *testing.T) {
-	repo, err := NewRepository(context.TODO(), "", "")
+	repo, err := NewRepository(context.TODO(), "", "", "")
 	if repo != nil || err == nil {
 		t.Fatal("invalid initialization")
 	}
-	repo, err = NewRepository(context.TODO(), "owner/test-repo", "token")
+	repo, err = NewRepository(context.TODO(), "", "owner/test-repo", "token")
 	if repo == nil || err != nil {
 		t.Fatal("invalid initialization")
 	}
+	repo, err = NewRepository(context.TODO(), "github.enterprise", "owner/test-repo", "token")
+	if repo.Client.BaseURL.Host != "github.enterprise" || err != nil {
+		t.Fatal("invalid enterprise initialization")
+	}
 }
 
 func createCommit(sha, message string) *github.RepositoryCommit {
@@ -95,7 +99,7 @@ func githubHandler(w http.ResponseWriter, r *http.Request) {
 }
 
 func getNewTestRepo(t *testing.T) (*Repository, *httptest.Server) {
-	repo, err := NewRepository(context.TODO(), "owner/test-repo", "token")
+	repo, err := NewRepository(context.TODO(), "", "owner/test-repo", "token")
 	if err != nil {
 		t.Fatal(err)
 		return nil, nil