datasources

Author: lantoli

internal/convert/convert.go

@@ -14,17 +14,20 @@ import (
 )
 
 const (
-	resourceType   = "resource"
-	cluster        = "mongodbatlas_cluster"
-	advCluster     = "mongodbatlas_advanced_cluster"
-	valClusterType = "REPLICASET"
-	valMaxPriority = 7
-	valMinPriority = 1
-	errFreeCluster = "free cluster (because no " + nRepSpecs + ")"
-	errRepSpecs    = "setting " + nRepSpecs
-	errConfigs     = "setting " + nConfig
-	errPriority    = "setting " + nPriority
-	errNumShards   = "setting " + nNumShards
+	resourceType     = "resource"
+	dataSourceType   = "data"
+	cluster          = "mongodbatlas_cluster"
+	advCluster       = "mongodbatlas_advanced_cluster"
+	clusterPlural    = "mongodbatlas_clusters"
+	advClusterPlural = "mongodbatlas_advanced_clusters"
+	valClusterType   = "REPLICASET"
+	valMaxPriority   = 7
+	valMinPriority   = 1
+	errFreeCluster   = "free cluster (because no " + nRepSpecs + ")"
+	errRepSpecs      = "setting " + nRepSpecs
+	errConfigs       = "setting " + nConfig
+	errPriority      = "setting " + nPriority
+	errNumShards     = "setting " + nNumShards
 )
 
 type attrVals struct {
@@ -43,34 +46,65 @@ func ClusterToAdvancedCluster(config []byte) ([]byte, error) {
 		return nil, err
 	}
 	for _, resource := range parser.Body().Blocks() {
-		labels := resource.Labels()
-		resourceName := labels[0]
-		if resource.Type() != resourceType || resourceName != cluster {
-			continue
-		}
-		resourceb := resource.Body()
-		if errDyn := checkDynamicBlock(resourceb); errDyn != nil {
-			return nil, errDyn
-		}
-		labels[0] = advCluster
-		resource.SetLabels(labels)
-
-		if resourceb.FirstMatchingBlock(nRepSpecs, nil) != nil {
-			err = fillCluster(resourceb)
-		} else {
-			err = fillFreeTierCluster(resourceb)
-		}
+		converted, err := convertResource(resource)
 		if err != nil {
 			return nil, err
 		}
-
-		resourceb.AppendNewline()
-		hcl.AppendComment(resourceb, "Generated by atlas-cli-plugin-terraform.")
-		hcl.AppendComment(resourceb, "Please confirm that all references to this resource are updated.")
+		converted = converted || convertDataSource(resource)
+		if converted {
+			resourceb := resource.Body()
+			resourceb.AppendNewline()
+			hcl.AppendComment(resourceb, "Generated by atlas-cli-plugin-terraform.")
+			hcl.AppendComment(resourceb, "Please confirm that all references to this resource are updated.")
+		}
 	}
 	return parser.Bytes(), nil
 }
 
+func convertResource(resource *hclwrite.Block) (bool, error) {
+	labels := resource.Labels()
+	resourceName := labels[0]
+	if resource.Type() != resourceType || resourceName != cluster {
+		return false, nil
+	}
+	resourceb := resource.Body()
+	if errDyn := checkDynamicBlock(resourceb); errDyn != nil {
+		return false, errDyn
+	}
+	labels[0] = advCluster
+	resource.SetLabels(labels)
+
+	var err error
+	if resourceb.FirstMatchingBlock(nRepSpecs, nil) != nil {
+		err = fillCluster(resourceb)
+	} else {
+		err = fillFreeTierCluster(resourceb)
+	}
+	if err != nil {
+		return false, err
+	}
+	return true, nil
+}
+
+func convertDataSource(resource *hclwrite.Block) bool {
+	if resource.Type() != dataSourceType {
+		return false
+	}
+	labels := resource.Labels()
+	resourceName := labels[0]
+	if resourceName == cluster {
+		labels[0] = advCluster
+		resource.SetLabels(labels)
+		return true
+	}
+	if resourceName == clusterPlural {
+		labels[0] = advClusterPlural
+		resource.SetLabels(labels)
+		return true
+	}
+	return false
+}
+
 // fillFreeTierCluster is the entry point to convert clusters in free tier
 func fillFreeTierCluster(resourceb *hclwrite.Body) error {
 	resourceb.SetAttributeValue(nClusterType, cty.StringVal(valClusterType))

internal/convert/testdata/clu2adv/data_sources.in.tf

@@ -0,0 +1,11 @@
+data "mongodbatlas_cluster" "singular" {
+  # data source content is kep, only data source type is changed - singular
+  project_id = mongodbatlas_advanced_cluster.example.project_id
+  name       = mongodbatlas_advanced_cluster.example.name
+  depends_on = [mongodbatlas_privatelink_endpoint_service.example_endpoint]
+}
+
+data "mongodbatlas_clusters" "plural" {
+  # data source content is kep, only data source type is changed - plural
+  project_id = mongodbatlas_advanced_cluster.example.project_id
+}

internal/convert/testdata/clu2adv/data_sources.out.tf

@@ -0,0 +1,17 @@
+data "mongodbatlas_advanced_cluster" "singular" {
+  # data source content is kep, only data source type is changed - singular
+  project_id = mongodbatlas_advanced_cluster.example.project_id
+  name       = mongodbatlas_advanced_cluster.example.name
+  depends_on = [mongodbatlas_privatelink_endpoint_service.example_endpoint]
+
+  # Generated by atlas-cli-plugin-terraform.
+  # Please confirm that all references to this resource are updated.
+}
+
+data "mongodbatlas_advanced_clusters" "plural" {
+  # data source content is kep, only data source type is changed - plural
+  project_id = mongodbatlas_advanced_cluster.example.project_id
+
+  # Generated by atlas-cli-plugin-terraform.
+  # Please confirm that all references to this resource are updated.
+}

internal/convert/testdata/clu2adv/free_cluster_with_count.in.tf

@@ -4,17 +4,13 @@ resource "resource1" "res1" {
 
 resource "mongodbatlas_cluster" "free_cluster" { # comment in the resource
   # comment in own line in the beginning
-  count                       = local.use_free_cluster ? 1 : 0
-  project_id                  = var.project_id # inline comment kept
-  name                        = var.cluster_name
+  count      = local.use_free_cluster ? 1 : 0
+  project_id = var.project_id # inline comment kept
+  name       = var.cluster_name
   # comment in own line in the middle is deleted
   provider_name               = "TENANT" # inline comment for attribute moved is not kept
   backing_provider_name       = "AWS"
   provider_region_name        = var.region
   provider_instance_size_name = "M0"
   # comment in own line at the end happens before replication_specs
 }
-
-data "mongodbatlas_cluster" "cluster2" {
-  name = "name4"
-}

internal/convert/testdata/clu2adv/free_cluster_with_count.out.tf

@@ -28,7 +28,3 @@ resource "mongodbatlas_advanced_cluster" "free_cluster" { # comment in the resou
   # Generated by atlas-cli-plugin-terraform.
   # Please confirm that all references to this resource are updated.
 }
-
-data "mongodbatlas_cluster" "cluster2" {
-  name = "name4"
-}