mirror of https://github.com/databricks/cli.git
39 lines
1.3 KiB
Go
39 lines
1.3 KiB
Go
package terraform
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/config/interpolation"
|
|
)
|
|
|
|
// Rewrite variable references to resources into Terraform compatible format.
|
|
func interpolateTerraformResourceIdentifiers(path string, lookup map[string]string) (string, error) {
|
|
parts := strings.Split(path, interpolation.Delimiter)
|
|
if parts[0] == "resources" {
|
|
switch parts[1] {
|
|
case "pipelines":
|
|
path = strings.Join(append([]string{"databricks_pipeline"}, parts[2:]...), interpolation.Delimiter)
|
|
return fmt.Sprintf("${%s}", path), nil
|
|
case "jobs":
|
|
path = strings.Join(append([]string{"databricks_job"}, parts[2:]...), interpolation.Delimiter)
|
|
return fmt.Sprintf("${%s}", path), nil
|
|
case "models":
|
|
path = strings.Join(append([]string{"databricks_mlflow_model"}, parts[2:]...), interpolation.Delimiter)
|
|
return fmt.Sprintf("${%s}", path), nil
|
|
case "experiments":
|
|
path = strings.Join(append([]string{"databricks_mlflow_experiment"}, parts[2:]...), interpolation.Delimiter)
|
|
return fmt.Sprintf("${%s}", path), nil
|
|
default:
|
|
panic("TODO: " + parts[1])
|
|
}
|
|
}
|
|
|
|
return interpolation.DefaultLookup(path, lookup)
|
|
}
|
|
|
|
func Interpolate() bundle.Mutator {
|
|
return interpolation.Interpolate(interpolateTerraformResourceIdentifiers)
|
|
}
|