update_cpi_config.go

package cmd

import (
	bosherr "github.com/cloudfoundry/bosh-utils/errors"

	boshdir "github.com/cloudfoundry/bosh-cli/director"
	boshtpl "github.com/cloudfoundry/bosh-cli/director/template"
	boshui "github.com/cloudfoundry/bosh-cli/ui"
)

type UpdateCPIConfigCmd struct {
	ui       boshui.UI
	director boshdir.Director
}

func NewUpdateCPIConfigCmd(ui boshui.UI, director boshdir.Director) UpdateCPIConfigCmd {
	return UpdateCPIConfigCmd{ui: ui, director: director}
}

func (c UpdateCPIConfigCmd) Run(opts UpdateCPIConfigOpts) error {
	tpl := boshtpl.NewTemplate(opts.Args.CPIConfig.Bytes)

	bytes, err := tpl.Evaluate(opts.VarFlags.AsVariables(), opts.OpsFlags.AsOp(), boshtpl.EvaluateOpts{})
	if err != nil {
		return bosherr.WrapErrorf(err, "Evaluating cpi config")
	}

	err = c.ui.AskForConfirmation()
	if err != nil {
		return err
	}

	return c.director.UpdateCPIConfig(bytes)
}