Add support for enterprise repos to github plugin (influxdata#6194)

plugins/inputs/github/README.md

@@ -14,6 +14,9 @@ alternative method for collecting repository information.
 
  ## Github API access token. Unauthenticated requests are limited to 60 per hour.
  # access_token = ""
+
+ ## Github API enterprise url. Github Enterprise accounts must specify their base url.
+ # enterprise_base_url = ""
 
  ## Timeout for HTTP requests.
  # http_timeout = "5s"

plugins/inputs/github/github.go

@@ -18,10 +18,11 @@ import (
 
 // GitHub - plugin main structure
 type GitHub struct {
- Repositories []string `toml:"repositories"`
- AccessToken string `toml:"access_token"`
- HTTPTimeout internal.Duration `toml:"http_timeout"`
- githubClient *github.Client
+ Repositories []string `toml:"repositories"`
+ AccessToken string `toml:"access_token"`
+ EnterpriseBaseURL string `toml:"enterprise_base_url"`
+ HTTPTimeout internal.Duration `toml:"http_timeout"`
+ githubClient *github.Client
 
  obfusticatedToken string
 
@@ -36,6 +37,9 @@ const sampleConfig = `
 
  ## Github API access token. Unauthenticated requests are limited to 60 per hour.
  # access_token = ""
+ 
+ ## Github API enterprise url. Github Enterprise accounts must specify their base url.
+ # enterprise_base_url = ""
 
  ## Timeout for HTTP requests.
  # http_timeout = "5s"
@@ -71,9 +75,16 @@ func (g *GitHub) createGitHubClient(ctx context.Context) (*github.Client, error)
 
  g.obfusticatedToken = g.AccessToken[0:4] + "..." + g.AccessToken[len(g.AccessToken)-3:]
 
- return github.NewClient(oauthClient), nil
+ return g.newGithubClient(oauthClient)
  }
 
+ return g.newGithubClient(httpClient)
+}
+
+func (g *GitHub) newGithubClient(httpClient *http.Client) (*github.Client, error) {
+ if g.EnterpriseBaseURL != "" {
+ return github.NewEnterpriseClient(g.EnterpriseBaseURL, "", httpClient)
+ }
  return github.NewClient(httpClient), nil
 }
 

plugins/inputs/github/github_test.go

@@ -1,13 +1,26 @@
 package github
 
 import (
+ "net/http"
  "reflect"
  "testing"
 
  gh "github.com/google/go-github/github"
  "github.com/stretchr/testify/require"
 )
 
+func TestNewGithubClient(t *testing.T) {
+ httpClient := &http.Client{}
+ g := &GitHub{}
+ client, err := g.newGithubClient(httpClient)
+ require.Nil(t, err)
+ require.Contains(t, client.BaseURL.String(), "api.github.com")
+ g.EnterpriseBaseURL = "api.example.com/"
+ enterpriseClient, err := g.newGithubClient(httpClient)
+ require.Nil(t, err)
+ require.Contains(t, enterpriseClient.BaseURL.String(), "api.example.com")
+}
+
 func TestSplitRepositoryNameWithWorkingExample(t *testing.T) {
  var validRepositoryNames = []struct {
  fullName string