Normalize hostname checking for tenancy or enterprise

This aligns the `cli/cli` behavior of IsTenancy and IsEnterprise with the counterparts here.
cli · Mar 18, 2024 · 7b29c8e · 7b29c8e
1 parent 6e9fb7c
commit 7b29c8e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pkg/auth/auth.go b/pkg/auth/auth.go
@@ -153,11 +153,13 @@ func defaultHost(cfg *config.Config) (string, string) {
 const tenancyHost = "ghe.com"
 
 func IsEnterprise(host string) bool {
-	return host != github && host != localhost && !IsTenancy(host)
+	normalizedHost := normalizeHostname(host)
+	return normalizedHost != github && normalizedHost != localhost && !IsTenancy(normalizedHost)
 }
 
 func IsTenancy(host string) bool {
-	return strings.HasSuffix(host, "."+tenancyHost)
+	normalizedHost := normalizeHostname(host)
+	return strings.HasSuffix(normalizedHost, "."+tenancyHost)
 }
 
 func normalizeHostname(host string) string {