Allow to specify prefix that will be used in all exported names

Author: alexott

docs/guides/experimental-exporter.md

@@ -36,6 +36,7 @@ All arguments are optional and they tune what code is being generated.
 * `-match` - Match resource names during listing operation. This filter applies to all resources that are getting listed, so if you want to import all dependencies of just one cluster, specify `-match=autoscaling -listing=compute`. By default is empty, which matches everything.
 * `-mounts` - List DBFS mount points, which is a extremely slow operation and would not trigger unless explicitly specified.
 * `-generateProviderDeclaration` - flag that toggles generation of `databricks.tf` file with declaration of the Databricks Terraform provider that is necessary for Terraform versions since Terraform 0.13 (disabled by default).
+* `-prefix` - optional prefix that will be added to the name of all exported resources - that's useful for exporting resources multiple workspaces for merging into single one.
 
 ## Services
 

exporter/command.go

@@ -77,6 +77,8 @@ func Run(args ...string) error {
 	flags.StringVar(&ic.match, "match", "", "Match resource names during listing operation. "+
 		"This filter applies to all resources that are getting listed, so if you want to import "+
 		"all dependencies of just one cluster, specify -listing=compute")
+	prefix := ""
+	flags.StringVar(&prefix, "prefix", "", "Prefix that will be added to the name of all exported resources")
 	newArgs := args
 	if len(args) > 1 && args[1] == "exporter" {
 		newArgs = args[2:]
@@ -85,6 +87,9 @@ func Run(args ...string) error {
 	if err != nil {
 		return err
 	}
+	if len(prefix) > 0 {
+		ic.prefix = prefix + "_"
+	}
 	if ic.debug {
 		logLevel = append(logLevel, "[DEBUG]")
 	}

exporter/context.go

@@ -72,6 +72,7 @@ type importContext struct {
 	lastActiveDays      int64
 	generateDeclaration bool
 	meAdmin             bool
+	prefix              string
 }
 
 type mount struct {
@@ -359,6 +360,7 @@ func (ic *importContext) ResourceName(r *resource) string {
 	if name == "" {
 		name = r.ID
 	}
+	name = ic.prefix + name
 	name = strings.ToLower(name)
 	name = ic.regexFix(name, ic.nameFixes)
 	// this is either numeric id or all-non-ascii

exporter/exporter_test.go

@@ -747,7 +747,7 @@ func TestImportingWithError(t *testing.T) {
 	err := Run("-directory", "/bin/sh", "-services", "groups,users")
 	assert.EqualError(t, err, "The path /bin/sh is not a directory")
 
-	err = Run("-directory", "/bin/abcd", "-services", "groups,users")
+	err = Run("-directory", "/bin/abcd", "-services", "groups,users", "-prefix", "abc")
 	assert.EqualError(t, err, "Can't create directory /bin/abcd")
 }
 

exporter/importables.go

@@ -47,7 +47,8 @@ var resourcesMap map[string]importable = map[string]importable{
 			if err != nil && !os.IsExist(err) {
 				return err
 			}
-			local, err := os.Create(fmt.Sprintf("%s/files/%s", ic.Directory, path.Base(r.ID)))
+			fileName := ic.prefix + path.Base(r.ID)
+			local, err := os.Create(fmt.Sprintf("%s/files/%s", ic.Directory, fileName))
 			if err != nil {
 				return err
 			}
@@ -58,7 +59,7 @@ var resourcesMap map[string]importable = map[string]importable{
 			}
 			// libraries installed with init scripts won't be exported.
 			b := body.AppendNewBlock("resource", []string{r.Resource, r.Name}).Body()
-			relativeFile := fmt.Sprintf("${path.module}/files/%s", path.Base(r.ID))
+			relativeFile := fmt.Sprintf("${path.module}/files/%s", fileName)
 			b.SetAttributeValue("path", cty.StringVal(strings.Replace(r.ID, "dbfs:", "", 1)))
 			b.SetAttributeRaw("source", hclwrite.Tokens{
 				&hclwrite.Token{Type: hclsyntax.TokenOQuote, Bytes: []byte{'"'}},
@@ -816,7 +817,8 @@ var resourcesMap map[string]importable = map[string]importable{
 			if err != nil && !os.IsExist(err) {
 				return err
 			}
-			local, err := os.Create(fmt.Sprintf("%s/files/gis-%s", ic.Directory, path.Base(r.Name)))
+			fileName := path.Base(r.Name)
+			local, err := os.Create(fmt.Sprintf("%s/files/gis-%s", ic.Directory, fileName))
 			if err != nil {
 				return err
 			}
@@ -829,7 +831,7 @@ var resourcesMap map[string]importable = map[string]importable{
 			if err != nil {
 				return err
 			}
-			relativeFile := fmt.Sprintf("${path.module}/files/gis-%s", path.Base(r.Name))
+			relativeFile := fmt.Sprintf("${path.module}/files/gis-%s", fileName)
 			b := body.AppendNewBlock("resource", []string{r.Resource, r.Name}).Body()
 			b.SetAttributeValue("name", cty.StringVal(gis.Name))
 			b.SetAttributeValue("enabled", cty.BoolVal(gis.Enabled))