Tag PowerVS Cluster Resources

Author: arshadd-b

cloud/scope/powervs_cluster.go

@@ -34,6 +34,7 @@ import (
 	cosSession "github.com/IBM/ibm-cos-sdk-go/aws/session"
 	"github.com/IBM/ibm-cos-sdk-go/service/s3"
 	tgapiv1 "github.com/IBM/networking-go-sdk/transitgatewayapisv1"
+	"github.com/IBM/platform-services-go-sdk/globaltaggingv1"
 	"github.com/IBM/platform-services-go-sdk/resourcecontrollerv2"
 	"github.com/IBM/platform-services-go-sdk/resourcemanagerv2"
 	"github.com/IBM/vpc-go-sdk/vpcv1"
@@ -50,6 +51,7 @@ import (
 	infrav1beta2 "sigs.k8s.io/cluster-api-provider-ibmcloud/api/v1beta2"
 	"sigs.k8s.io/cluster-api-provider-ibmcloud/pkg/cloud/services/authenticator"
 	"sigs.k8s.io/cluster-api-provider-ibmcloud/pkg/cloud/services/cos"
+	"sigs.k8s.io/cluster-api-provider-ibmcloud/pkg/cloud/services/globaltagging"
 	"sigs.k8s.io/cluster-api-provider-ibmcloud/pkg/cloud/services/powervs"
 	"sigs.k8s.io/cluster-api-provider-ibmcloud/pkg/cloud/services/resourcecontroller"
 	"sigs.k8s.io/cluster-api-provider-ibmcloud/pkg/cloud/services/resourcemanager"
@@ -79,6 +81,7 @@ const (
 	// vpcSubnetIPAddressCount is the total IP Addresses for the subnet.
 	// Support for custom address prefixes will be added at a later time. Currently, we use the ip count for subnet creation.
 	vpcSubnetIPAddressCount int64 = 256
+	tagKey                        = "powervs.cluster.x-k8s.io-resource-owner:"
 )
 
 // PowerVSClusterScopeParams defines the input parameters used to create a new PowerVSClusterScope.
@@ -101,6 +104,7 @@ type ClientFactory struct {
 	TransitGatewayFactory     func() (transitgateway.TransitGateway, error)
 	ResourceControllerFactory func() (resourcecontroller.ResourceController, error)
 	ResourceManagerFactory    func() (resourcemanager.ResourceManager, error)
+	GlobalTaggingFactory      func() (globaltagging.GlobalTagging, error)
 }
 
 // PowerVSClusterScope defines a scope defined around a Power VS Cluster.
@@ -110,6 +114,7 @@ type PowerVSClusterScope struct {
 	patchHelper *patch.Helper
 
 	IBMPowerVSClient      powervs.PowerVS
+	GlobalTaggingClient   globaltagging.GlobalTagging
 	IBMVPCClient          vpc.Vpc
 	TransitGatewayClient  transitgateway.TransitGateway
 	ResourceClient        resourcecontroller.ResourceController
@@ -254,6 +259,18 @@ func NewPowerVSClusterScope(params PowerVSClusterScopeParams) (*PowerVSClusterSc
 		Authenticator: auth,
 	}
 
+	// Create Global Tagging client.
+	gtOptions := globaltagging.ServiceOptions{
+		GlobalTaggingV1Options: &globaltaggingv1.GlobalTaggingV1Options{
+			Authenticator: auth,
+		},
+	}
+
+	globalTaggingClient, err := params.getGlobalTaggingClient(gtOptions)
+	if err != nil {
+		return nil, fmt.Errorf("failed to create global tagging client: %w", err)
+	}
+
 	rmClient, err := params.getResourceManagerClient(rcManagerOptions)
 	if err != nil {
 		return nil, fmt.Errorf("failed to create resource manager client: %w", err)
@@ -269,6 +286,7 @@ func NewPowerVSClusterScope(params PowerVSClusterScopeParams) (*PowerVSClusterSc
 		IBMPowerVSClient:      powerVSClient,
 		IBMVPCClient:          vpcClient,
 		TransitGatewayClient:  tgClient,
+		GlobalTaggingClient:   globalTaggingClient,
 		ResourceClient:        resourceClient,
 		ResourceManagerClient: rmClient,
 	}
@@ -296,6 +314,18 @@ func (params PowerVSClusterScopeParams) getPowerVSClient(options powervs.Service
 	return powervs.NewService(options)
 }
 
+func (params PowerVSClusterScopeParams) getGlobalTaggingClient(gtOptions globaltagging.ServiceOptions) (globaltagging.GlobalTagging, error) {
+	if params.GlobalTaggingFactory != nil {
+		return params.GlobalTaggingFactory()
+	}
+
+	if gtEndpoint := endpoints.FetchEndpoints(string(endpoints.GlobalTagging), params.ServiceEndpoint); gtEndpoint != "" {
+		params.Logger.V(3).Info("Overriding the default global tagging endpoint", "GlobaTaggingEndpoint", gtEndpoint)
+		gtOptions.URL = gtEndpoint
+	}
+	return globaltagging.NewService(gtOptions)
+}
+
 func (params PowerVSClusterScopeParams) getVPCClient() (vpc.Vpc, error) {
 	if params.Logger.V(DEBUGLEVEL).Enabled() {
 		core.SetLoggingLevel(core.LevelDebug)
@@ -852,10 +882,13 @@ func (s *PowerVSClusterScope) createServiceInstance() (*resourcecontrollerv2.Res
 	if zone == nil {
 		return nil, fmt.Errorf("PowerVS zone is not set")
 	}
+
+	tag := tagKey + s.Name()
 	serviceInstance, _, err := s.ResourceClient.CreateResourceInstance(&resourcecontrollerv2.CreateResourceInstanceOptions{
 		Name:           s.GetServiceName(infrav1beta2.ResourceTypeServiceInstance),
 		Target:         zone,
 		ResourceGroup:  &resourceGroupID,
+		Tags:           append(make([]string, 0), tag),
 		ResourcePlanID: ptr.To(resourcecontroller.PowerVSResourcePlanID),
 	})
 	if err != nil {
@@ -1072,6 +1105,24 @@ func (s *PowerVSClusterScope) createDHCPServer() (*string, error) {
 	return dhcpServer.ID, nil
 }
 
+// TagResource will attach a user Tag to a resource.
+func (s *PowerVSClusterScope) TagResource(tagName string, resourceCRN *string) error {
+	tagOptions := &globaltaggingv1.AttachTagOptions{}
+	tagOptions.SetResources([]globaltaggingv1.Resource{
+		{
+			ResourceID: resourceCRN,
+		},
+	})
+
+	tagOptions.SetTagName(tagName)
+	tagOptions.SetTagType(globaltaggingv1.AttachTagOptionsTagTypeUserConst)
+	if _, _, err := s.GlobalTaggingClient.AttachTag(tagOptions); err != nil {
+		return fmt.Errorf("failure tagging resource: %w", err)
+	}
+
+	return nil
+}
+
 // ReconcileVPC reconciles VPC.
 func (s *PowerVSClusterScope) ReconcileVPC() (bool, error) {
 	// if VPC server id is set means the VPC is already created
@@ -1172,6 +1223,11 @@ func (s *PowerVSClusterScope) createVPC() (*string, error) {
 		return nil, err
 	}
 
+	tag := tagKey + s.Name()
+	if err = s.TagResource(tag, vpcDetails.CRN); err != nil {
+		s.Error(err, "failed to tag vpc")
+	}
+
 	// set security group for vpc
 	options := &vpcv1.CreateSecurityGroupRuleOptions{}
 	options.SetSecurityGroupID(*vpcDetails.DefaultSecurityGroup.ID)
@@ -1325,6 +1381,12 @@ func (s *PowerVSClusterScope) createVPCSubnet(subnet infrav1beta2.Subnet) (*stri
 	if subnetDetails == nil {
 		return nil, fmt.Errorf("create VPC subnet is nil")
 	}
+
+	tag := tagKey + s.Name()
+	err = s.TagResource(tag, subnetDetails.CRN)
+	if err != nil {
+		s.Error(err, "failed to tag subnet")
+	}
 	return subnetDetails.ID, nil
 }
 
@@ -1542,7 +1604,12 @@ func (s *PowerVSClusterScope) createVPCSecurityGroup(spec infrav1beta2.VPCSecuri
 	if err != nil {
 		return nil, err
 	}
-	// To-Do: Add tags to VPC security group, need to implement the client for "github.com/IBM/platform-services-go-sdk/globaltaggingv1".
+
+	tag := tagKey + s.Name()
+	err = s.TagResource(tag, securityGroup.CRN)
+	if err != nil {
+		s.Error(err, "failed to tag security group")
+	}
 	return securityGroup.ID, nil
 }
 
@@ -2002,6 +2069,12 @@ func (s *PowerVSClusterScope) createTransitGateway() error {
 		return err
 	}
 
+	tag := tagKey + s.Name()
+	err = s.TagResource(tag, tg.Crn)
+	if err != nil {
+		s.Error(err, "failed to tag transitGateway")
+	}
+
 	s.SetTransitGatewayStatus(tg.ID, ptr.To(true))
 
 	vpcCRN, err := s.fetchVPCCRN()
@@ -2217,6 +2290,12 @@ func (s *PowerVSClusterScope) createLoadBalancer(lb infrav1beta2.VPCLoadBalancer
 	if err != nil {
 		return nil, err
 	}
+
+	tag := tagKey + s.Name()
+	if err = s.TagResource(tag, loadBalancer.CRN); err != nil {
+		s.Error(err, "failed to tag load balancer")
+	}
+
 	lbState := infrav1beta2.VPCLoadBalancerState(*loadBalancer.ProvisioningStatus)
 	return &infrav1beta2.VPCLoadBalancerStatus{
 		ID:                loadBalancer.ID,
@@ -2373,12 +2452,14 @@ func (s *PowerVSClusterScope) createCOSServiceInstance() (*resourcecontrollerv2.
 		return nil, fmt.Errorf("failed to fetch resource group ID for resource group %v, ID is empty", s.ResourceGroup())
 	}
 
+	tag := tagKey + s.Name()
 	target := "Global"
 	// create service instance
 	serviceInstance, _, err := s.ResourceClient.CreateResourceInstance(&resourcecontrollerv2.CreateResourceInstanceOptions{
 		Name:           s.GetServiceName(infrav1beta2.ResourceTypeCOSInstance),
 		Target:         &target,
 		ResourceGroup:  &resourceGroupID,
+		Tags:           append(make([]string, 0), tag),
 		ResourcePlanID: ptr.To(resourcecontroller.CosResourcePlanID),
 	})
 	if err != nil {