From 3480f0322bd9497605b6db760641884491838ba9 Mon Sep 17 00:00:00 2001
From: Leonardo Ramos dos Santos <leonardo.ramosantos@yahoo.com.br>
Date: Thu, 20 Nov 2025 15:15:27 +0000
Subject: [PATCH] Allow usage with GithubEnterprise

---
 config/config.go      |  5 ++++
 config/config_test.go | 53 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 58 insertions(+)

diff --git a/config/config.go b/config/config.go
index 7fa5cb0..b3b296e 100644
--- a/config/config.go
+++ b/config/config.go
@@ -101,6 +101,11 @@ func (c *Config) GetClient() (*github.Client, error) {
 
 	client := github.NewClient(paginator)
 
+	if c.ApiUrl != nil && c.ApiUrl.String() != "https://api.github.com" {
+		// Don't need to validate error as it's checked in envconfig
+		client, _ = client.WithEnterpriseURLs(c.ApiUrl.String(), c.ApiUrl.String())
+	}
+
 	if c.GithubToken != "" {
 		client = client.WithAuthToken(c.GithubToken)
 	}
diff --git a/config/config_test.go b/config/config_test.go
index d77f346..166937e 100644
--- a/config/config_test.go
+++ b/config/config_test.go
@@ -142,3 +142,56 @@ func TestConfig(t *testing.T) {
 		})
 	}
 }
+
+func TestGetClient(t *testing.T) {
+	tests := []struct {
+		name         string
+		envVars      map[string]string
+		expectedHost string
+		expectedPath string
+		expectedErr  error
+	}{
+		{
+			name:         "default config",
+			expectedHost: "api.github.com",
+			expectedPath: "/",
+			expectedErr:  nil,
+		},
+		{
+			name: "non-default config",
+			envVars: map[string]string{
+				"METRICS_PATH":                "/otherendpoint",
+				"LISTEN_PORT":                 "1111",
+				"LOG_LEVEL":                   "DEBUG",
+				"API_URL":                     "https://example.com",
+				"REPOS":                       "repo1, repo2",
+				"ORGS":                        "org1,org2 ",
+				"USERS":                       " user1, user2 ",
+				"GITHUB_RESULTS_PER_PAGE":     "50",
+				"GITHUB_TOKEN":                "token",
+				"GITHUB_RATE_LIMIT_ENABLED":   "false",
+				"FETCH_REPO_RELEASES_ENABLED": "false",
+				"FETCH_ORGS_CONCURRENCY":      "2",
+				"FETCH_ORG_REPOS_CONCURRENCY": "3",
+				"FETCH_REPOS_CONCURRENCY":     "4",
+				"FETCH_USERS_CONCURRENCY":     "5",
+			},
+			expectedHost: "example.com",
+			expectedPath: "/api/v3/",
+			expectedErr:  nil,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			for k, v := range tt.envVars {
+				t.Setenv(k, v)
+			}
+
+			cfg, _ := Init()
+			client, _ := cfg.GetClient()
+
+			assert.Equal(t, client.BaseURL.Host, tt.expectedHost)
+			assert.Equal(t, client.BaseURL.Path, tt.expectedPath)
+		})
+	}
+}