includeMoved

Author: lantoli

README.md

@@ -32,9 +32,11 @@ you can also use shorter aliases, e.g.:
 atlas tf clu2adv -f in.tf -o out.tf
 ```
 
-If you want to overwrite the output file if it exists, or even use the same output file as the input file, use the `--replaceOutput true` or the `-r` flag.
+If you want to include the `moved blocks` in the output file, use the `--includeMoved` or the `-m` flag.
 
-You can use the `--watch true` or the `-w` flag to keep the plugin running and watching for changes in the input file. You can have input and output files open in an editor and see easily how changes to the input file affect the output file.
+If you want to overwrite the output file if it exists, or even use the same output file as the input file, use the `--replaceOutput` or the `-r` flag.
+
+You can use the `--watch` or the `-w` flag to keep the plugin running and watching for changes in the input file. You can have input and output files open in an editor and see easily how changes to the input file affect the output file.
 
 ### Limitations
 

internal/cli/clu2adv/opts.go

@@ -16,6 +16,7 @@ type opts struct {
 	output        string
 	replaceOutput bool
 	watch         bool
+	includeMoved  bool
 }
 
 func (o *opts) PreRun() error {
@@ -43,7 +44,7 @@ func (o *opts) generateFile(allowParseErrors bool) error {
 	if err != nil {
 		return fmt.Errorf("failed to read file %s: %w", o.file, err)
 	}
-	outConfig, err := convert.ClusterToAdvancedCluster(inConfig)
+	outConfig, err := convert.ClusterToAdvancedCluster(inConfig, o.includeMoved)
 	if err != nil {
 		if allowParseErrors {
 			outConfig = []byte("# CONVERT ERROR: " + err.Error() + "\n\n")

internal/convert/const_names.go

@@ -48,4 +48,7 @@ const (
 	nKey                        = "key"
 	nValue                      = "value"
 	nUseRepSpecsPerShard        = "use_replication_spec_per_shard"
+	nMoved                      = "moved"
+	nFrom                       = "from"
+	nTo                         = "to"
 )

internal/convert/convert.go

@@ -40,24 +40,33 @@ type attrVals struct {
 // All other resources and data sources are left untouched.
 // Note: hclwrite.Tokens are used instead of cty.Value so expressions with interpolations like var.region can be preserved.
 // cty.Value only supports literal expressions.
-func ClusterToAdvancedCluster(config []byte) ([]byte, error) {
+func ClusterToAdvancedCluster(config []byte, includeMoved bool) ([]byte, error) {
+	var moveLabels []string
 	parser, err := hcl.GetParser(config)
 	if err != nil {
 		return nil, err
 	}
-	for _, block := range parser.Body().Blocks() {
-		converted, err := convertResource(block)
+	parserb := parser.Body()
+	for _, block := range parserb.Blocks() {
+		convertedResource, err := convertResource(block)
 		if err != nil {
-			return nil, err
+			return nil,
+				err
 		}
-		converted = converted || convertDataSource(block)
-		if converted {
+		if includeMoved && convertedResource {
+			if moveLabel := getResourceLabel(block); moveLabel != "" {
+				moveLabels = append(moveLabels, moveLabel)
+			}
+		}
+		convertedDataSource := convertDataSource(block)
+		if convertedResource || convertedDataSource {
 			blockb := block.Body()
 			blockb.AppendNewline()
 			hcl.AppendComment(blockb, "Generated by atlas-cli-plugin-terraform.")
 			hcl.AppendComment(blockb, "Please confirm that all references to this resource are updated.")
 		}
 	}
+	fillMovedBlocks(parserb, moveLabels)
 	return parser.Bytes(), nil
 }
 
@@ -99,6 +108,24 @@ func convertDataSource(block *hclwrite.Block) bool {
 	return false
 }
 
+func fillMovedBlocks(body *hclwrite.Body, moveLabels []string) {
+	if len(moveLabels) == 0 {
+		return
+	}
+	body.AppendNewline()
+	hcl.AppendComment(body, "Moved blocks")
+	body.AppendNewline()
+	for i, moveLabel := range moveLabels {
+		block := body.AppendNewBlock(nMoved, nil)
+		blockb := block.Body()
+		hcl.SetAttrExpr(blockb, nFrom, fmt.Sprintf("%s.%s", cluster, moveLabel))
+		hcl.SetAttrExpr(blockb, nTo, fmt.Sprintf("%s.%s", advCluster, moveLabel))
+		if i < len(moveLabels)-1 {
+			body.AppendNewline()
+		}
+	}
+}
+
 // fillFreeTierCluster is the entry point to convert clusters in free tier
 func fillFreeTierCluster(resourceb *hclwrite.Body) error {
 	resourceb.SetAttributeValue(nClusterType, cty.StringVal(valClusterType))
@@ -346,6 +373,14 @@ func getResourceName(resource *hclwrite.Block) string {
 	return labels[0]
 }
 
+func getResourceLabel(resource *hclwrite.Block) string {
+	labels := resource.Labels()
+	if len(labels) <= 1 {
+		return ""
+	}
+	return labels[1]
+}
+
 func checkDynamicBlock(body *hclwrite.Body) error {
 	for _, block := range body.Blocks() {
 		if block.Type() == "dynamic" {

internal/convert/convert_test.go

@@ -38,7 +38,8 @@ func TestClusterToAdvancedCluster(t *testing.T) {
 		t.Run(testName, func(t *testing.T) {
 			inConfig, err := afero.ReadFile(fs, inputFile)
 			require.NoError(t, err)
-			outConfig, err := convert.ClusterToAdvancedCluster(inConfig)
+			includeMoved := strings.Contains(testName, "includeMoved")
+			outConfig, err := convert.ClusterToAdvancedCluster(inConfig, includeMoved)
 			if err == nil {
 				g.Assert(t, testName, outConfig)
 			} else {

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

@@ -0,0 +1,71 @@
+resource "mongodbatlas_cluster" "cluster1" {
+  project_id                  = var.project_id
+  name                        = "clu1"
+  cluster_type                = "REPLICASET"
+  provider_name               = "AWS"
+  provider_instance_size_name = "M10"
+  replication_specs {
+    num_shards = 1
+    regions_config {
+      region_name     = "US_EAST_1"
+      electable_nodes = 3
+      priority        = 7
+    }
+  }
+}
+
+resource "mongodbatlas_cluster" "cluster2" {
+  project_id                  = var.project_id
+  name                        = "clu2"
+  cluster_type                = "REPLICASET"
+  provider_name               = "AWS"
+  provider_instance_size_name = "M30"
+  replication_specs {
+    num_shards = 1
+    regions_config {
+      region_name     = "US_WEST_2"
+      electable_nodes = 3
+      priority        = 7
+    }
+  }
+}
+
+resource "mongodbatlas_cluster" "count" {
+  # count doesn't affect moved blocks, it works in the same way
+  count                       = local.create_cluster ? 1 : 0
+  project_id                  = var.project_id
+  name                        = "count"
+  cluster_type                = "REPLICASET"
+  provider_name               = "AWS"
+  provider_instance_size_name = "M30"
+  replication_specs {
+    num_shards = 1
+    regions_config {
+      region_name     = "US_WEST_2"
+      electable_nodes = 3
+      priority        = 7
+    }
+  }
+}
+
+resource "mongodbatlas_cluster" "forEach" {
+  # for_each doesn't affect moved blocks, it works in the same way
+  for_each                    = toset(["clu1", "clu2", "clu3"])
+  project_id                  = var.project_id
+  name                        = each.key
+  cluster_type                = "REPLICASET"
+  provider_name               = "AWS"
+  provider_instance_size_name = "M30"
+  replication_specs {
+    num_shards = 1
+    regions_config {
+      region_name     = "US_WEST_2"
+      electable_nodes = 3
+      priority        = 7
+    }
+  }
+}
+
+resource "another_resource" "another" {
+  hello = "there"
+}

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

@@ -0,0 +1,125 @@
+resource "mongodbatlas_advanced_cluster" "cluster1" {
+  project_id   = var.project_id
+  name         = "clu1"
+  cluster_type = "REPLICASET"
+  replication_specs = [
+    {
+      region_configs = [
+        {
+          provider_name = "AWS"
+          region_name   = "US_EAST_1"
+          priority      = 7
+          electable_specs = {
+            node_count    = 3
+            instance_size = "M10"
+          }
+        }
+      ]
+    }
+  ]
+
+  # Generated by atlas-cli-plugin-terraform.
+  # Please confirm that all references to this resource are updated.
+}
+
+resource "mongodbatlas_advanced_cluster" "cluster2" {
+  project_id   = var.project_id
+  name         = "clu2"
+  cluster_type = "REPLICASET"
+  replication_specs = [
+    {
+      region_configs = [
+        {
+          provider_name = "AWS"
+          region_name   = "US_WEST_2"
+          priority      = 7
+          electable_specs = {
+            node_count    = 3
+            instance_size = "M30"
+          }
+        }
+      ]
+    }
+  ]
+
+  # Generated by atlas-cli-plugin-terraform.
+  # Please confirm that all references to this resource are updated.
+}
+
+resource "mongodbatlas_advanced_cluster" "count" {
+  # count doesn't affect moved blocks, it works in the same way
+  count        = local.create_cluster ? 1 : 0
+  project_id   = var.project_id
+  name         = "count"
+  cluster_type = "REPLICASET"
+  replication_specs = [
+    {
+      region_configs = [
+        {
+          provider_name = "AWS"
+          region_name   = "US_WEST_2"
+          priority      = 7
+          electable_specs = {
+            node_count    = 3
+            instance_size = "M30"
+          }
+        }
+      ]
+    }
+  ]
+
+  # Generated by atlas-cli-plugin-terraform.
+  # Please confirm that all references to this resource are updated.
+}
+
+resource "mongodbatlas_advanced_cluster" "forEach" {
+  # for_each doesn't affect moved blocks, it works in the same way
+  for_each     = toset(["clu1", "clu2", "clu3"])
+  project_id   = var.project_id
+  name         = each.key
+  cluster_type = "REPLICASET"
+  replication_specs = [
+    {
+      region_configs = [
+        {
+          provider_name = "AWS"
+          region_name   = "US_WEST_2"
+          priority      = 7
+          electable_specs = {
+            node_count    = 3
+            instance_size = "M30"
+          }
+        }
+      ]
+    }
+  ]
+
+  # Generated by atlas-cli-plugin-terraform.
+  # Please confirm that all references to this resource are updated.
+}
+
+resource "another_resource" "another" {
+  hello = "there"
+}
+
+# Moved blocks
+
+moved {
+  from = mongodbatlas_cluster.cluster1
+  to   = mongodbatlas_advanced_cluster.cluster1
+}
+
+moved {
+  from = mongodbatlas_cluster.cluster2
+  to   = mongodbatlas_advanced_cluster.cluster2
+}
+
+moved {
+  from = mongodbatlas_cluster.count
+  to   = mongodbatlas_advanced_cluster.count
+}
+
+moved {
+  from = mongodbatlas_cluster.forEach
+  to   = mongodbatlas_advanced_cluster.forEach
+}

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

@@ -0,0 +1,15 @@
+resource "mongodbatlas_cluster" "cluster" {
+  project_id                  = var.project_id
+  name                        = "clu"
+  cluster_type                = "REPLICASET"
+  provider_name               = "AWS"
+  provider_instance_size_name = "M10"
+  replication_specs {
+    num_shards = 1
+    regions_config {
+      region_name     = "US_EAST_1"
+      electable_nodes = 3
+      priority        = 7
+    }
+  }
+}

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

@@ -0,0 +1,30 @@
+resource "mongodbatlas_advanced_cluster" "cluster" {
+  project_id   = var.project_id
+  name         = "clu"
+  cluster_type = "REPLICASET"
+  replication_specs = [
+    {
+      region_configs = [
+        {
+          provider_name = "AWS"
+          region_name   = "US_EAST_1"
+          priority      = 7
+          electable_specs = {
+            node_count    = 3
+            instance_size = "M10"
+          }
+        }
+      ]
+    }
+  ]
+
+  # Generated by atlas-cli-plugin-terraform.
+  # Please confirm that all references to this resource are updated.
+}
+
+# Moved blocks
+
+moved {
+  from = mongodbatlas_cluster.cluster
+  to   = mongodbatlas_advanced_cluster.cluster
+}

internal/hcl/hcl.go

@@ -30,6 +30,12 @@ func PopAttr(body *hclwrite.Body, attrName, errPrefix string) (hclwrite.Tokens,
 	return tokens, nil
 }
 
+// SetAttrExpr sets an attribute to an expression (possibly with interpolations) without quotes.
+func SetAttrExpr(body *hclwrite.Body, attrName, expresion string) {
+	tokens := hclwrite.Tokens{{Type: hclsyntax.TokenIdent, Bytes: []byte(expresion)}}
+	body.SetAttributeRaw(attrName, tokens)
+}
+
 // SetAttrInt sets an attribute to a number.
 func SetAttrInt(body *hclwrite.Body, attrName string, number int) {
 	tokens := hclwrite.Tokens{