mirror of https://github.com/databricks/cli.git
80 lines
2.0 KiB
Go
80 lines
2.0 KiB
Go
package terraform
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/libs/cmdio"
|
|
"github.com/databricks/cli/libs/diag"
|
|
"github.com/databricks/cli/libs/plan"
|
|
)
|
|
|
|
type planDeploy struct{}
|
|
|
|
func (p *planDeploy) Name() string {
|
|
return "terraform.PlanDeploy"
|
|
}
|
|
|
|
// TODO: We need end to end tests for the approval flow.
|
|
func (p *planDeploy) Apply(ctx context.Context, b *bundle.Bundle) diag.Diagnostics {
|
|
tf := b.Terraform
|
|
if tf == nil {
|
|
return diag.Errorf("terraform not initialized")
|
|
}
|
|
|
|
// Get terraform project root.
|
|
tfDir, err := Dir(ctx, b)
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
// Set plan in main bundle struct for downstream mutators
|
|
b.Plan, err = plan.NewTerraformPlan(ctx, plan.TerraformPlanOpts{
|
|
Executable: tf,
|
|
Root: tfDir,
|
|
IsDestroy: false,
|
|
})
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
// If the `--auto-approve` flag is set, we don't need to ask for approval.
|
|
if b.AutoApprove {
|
|
return nil
|
|
}
|
|
|
|
destructiveActions := b.Plan.ActionsByTypes(plan.ActionTypeRecreate, plan.ActionTypeDelete)
|
|
|
|
// If there are no deletes or recreates, we don't need to ask for approval.
|
|
if len(destructiveActions) == 0 {
|
|
return nil
|
|
}
|
|
|
|
// display information the user needs to know about the deploy plan.
|
|
cmdio.LogString(ctx, "The following resources will be deleted or recreated:")
|
|
for _, a := range destructiveActions {
|
|
cmdio.Log(ctx, a)
|
|
}
|
|
|
|
approved, err := cmdio.AskYesOrNo(ctx, "Would you like to proceed?")
|
|
if err != nil {
|
|
return diag.FromErr(err)
|
|
}
|
|
|
|
// TODO: Remove the previous flag for deployment in the bundle tree?
|
|
if !approved {
|
|
// We error here to terminate the control flow and prevent the current
|
|
// process from modifying any deployment state.
|
|
cmdio.LogString(ctx, "No changes are being made...")
|
|
return diag.FromErr(bundle.ErrorBreakSequence)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// PlanDeploy returns a [bundle.Mutator] that runs the equivalent of `terraform plan -out ./plan`
|
|
// from the bundle's ephemeral working directory for Terraform.
|
|
func PlanDeploy() bundle.Mutator {
|
|
return &planDeploy{}
|
|
}
|