mirror of https://github.com/databricks/cli.git
122 lines
2.9 KiB
Go
122 lines
2.9 KiB
Go
package terraform
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/databricks/bricks/bundle/config"
|
|
"github.com/databricks/bricks/bundle/internal/tf/schema"
|
|
tfjson "github.com/hashicorp/terraform-json"
|
|
)
|
|
|
|
func conv(from any, to any) {
|
|
buf, _ := json.Marshal(from)
|
|
json.Unmarshal(buf, &to)
|
|
}
|
|
|
|
// BundleToTerraform converts resources in a bundle configuration
|
|
// to the equivalent Terraform JSON representation.
|
|
//
|
|
// NOTE: THIS IS CURRENTLY A HACK. WE NEED A BETTER WAY TO
|
|
// CONVERT TO/FROM TERRAFORM COMPATIBLE FORMAT.
|
|
func BundleToTerraform(config *config.Root) *schema.Root {
|
|
tfroot := schema.NewRoot()
|
|
tfroot.Provider = schema.NewProviders()
|
|
tfroot.Provider.Databricks.Profile = config.Workspace.Profile
|
|
tfroot.Resource = schema.NewResources()
|
|
|
|
for k, src := range config.Resources.Jobs {
|
|
var dst schema.ResourceJob
|
|
conv(src, &dst)
|
|
|
|
for _, v := range src.Tasks {
|
|
var t schema.ResourceJobTask
|
|
conv(v, &t)
|
|
|
|
for _, v_ := range v.Libraries {
|
|
var l schema.ResourceJobTaskLibrary
|
|
conv(v_, &l)
|
|
t.Library = append(t.Library, l)
|
|
}
|
|
|
|
dst.Task = append(dst.Task, t)
|
|
}
|
|
|
|
for _, v := range src.JobClusters {
|
|
var t schema.ResourceJobJobCluster
|
|
conv(v, &t)
|
|
dst.JobCluster = append(dst.JobCluster, t)
|
|
}
|
|
|
|
// Unblock downstream work. To be addressed more generally later.
|
|
if git := src.GitSource; git != nil {
|
|
dst.GitSource = &schema.ResourceJobGitSource{
|
|
Url: git.GitUrl,
|
|
Branch: git.GitBranch,
|
|
Commit: git.GitCommit,
|
|
Provider: string(git.GitProvider),
|
|
Tag: git.GitTag,
|
|
}
|
|
}
|
|
|
|
tfroot.Resource.Job[k] = &dst
|
|
}
|
|
|
|
for k, src := range config.Resources.Pipelines {
|
|
var dst schema.ResourcePipeline
|
|
conv(src, &dst)
|
|
|
|
for _, v := range src.Libraries {
|
|
var l schema.ResourcePipelineLibrary
|
|
conv(v, &l)
|
|
dst.Library = append(dst.Library, l)
|
|
}
|
|
|
|
for _, v := range src.Clusters {
|
|
var l schema.ResourcePipelineCluster
|
|
conv(v, &l)
|
|
dst.Cluster = append(dst.Cluster, l)
|
|
}
|
|
|
|
tfroot.Resource.Pipeline[k] = &dst
|
|
}
|
|
|
|
return tfroot
|
|
}
|
|
|
|
func TerraformToBundle(state *tfjson.State, config *config.Root) error {
|
|
if state.Values == nil {
|
|
return fmt.Errorf("state.Values not set")
|
|
}
|
|
|
|
if state.Values.RootModule == nil {
|
|
return fmt.Errorf("state.Values.RootModule not set")
|
|
}
|
|
|
|
for _, resource := range state.Values.RootModule.Resources {
|
|
// Limit to resources.
|
|
if resource.Mode != tfjson.ManagedResourceMode {
|
|
continue
|
|
}
|
|
|
|
switch resource.Type {
|
|
case "databricks_job":
|
|
var tmp schema.ResourceJob
|
|
conv(resource.AttributeValues, &tmp)
|
|
cur := config.Resources.Jobs[resource.Name]
|
|
conv(tmp, &cur)
|
|
config.Resources.Jobs[resource.Name] = cur
|
|
case "databricks_pipeline":
|
|
var tmp schema.ResourcePipeline
|
|
conv(resource.AttributeValues, &tmp)
|
|
cur := config.Resources.Pipelines[resource.Name]
|
|
conv(tmp, &cur)
|
|
config.Resources.Pipelines[resource.Name] = cur
|
|
default:
|
|
return fmt.Errorf("missing mapping for %s", resource.Type)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|