From 337eb607fc0b68b166b34abbe065be3c453bf3c2 Mon Sep 17 00:00:00 2001
From: Amr Tamimi <amrnt0@gmail.com>
Date: Sun, 3 Jun 2018 18:28:19 +0200
Subject: [PATCH] fixes

---
 _site/content/posts/modules/github.md |  9 +++++++--
 github/github_repo.go                 | 17 ++++++++++++-----
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/_site/content/posts/modules/github.md b/_site/content/posts/modules/github.md
index b5a39bc0..f1339488 100644
--- a/_site/content/posts/modules/github.md
+++ b/_site/content/posts/modules/github.md
@@ -23,10 +23,15 @@ All open pull requests created by you.
 wtf/github/
 ```
 
-## Required ENV Variables
+## Github Required ENV Variables
 
 <span class="caption">Key:</span> `WTF_GITHUB_TOKEN` <br />
-<span class="caption">Action:</span> Your <a href="https://developer.github.com/v3/oauth_authorizations/#create-a-new-authorization">Github API</a> token or <a href="https://developer.github.com/enterprise/2.13/v3/enterprise-admin/">Github Enterprise</a> API token.
+<span class="caption">Action:</span> Your <a href="https://developer.github.com/v3/oauth_authorizations/#create-a-new-authorization">Github API</a> token.
+
+## GitHub Enterprise Required ENV Variables
+
+<span class="caption">Key:</span> `WTF_GITHUB_TOKEN` <br />
+<span class="caption">Action:</span> Your <a href="https://developer.github.com/v3/oauth_authorizations/#create-a-new-authorization">Github API</a> token.
 
 <span class="caption">Key:</span> `WTF_GITHUB_BASE_URL` <br />
 <span class="caption">Action:</span> Your <a href="https://developer.github.com/enterprise/2.13/v3/enterprise-admin/">Github Enterprise</a> API URL.
diff --git a/github/github_repo.go b/github/github_repo.go
index 532273d0..af2db77e 100644
--- a/github/github_repo.go
+++ b/github/github_repo.go
@@ -63,6 +63,16 @@ func (repo *GithubRepo) StarCount() int {
 
 /* -------------------- Unexported Functions -------------------- */
 
+func (repo *GithubRepo) isGitHubEnterprise() bool {
+	if len(repo.baseURL) > 0 {
+		if len(repo.uploadURL) == 0 {
+			repo.uploadURL = repo.baseURL
+		}
+		return true
+	}
+	return false
+}
+
 func (repo *GithubRepo) oauthClient() *http.Client {
 	tokenService := oauth2.StaticTokenSource(
 		&oauth2.Token{AccessToken: repo.apiKey},
@@ -74,11 +84,8 @@ func (repo *GithubRepo) oauthClient() *http.Client {
 func (repo *GithubRepo) githubClient() (*ghb.Client, error) {
 	oauthClient := repo.oauthClient()
 
-	if len(repo.baseURL) > 0 {
-		if len(repo.uploadURL) == 0 {
-			repo.uploadURL = repo.baseURL
-		}
-		return ghb.NewEnterpriseClient(repo.baseURL, repo.baseURL, oauthClient)
+	if repo.isGitHubEnterprise() {
+		return ghb.NewEnterpriseClient(repo.baseURL, repo.uploadURL, oauthClient)
 	}
 
 	return ghb.NewClient(oauthClient), nil