Add IsCloud and IsEnterprise methods to client struct

Author: sebasslash

tfe.go

@@ -35,6 +35,7 @@ const (
 	_userAgent         = "go-tfe"
 	_headerRateLimit   = "X-RateLimit-Limit"
 	_headerRateReset   = "X-RateLimit-Reset"
+	_headerAppName     = "TFP-AppName"
 	_headerAPIVersion  = "TFP-API-Version"
 	_headerTFEVersion  = "X-TFE-Version"
 	_includeQueryParam = "include"
@@ -120,6 +121,7 @@ type Client struct {
 	retryServerErrors bool
 	remoteAPIVersion  string
 	remoteTFEVersion  string
+	appName           string
 
 	Admin                      Admin
 	Agents                     Agents
@@ -361,6 +363,9 @@ func NewClient(cfg *Config) (*Client, error) {
 	// Save the TFE version
 	client.remoteTFEVersion = meta.TFEVersion
 
+	// Save the app name
+	client.appName = meta.AppName
+
 	// Create Admin
 	client.Admin = Admin{
 		Organizations:     &adminOrganizations{client: client},
@@ -431,6 +436,23 @@ func NewClient(cfg *Config) (*Client, error) {
 	return client, nil
 }
 
+// IsCloud returns true if the client is configured against a Terraform Cloud
+// instance.
+//
+// Whether an instance is TFC or TFE is derived from the TFP-AppName header.
+func (c Client) IsCloud() bool {
+	return c.appName == "Terraform Cloud"
+}
+
+// IsEnterprise returns true if the client is configured against a Terraform
+// Enterprise instance.
+//
+// Whether an instance is TFC or TFE is derived from the TFP-AppName header. Note:
+// not all TFE releases include this header in API responses.
+func (c Client) IsEnterprise() bool {
+	return !c.IsCloud()
+}
+
 // RemoteAPIVersion returns the server's declared API version string.
 //
 // A Terraform Cloud or Enterprise API server returns its API version in an
@@ -565,6 +587,9 @@ type rawAPIMetadata struct {
 	// X-RateLimit-Limit response header, or an empty string if that header
 	// field was not included in the response.
 	RateLimit string
+
+	// AppName is either 'Terraform Cloud' or 'Terraform Enterprise'
+	AppName string
 }
 
 func (c *Client) getRawAPIMetadata() (rawAPIMetadata, error) {
@@ -597,6 +622,7 @@ func (c *Client) getRawAPIMetadata() (rawAPIMetadata, error) {
 	meta.APIVersion = resp.Header.Get(_headerAPIVersion)
 	meta.RateLimit = resp.Header.Get(_headerRateLimit)
 	meta.TFEVersion = resp.Header.Get(_headerTFEVersion)
+	meta.AppName = resp.Header.Get(_headerAppName)
 
 	return meta, nil
 }

tfe_integration_test.go

@@ -25,6 +25,11 @@ func TestClient_newClient(t *testing.T) {
 		w.Header().Set("X-RateLimit-Limit", "30")
 		w.Header().Set("TFP-API-Version", "34.21.9")
 		w.Header().Set("X-TFE-Version", "202205-1")
+		if enterpriseEnabled() {
+			w.Header().Set("TFP-AppName", "Terraform Enterprise")
+		} else {
+			w.Header().Set("TFP-AppName", "Terraform Cloud")
+		}
 		w.WriteHeader(204) // We query the configured ping URL which should return a 204.
 	}))
 	defer ts.Close()
@@ -85,6 +90,12 @@ func TestClient_newClient(t *testing.T) {
 			t.Errorf("unexpected remote TFE version %q; want %q", client.RemoteTFEVersion(), want)
 		}
 
+		if enterpriseEnabled() {
+			assert.True(t, client.IsEnterprise())
+		} else {
+			assert.True(t, client.IsCloud())
+		}
+
 		client.SetFakeRemoteAPIVersion("1.0")
 
 		if want := "1.0"; client.RemoteAPIVersion() != want {