mirror of https://github.com/databricks/cli.git
110 lines
2.6 KiB
Go
110 lines
2.6 KiB
Go
package terraform
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/libs/cmdio"
|
|
"github.com/databricks/cli/libs/diag"
|
|
"github.com/hashicorp/terraform-exec/tfexec"
|
|
)
|
|
|
|
type BindOptions struct {
|
|
AutoApprove bool
|
|
ResourceType string
|
|
ResourceKey string
|
|
ResourceId string
|
|
}
|
|
|
|
type importResource struct {
|
|
opts *BindOptions
|
|
}
|
|
|
|
// Apply implements bundle.Mutator.
|
|
func (m *importResource) Apply(ctx context.Context, b *bundle.Bundle) diag.Diagnostics {
|
|
dir, err := Dir(ctx, b)
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
tf := b.Terraform
|
|
if tf == nil {
|
|
return diag.Errorf("terraform not initialized")
|
|
}
|
|
|
|
err = tf.Init(ctx, tfexec.Upgrade(true))
|
|
if err != nil {
|
|
return diag.Errorf("terraform init: %v", err)
|
|
}
|
|
tmpDir, err := os.MkdirTemp("", "state-*")
|
|
if err != nil {
|
|
return diag.Errorf("terraform init: %v", err)
|
|
}
|
|
tmpState := filepath.Join(tmpDir, TerraformStateFileName)
|
|
|
|
importAddress := fmt.Sprintf("%s.%s", m.opts.ResourceType, m.opts.ResourceKey)
|
|
err = tf.Import(ctx, importAddress, m.opts.ResourceId, tfexec.StateOut(tmpState))
|
|
if err != nil {
|
|
return diag.Errorf("terraform import: %v", err)
|
|
}
|
|
|
|
buf := bytes.NewBuffer(nil)
|
|
tf.SetStdout(buf)
|
|
|
|
//lint:ignore SA1019 We use legacy -state flag for now to plan the import changes based on temporary state file
|
|
changed, err := tf.Plan(ctx, tfexec.State(tmpState), tfexec.Target(importAddress))
|
|
if err != nil {
|
|
return diag.Errorf("terraform plan: %v", err)
|
|
}
|
|
|
|
defer os.RemoveAll(tmpDir)
|
|
|
|
if changed && !m.opts.AutoApprove {
|
|
output := buf.String()
|
|
// Remove output starting from Warning until end of output
|
|
output = output[:bytes.Index([]byte(output), []byte("Warning:"))]
|
|
cmdio.LogString(ctx, output)
|
|
ans, err := cmdio.AskYesOrNo(ctx, "Confirm import changes? Changes will be remotely applied only after running 'bundle deploy'.")
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
if !ans {
|
|
return diag.Errorf("import aborted")
|
|
}
|
|
}
|
|
|
|
// If user confirmed changes, move the state file from temp dir to state location
|
|
f, err := os.Create(filepath.Join(dir, TerraformStateFileName))
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
defer f.Close()
|
|
|
|
tmpF, err := os.Open(tmpState)
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
defer tmpF.Close()
|
|
|
|
_, err = io.Copy(f, tmpF)
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Name implements bundle.Mutator.
|
|
func (*importResource) Name() string {
|
|
return "terraform.Import"
|
|
}
|
|
|
|
func Import(opts *BindOptions) bundle.Mutator {
|
|
return &importResource{opts: opts}
|
|
}
|