CLOUDP-62037: Add Labels when Creating/Updating cluster with mongocli Atlas

internal/cli/atlas_clusters_create.go

@@ -85,10 +85,29 @@ func (opts *atlasClustersCreateOpts) newCluster() (*atlas.Cluster, error) {
 		cluster.Name = opts.name
 	}
 
+	updateLabels(cluster)
+
 	cluster.GroupID = opts.ProjectID()
 	return cluster, nil
 }
 
+func updateLabels(out *atlas.Cluster) {
+	found := false
+	for _, v := range out.Labels {
+		if v.Key == labelKey && v.Value == labelValue {
+			found = true
+			break
+		}
+	}
+
+	if !found {
+		out.Labels = append(out.Labels, atlas.Label{
+			Key:   labelKey,
+			Value: labelValue,
+		})
+	}
+}
+
 func (opts *atlasClustersCreateOpts) applyOpts(out *atlas.Cluster) {
 	replicationSpec := opts.newReplicationSpec()
 	out.ProviderBackupEnabled = &opts.backup

internal/cli/atlas_clusters_update.go

@@ -98,6 +98,8 @@ func (opts *atlasClustersUpdateOpts) patchOpts(out *atlas.Cluster) {
 	if opts.tier != "" {
 		out.ProviderSettings.InstanceSizeName = opts.tier
 	}
+
+	updateLabels(out)
 }
 
 // mongocli atlas cluster(s) update name --projectId projectId [--tier M#] [--diskSizeGB N] [--mdbVersion]

internal/cli/cli.go

@@ -30,6 +30,8 @@ import (
 const (
 	fallbackSuccessMessage = "'%s' deleted\n"
 	fallbackFailMessage    = "entry not deleted"
+	labelKey               = "Infrastructure Tool"
+	labelValue             = "mongoCLI"
 )
 
 type globalOpts struct {