add project support

Author: AkbaraliShaikh

internal/authctx/client.go

@@ -14,6 +14,7 @@ import (
 
 const (
 	ServerEndpointEnvVar = "TMC_ENDPOINT"
+	ProjectIDEnvVar      = "PROJECT_ID"
 
 	// TMC SaaS env variables.
 	VMWCloudEndpointEnvVar = "VMW_CLOUD_ENDPOINT"
@@ -37,6 +38,7 @@ const (
 type TanzuContext struct {
 	SelfManaged      bool
 	ServerEndpoint   string
+	ProjectID        string
 	SMUsername       string
 	Token            string // selfmanaged password is stored here
 	VMWCloudEndPoint string // selfmanaged odic issuer is stored here
@@ -75,6 +77,10 @@ func setup(cfg *TanzuContext) (err error) {
 	cfg.TMCConnection.WithHost(cfg.ServerEndpoint)
 	cfg.TMCConnection.Headers.Set("Host", cfg.ServerEndpoint)
 
+	if cfg.ProjectID != "" {
+		cfg.TMCConnection.Headers.Set("X-Project-Id", cfg.ProjectID)
+	}
+
 	if cfg.IsSelfManaged() {
 		// We need to add this only for self-managed flow because the SaaS token has a longer ttl.
 		cfg.TMCConnection.WithRefreshAuthCtx(fetchAuthHeaders)

internal/authctx/constants.go

@@ -7,6 +7,7 @@ package authctx
 
 const (
 	endpoint           = "endpoint"
+	projectID          = "project_id"
 	vmwCloudEndpoint   = "vmw_cloud_endpoint"
 	vmwCloudAPIToken   = "vmw_cloud_api_token"
 	defaultCSPEndpoint = "console.cloud.vmware.com"

internal/authctx/helper.go

@@ -25,6 +25,11 @@ func ProviderAuthSchema() map[string]*schema.Schema {
 			Required:    true,
 			DefaultFunc: schema.EnvDefaultFunc(ServerEndpointEnvVar, ""),
 		},
+		projectID: {
+			Type:        schema.TypeString,
+			Optional:    true,
+			DefaultFunc: schema.EnvDefaultFunc(ProjectIDEnvVar, ""),
+		},
 		vmwCloudEndpoint: {
 			Type:        schema.TypeString,
 			Required:    true,
@@ -131,6 +136,7 @@ func ProviderConfigureContext(_ context.Context, d *schema.ResourceData) (interf
 	_, smAuth := d.GetOk(selfManaged)
 
 	config.ServerEndpoint, _ = d.Get(endpoint).(string)
+	config.ProjectID, _ = d.Get(projectID).(string)
 	config.VMWCloudEndPoint, _ = d.Get(vmwCloudEndpoint).(string)
 	config.Token, _ = d.Get(vmwCloudAPIToken).(string)